fix: fix lint issues (isort, pycodestyle)

jsonbailey · claude · jsonbailey · commit 5527667af0c0 · 2026-05-04T10:11:55.000-05:00
Co-Authored-By: Claude Sonnet 4.6 &lt;noreply@anthropic.com&gt;
diff --git a/packages/sdk/server-ai/src/ldai/tracker.py b/packages/sdk/server-ai/src/ldai/tracker.py
@@ -680,7 +680,9 @@ def track_invocation_success(self) -> None:
         Track a successful graph invocation.
         """
         if self._summary.success is not None:
-            log.warning("Invocation status has already been tracked for this graph execution. %s", self.__get_track_data())
+            log.warning(
+                "Invocation status has already been tracked for this graph execution. %s",
+                self.__get_track_data())
             return
         self._summary.success = True
         self._ld_client.track(
@@ -695,7 +697,9 @@ def track_invocation_failure(self) -> None:
         Track an unsuccessful graph invocation.
         """
         if self._summary.success is not None:
-            log.warning("Invocation status has already been tracked for this graph execution. %s", self.__get_track_data())
+            log.warning(
+                "Invocation status has already been tracked for this graph execution. %s",
+                self.__get_track_data())
             return
         self._summary.success = False
         self._ld_client.track(
diff --git a/packages/sdk/server-ai/tests/test_agent_graph.py b/packages/sdk/server-ai/tests/test_agent_graph.py
@@ -1,15 +1,15 @@
+from unittest.mock import MagicMock
+
 import pytest
 from ldclient import Config, Context, LDClient
 from ldclient.integrations.test_data import TestData
 
-from unittest.mock import MagicMock
-
 from ldai import (
-    LDAIClient,
-    AIAgentGraphConfig,
     AgentGraphDefinition,
     AIAgentConfig,
+    AIAgentGraphConfig,
     Edge,
+    LDAIClient,
 )
 from ldai.evaluator import Evaluator
 
@@ -124,7 +124,10 @@ def td() -> TestData:
                     "parameters": {"temperature": 0.3, "maxTokens": 2048},
                 },
                 "provider": {"name": "openai"},
-                "instructions": "You are a helpful customer support agent for {{company_name}}. Always be polite and professional.",
+                "instructions": (
+                    "You are a helpful customer support agent for {{company_name}}."
+                    " Always be polite and professional."
+                ),
                 "_ldMeta": {
                     "enabled": True,
                     "variationKey": "agent-v1",
@@ -142,7 +145,10 @@ def td() -> TestData:
         .variations(
             {
                 "model": {"name": "claude-3", "parameters": {"temperature": 0.5}},
-                "instructions": "Hello {{ldctx.name}}! I am your personal assistant. Your user key is {{ldctx.key}}.",
+                "instructions": (
+                    "Hello {{ldctx.name}}! I am your personal assistant."
+                    " Your user key is {{ldctx.key}}."
+                ),
                 "_ldMeta": {
                     "enabled": True,
                     "variationKey": "personal-v1",
@@ -160,7 +166,10 @@ def td() -> TestData:
         .variations(
             {
                 "model": {"name": "gpt-3.5-turbo"},
-                "instructions": "Welcome {{ldctx.user.name}} from {{ldctx.org.name}}! Your organization tier is {{ldctx.org.tier}}.",
+                "instructions": (
+                    "Welcome {{ldctx.user.name}} from {{ldctx.org.name}}!"
+                    " Your organization tier is {{ldctx.org.tier}}."
+                ),
                 "_ldMeta": {
                     "enabled": True,
                     "variationKey": "multi-v1",
diff --git a/packages/sdk/server-ai/tests/test_agents.py b/packages/sdk/server-ai/tests/test_agents.py
@@ -2,8 +2,13 @@
 from ldclient import Config, Context, LDClient
 from ldclient.integrations.test_data import TestData
 
-from ldai import (LDAIAgentConfig, LDAIAgentDefaults, LDAIClient, ModelConfig,
-                  ProviderConfig)
+from ldai import (
+    LDAIAgentConfig,
+    LDAIAgentDefaults,
+    LDAIClient,
+    ModelConfig,
+    ProviderConfig,
+)
 
 
 @pytest.fixture
@@ -17,7 +22,10 @@ def td() -> TestData:
             {
                 'model': {'name': 'gpt-4', 'parameters': {'temperature': 0.3, 'maxTokens': 2048}},
                 'provider': {'name': 'openai'},
-                'instructions': 'You are a helpful customer support agent for {{company_name}}. Always be polite and professional.',
+                'instructions': (
+                    'You are a helpful customer support agent for {{company_name}}.'
+                    ' Always be polite and professional.'
+                ),
                 '_ldMeta': {'enabled': True, 'variationKey': 'agent-v1', 'version': 1, 'mode': 'agent'},
             }
         )
@@ -30,7 +38,10 @@ def td() -> TestData:
         .variations(
             {
                 'model': {'name': 'claude-3', 'parameters': {'temperature': 0.5}},
-                'instructions': 'Hello {{ldctx.name}}! I am your personal assistant. Your user key is {{ldctx.key}}.',
+                'instructions': (
+                    'Hello {{ldctx.name}}! I am your personal assistant.'
+                    ' Your user key is {{ldctx.key}}.'
+                ),
                 '_ldMeta': {'enabled': True, 'variationKey': 'personal-v1', 'version': 2, 'mode': 'agent'},
             }
         )
@@ -43,7 +54,10 @@ def td() -> TestData:
         .variations(
             {
                 'model': {'name': 'gpt-3.5-turbo'},
-                'instructions': 'Welcome {{ldctx.user.name}} from {{ldctx.org.name}}! Your organization tier is {{ldctx.org.tier}}.',
+                'instructions': (
+                    'Welcome {{ldctx.user.name}} from {{ldctx.org.name}}!'
+                    ' Your organization tier is {{ldctx.org.tier}}.'
+                ),
                 '_ldMeta': {'enabled': True, 'variationKey': 'multi-v1', 'version': 1, 'mode': 'agent'},
             }
         )
@@ -82,7 +96,10 @@ def td() -> TestData:
             {
                 'model': {'name': 'gpt-4', 'parameters': {'temperature': 0.7}},
                 'provider': {'name': 'openai'},
-                'instructions': 'You are a sales assistant for {{company_name}}. Help customers find the right products.',
+                'instructions': (
+                    'You are a sales assistant for {{company_name}}.'
+                    ' Help customers find the right products.'
+                ),
                 '_ldMeta': {'enabled': True, 'variationKey': 'sales-v1', 'version': 1, 'mode': 'agent'},
             }
         )
@@ -96,7 +113,10 @@ def td() -> TestData:
             {
                 'model': {'name': 'gpt-4', 'parameters': {'temperature': 0.2, 'maxTokens': 3000}},
                 'provider': {'name': 'openai'},
-                'instructions': 'You are a research assistant specializing in {{topic}}. Your expertise level should match {{ldctx.expertise}}.',
+                'instructions': (
+                    'You are a research assistant specializing in {{topic}}.'
+                    ' Your expertise level should match {{ldctx.expertise}}.'
+                ),
                 '_ldMeta': {'enabled': True, 'variationKey': 'research-v1', 'version': 1, 'mode': 'agent'},
             }
         )
@@ -139,7 +159,10 @@ def test_single_agent_method(ldai_client: LDAIClient):
     assert agent.model.get_parameter('maxTokens') == 3000
     assert agent.provider is not None
     assert agent.provider.name == 'openai'
-    assert agent.instructions == 'You are a research assistant specializing in quantum computing. Your expertise level should match advanced.'
+    assert agent.instructions == (
+        'You are a research assistant specializing in quantum computing.'
+        ' Your expertise level should match advanced.'
+    )
     assert callable(agent.create_tracker)
 
 
@@ -237,7 +260,10 @@ def test_agents_method_different_variables_per_agent(ldai_client: LDAIClient):
     assert personal_agent.instructions == 'Hello Alice! I am your personal assistant. Your user key is user-key.'
 
     support_agent = agents['customer-support-agent']
-    assert support_agent.instructions == 'You are a helpful customer support agent for TechStart Inc. Always be polite and professional.'
+    assert support_agent.instructions == (
+        'You are a helpful customer support agent for TechStart Inc.'
+        ' Always be polite and professional.'
+    )
 
 
 def test_agents_with_multi_context_interpolation(ldai_client: LDAIClient):
diff --git a/packages/sdk/server-ai/tests/test_judge.py b/packages/sdk/server-ai/tests/test_judge.py
@@ -8,7 +8,13 @@
 
 from ldai.judge import Judge
 from ldai.judge.evaluation_schema_builder import EvaluationSchemaBuilder
-from ldai.models import AIJudgeConfig, AIJudgeConfigDefault, LDMessage, ModelConfig, ProviderConfig
+from ldai.models import (
+    AIJudgeConfig,
+    AIJudgeConfigDefault,
+    LDMessage,
+    ModelConfig,
+    ProviderConfig,
+)
 from ldai.providers.types import JudgeResult, LDAIMetrics, RunnerResult
 from ldai.tracker import LDAIConfigTracker
 
diff --git a/packages/sdk/server-ai/tests/test_managed_agent.py b/packages/sdk/server-ai/tests/test_managed_agent.py
@@ -1,17 +1,17 @@
 """Tests for ManagedAgent."""
 
-import pytest
 from unittest.mock import AsyncMock, MagicMock
 
+import pytest
+from ldclient import Config, Context, LDClient
+from ldclient.integrations.test_data import TestData
+
 from ldai import LDAIClient, ManagedAgent
 from ldai.managed_agent import ManagedAgent
 from ldai.models import AIAgentConfig, AIAgentConfigDefault, ModelConfig, ProviderConfig
 from ldai.providers.types import LDAIMetrics, ManagedResult, RunnerResult
 from ldai.tracker import LDAIMetricSummary
 
-from ldclient import Config, Context, LDClient
-from ldclient.integrations.test_data import TestData
-
 
 def _make_summary(success: bool = True) -> LDAIMetricSummary:
     summary = LDAIMetricSummary()
diff --git a/packages/sdk/server-ai/tests/test_managed_agent_graph.py b/packages/sdk/server-ai/tests/test_managed_agent_graph.py
@@ -1,18 +1,24 @@
 """Tests for ManagedAgentGraph and LDAIClient.create_agent_graph()."""
 
-import pytest
 from unittest.mock import AsyncMock, MagicMock, patch
+
+import pytest
 from ldclient import Config, Context, LDClient
 from ldclient.integrations.test_data import TestData
 
 from ldai import LDAIClient, LDAIMetricSummary, ManagedAgentGraph, ManagedGraphResult
-from ldai.providers.types import AgentGraphRunnerResult, GraphMetrics, GraphMetricSummary, LDAIMetrics
 from ldai.providers import AgentGraphRunner, ToolRegistry
+from ldai.providers.types import (
+    AgentGraphRunnerResult,
+    GraphMetrics,
+    GraphMetricSummary,
+    LDAIMetrics,
+)
 from ldai.tracker import TokenUsage
 
-
 # --- Test doubles ---
 
+
 class StubAgentGraphRunner(AgentGraphRunner):
     """Runner that returns AgentGraphRunnerResult (new shape)."""
     def __init__(self, content: str = "stub output"):
diff --git a/packages/sdk/server-ai/tests/test_managed_model.py b/packages/sdk/server-ai/tests/test_managed_model.py
@@ -13,7 +13,6 @@
 from ldai.tracker import LDAIConfigTracker, LDAIMetricSummary
 
 
-
 def _make_runner_result(content: str = 'response text') -> RunnerResult:
     return RunnerResult(
         content=content,
diff --git a/packages/sdk/server-ai/tests/test_model_config.py b/packages/sdk/server-ai/tests/test_model_config.py
@@ -3,8 +3,11 @@
 from ldclient.integrations.test_data import TestData
 
 from ldai import LDAIClient, LDMessage, ModelConfig
-from ldai.models import (AIAgentConfigDefault, AICompletionConfigDefault,
-                         AIJudgeConfigDefault)
+from ldai.models import (
+    AIAgentConfigDefault,
+    AICompletionConfigDefault,
+    AIJudgeConfigDefault,
+)
 
 
 @pytest.fixture
@@ -14,7 +17,11 @@ def td() -> TestData:
         td.flag('model-config')
         .variations(
             {
-                'model': {'name': 'fakeModel', 'parameters': {'temperature': 0.5, 'maxTokens': 4096}, 'custom': {'extra-attribute': 'value'}},
+                'model': {
+                    'name': 'fakeModel',
+                    'parameters': {'temperature': 0.5, 'maxTokens': 4096},
+                    'custom': {'extra-attribute': 'value'},
+                },
                 'provider': {'name': 'fakeProvider'},
                 'messages': [{'role': 'system', 'content': 'Hello, {{name}}!'}],
                 '_ldMeta': {'enabled': True, 'variationKey': 'abcd', 'version': 1},
@@ -44,7 +51,9 @@ def td() -> TestData:
         td.flag('ctx-interpolation')
         .variations(
             {
-                'model': {'name': 'fakeModel', 'parameters': {'extra-attribute': 'I can be anything I set my mind/type to'}},
+                'model': {'name': 'fakeModel', 'parameters': {
+                    'extra-attribute': 'I can be anything I set my mind/type to',
+                }},
                 'messages': [{'role': 'system', 'content': 'Hello, {{ldctx.name}}! Is your last name {{ldctx.last}}?'}],
                 '_ldMeta': {'enabled': True, 'variationKey': 'abcd', 'version': 1},
             }
@@ -56,8 +65,12 @@ def td() -> TestData:
         td.flag('multi-ctx-interpolation')
         .variations(
             {
-                'model': {'name': 'fakeModel', 'parameters': {'extra-attribute': 'I can be anything I set my mind/type to'}},
-                'messages': [{'role': 'system', 'content': 'Hello, {{ldctx.user.name}}! Do you work for {{ldctx.org.shortname}}?'}],
+                'model': {'name': 'fakeModel', 'parameters': {
+                    'extra-attribute': 'I can be anything I set my mind/type to',
+                }},
+                'messages': [{'role': 'system', 'content': (
+                    'Hello, {{ldctx.user.name}}! Do you work for {{ldctx.org.shortname}}?'
+                )}],
                 '_ldMeta': {'enabled': True, 'variationKey': 'abcd', 'version': 1},
             }
         )
diff --git a/packages/sdk/server-ai/tests/test_runner_abcs.py b/packages/sdk/server-ai/tests/test_runner_abcs.py
@@ -1,11 +1,17 @@
 import pytest
 
-from ldai.providers import AgentGraphResult, AgentGraphRunner, AgentGraphRunnerResult, AgentRunner, ToolRegistry
+from ldai.providers import (
+    AgentGraphResult,
+    AgentGraphRunner,
+    AgentGraphRunnerResult,
+    AgentRunner,
+    ToolRegistry,
+)
 from ldai.providers.types import GraphMetrics, LDAIMetrics, RunnerResult
 
-
 # --- Concrete test doubles ---
 
+
 class ConcreteAgentRunner:
     async def run(self, input):
         return RunnerResult(
diff --git a/packages/sdk/server-ai/tests/test_tools.py b/packages/sdk/server-ai/tests/test_tools.py
@@ -2,7 +2,7 @@
 from ldclient import Config, Context, LDClient
 from ldclient.integrations.test_data import TestData
 
-from ldai import LDTool, LDAIClient
+from ldai import LDAIClient, LDTool
 from ldai.models import AIAgentConfigDefault, AICompletionConfigDefault
 
 
@@ -256,7 +256,9 @@ def test_completion_config_model_params_tools_as_list_returns_none(client, conte
 
 
 def test_completion_config_model_params_tools_skips_bad_entries_silently(client, context):
-    result = client.completion_config('completion-model-params-tools-missing-name', context, AICompletionConfigDefault())
+    result = client.completion_config(
+        'completion-model-params-tools-missing-name', context, AICompletionConfigDefault()
+    )
 
     assert result.tools is not None
     assert 'valid-tool' in result.tools
diff --git a/packages/sdk/server-ai/tests/test_tracker.py b/packages/sdk/server-ai/tests/test_tracker.py
