made querying async

This commit is contained in:
bolade
2025-10-28 21:09:47 +01:00
parent bb03f6ade4
commit 02c8bb816f
7 changed files with 27 additions and 14 deletions
+4 -4
View File
@@ -1,6 +1,4 @@
import io
import logging
import os
import pandas as pd
from db.db import Base, db_dependency, engine
@@ -44,6 +42,7 @@ class QueryRequest(BaseModel):
}
}
class CompanyQueryRequest(BaseModel):
question: str
@@ -54,6 +53,7 @@ class CompanyQueryRequest(BaseModel):
}
}
@app.get("/")
def health():
return {"Hello": "World"}
@@ -122,7 +122,7 @@ async def query_investors(request: QueryRequest):
- "Healthcare investors in Europe"
"""
processor = QueryProcessor()
results = processor.process_query(request.question)
results = await processor.process_query(request.question)
return results
@@ -143,7 +143,7 @@ async def query_companies(request: CompanyQueryRequest):
- "European startups founded after 2019"
"""
processor = CompanyQueryProcessor()
results = processor.process_query(request.question)
results = await processor.process_query(request.question)
return results