diff --git a/backend/ee/danswer/server/query_and_chat/chat_backend.py b/backend/ee/danswer/server/query_and_chat/chat_backend.py index 0c5149a5c90..26aff2774e3 100644 --- a/backend/ee/danswer/server/query_and_chat/chat_backend.py +++ b/backend/ee/danswer/server/query_and_chat/chat_backend.py @@ -16,6 +16,7 @@ from danswer.db.engine import get_session from danswer.db.models import User from danswer.search.models import OptionalSearchSetting 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.utils.logger import setup_logger 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) elif isinstance(packet, StreamingError): response.error_msg = packet.error + elif isinstance(packet, ChatMessageDetail): + response.message_id = packet.message_id response.answer = answer if answer: diff --git a/backend/ee/danswer/server/query_and_chat/models.py b/backend/ee/danswer/server/query_and_chat/models.py index 1876f3f4648..ec5ec5ac411 100644 --- a/backend/ee/danswer/server/query_and_chat/models.py +++ b/backend/ee/danswer/server/query_and_chat/models.py @@ -47,3 +47,4 @@ class ChatBasicResponse(BaseModel): answer_citationless: str | None = None simple_search_docs: list[SimpleDoc] | None = None error_msg: str | None = None + message_id: int | None = None