From a8d10750c1be4651ac23ef2bd18235a2c3ace822 Mon Sep 17 00:00:00 2001 From: evan-danswer Date: Thu, 27 Feb 2025 11:56:51 -0800 Subject: [PATCH] fix propagation of is_agentic (#4150) --- backend/onyx/db/chat.py | 1 + backend/onyx/server/query_and_chat/models.py | 1 + web/src/app/chat/lib.tsx | 1 + 3 files changed, 3 insertions(+) diff --git a/backend/onyx/db/chat.py b/backend/onyx/db/chat.py index 62d29827a..5335d25b8 100644 --- a/backend/onyx/db/chat.py +++ b/backend/onyx/db/chat.py @@ -962,6 +962,7 @@ def translate_db_message_to_chat_message_detail( chat_message.sub_questions ), refined_answer_improvement=chat_message.refined_answer_improvement, + is_agentic=chat_message.is_agentic, error=chat_message.error, ) diff --git a/backend/onyx/server/query_and_chat/models.py b/backend/onyx/server/query_and_chat/models.py index d7db7cf8c..132be33ca 100644 --- a/backend/onyx/server/query_and_chat/models.py +++ b/backend/onyx/server/query_and_chat/models.py @@ -242,6 +242,7 @@ class ChatMessageDetail(BaseModel): files: list[FileDescriptor] tool_call: ToolCallFinalResult | None refined_answer_improvement: bool | None = None + is_agentic: bool | None = None error: str | None = None def model_dump(self, *args: list, **kwargs: dict[str, Any]) -> dict[str, Any]: # type: ignore diff --git a/web/src/app/chat/lib.tsx b/web/src/app/chat/lib.tsx index e2ea1b4a5..a95c64d14 100644 --- a/web/src/app/chat/lib.tsx +++ b/web/src/app/chat/lib.tsx @@ -501,6 +501,7 @@ export function processRawChatHistory( sub_questions: subQuestions, isImprovement: (messageInfo.refined_answer_improvement as unknown as boolean) || false, + is_agentic: messageInfo.is_agentic, }; messages.set(messageInfo.message_id, message);