Add CompanyTable model and refactor query handling; update requirements for new dependencies

This commit is contained in:
bolade
2025-09-02 12:22:50 +01:00
parent 74931f235e
commit 65b5df3a43
19 changed files with 166 additions and 216 deletions
+11 -13
View File
@@ -1,16 +1,14 @@
import io
from api import investors
import pandas as pd
from app.db.db import db_dependency, init_database
from db.db import db_dependency
from fastapi import FastAPI, File, UploadFile
from app.services.openrouter import InvestorProcessor
from app.pydantic_schemas import QueryRequest, QueryResponseList
from app.services.langgraph_agent import LangGraphQueryAgent
from services.openrouter import InvestorProcessor
from services.querying import QueryProcessor
app = FastAPI()
init_database()
app.include_router(investors.router)
# init_database()
@app.get("/")
@@ -32,11 +30,11 @@ async def parse_csv(db: db_dependency, file: UploadFile = File(...)):
return {"results": [r.dict() for r in results]}
@app.post("/query", response_model=QueryResponseList)
async def query_investors(db: db_dependency, request: QueryRequest):
agent = LangGraphQueryAgent(sql_session=db)
result = agent.run(request.question)
return result
@app.post("/query")
async def query_investors(db: db_dependency, question: str):
processor = QueryProcessor(sql_session=db)
results = processor.process_query(question)
return {"results": results}
if __name__ == "__main__":