fix(backend): fix chat endpoint when chat id does not exists

pingcap · Jan 7, 2025 · 414d9c3 · 414d9c3
1 parent 25d01a2
commit 414d9c3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/backend/app/exceptions.py b/backend/app/exceptions.py
@@ -1,4 +1,6 @@
 from http import HTTPStatus
+from uuid import UUID
+
 from fastapi import HTTPException
 
 # Common
@@ -19,7 +21,7 @@ class ChatException(HTTPException):
 class ChatNotFound(ChatException):
     status_code = 404
 
-    def __init__(self, chat_id: int):
+    def __init__(self, chat_id: UUID):
         self.detail = f"chat #{chat_id} is not found"
 
 

diff --git a/backend/app/rag/chat.py b/backend/app/rag/chat.py
@@ -113,7 +113,7 @@ def __init__(
             #   anonymous user can only access anonymous chat by track_id
             self.db_chat_obj = chat_repo.get(self.db_session, chat_id)
             if not self.db_chat_obj:
-                raise ChatNotFound()
+                raise ChatNotFound(chat_id)
             try:
                 self.chat_engine_config = ChatEngineConfig.load_from_db(
                     db_session, self.db_chat_obj.engine.name