Update main.py

main.py

@@ -10,60 +10,68 @@ from agent import run_agent_query
 
 app = FastAPI(title="CFO Bot - Agent API")
 
-# Allow local dev; modify in production
+# ---------------------------------------------------------
+# CORS (adjust for production)
+# ---------------------------------------------------------
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],  # replace with exact domain in production
+    allow_origins=["*"],  # replace with your frontend domain
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
 )
 
-# -------------------------------
-# Models
-# -------------------------------
-
+# ---------------------------------------------------------
+# Request / Response Models
+# ---------------------------------------------------------
 class QueryRequest(BaseModel):
     query: str
-    thread_id: str | None = None
+    thread_id: str  # REQUIRED — not optional
 
 
 class QueryResponse(BaseModel):
     output: str
 
 
-# -------------------------------
-# Routes
-# -------------------------------
-
+# ---------------------------------------------------------
+# Health check
+# ---------------------------------------------------------
 @app.get("/")
 def root():
     return {"status": "CFO Bot API running", "version": "1.0.0"}
 
 
+# ---------------------------------------------------------
+# Query Route
+# ---------------------------------------------------------
 @app.post("/query", response_model=QueryResponse)
 async def query_endpoint(req: QueryRequest):
 
-    if not req.query or req.query.strip() == "":
+    # Validate query
+    if not req.query or not req.query.strip():
         raise HTTPException(status_code=400, detail="Query cannot be empty.")
 
+    # Validate thread_id
+    if not req.thread_id or not req.thread_id.strip():
+        raise HTTPException(status_code=400, detail="thread_id is required.")
+
     try:
-        # Call the agent runner
+        # Run agent — NO DEFAULT thread_id
         result = await run_agent_query(
             req.query,
-            thread_id=req.thread_id or "default"
+            thread_id=req.thread_id
         )
 
     except Exception as exc:
-        raise HTTPException(status_code=500, detail=f"Agent failed: {exc}")
+        raise HTTPException(status_code=500, detail=f"Agent failed: {str(exc)}")
 
-    # Extract final output from agent SDK result
+    # Extract output properly
     final_output = (
         getattr(result, "final_output", None)
         or (result.get("final_output") if isinstance(result, dict) else None)
     )
 
-    if final_output is None:
+    if not final_output:
         final_output = "No response generated."
 
     return QueryResponse(output=final_output)