Skip to content

Commit 233c5e8

Browse files
Update test_foundry_integration.py
1 parent d29df6f commit 233c5e8

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

src/tests/agents/test_foundry_integration.py

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,11 +3,19 @@
33
Tests Bing search, RAG, MCP tools, and Code Interpreter capabilities.
44
"""
55
# pylint: disable=E0401, E0611, C0413
6+
7+
import sys
8+
from pathlib import Path
9+
610
import pytest
711

12+
# Add the backend path to sys.path so we can import v3 modules
13+
backend_path = Path(__file__).parent.parent.parent / "backend"
14+
sys.path.insert(0, str(backend_path))
15+
816
# Now import from the v3 package
9-
from src.backend.v3.magentic_agents.foundry_agent import FoundryAgentTemplate
10-
from src.backend.v3.magentic_agents.models.agent_models import (MCPConfig,
17+
from v3.magentic_agents.foundry_agent import FoundryAgentTemplate
18+
from v3.magentic_agents.models.agent_models import (BingConfig, MCPConfig,
1119
SearchConfig)
1220

1321

@@ -266,4 +274,4 @@ async def test_multiple_capabilities_together(self):
266274

267275
if __name__ == "__main__":
268276
"""Run the tests directly for debugging."""
269-
pytest.main([__file__, "-v", "-s"])
277+
pytest.main([__file__, "-v", "-s"])

0 commit comments

Comments
 (0)