modelcontextprotocol/python-sdk
Watch
1
0
Fork
You've already forked python-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix typing issues

Browse source
This commit is contained in:
David Soria Parra 2025-01-23 13:25:21 +00:00
commit c14ec2e0d7
No known key found for this signature in database
2 changed files with 10 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

13
examples/clients/simple-chatbot/mcp_simple_chatbot/main.py
View file

@ -91,10 +91,15 @@ class Server:
else None, else None,
) )
try: try:
stdio_transport = await self.exit_stack.enter_async_context(stdio_client(server_params)) stdio_transport = await self.exit_stack.enter_async_context(
stdio_client(server_params)
)
read, write = stdio_transport read, write = stdio_transport
self.session = await self.exit_stack.enter_async_context(ClientSession(read, write)) session = await self.exit_stack.enter_async_context(
await self.session.initialize() ClientSession(read, write)
)
await session.initialize()
self.session = session
except Exception as e: except Exception as e:
logging.error(f"Error initializing server {self.name}: {e}") logging.error(f"Error initializing server {self.name}: {e}")
await self.cleanup() await self.cleanup()
@ -257,7 +262,7 @@ class LLMClient:
error_message = f"Error getting LLM response: {str(e)}" error_message = f"Error getting LLM response: {str(e)}"
logging.error(error_message) logging.error(error_message)
if hasattr(e, 'response'): if isinstance(e, httpx.HTTPStatusError):
status_code = e.response.status_code status_code = e.response.status_code
logging.error(f"Status code: {status_code}") logging.error(f"Status code: {status_code}")
logging.error(f"Response details: {e.response.text}") logging.error(f"Response details: {e.response.text}")

2
examples/clients/simple-chatbot/pyproject.toml
View file

@ -45,4 +45,4 @@ line-length = 88
target-version = "py310" target-version = "py310"
[tool.uv] [tool.uv]
dev-dependencies = ["pyright>=1.1.378", "pytest>=8.3.3", "ruff>=0.6.9"] dev-dependencies = ["pyright>=1.1.379", "pytest>=8.3.3", "ruff>=0.6.9"]