endpoint changed

app.py

@@ -13,6 +13,8 @@ from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import StreamingResponse
 from pydantic import BaseModel
 import uvicorn
+import asyncio
+import json
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -319,5 +321,28 @@ async def list_docs():
     
     return {"documents": docs_data}
 
+@app.get("/mcp/sse")
+async def mcp_sse():
+    """SSE endpoint for MCP communication"""
+    async def event_generator():
+        # Send initial connection event
+        yield f"data: {json.dumps({'type': 'connection', 'status': 'connected'})}\n\n"
+        
+        # Keep connection alive
+        while True:
+            await asyncio.sleep(30)  # Send heartbeat every 30 seconds
+            yield f"data: {json.dumps({'type': 'heartbeat', 'timestamp': asyncio.get_event_loop().time()})}\n\n"
+    
+    return StreamingResponse(
+        event_generator(),
+        media_type="text/event-stream",
+        headers={
+            "Cache-Control": "no-cache",
+            "Connection": "keep-alive",
+            "Access-Control-Allow-Origin": "*",
+            "Access-Control-Allow-Headers": "*",
+        }
+    )
+
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)