Update app.py

app.py

@@ -1,48 +1,89 @@
-import os
-import json
+from mcp.server.fastmcp import FastMCP
 import httpx
-from fastapi import FastAPI, Request, HTTPException
-from mcp.server import Server
-from mcp.server.sse import sse_server
-from mcp.types import Tool, TextContent
-
-# Create MCP Server
-server = Server("sap-mcp-proxy")
-
-# SAP function
-async def fetch_supplier_invoices():
-    api_key = os.getenv("API_KEY")
-    if not api_key:
-        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}
+import os
 
-# Register tools
-@server.list_tools()
-async def list_tools() -> list[Tool]:
-    return [Tool(name="get_supplier_invoices", description="Fetch supplier invoices from SAP", inputSchema={"type": "object", "properties": {}})]
+# SAP API Configuration
+SAP_API_URL = "https://sandbox.api.sap.com/s4hanacloud/sap/opu/odata4/sap/api_purchaserequisition_2/srvd_a2x/sap/purchaserequisition/0001/PurchaseReqn"
+SAP_API_KEY = os.getenv("SAP_API_KEY", "your_sap_api_key_here")
 
-@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}")
+# Initialize FastMCP server
+mcp = FastMCP("SAP_PurchaseRequisition")
 
-# Create app using MCP's built-in SSE server helper
-app = sse_server(server)
+@mcp.tool()
+async def get_purchase_requisitions(top: int = 50) -> dict:
+    """
+    Fetch purchase requisitions from SAP S/4HANA Cloud API.
+    
+    Args:
+        top: Number of records to fetch (default 50, max 100)
+    
+    Returns:
+        Dictionary containing purchase requisition data
+    """
+    headers = {
+        "APIKey": SAP_API_KEY,
+        "Accept": "application/json"
+    }
+    
+    params = {
+        "$top": min(top, 100)  # Limit to max 100 records
+    }
+    
+    async with httpx.AsyncClient(timeout=30.0) as client:
+        try:
+            response = await client.get(
+                SAP_API_URL,
+                headers=headers,
+                params=params
+            )
+            response.raise_for_status()
+            data = response.json()
+            return {
+                "status": "success",
+                "data": data,
+                "count": len(data.get("value", []))
+            }
+        except httpx.HTTPStatusError as e:
+            return {
+                "status": "error",
+                "message": f"HTTP error occurred: {e.response.status_code}",
+                "details": str(e)
+            }
+        except Exception as e:
+            return {
+                "status": "error",
+                "message": "Failed to fetch purchase requisitions",
+                "details": str(e)
+            }
 
-# Add auth middleware
-API_GATEWAY_KEY = os.getenv("API_GATEWAY_KEY")
+@mcp.tool()
+async def get_purchase_requisition_summary(top: int = 10) -> str:
+    """
+    Get a summary of recent purchase requisitions.
+    
+    Args:
+        top: Number of records to summarize (default 10)
+    
+    Returns:
+        Formatted summary string
+    """
+    result = await get_purchase_requisitions(top)
+    
+    if result["status"] == "error":
+        return f"Error: {result['message']}"
+    
+    items = result["data"].get("value", [])
+    if not items:
+        return "No purchase requisitions found."
+    
+    summary = f"Found {result['count']} purchase requisitions:\n\n"
+    for idx, item in enumerate(items, 1):
+        pr_number = item.get("PurchaseRequisition", "N/A")
+        description = item.get("PurchaseRequisitionDescription", "No description")
+        summary += f"{idx}. PR#{pr_number}: {description}\n"
+    
+    return summary
 
-@app.middleware("http")
-async def check_api_key(request: Request, call_next):
-    if request.url.path in ["/", "/health"]:
-        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="Unauthorized")
-    return await call_next(request)
+if __name__ == "__main__":
+    # Run with streamable HTTP transport for Hugging Face Spaces
+    mcp.run(transport="streamable-http", port=7860)