Fix uvicorn to run for all types of transport
This commit is contained in:
parent
ef177147c4
commit
d0945d99a3
1 changed files with 6 additions and 17 deletions
|
|
@ -105,23 +105,12 @@ def main():
|
||||||
)
|
)
|
||||||
|
|
||||||
# Run the server with the modified app (with middleware)
|
# Run the server with the modified app (with middleware)
|
||||||
# Note: We cannot use mcp.run() as it creates a new app instance without our middleware
|
|
||||||
if settings.TRANSPORT == "streamable-http":
|
|
||||||
# Run uvicorn directly with our middleware-enhanced app
|
|
||||||
uvicorn.run(
|
uvicorn.run(
|
||||||
mcp_app,
|
mcp_app,
|
||||||
host=settings.HOST,
|
host=settings.HOST,
|
||||||
port=settings.PORT,
|
port=settings.PORT,
|
||||||
log_level=settings.LOG_LEVEL.lower(),
|
log_level=settings.LOG_LEVEL.lower(),
|
||||||
)
|
)
|
||||||
else:
|
|
||||||
# For other transports (SSE), use the default run method
|
|
||||||
# Note: SSE transport may not support custom middleware
|
|
||||||
logger.warning(
|
|
||||||
"Using mcp.run() for non-streamable-http transport. "
|
|
||||||
"CORS middleware may not be applied."
|
|
||||||
)
|
|
||||||
mcp.run(transport=settings.TRANSPORT)
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue