mcp asyncio call loop error

pull/1065/head
Aksh Parekh 2 months ago
parent 7b6b7c4980
commit 6a97d346d5

@ -117,4 +117,4 @@ def get_okx_crypto_volume(symbol: str) -> str:
if __name__ == "__main__": if __name__ == "__main__":
# Run the server on port 8000 (you can change this to any available port) # Run the server on port 8000 (you can change this to any available port)
mcp.run(transport="sse") mcp.run(transport="streamable-http")

@ -11,7 +11,6 @@ from typing import Any, Dict, List, Literal, Optional, Union
from litellm.types.utils import ChatCompletionMessageToolCall from litellm.types.utils import ChatCompletionMessageToolCall
from loguru import logger from loguru import logger
from mcp import ClientSession from mcp import ClientSession
from mcp.client.sse import sse_client
try: try:
from mcp.client.streamable_http import streamablehttp_client from mcp.client.streamable_http import streamablehttp_client
@ -315,7 +314,7 @@ def get_mcp_client(transport, url, headers=None, timeout=5, **kwargs):
url, headers=headers, timeout=timeout, **kwargs url, headers=headers, timeout=timeout, **kwargs
) )
else: else:
return sse_client( return streamablehttp_client(
url, headers=headers, timeout=timeout, **kwargs url, headers=headers, timeout=timeout, **kwargs
) )

Loading…
Cancel
Save