Create main.py

main.py

@@ -0,0 +1,57 @@
+# main.py
+import os
+import asyncio
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+from fastapi.middleware.cors import CORSMiddleware
+
+from agent import run_agent_query
+
+app = FastAPI(title="CFO Bot - Agent API")
+
+# Allow local dev; lock down in production
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # change to your frontend origin
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+class QueryRequest(BaseModel):
+    query: str
+    thread_id: str | None = None
+    max_steps: int | None = 4
+
+class QueryResponse(BaseModel):
+    output: str
+    tool_calls: dict | None = None
+    session_id: str | None = None
+
+@app.post("/query", response_model=QueryResponse)
+async def query_endpoint(req: QueryRequest):
+    if not req.query or req.query.strip() == "":
+        raise HTTPException(status_code=400, detail="Query cannot be empty.")
+    # run agent
+    try:
+        result = await run_agent_query(req.query, thread_id=req.thread_id or "default", max_steps=req.max_steps or 4)
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Agent run failed: {e}")
+
+    # Extract useful fields from result, be resilient if SDK result shape differs
+    final_output = getattr(result, "final_output", None) or result.get("final_output") if isinstance(result, dict) else None
+    tool_calls = getattr(result, "tool_calls", None) or result.get("tool_calls") if isinstance(result, dict) else None
+
+    # Fallback: result may be an object with str() representation
+    if final_output is None:
+        try:
+            final_output = str(result)
+        except Exception:
+            final_output = "No output."
+
+    return QueryResponse(output=final_output, tool_calls=tool_calls, session_id=req.thread_id or "default")
+
+# Run with: uvicorn main:app --host 0.0.0.0 --port 8080
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run("main:app", host="0.0.0.0", port=int(os.environ.get("PORT", 8080)), log_level="info")