fix: Update provider packages for tracker factory pattern

Migrate langchain and openai provider packages from config.tracker
to config.create_tracker() and fix test signatures to match.

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

Author: jsonbailey

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

@@ -200,7 +200,7 @@ def flush(self, graph: AgentGraphDefinition) -> None:
             node = graph.get_node(node_key)
             if not node:
                 continue
-            config_tracker = node.get_config().tracker
+            config_tracker = node.get_config().create_tracker()
             if not config_tracker:
                 continue
 

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

@@ -536,7 +536,6 @@ def sync_tool(x: str = '') -> str:
             ),
             provider=ProviderConfig(name='openai'),
             instructions='',
-            tracker=MagicMock(),
         )
         tools = build_structured_tools(cfg, {'my_tool': sync_tool})
         assert len(tools) == 1
@@ -559,7 +558,6 @@ async def async_tool(x: str = '') -> str:
             ),
             provider=ProviderConfig(name='openai'),
             instructions='',
-            tracker=MagicMock(),
         )
         tools = build_structured_tools(cfg, {'my_tool': async_tool})
         assert len(tools) == 1

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

@@ -17,7 +17,6 @@ def _make_graph(enabled: bool = True) -> AgentGraphDefinition:
         model=ModelConfig(name='gpt-4'),
         provider=ProviderConfig(name='openai'),
         instructions='You are a helpful assistant.',
-        tracker=MagicMock(),
     )
     graph_config = AIAgentGraphConfig(
         key='test-graph',

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

@@ -51,7 +51,7 @@ def _make_graph(mock_ld_client: MagicMock, node_key: str = 'root-agent', graph_k
         model=ModelConfig(name='gpt-4', parameters={}),
         provider=ProviderConfig(name='openai'),
         instructions='Be helpful.',
-        tracker=node_tracker,
+        create_tracker=lambda: node_tracker,
     )
     graph_config = AIAgentGraphConfig(
         key=graph_key,
@@ -411,7 +411,7 @@ def test_flush_with_no_graph_key_on_node_tracker():
         model=ModelConfig(name='gpt-4', parameters={}),
         provider=ProviderConfig(name='openai'),
         instructions='Be helpful.',
-        tracker=node_tracker,
+        create_tracker=lambda: node_tracker,
     )
     graph_config = AIAgentGraphConfig(
         key='test-graph',
@@ -465,7 +465,6 @@ def test_flush_skips_node_without_tracker():
         model=ModelConfig(name='gpt-4', parameters={}),
         provider=ProviderConfig(name='openai'),
         instructions='',
-        tracker=None,
     )
     graph_config = AIAgentGraphConfig(
         key='g', root_config_key='no-track', edges=[], enabled=True

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

@@ -69,7 +69,7 @@ def _make_graph(
         model=ModelConfig(name='gpt-4', parameters={'tools': tool_defs} if tool_defs else {}),
         provider=ProviderConfig(name='openai'),
         instructions='You are a helpful assistant.',
-        tracker=node_tracker,
+        create_tracker=lambda: node_tracker,
     )
 
     graph_config = AIAgentGraphConfig(
@@ -171,15 +171,15 @@ def _make_two_node_graph(mock_ld_client: MagicMock) -> 'AgentGraphDefinition':
         model=ModelConfig(name='gpt-4', parameters={}),
         provider=ProviderConfig(name='openai'),
         instructions='You are root.',
-        tracker=root_tracker,
+        create_tracker=lambda: root_tracker,
     )
     child_config = AIAgentConfig(
         key='child-agent',
         enabled=True,
         model=ModelConfig(name='gpt-4', parameters={}),
         provider=ProviderConfig(name='openai'),
         instructions='You are child.',
-        tracker=child_tracker,
+        create_tracker=lambda: child_tracker,
     )
 
     edge = Edge(key='root-to-child', source_config='root-agent', target_config='child-agent')
@@ -517,6 +517,7 @@ def _make_multi_child_graph(mock_ld_client: MagicMock) -> 'AgentGraphDefinition'
     def _node_tracker(key: str) -> LDAIConfigTracker:
         return LDAIConfigTracker(
             ld_client=mock_ld_client,
+            run_id='test-run-id',
             variation_key='test-variation',
             config_key=key,
             version=1,
@@ -541,23 +542,23 @@ def _node_tracker(key: str) -> LDAIConfigTracker:
             model=ModelConfig(name='gpt-4', parameters={}),
             provider=ProviderConfig(name='openai'),
             instructions='Route to the appropriate specialist agent.',
-            tracker=_node_tracker('orchestrator'),
+            create_tracker=lambda: _node_tracker('orchestrator'),
         ),
         'agent-a': AIAgentConfig(
             key='agent-a',
             enabled=True,
             model=ModelConfig(name='gpt-4', parameters={}),
             provider=ProviderConfig(name='openai'),
             instructions='You handle topic A.',
-            tracker=_node_tracker('agent-a'),
+            create_tracker=lambda: _node_tracker('agent-a'),
         ),
         'agent-b': AIAgentConfig(
             key='agent-b',
             enabled=True,
             model=ModelConfig(name='gpt-4', parameters={}),
             provider=ProviderConfig(name='openai'),
             instructions='You handle topic B.',
-            tracker=_node_tracker('agent-b'),
+            create_tracker=lambda: _node_tracker('agent-b'),
         ),
     }
 
@@ -628,6 +629,7 @@ def _make_multi_child_graph_with_tools(mock_ld_client: MagicMock, tool_names: li
     def _node_tracker(key: str) -> LDAIConfigTracker:
         return LDAIConfigTracker(
             ld_client=mock_ld_client,
+            run_id='test-run-id',
             variation_key='test-variation',
             config_key=key,
             version=1,
@@ -653,23 +655,23 @@ def _node_tracker(key: str) -> LDAIConfigTracker:
             model=ModelConfig(name='gpt-4', parameters={'tools': tool_defs}),
             provider=ProviderConfig(name='openai'),
             instructions='Route to a specialist after gathering info.',
-            tracker=_node_tracker('orchestrator'),
+            create_tracker=lambda: _node_tracker('orchestrator'),
         ),
         'agent-a': AIAgentConfig(
             key='agent-a',
             enabled=True,
             model=ModelConfig(name='gpt-4', parameters={}),
             provider=ProviderConfig(name='openai'),
             instructions='You handle topic A.',
-            tracker=_node_tracker('agent-a'),
+            create_tracker=lambda: _node_tracker('agent-a'),
         ),
         'agent-b': AIAgentConfig(
             key='agent-b',
             enabled=True,
             model=ModelConfig(name='gpt-4', parameters={}),
             provider=ProviderConfig(name='openai'),
             instructions='You handle topic B.',
-            tracker=_node_tracker('agent-b'),
+            create_tracker=lambda: _node_tracker('agent-b'),
         ),
     }
 

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

@@ -148,7 +148,7 @@ def _build_agents(self, path: List[str], state: _RunState) -> Any:
 
         def build_node(node: AgentGraphNode, ctx: dict) -> Any:
             node_config = node.get_config()
-            config_tracker = node_config.tracker
+            config_tracker = node_config.create_tracker()
             model = node_config.model
 
             if not model:
@@ -266,7 +266,7 @@ def _flush_final_segment(
         node = self._graph.get_node(state.last_node_key)
         if node is None:
             return
-        config_tracker = node.get_config().tracker
+        config_tracker = node.get_config().create_tracker()
         if config_tracker is None:
             return
 
@@ -296,6 +296,6 @@ def _track_tool_calls(self, result: Any) -> None:
             node = self._graph.get_node(agent_key)
             if node is None:
                 continue
-            config_tracker = node.get_config().tracker
+            config_tracker = node.get_config().create_tracker()
             if config_tracker is not None:
                 config_tracker.track_tool_call(tool_name)

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

@@ -12,13 +12,14 @@
 
 def _make_graph(enabled: bool = True) -> AgentGraphDefinition:
     """Build a minimal single-node AgentGraphDefinition for testing."""
+    node_tracker = MagicMock()
     root_config = AIAgentConfig(
         key='root-agent',
         enabled=enabled,
         model=ModelConfig(name='gpt-4'),
         provider=ProviderConfig(name='openai'),
         instructions='You are a helpful assistant.',
-        tracker=MagicMock(),
+        create_tracker=lambda: node_tracker,
     )
     graph_config = AIAgentGraphConfig(
         key='test-graph',
@@ -136,7 +137,7 @@ async def test_openai_agent_graph_runner_run_success():
     tracker.track_path.assert_called_once()
     tracker.track_latency.assert_called_once()
 
-    root_tracker = graph.get_node('root-agent').get_config().tracker
+    root_tracker = graph.get_node('root-agent').get_config().create_tracker()
     root_tracker.track_duration.assert_called_once()
     root_tracker.track_tokens.assert_called_once()
     root_tracker.track_success.assert_called_once()

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

@@ -64,7 +64,7 @@ def _make_graph(
         model=ModelConfig(name='gpt-4', parameters={'tools': tool_defs} if tool_defs else {}),
         provider=ProviderConfig(name='openai'),
         instructions='You are a helpful assistant.',
-        tracker=node_tracker,
+        create_tracker=lambda: node_tracker,
     )
 
     graph_config = AIAgentGraphConfig(
@@ -208,15 +208,15 @@ def _make_two_node_graph(mock_ld_client: MagicMock) -> AgentGraphDefinition:
         model=ModelConfig(name='gpt-4', parameters={}),
         provider=ProviderConfig(name='openai'),
         instructions='You are root.',
-        tracker=root_tracker,
+        create_tracker=lambda: root_tracker,
     )
     child_config = AIAgentConfig(
         key='child-agent',
         enabled=True,
         model=ModelConfig(name='gpt-4', parameters={}),
         provider=ProviderConfig(name='openai'),
         instructions='You are child.',
-        tracker=child_tracker,
+        create_tracker=lambda: child_tracker,
     )
 
     edge = Edge(key='root-to-child', source_config='root-agent', target_config='child-agent')