Update app.py

app.py

@@ -1,53 +1,62 @@
 import os
 import httpx
+import asyncio
+from mcp.server.fastapi import FastMCPServer
+from mcp.server import Server
+from mcp.types import Tool, TextContent
 from fastapi import FastAPI, Request, HTTPException
-from fastmcp import FastMCP
 
-# Create separate FastAPI app for our endpoints
-app = FastAPI()
+# Initialize MCP Server
+mcp_server = Server("sap-mcp-proxy")
+
+# SAP API helper
+async def fetch_supplier_invoices():
+    api_key = os.getenv("API_KEY")
+    if not api_key:
+        return {"error": "SAP 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}
+
+# Register MCP tool
+@mcp_server.list_tools()
+async def list_tools():
+    return [
+        Tool(
+            name="get_supplier_invoices",
+            description="Fetch 50 supplier invoices from SAP S/4HANA OData API",
+            inputSchema={"type": "object", "properties": {}, "required": []}
+        )
+    ]
 
-# Create MCP server
-mcp = FastMCP(name="SAP MCP Proxy", version="0.1")
+@mcp_server.call_tool()
+async def call_tool(name: str, arguments: dict):
+    if name == "get_supplier_invoices":
+        result = await fetch_supplier_invoices()
+        return [TextContent(type="text", text=str(result))]
+    raise ValueError(f"Unknown tool: {name}")
 
+# Create FastAPI app with MCP endpoints
+app = FastAPI()
+mcp_app = FastMCPServer(mcp_server, path="/mcp")
+app.mount("/mcp", mcp_app)
+
+# Auth middleware
 API_GATEWAY_KEY = os.getenv("API_GATEWAY_KEY")
 
-# Authentication middleware
 @app.middleware("http")
 async def check_api_key(request: Request, call_next):
-    # Allow unauthenticated access to health check and docs
-    if request.url.path in ["/", "/docs", "/redoc", "/routes"]:
+    if request.url.path in ["/", "/docs"]:
         return await call_next(request)
     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")
-    response = await call_next(request)
-    return response
+    return await call_next(request)
 
-# Define MCP tool
-@mcp.tool(description="Fetch 50 supplier invoices from SAP S/4HANA OData API")
-async def get_supplier_invoices():
-    api_key = os.getenv("API_KEY")
-    if not api_key:
-        return {"error": "SAP 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)
-        if resp.status_code == 200:
-            return resp.json()
-        else:
-            return {"error": resp.status_code, "body": resp.text}
-
-# Health check endpoint
 @app.get("/")
 def root():
-    return {"message": "SAP MCP Proxy is running"}
-
-# Debug endpoint
-@app.get("/routes")
-def get_routes():
-    return [route.path for route in app.routes]
+    return {"message": "SAP MCP Proxy is running", "mcp_endpoint": "/mcp"}