Create gaia_api.py

gaia_api.py

@@ -0,0 +1,40 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from app import run_agent
+
+app = FastAPI()
+
+class GaiaRequest(BaseModel):
+    prompt: str
+
+class GaiaResponse(BaseModel):
+    output: str
+
+@app.post("/predict", response_model=GaiaResponse)
+def predict(req: GaiaRequest):
+    """
+    GAIA evaluation endpoint
+    Takes a prompt and returns the agent's response
+    """
+    try:
+        result = run_agent(req.prompt)
+        return GaiaResponse(output=result)
+    except Exception as e:
+        return GaiaResponse(output=f"Error: {str(e)}")
+
+@app.get("/health")
+def health_check():
+    """Health check endpoint"""
+    return {"status": "healthy"}
+
+@app.get("/")
+def root():
+    """Root endpoint with basic info"""
+    return {
+        "name": "GAIA Test Agent",
+        "description": "AI Navigation and General Assistant Agent",
+        "endpoints": {
+            "predict": "/predict",
+            "health": "/health"
+        }
+    }