mcp change

app.py

@@ -321,16 +321,53 @@ async def list_docs():
     
     return {"documents": docs_data}
 
-@app.get("/mcp/sse")
+@app.get("/mcp")
 async def mcp_sse():
-    """SSE endpoint for MCP communication"""
+    """MCP SSE endpoint for mcp-remote"""
     async def event_generator():
-        # Send initial connection event
-        yield f"data: {json.dumps({'type': 'connection', 'status': 'connected'})}\n\n"
+        # Send initial MCP server info
+        yield f"data: {json.dumps({'type': 'mcp_server', 'name': 'mcp-docs-server', 'version': '1.0.0'})}\n\n"
+        
+        # Send available tools
+        tools = [
+            {
+                "name": "search_docs",
+                "description": "Search through MCP documentation chunks",
+                "inputSchema": {
+                    "type": "object",
+                    "properties": {
+                        "query": {"type": "string", "description": "Search query"},
+                        "limit": {"type": "integer", "description": "Maximum number of results", "default": 5}
+                    },
+                    "required": ["query"]
+                }
+            },
+            {
+                "name": "get_chunk",
+                "description": "Get a specific documentation chunk by ID",
+                "inputSchema": {
+                    "type": "object",
+                    "properties": {
+                        "chunk_id": {"type": "string", "description": "Chunk ID to retrieve"}
+                    },
+                    "required": ["chunk_id"]
+                }
+            },
+            {
+                "name": "list_docs",
+                "description": "List all available documents",
+                "inputSchema": {
+                    "type": "object",
+                    "properties": {}
+                }
+            }
+        ]
+        
+        yield f"data: {json.dumps({'type': 'tools', 'tools': tools})}\n\n"
         
         # Keep connection alive
         while True:
-            await asyncio.sleep(30)  # Send heartbeat every 30 seconds
+            await asyncio.sleep(30)
             yield f"data: {json.dumps({'type': 'heartbeat', 'timestamp': asyncio.get_event_loop().time()})}\n\n"
     
     return StreamingResponse(