Fix: Use query parameters instead of body for API endpoints

env/server.py

@@ -5,7 +5,7 @@ Provides REST and WebSocket endpoints for HuggingFace Spaces deployment.
 """
 
 from fastapi import FastAPI, WebSocket, WebSocketDisconnect
-from pydantic import BaseModel
+from pydantic import BaseModel, Field
 from typing import Optional, Dict, Any
 import json
 import asyncio
@@ -19,7 +19,7 @@ envs: Dict[str, SQLAnalystEnv] = {}
 
 
 class ResetRequest(BaseModel):
-    task_id: str = "monthly_signups"
+    task_id: str = Field(default="monthly_signups")
 
 
 class StepRequest(BaseModel):
@@ -42,10 +42,10 @@ async def root():
 
 
 @app.post("/reset")
-async def reset(req: ResetRequest) -> Dict[str, Any]:
-    session_id = req.task_id
+async def reset(task_id: str = "monthly_signups") -> Dict[str, Any]:
+    session_id = task_id
 
-    env = SQLAnalystEnv(task_id=req.task_id)
+    env = SQLAnalystEnv(task_id=task_id)
     result = env.reset()
     envs[session_id] = env
 
@@ -65,15 +65,17 @@ async def reset(req: ResetRequest) -> Dict[str, Any]:
 
 
 @app.post("/step")
-async def step(req: StepRequest) -> Dict[str, Any]:
-    session_id = req.session_id
-
+async def step(
+    session_id: str,
+    sql_query: Optional[str] = None,
+    submit_answer: Optional[str] = None,
+) -> Dict[str, Any]:
     if session_id not in envs:
         return {"error": "Session not found. Call /reset first."}
 
     env = envs[session_id]
 
-    action = Action(sql_query=req.sql_query, submit_answer=req.submit_answer)
+    action = Action(sql_query=sql_query, submit_answer=submit_answer)
 
     result = env.step(action)
 
@@ -108,9 +110,7 @@ async def step(req: StepRequest) -> Dict[str, Any]:
 
 
 @app.post("/state")
-async def state(req: StateRequest) -> Dict[str, Any]:
-    session_id = req.session_id
-
+async def state(session_id: str) -> Dict[str, Any]:
     if session_id not in envs:
         return {"error": "Session not found. Call /reset first."}
 
@@ -129,9 +129,7 @@ async def state(req: StateRequest) -> Dict[str, Any]:
 
 
 @app.post("/delete")
-async def delete_session(req: StateRequest) -> Dict[str, str]:
-    session_id = req.session_id
-
+async def delete_session(session_id: str) -> Dict[str, str]:
     if session_id in envs:
         del envs[session_id]
         return {"status": "deleted", "session_id": session_id}