langgraph

.gitignore

@@ -20,5 +20,4 @@ app/__pycache__/
 pnlbs/__pycache__/
 AGENT_GUIDE.md
 docker-compose.dev.yml
-agents/financial_agents.py
-agents/financial_tools.py
+file_cleanup.py

agents/langgraph_routes.py

@@ -0,0 +1,60 @@
+from typing import TypedDict, Dict, Any, List, Annotated
+import time, uuid, os
+from langgraph.graph import StateGraph, END
+from langchain_core.messages import HumanMessage, AIMessage, BaseMessage
+from agents.simple_tools import (
+    generate_notes_full_pipeline_from_path,
+    generate_balance_sheet,
+    generate_pnl_statement,
+    generate_cash_flow_statement,
+)
+
+class FinancialAgentState(TypedDict):
+    messages: Annotated[List[BaseMessage], "History"]
+    file_path: str
+    result: Dict[str, Any]
+    status: str
+    start_time: float
+    end_time: float
+    error: str
+
+def make_workflow(tool_func):
+    def node(state: FinancialAgentState) -> FinancialAgentState:
+        state["start_time"] = time.time()
+        try:
+            # Use .invoke() to avoid deprecation warning
+            result = tool_func.invoke({"file_path": state["file_path"]})
+            state["result"] = result
+            state["status"] = "success" if result.get("status") == "success" else "error"
+            state["error"] = result.get("error", "")
+        except Exception as e:
+            state["status"] = "error"
+            state["error"] = str(e)
+        state["end_time"] = time.time()
+        return state
+
+    wf = StateGraph(FinancialAgentState)
+    wf.add_node("run", node)
+    wf.set_entry_point("run")
+    wf.add_edge("run", END)
+    return wf.compile()
+
+workflows = {
+    "notes": make_workflow(generate_notes_full_pipeline_from_path),
+    "pnl": make_workflow(generate_pnl_statement),
+    "bs": make_workflow(generate_balance_sheet),
+    "cf": make_workflow(generate_cash_flow_statement),
+}
+
+def run_workflow(file_path: str, kind: str) -> Dict[str, Any]:
+    state = FinancialAgentState(
+        messages=[HumanMessage(content=f"Run {kind} for {file_path}")],
+        file_path=file_path,
+        result={},
+        status="",
+        start_time=0,
+        end_time=0,
+        error="",
+    )
+    final = workflows[kind].invoke(state)
+    return final

app.py

@@ -1,4 +1,3 @@
-
 from fastapi import FastAPI, APIRouter, UploadFile, File, Form, HTTPException
 from fastapi.responses import JSONResponse, PlainTextResponse, FileResponse
 from typing import Optional, Dict, Any
@@ -16,13 +15,13 @@ from notes.llm_notes_generator import FlexibleFinancialNoteGenerator
 from notes.notes_generator import process_json
 from notes.json_to_excel import json_to_xlsx
 from utils.utils_normalize import normalize_llm_note_json, normalize_llm_notes_json
+from agents.langgraph_routes import run_workflow
 
 # Configure logging for the application
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger("financial_notes_api")
 
-# Import agent system
-from agents.simple_agent import FinancialStatementAgent
+
 
 app = FastAPI(
     title="Financial Notes Generator API",
@@ -400,6 +399,60 @@ async def agent_generate_statements(
         logger.error(f"Error in agent statement generation: {e}")
         raise HTTPException(status_code=500, detail=f"Agent generation failed: {str(e)}")
 
+@router.post("/notes")
+async def notes_route(file: UploadFile = File(...)):
+    file_path = f"data/input/{file.filename}"
+    os.makedirs("data/input", exist_ok=True)
+    with open(file_path, "wb") as buffer:
+        shutil.copyfileobj(file.file, buffer)
+    result = run_workflow(file_path, "notes")
+    if result["status"] == "success":
+        return FileResponse(result["result"]["output_xlsx_path"], filename=os.path.basename(result["result"]["output_xlsx_path"]))
+    raise HTTPException(status_code=500, detail=result["error"])
+
+@router.post("/pnl")
+async def pnl_route(file: UploadFile = File(...)):
+    file_path = f"data/input/{file.filename}"
+    os.makedirs("data/input", exist_ok=True)
+    with open(file_path, "wb") as buffer:
+        shutil.copyfileobj(file.file, buffer)
+    result = run_workflow(file_path, "pnl")
+    if result["status"] == "success":
+        return FileResponse(result["result"].get("output_path", "data/pnl_statement.xlsx"), filename=os.path.basename(result["result"].get("output_path", "data/pnl_statement.xlsx")))
+    raise HTTPException(status_code=500, detail=result["error"])
+
+@router.post("/bs")
+async def bs_route(file: UploadFile = File(...)):
+    file_path = f"data/input/{file.filename}"
+    os.makedirs("data/input", exist_ok=True)
+    with open(file_path, "wb") as buffer:
+        shutil.copyfileobj(file.file, buffer)
+    result = run_workflow(file_path, "bs")
+    if result["status"] == "success":
+        # Use first xlsx file in output dir if present
+        output_file = result["result"].get("output_path")
+        if not output_file or not os.path.isfile(output_file):
+            # Try to find the first .xlsx file in data/output/ (ensure it's a file, not a directory)
+            output_dir = "data/output/"
+            xlsx_files = [f for f in os.listdir(output_dir) if f.endswith('.xlsx') and os.path.isfile(os.path.join(output_dir, f))]
+            if xlsx_files:
+                output_file = os.path.join(output_dir, xlsx_files[0])
+            else:
+                raise HTTPException(status_code=500, detail="No balance sheet Excel file produced")
+        return FileResponse(output_file, filename=os.path.basename(output_file))
+    else:
+        raise HTTPException(status_code=500, detail=result["error"])
+
+@router.post("/cf")
+async def cf_route(file: UploadFile = File(...)):
+    file_path = f"data/input/{file.filename}"
+    os.makedirs("data/input", exist_ok=True)
+    with open(file_path, "wb") as buffer:
+        shutil.copyfileobj(file.file, buffer)
+    result = run_workflow(file_path, "cf")
+    if result["status"] == "success":
+        return FileResponse(result["result"].get("output_path", "data/cash_flow_statements.xlsx"), filename=os.path.basename(result["result"].get("output_path", "data/cash_flow_statements.xlsx")))
+    raise HTTPException(status_code=500, detail=result["error"])
 app.include_router(router)
 
 if __name__ == "__main__":

requirements.txt

@@ -13,4 +13,4 @@ langchain
 langchain-openai
 langchain-community
 langchain-core
-langsmith
+langsmithz