Add missing /v1/ground/chat/completions endpoint for Agent-S

app.py

@@ -1,4 +1,3 @@
-# app.py - Add Custom Endpoint for Agent-S
 import gradio as gr
 from transformers import AutoProcessor, AutoModel
 import torch
@@ -123,17 +122,18 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# Custom endpoint specifically for Agent-S
-@app.post("/v1/ground")
-async def agent_s_grounding(request: Request):
+# CRITICAL: Add the missing endpoint that Agent-S expects
+@app.post("/v1/ground/chat/completions")
+async def chat_completions(request: Request):
     """
-    Custom endpoint specifically designed for Agent-S
+    Chat completions endpoint that Agent-S expects
     """
     try:
         # Parse the request body
         body = await request.json()
         
-        # Agent-S typically sends data in this format
+        # Extract image and prompt from the request
+        # Agent-S might send data in different formats
         if "data" in body and len(body["data"]) >= 2:
             image = body["data"][0]  # First element is image
             prompt = body["data"][1]  # Second element is prompt
@@ -157,21 +157,26 @@ async def agent_s_grounding(request: Request):
             content={"error": f"Internal server error: {str(e)}", "status": "failed"}
         )
 
-# Alternative endpoint names for compatibility
+# Keep existing endpoints for compatibility
+@app.post("/v1/ground")
+async def agent_s_grounding(request: Request):
+    """Custom endpoint specifically designed for Agent-S"""
+    return await chat_completions(request)
+
 @app.post("/api/ground")
 async def api_ground(request: Request):
     """Alternative endpoint name for compatibility"""
-    return await agent_s_grounding(request)
+    return await chat_completions(request)
 
 @app.post("/predict")
 async def predict(request: Request):
     """Alternative endpoint name for compatibility"""
-    return await agent_s_grounding(request)
+    return await chat_completions(request)
 
 @app.post("/")
 async def root_endpoint(request: Request):
     """Root endpoint for compatibility"""
-    return await agent_s_grounding(request)
+    return await chat_completions(request)
 
 # Create Gradio interface
 iface = gr.Interface(