highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-10-10 05:05:34 +02:00
Code Issues Packages Projects Releases Wiki Activity

Save Retrieved Docs for One Shot Flows (#1259)

Browse Source
This commit is contained in:
Yuhong Sun
2024-03-26 22:48:40 -07:00
committed by GitHub
parent efc7d6e098
commit fbff5b5784
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
backend/danswer/one_shot_answer/answer_question.py
View File

@@ -16,6 +16,7 @@ from danswer.configs.chat_configs import MAX_CHUNKS_FED_TO_CHAT
from danswer.configs.chat_configs import QA_TIMEOUT from danswer.configs.chat_configs import QA_TIMEOUT
from danswer.configs.constants import MessageType from danswer.configs.constants import MessageType
from danswer.db.chat import create_chat_session from danswer.db.chat import create_chat_session
from danswer.db.chat import create_db_search_doc
from danswer.db.chat import create_new_chat_message from danswer.db.chat import create_new_chat_message
from danswer.db.chat import get_or_create_root_message from danswer.db.chat import get_or_create_root_message
from danswer.db.chat import get_prompt_by_id from danswer.db.chat import get_prompt_by_id
@@ -202,6 +203,11 @@ def stream_answer_objects(
) )
yield from answer.processed_streamed_output yield from answer.processed_streamed_output
reference_db_search_docs = [
create_db_search_doc(server_search_doc=top_doc, db_session=db_session)
for top_doc in top_docs
]
# Saving Gen AI answer and responding with message info # Saving Gen AI answer and responding with message info
gen_ai_response_message = create_new_chat_message( gen_ai_response_message = create_new_chat_message(
chat_session_id=chat_session.id, chat_session_id=chat_session.id,
@@ -211,7 +217,7 @@ def stream_answer_objects(
token_count=len(llm_tokenizer(answer.llm_answer)), token_count=len(llm_tokenizer(answer.llm_answer)),
message_type=MessageType.ASSISTANT, message_type=MessageType.ASSISTANT,
error=None, error=None,
reference_docs=None, # Don't need to save reference docs for one shot flow reference_docs=reference_db_search_docs,
db_session=db_session, db_session=db_session,
commit=True, commit=True,
) )