From 37cdee51010322ab68b5376904412df54596a8d6 Mon Sep 17 00:00:00 2001 From: takatost Date: Sat, 9 Mar 2024 00:58:12 +0800 Subject: [PATCH] fix generate bug --- api/core/app/apps/advanced_chat/app_generator.py | 4 ++-- api/core/app/apps/workflow/app_generator.py | 2 -- api/core/workflow/workflow_engine_manager.py | 4 ++-- 3 files changed, 4 insertions(+), 6 deletions(-) diff --git a/api/core/app/apps/advanced_chat/app_generator.py b/api/core/app/apps/advanced_chat/app_generator.py index ed45e2ba8a..a0f197ec37 100644 --- a/api/core/app/apps/advanced_chat/app_generator.py +++ b/api/core/app/apps/advanced_chat/app_generator.py @@ -216,5 +216,5 @@ class AdvancedChatAppGenerator(MessageBasedAppGenerator): else: logger.exception(e) raise e - finally: - db.session.remove() + # finally: + # db.session.remove() diff --git a/api/core/app/apps/workflow/app_generator.py b/api/core/app/apps/workflow/app_generator.py index d3303047ca..b1a70a83ba 100644 --- a/api/core/app/apps/workflow/app_generator.py +++ b/api/core/app/apps/workflow/app_generator.py @@ -168,5 +168,3 @@ class WorkflowAppGenerator(BaseAppGenerator): else: logger.exception(e) raise e - finally: - db.session.remove() diff --git a/api/core/workflow/workflow_engine_manager.py b/api/core/workflow/workflow_engine_manager.py index 628df4ac5f..c5af015e87 100644 --- a/api/core/workflow/workflow_engine_manager.py +++ b/api/core/workflow/workflow_engine_manager.py @@ -354,12 +354,12 @@ class WorkflowEngineManager: if workflow_run_state.workflow.type == WorkflowType.CHAT.value and node.node_type == NodeType.END: workflow_nodes_and_result_before_end = workflow_run_state.workflow_nodes_and_results[-2] if workflow_nodes_and_result_before_end: - if workflow_nodes_and_result_before_end.node.node_type == NodeType.LLM.value: + if workflow_nodes_and_result_before_end.node.node_type == NodeType.LLM: if not node_run_result.outputs: node_run_result.outputs = {} node_run_result.outputs['text'] = workflow_nodes_and_result_before_end.result.outputs.get('text') - elif workflow_nodes_and_result_before_end.node.node_type == NodeType.DIRECT_ANSWER.value: + elif workflow_nodes_and_result_before_end.node.node_type == NodeType.DIRECT_ANSWER: if not node_run_result.outputs: node_run_result.outputs = {}