Update gaia_api.py

gaia_api.py

@@ -1,40 +1,105 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
-from app import run_agent
+from app import run_agent  # Import from your main app
+import logging
 
-app = FastAPI()
+# Set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+app = FastAPI(
+    title="GAIA Test Agent API",
+    description="API endpoint for GAIA benchmark evaluation",
+    version="1.0.0"
+)
 
 class GaiaRequest(BaseModel):
     prompt: str
-
+    
 class GaiaResponse(BaseModel):
     output: str
 
 @app.post("/predict", response_model=GaiaResponse)
-def predict(req: GaiaRequest):
+async def predict(request: GaiaRequest):
     """
-    GAIA evaluation endpoint
-    Takes a prompt and returns the agent's response
+    Main prediction endpoint for GAIA evaluation
+    This is the endpoint that GAIA will call to get answers
     """
     try:
-        result = run_agent(req.prompt)
-        return GaiaResponse(output=result)
+        logger.info(f"Received question: {request.prompt[:100]}...")
+        
+        # Get answer from your agent
+        answer = run_agent(request.prompt)
+        
+        logger.info(f"Generated answer: {answer[:100]}...")
+        
+        return GaiaResponse(output=answer)
+        
     except Exception as e:
-        return GaiaResponse(output=f"Error: {str(e)}")
+        logger.error(f"Error processing request: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
 
 @app.get("/health")
-def health_check():
+async def health_check():
     """Health check endpoint"""
-    return {"status": "healthy"}
+    return {
+        "status": "healthy",
+        "message": "GAIA Test Agent is running"
+    }
 
 @app.get("/")
-def root():
-    """Root endpoint with basic info"""
+async def root():
+    """Root endpoint with API information"""
     return {
         "name": "GAIA Test Agent",
-        "description": "AI Navigation and General Assistant Agent",
+        "description": "AI Agent for GAIA Benchmark Evaluation",
         "endpoints": {
-            "predict": "/predict",
-            "health": "/health"
+            "predict": "/predict - Main prediction endpoint (POST)",
+            "health": "/health - Health check (GET)",
+            "docs": "/docs - Interactive API documentation (GET)"
+        },
+        "usage": {
+            "predict": {
+                "method": "POST",
+                "body": {
+                    "prompt": "Your question here"
+                },
+                "response": {
+                    "output": "Agent's answer"
+                }
+            }
         }
-    }
+    }
+
+@app.get("/info")
+async def info():
+    """Get agent information"""
+    return {
+        "agent_type": "General AI Assistant",
+        "model": "DeepSeek V3.1 Terminus via OpenRouter",
+        "capabilities": [
+            "General question answering",
+            "Mathematical calculations", 
+            "Factual queries",
+            "Yes/No questions",
+            "Reasoning tasks"
+        ],
+        "optimized_for": "GAIA benchmark evaluation"
+    }
+
+# For debugging - remove in production
+@app.get("/test")
+async def test_endpoint():
+    """Test endpoint to verify the agent works"""
+    try:
+        test_answer = run_agent("What is 2 + 2?")
+        return {
+            "test_question": "What is 2 + 2?",
+            "test_answer": test_answer,
+            "status": "Agent working correctly"
+        }
+    except Exception as e:
+        return {
+            "status": "Error",
+            "error": str(e)
+        }