From 41cc85a7786674645eb9b6fff379d13a1babbed8 Mon Sep 17 00:00:00 2001
From: hjlarry <hjlarry@163.com>
Date: Mon, 12 May 2025 14:07:27 +0800
Subject: [PATCH] fix

---
 api/core/app/apps/advanced_chat/generate_task_pipeline.py | 2 +-
 api/services/conversation_service.py                      | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/api/core/app/apps/advanced_chat/generate_task_pipeline.py b/api/core/app/apps/advanced_chat/generate_task_pipeline.py
index 6542dd2a42..f71c49d112 100644
--- a/api/core/app/apps/advanced_chat/generate_task_pipeline.py
+++ b/api/core/app/apps/advanced_chat/generate_task_pipeline.py
@@ -116,7 +116,7 @@ class AdvancedChatAppGenerateTaskPipeline:
         self._workflow_cycle_manager = WorkflowCycleManager(
             application_generate_entity=application_generate_entity,
             workflow_system_variables={
-                SystemVariableKey.QUERY: db.session.query(Message),
+                SystemVariableKey.QUERY: message.query,
                 SystemVariableKey.FILES: application_generate_entity.files,
                 SystemVariableKey.CONVERSATION_ID: conversation.id,
                 SystemVariableKey.USER_ID: user_session_id,
diff --git a/api/services/conversation_service.py b/api/services/conversation_service.py
index b2295a70c7..afdaa49465 100644
--- a/api/services/conversation_service.py
+++ b/api/services/conversation_service.py
@@ -134,7 +134,7 @@ class ConversationService:
         # generate conversation name
         try:
             name = LLMGenerator.generate_conversation_name(
-                app_model.tenant_id, db.session.query(Message), conversation.id, app_model.id
+                app_model.tenant_id, message.query, conversation.id, app_model.id
             )
             conversation.name = name
         except: