|
2 | 2 | from ldclient import Config, Context, LDClient |
3 | 3 | from ldclient.integrations.test_data import TestData |
4 | 4 |
|
| 5 | +from unittest.mock import MagicMock |
| 6 | + |
5 | 7 | from ldai import ( |
6 | 8 | LDAIClient, |
7 | 9 | AIAgentGraphConfig, |
@@ -268,13 +270,17 @@ def test_agent_graph_build_nodes(ldai_client: LDAIClient): |
268 | 270 | ai_graph_config, |
269 | 271 | { |
270 | 272 | "customer-support-agent": AIAgentConfig( |
271 | | - key="customer-support-agent", enabled=True |
| 273 | + key="customer-support-agent", enabled=True, create_tracker=MagicMock(), |
| 274 | + ), |
| 275 | + "personalized-agent": AIAgentConfig( |
| 276 | + key="personalized-agent", enabled=True, create_tracker=MagicMock(), |
272 | 277 | ), |
273 | | - "personalized-agent": AIAgentConfig(key="personalized-agent", enabled=True), |
274 | 278 | "multi-context-agent": AIAgentConfig( |
275 | | - key="multi-context-agent", enabled=True |
| 279 | + key="multi-context-agent", enabled=True, create_tracker=MagicMock(), |
| 280 | + ), |
| 281 | + "minimal-agent": AIAgentConfig( |
| 282 | + key="minimal-agent", enabled=True, create_tracker=MagicMock(), |
276 | 283 | ), |
277 | | - "minimal-agent": AIAgentConfig(key="minimal-agent", enabled=True), |
278 | 284 | }, |
279 | 285 | ) |
280 | 286 |
|
|
0 commit comments