fix generate bug

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()

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()

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 = {}