From 1fef9b078379c8991f6029c34ac10d4cbdc5a44d Mon Sep 17 00:00:00 2001 From: Joey Wang Date: Tue, 15 Aug 2023 06:59:30 -0700 Subject: [PATCH] fix(chatMessages): Fix error on answering question (#953) --- backend/core/llm/qa_base.py | 2 +- backend/core/repository/brain/get_brain_prompt_id.py | 2 +- backend/core/repository/brain/get_default_user_brain.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/backend/core/llm/qa_base.py b/backend/core/llm/qa_base.py index 6c0147ef4d94..dcd8ffa77272 100644 --- a/backend/core/llm/qa_base.py +++ b/backend/core/llm/qa_base.py @@ -177,7 +177,7 @@ def generate_answer( **{ "chat_id": chat_id, "user_message": question.question, - "assistant": "", + "assistant": answer, "message_time": new_chat.message_time, "prompt_title": prompt.title if prompt else None, "brain_name": brain.name if brain else None, diff --git a/backend/core/repository/brain/get_brain_prompt_id.py b/backend/core/repository/brain/get_brain_prompt_id.py index 7ea844e462c6..9aa67b45fe40 100644 --- a/backend/core/repository/brain/get_brain_prompt_id.py +++ b/backend/core/repository/brain/get_brain_prompt_id.py @@ -5,6 +5,6 @@ def get_brain_prompt_id(brain_id: UUID) -> UUID | None: brain = get_brain_by_id(brain_id) - prompt_id = brain.brain_id if brain else None + prompt_id = brain.prompt_id if brain else None return prompt_id diff --git a/backend/core/repository/brain/get_default_user_brain.py b/backend/core/repository/brain/get_default_user_brain.py index 5ff5c63bc6ec..cbf7d85df247 100644 --- a/backend/core/repository/brain/get_default_user_brain.py +++ b/backend/core/repository/brain/get_default_user_brain.py @@ -12,7 +12,7 @@ def get_user_default_brain(user_id: UUID) -> BrainEntity | None: supabase_db = get_supabase_db() brain_id = supabase_db.get_default_user_brain_id(user_id) - logger.info("Default brain response:", brain_id) + logger.info(f"Default brain response: {brain_id}") if brain_id is None: return None