tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-13 06:23:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix test

Browse Source
This commit is contained in:
Ishaan Jaff
2025-03-29 17:11:46 -07:00
parent 3e378f2bec
commit 4e106ce217
1 changed files with 9 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/mcp_tests/test_mcp_server.py
+9 -12
View File
@@ -7,21 +7,24 @@ sys.path.insert(
0, os.path.abspath("../../..")
) # Adds the parent directory to the system path
from litellm.proxy._experimental.mcp_server.mcp_client_manager import (
from litellm.proxy._experimental.mcp_server.mcp_server_manager import (
MCPServerManager,
MCPSSEServer,
)
MCP_SERVERS = [
MCPSSEServer(name="zapier_mcp_server", url=os.environ.get("ZAPIER_MCP_SERVER_URL")),
]
mcp_server_manager = MCPServerManager(mcp_servers=MCP_SERVERS)
mcp_server_manager = MCPServerManager()
@pytest.mark.asyncio
async def test_mcp_server_manager():
mcp_server_manager.load_servers_from_config(
{
"zapier_mcp_server": {
"url": os.environ.get("ZAPIER_MCP_SERVER_URL"),
}
}
)
tools = await mcp_server_manager.list_tools()
print("TOOLS FROM MCP SERVER MANAGER== ", tools)
@@ -29,9 +32,3 @@ async def test_mcp_server_manager():
name="gmail_send_email", arguments={"body": "Test"}
)
print("RESULT FROM CALLING TOOL FROM MCP SERVER MANAGER== ", result)
"""
TODO test with multiple MCP servers and calling a specific
"""