fix: resolve lint errors from rebase onto main

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Author: jsonbailey

packages/ai-providers/server-ai-langchain/src/ldai_langchain/langchain_runner_factory.py

@@ -18,7 +18,9 @@ def create_agent_graph(self, graph_def: Any, tools: ToolRegistry) -> Any:
         :param tools: Registry mapping tool names to callables (langchain-compatible)
         :return: LangGraphAgentGraphRunner ready to execute the graph
         """
-        from ldai_langchain.langgraph_agent_graph_runner import LangGraphAgentGraphRunner
+        from ldai_langchain.langgraph_agent_graph_runner import (
+            LangGraphAgentGraphRunner,
+        )
         return LangGraphAgentGraphRunner(graph_def, tools)
 
     def create_model(self, config: AIConfigKind) -> LangChainModelRunner:

packages/ai-providers/server-ai-langchain/src/ldai_langchain/langgraph_agent_graph_runner.py

@@ -6,11 +6,15 @@
 
 from ldai import log
 from ldai.agent_graph import AgentGraphDefinition, AgentGraphNode
+from ldai.providers import AgentGraphResult, AgentGraphRunner, ToolRegistry
 from ldai.providers.types import LDAIMetrics
-from ldai.runners.agent_graph_runner import AgentGraphRunner
-from ldai.runners.types import AgentGraphResult, ToolRegistry
 
-from ldai_langchain.langchain_helper import LangChainHelper
+from ldai_langchain.langchain_helper import (
+    get_ai_metrics_from_response,
+    get_ai_usage_from_response,
+    get_tool_calls_from_response,
+    sum_token_usage_from_messages,
+)
 
 
 class LangGraphAgentGraphRunner(AgentGraphRunner):
@@ -53,7 +57,7 @@ async def run(self, input: Any) -> AgentGraphResult:
             from typing_extensions import TypedDict
 
             class WorkflowState(TypedDict):
-                messages: Annotated[List[AnyMessage], operator.add]
+                messages: Annotated[List[Any], operator.add]
 
             agent_builder: StateGraph = StateGraph(WorkflowState)
             root_node = self._graph.root()
@@ -87,11 +91,11 @@ def invoke(state: WorkflowState) -> WorkflowState:
                     if node_tracker:
                         response = node_tracker.track_metrics_of(
                             lambda: model.invoke(state['messages']),
-                            LangChainHelper.get_ai_metrics_from_response,
+                            get_ai_metrics_from_response,
                             graph_key=gk,
                         )
                         node_tracker.track_tool_calls(
-                            LangChainHelper.get_tool_calls_from_response(response),
+                            get_tool_calls_from_response(response),
                             graph_key=tracker.graph_key if tracker is not None else None,
                         )
                     else:
@@ -101,7 +105,7 @@ def invoke(state: WorkflowState) -> WorkflowState:
 
                 invoke.__name__ = node_key
 
-                agent_builder.add_node(name=node_key, node=invoke)
+                agent_builder.add_node(node_key, invoke)
 
                 if node_key == root_key:
                     agent_builder.add_edge(START, node_key)
@@ -134,7 +138,7 @@ def invoke(state: WorkflowState) -> WorkflowState:
                 tracker.track_latency(duration)
                 tracker.track_invocation_success()
                 tracker.track_total_tokens(
-                    LangChainHelper.sum_token_usage_from_messages(messages)
+                    sum_token_usage_from_messages(messages)
                 )
 
             return AgentGraphResult(

packages/ai-providers/server-ai-langchain/tests/test_langgraph_agent_graph_runner.py

@@ -5,7 +5,7 @@
 
 from ldai.agent_graph import AgentGraphDefinition
 from ldai.models import AIAgentGraphConfig, AIAgentConfig, ModelConfig, ProviderConfig
-from ldai.runners.types import AgentGraphResult, ToolRegistry
+from ldai.providers import AgentGraphResult, ToolRegistry
 from ldai_langchain.langgraph_agent_graph_runner import LangGraphAgentGraphRunner
 from ldai_langchain.langchain_runner_factory import LangChainRunnerFactory
 

packages/ai-providers/server-ai-openai/src/ldai_openai/openai_agent_graph_runner.py

@@ -5,9 +5,8 @@
 
 from ldai import log
 from ldai.agent_graph import AgentGraphDefinition, AgentGraphNode
+from ldai.providers import AgentGraphResult, AgentGraphRunner, ToolRegistry
 from ldai.providers.types import LDAIMetrics
-from ldai.runners.agent_graph_runner import AgentGraphRunner
-from ldai.runners.types import AgentGraphResult, ToolRegistry
 from ldai.tracker import TokenUsage
 
 
@@ -35,7 +34,12 @@ def _log_run_result_shape(result: Any) -> None:
 
 def _build_native_tool_map() -> dict:
     try:
-        from agents import CodeInterpreterTool, FileSearchTool, ImageGenerationTool, WebSearchTool
+        from agents import (
+            CodeInterpreterTool,
+            FileSearchTool,
+            ImageGenerationTool,
+            WebSearchTool,
+        )
         return {
             'web_search_tool': lambda _: WebSearchTool(),
             'file_search_tool': lambda _: FileSearchTool(),
@@ -190,7 +194,14 @@ def _build_agents(self, path: List[str]) -> Any:
         :return: The root Agent instance
         """
         try:
-            from agents import Agent, FunctionTool, Handoff, RunContextWrapper, Tool, handoff
+            from agents import (
+                Agent,
+                FunctionTool,
+                Handoff,
+                RunContextWrapper,
+                Tool,
+                handoff,
+            )
             from agents.extensions.handoff_prompt import RECOMMENDED_PROMPT_PREFIX
             from agents.tool_context import ToolContext
         except ImportError as exc:

packages/ai-providers/server-ai-openai/src/ldai_openai/openai_runner_factory.py

@@ -2,8 +2,7 @@
 from typing import Any, Optional
 
 from ldai.models import AIConfigKind
-from ldai.providers import AIProvider
-from ldai.runners.types import ToolRegistry
+from ldai.providers import AIProvider, ToolRegistry
 from openai import AsyncOpenAI
 
 from ldai_openai.openai_model_runner import OpenAIModelRunner

packages/ai-providers/server-ai-openai/tests/test_openai_agent_graph_runner.py

@@ -5,7 +5,7 @@
 
 from ldai.agent_graph import AgentGraphDefinition
 from ldai.models import AIAgentGraphConfig, AIAgentConfig, Edge, ModelConfig, ProviderConfig
-from ldai.runners.types import AgentGraphResult, ToolRegistry
+from ldai.providers import AgentGraphResult, ToolRegistry
 from ldai_openai.openai_agent_graph_runner import OpenAIAgentGraphRunner
 from ldai_openai.openai_runner_factory import OpenAIRunnerFactory
 

packages/sdk/server-ai/src/ldai/client.py

@@ -25,8 +25,8 @@
     ModelConfig,
     ProviderConfig,
 )
+from ldai.providers import ToolRegistry
 from ldai.providers.runner_factory import RunnerFactory
-from ldai.runners.types import ToolRegistry
 from ldai.sdk_info import AI_SDK_LANGUAGE, AI_SDK_NAME, AI_SDK_VERSION
 from ldai.tracker import AIGraphTracker, LDAIConfigTracker
 
@@ -653,7 +653,7 @@ async def create_agent_graph(
         if not graph.enabled:
             return None
 
-        runner = await RunnerFactory.create_agent_graph(
+        runner = RunnerFactory.create_agent_graph(
             graph, tools or {}, default_ai_provider
         )
         if not runner:

packages/sdk/server-ai/src/ldai/managed_agent_graph.py

@@ -2,8 +2,7 @@
 
 from typing import Any, Optional
 
-from ldai.runners.agent_graph_runner import AgentGraphRunner
-from ldai.runners.types import AgentGraphResult
+from ldai.providers import AgentGraphResult, AgentGraphRunner
 from ldai.tracker import AIGraphTracker
 
 

packages/sdk/server-ai/tests/test_managed_agent_graph.py

@@ -7,8 +7,7 @@
 
 from ldai import LDAIClient, ManagedAgentGraph
 from ldai.providers.types import LDAIMetrics
-from ldai.runners.agent_graph_runner import AgentGraphRunner
-from ldai.runners.types import AgentGraphResult, ToolRegistry
+from ldai.providers import AgentGraphResult, AgentGraphRunner, ToolRegistry
 from ldai.tracker import AIGraphTracker
 
 
@@ -125,7 +124,7 @@ async def test_create_agent_graph_returns_managed_agent_graph(ldai_client: LDAIC
 
     with patch(
         'ldai.providers.runner_factory.RunnerFactory.create_agent_graph',
-        new=AsyncMock(return_value=stub_runner),
+        new=MagicMock(return_value=stub_runner),
     ):
         managed = await ldai_client.create_agent_graph('travel-graph', context)
 
@@ -147,7 +146,7 @@ async def test_create_agent_graph_returns_none_when_runner_factory_fails(ldai_cl
 
     with patch(
         'ldai.providers.runner_factory.RunnerFactory.create_agent_graph',
-        new=AsyncMock(return_value=None),
+        new=MagicMock(return_value=None),
     ):
         managed = await ldai_client.create_agent_graph('travel-graph', context)
 
@@ -160,7 +159,7 @@ async def test_create_agent_graph_passes_tools_to_factory(ldai_client: LDAIClien
     tools: ToolRegistry = {'search': lambda q: f'results for {q}'}
     captured = {}
 
-    async def fake_create_agent_graph(graph_def, tools_arg, default_ai_provider=None):
+    def fake_create_agent_graph(graph_def, tools_arg, default_ai_provider=None):
         captured['tools'] = tools_arg
         return StubAgentGraphRunner()
 
@@ -179,7 +178,7 @@ async def test_create_agent_graph_run_produces_result(ldai_client: LDAIClient):
 
     with patch(
         'ldai.providers.runner_factory.RunnerFactory.create_agent_graph',
-        new=AsyncMock(return_value=StubAgentGraphRunner("final answer")),
+        new=MagicMock(return_value=StubAgentGraphRunner("final answer")),
     ):
         managed = await ldai_client.create_agent_graph('travel-graph', context)