Update app.py

app.py

@@ -1,86 +1,48 @@
 import os
 import json
 import httpx
-from typing import Any
-from fastapi import FastAPI, Request, Response, HTTPException
-from fastapi.responses import StreamingResponse
+from fastapi import FastAPI, Request, HTTPException
 from mcp.server import Server
-from mcp.server.sse import SseServerTransport
+from mcp.server.sse import sse_server
 from mcp.types import Tool, TextContent
 
 # Create MCP Server
-mcp_server = Server("sap-mcp-proxy")
+server = Server("sap-mcp-proxy")
 
-# SAP API function
+# SAP function
 async def fetch_supplier_invoices():
     api_key = os.getenv("API_KEY")
     if not api_key:
-        return {"error": "SAP API_KEY not configured"}
-    
+        return {"error": "API_KEY not configured"}
     url = "https://sandbox.api.sap.com/s4hanacloud/sap/opu/odata/sap/API_SUPPLIERINVOICE_PROCESS_SRV/A_BR_SupplierInvoiceNFDocument?$top=50&$inlinecount=allpages"
     headers = {"APIKey": api_key, "Accept": "application/json"}
-    
     async with httpx.AsyncClient() as client:
         resp = await client.get(url, headers=headers)
-        return resp.json() if resp.status_code == 200 else {"error": resp.status_code, "body": resp.text}
+        return resp.json() if resp.status_code == 200 else {"error": resp.status_code}
 
-# Register MCP tools
-@mcp_server.list_tools()
+# Register tools
+@server.list_tools()
 async def list_tools() -> list[Tool]:
-    return [
-        Tool(
-            name="get_supplier_invoices",
-            description="Fetch 50 supplier invoices from SAP S/4HANA OData API",
-            inputSchema={"type": "object", "properties": {}, "required": []}
-        )
-    ]
+    return [Tool(name="get_supplier_invoices", description="Fetch supplier invoices from SAP", inputSchema={"type": "object", "properties": {}})]
 
-@mcp_server.call_tool()
+@server.call_tool()
 async def call_tool(name: str, arguments: dict) -> list[TextContent]:
     if name == "get_supplier_invoices":
         result = await fetch_supplier_invoices()
         return [TextContent(type="text", text=json.dumps(result))]
     raise ValueError(f"Unknown tool: {name}")
 
-# Create FastAPI app
-app = FastAPI(title="SAP MCP Proxy")
+# Create app using MCP's built-in SSE server helper
+app = sse_server(server)
+
+# Add auth middleware
 API_GATEWAY_KEY = os.getenv("API_GATEWAY_KEY")
 
-# Auth middleware
 @app.middleware("http")
 async def check_api_key(request: Request, call_next):
-    if request.url.path in ["/", "/docs", "/redoc", "/openapi.json"]:
+    if request.url.path in ["/", "/health"]:
         return await call_next(request)
-    token = request.headers.get("X-API-Key") or request.headers.get("Authorization", "").replace("Bearer ", "")
+    token = request.headers.get("X-API-Key")
     if not token or token != API_GATEWAY_KEY:
-        raise HTTPException(status_code=401, detail="Invalid or missing API key")
+        raise HTTPException(status_code=401, detail="Unauthorized")
     return await call_next(request)
-
-# MCP SSE endpoint
-@app.get("/sse")
-@app.post("/sse")
-async def handle_sse(request: Request):
-    async def event_generator():
-        transport = SseServerTransport("/messages")
-        async with mcp_server.run(transport.read, transport.write):
-            async for message in transport.connect():
-                yield message
-    
-    return StreamingResponse(event_generator(), media_type="text/event-stream")
-
-# MCP message endpoint
-@app.post("/messages")
-async def handle_message(request: Request):
-    body = await request.json()
-    # Process MCP JSON-RPC message
-    response = await mcp_server.handle_message(body)
-    return response
-
-# Health check
-@app.get("/")
-def root():
-    return {
-        "message": "SAP MCP Proxy is running",
-        "mcp_sse_endpoint": "/sse",
-        "mcp_messages_endpoint": "/messages"
-    }