bring logging in line with rest of quickstart

This commit is contained in:
David Soria Parra 2024-11-21 20:53:20 +00:00
commit 0cb430ab3d
No known key found for this signature in database

View file

@ -626,28 +626,24 @@ uv run pyright src
<Step title="Add logging support"> <Step title="Add logging support">
```python ```python
import logging # Set up logging
logger = logging.getLogger("weather-server")
logger.setLevel(logging.INFO)
class WeatherServer(Server): @app.set_logging_level()
def __init__(self): async def set_logging_level(level: LoggingLevel) -> EmptyResult:
super().__init__("example-weather-server") logger.setLevel(level.upper())
self.logger = logging.getLogger("weather-server") await app.request_context.session.send_log_message(
self.logger.setLevel(logging.INFO) level="info",
data=f"Log level set to {level}",
logger="weather-server"
)
return EmptyResult()
@self.set_logging_level() # Use logger throughout the code
async def set_logging_level(self, level: LoggingLevel) -> EmptyResult: # For example:
self.logger.setLevel(level.upper()) # logger.info("Weather data fetched successfully")
await self.request_context.session.send_log_message( # logger.error(f"Error fetching weather data: {str(e)}")
level="info",
data=f"Log level set to {level}",
logger="weather-server"
)
return EmptyResult()
# Use self.logger for logging throughout the class methods
# For example:
# self.logger.info("Weather data fetched successfully")
# self.logger.error(f"Error fetching weather data: {str(e)}")
``` ```
</Step> </Step>
@ -803,4 +799,4 @@ uv run pyright src
<Card title="Python SDK" icon="python" href="/api-reference/python/class-reference"> <Card title="Python SDK" icon="python" href="/api-reference/python/class-reference">
Read more about the Python SDK Read more about the Python SDK
</Card> </Card>
</CardGroup> </CardGroup>