Add CompanyTable model and refactor query handling; update requirements for new dependencies
This commit is contained in:
+11
-13
@@ -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__":
|
||||
|
||||
Reference in New Issue
Block a user