aljaz/mcp-python-sdk
1
0
Fork 0
mirror of https://github.com/aljazceru/mcp-python-sdk.git synced 2025-12-19 14:54:24 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #172 from sheffler/fixAsyncLog

Browse Source 
async and await for send_log_message in fastmcp
This commit is contained in:
David Soria Parra
2025-01-27 10:40:10 +00:00
committed by GitHub
parent 57c3aa2f0d 7a64d8f9af
commit 91840fd9c0
1 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/mcp/server/fastmcp/server.py
View File

@@ -620,7 +620,7 @@ class Context(BaseModel):
), "Context is not available outside of a request" ), "Context is not available outside of a request"
return await self._fastmcp.read_resource(uri) return await self._fastmcp.read_resource(uri)
def log( async def log(
self, self,
level: Literal["debug", "info", "warning", "error"], level: Literal["debug", "info", "warning", "error"],
message: str, message: str,
@@ -635,7 +635,7 @@ class Context(BaseModel):
logger_name: Optional logger name logger_name: Optional logger name
**extra: Additional structured data to include **extra: Additional structured data to include
""" """
self.request_context.session.send_log_message( await self.request_context.session.send_log_message(
level=level, data=message, logger=logger_name level=level, data=message, logger=logger_name
) )
@@ -659,18 +659,18 @@ class Context(BaseModel):
return self.request_context.session return self.request_context.session
# Convenience methods for common log levels # Convenience methods for common log levels
def debug(self, message: str, **extra: Any) -> None: async def debug(self, message: str, **extra: Any) -> None:
"""Send a debug log message.""" """Send a debug log message."""
self.log("debug", message, **extra) await self.log("debug", message, **extra)
def info(self, message: str, **extra: Any) -> None: async def info(self, message: str, **extra: Any) -> None:
"""Send an info log message.""" """Send an info log message."""
self.log("info", message, **extra) await self.log("info", message, **extra)
def warning(self, message: str, **extra: Any) -> None: async def warning(self, message: str, **extra: Any) -> None:
"""Send a warning log message.""" """Send a warning log message."""
self.log("warning", message, **extra) await self.log("warning", message, **extra)
def error(self, message: str, **extra: Any) -> None: async def error(self, message: str, **extra: Any) -> None:
"""Send an error log message.""" """Send an error log message."""
self.log("error", message, **extra) await self.log("error", message, **extra)