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)
|
||||
# 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(
|
||||
mcp_app,
|
||||
host=settings.HOST,
|
||||
port=settings.PORT,
|
||||
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)
|
||||
uvicorn.run(
|
||||
mcp_app,
|
||||
host=settings.HOST,
|
||||
port=settings.PORT,
|
||||
log_level=settings.LOG_LEVEL.lower(),
|
||||
)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue