made querying async
This commit is contained in:
+4
-4
@@ -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
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user