|
37 | 37 | invalid_mermaid_diagram_hard,
|
38 | 38 | valid_mermaid_diagram,
|
39 | 39 | )
|
40 |
| -from mcp_servers.mermaid_validator import validate_mermaid_diagram |
| 40 | +from agents_mcp_usage.utils import get_mcp_server_path |
| 41 | +import sys |
| 42 | +import importlib.util |
41 | 43 |
|
42 | 44 | load_dotenv()
|
43 | 45 |
|
|
57 | 59 | BASE_RETRY_DELAY = 1.0 # seconds
|
58 | 60 | MAX_RETRY_DELAY = 30.0 # seconds
|
59 | 61 |
|
| 62 | +# Dynamically import validate_mermaid_diagram from the mermaid_validator.py file |
| 63 | +mermaid_validator_path = str(get_mcp_server_path("mermaid_validator.py")) |
| 64 | +spec = importlib.util.spec_from_file_location("mermaid_validator", mermaid_validator_path) |
| 65 | +mermaid_validator = importlib.util.module_from_spec(spec) |
| 66 | +sys.modules["mermaid_validator"] = mermaid_validator |
| 67 | +spec.loader.exec_module(mermaid_validator) |
| 68 | +validate_mermaid_diagram = mermaid_validator.validate_mermaid_diagram |
| 69 | + |
60 | 70 | # ============================================================================
|
61 | 71 | # Retry Utilities
|
62 | 72 | # ============================================================================
|
@@ -177,15 +187,15 @@ def get_mcp_servers() -> List[MCPServerStdio]:
|
177 | 187 | command="uv",
|
178 | 188 | args=[
|
179 | 189 | "run",
|
180 |
| - "mcp_servers/example_server.py", |
| 190 | + str(get_mcp_server_path("example_server.py")), |
181 | 191 | "stdio",
|
182 | 192 | ],
|
183 | 193 | )
|
184 | 194 | mermaid_server = MCPServerStdio(
|
185 | 195 | command="uv",
|
186 | 196 | args=[
|
187 | 197 | "run",
|
188 |
| - "mcp_servers/mermaid_validator.py", |
| 198 | + str(get_mcp_server_path("mermaid_validator.py")), |
189 | 199 | ],
|
190 | 200 | )
|
191 | 201 | return [local_server, mermaid_server]
|
|
0 commit comments