pgurazada1 commited on
Commit
7737dc2
·
verified ·
1 Parent(s): bedf4dd

Update server.py

Browse files
Files changed (1) hide show
  1. server.py +4 -3
server.py CHANGED
@@ -224,8 +224,9 @@ async def list_tools(request: Request) -> JSONResponse:
224
  """Return all registered tool metadata as JSON."""
225
  return JSONResponse(tool_registry)
226
 
227
- # Add the Hugging Face token auth middleware
228
- mcp.streamable_http_app.add_middleware(HuggingFaceTokenAuthMiddleware)
 
229
 
230
  if __name__ == "__main__":
231
- uvicorn.run(mcp.streamable_http_app, host="0.0.0.0", port=8000)
 
224
  """Return all registered tool metadata as JSON."""
225
  return JSONResponse(tool_registry)
226
 
227
+ # Add the Hugging Face token auth middleware at app construction time
228
+ app = mcp.streamable_http_app()
229
+ app.add_middleware(HuggingFaceTokenAuthMiddleware)
230
 
231
  if __name__ == "__main__":
232
+ uvicorn.run(app, host="0.0.0.0", port=8000)