highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-09-19 12:03:54 +02:00
Code Issues Packages Projects Releases Wiki Activity

Initial document display

Browse Source
This commit is contained in:
Weves
2023-05-12 00:49:56 -07:00
committed by Chris Weaver
parent 6028523198
commit 66130c8845
6 changed files with 68 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend/danswer/main.py
View File

@@ -90,7 +90,7 @@ def get_application() -> FastAPI:
)
@application.on_event("startup")
async def startup_event() -> None:
def startup_event() -> None:
from danswer.semantic_search.semantic_search import (
warm_up_models,
) # To avoid circular imports

6
backend/danswer/server/search_backend.py
View File

@@ -81,7 +81,7 @@ async def promote_admin(
@router.get("/direct-qa", response_model=QAResponse)
def direct_qa(question: QAQuestion):
def direct_qa(question: QAQuestion = Depends()):
start_time = time.time()
query = question.query
@@ -104,7 +104,7 @@ def direct_qa(question: QAQuestion):
@router.get("/stream-direct-qa")
def stream_direct_qa(question: QAQuestion):
def stream_direct_qa(question: QAQuestion = Depends()):
top_documents_key = "top_documents"
answer_key = "answer"
quotes_key = "quotes"
@@ -147,7 +147,7 @@ def stream_direct_qa(question: QAQuestion):
@router.get("/keyword-search", response_model=KeywordResponse)
def keyword_search(question: QAQuestion):
def keyword_search(question: QAQuestion = Depends()):
ts_client = TSClient.get_instance()
query = question.query
collection = question.collection