highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-09-27 20:38:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add message id to simple message endpoint (#69)

Browse Source
This commit is contained in:
Yuhong Sun
2024-04-14 19:24:45 -07:00
committed by Chris Weaver
parent c00bd44bcc
commit 50f006557f
2 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
backend/ee/danswer/server/query_and_chat/chat_backend.py
View File

@@ -16,6 +16,7 @@ from danswer.db.engine import get_session
from danswer.db.models import User from danswer.db.models import User
from danswer.search.models import OptionalSearchSetting from danswer.search.models import OptionalSearchSetting
from danswer.search.models import RetrievalDetails from danswer.search.models import RetrievalDetails
from danswer.server.query_and_chat.models import ChatMessageDetail
from danswer.server.query_and_chat.models import CreateChatMessageRequest from danswer.server.query_and_chat.models import CreateChatMessageRequest
from danswer.utils.logger import setup_logger from danswer.utils.logger import setup_logger
from ee.danswer.server.query_and_chat.models import BasicCreateChatMessageRequest from ee.danswer.server.query_and_chat.models import BasicCreateChatMessageRequest
@@ -111,6 +112,8 @@ def handle_simplified_chat_message(
response.simple_search_docs = translate_doc_response_to_simple_doc(packet) response.simple_search_docs = translate_doc_response_to_simple_doc(packet)
elif isinstance(packet, StreamingError): elif isinstance(packet, StreamingError):
response.error_msg = packet.error response.error_msg = packet.error
elif isinstance(packet, ChatMessageDetail):
response.message_id = packet.message_id
response.answer = answer response.answer = answer
if answer: if answer:

1
backend/ee/danswer/server/query_and_chat/models.py
View File

@@ -47,3 +47,4 @@ class ChatBasicResponse(BaseModel):
answer_citationless: str | None = None answer_citationless: str | None = None
simple_search_docs: list[SimpleDoc] | None = None simple_search_docs: list[SimpleDoc] | None = None
error_msg: str | None = None error_msg: str | None = None
message_id: int | None = None