aakashdg commited on
Commit
864b90d
·
verified ·
1 Parent(s): 62a372d

testing SSE fix

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -20,6 +20,8 @@ from src.servers.elevation import ElevationServer
20
  from src.servers.pests import PestsServer
21
 
22
  from mcp_server import sse_app
 
 
23
 
24
 
25
  # ============================================================================
@@ -344,9 +346,10 @@ app.add_middleware(
344
  allow_headers=["*"],
345
  )
346
 
 
347
 
348
  # Exposing endpoint for sse
349
- app.mount("/mcp", sse_app)
350
 
351
  # ============================================================================
352
  # API ENDPOINTS
 
20
  from src.servers.pests import PestsServer
21
 
22
  from mcp_server import sse_app
23
+ from mcp.server.transport import SSETransport
24
+ from starlette.applications import Starlette
25
 
26
 
27
  # ============================================================================
 
346
  allow_headers=["*"],
347
  )
348
 
349
+ sse_transport = SSETransport()
350
 
351
  # Exposing endpoint for sse
352
+ app.mount("/mcp", sse_transport.app)
353
 
354
  # ============================================================================
355
  # API ENDPOINTS