opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-03 07:46:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix generate bug

Browse Source
This commit is contained in:
takatost 2024-03-09 00:58:12 +08:00
parent b5366cba03
commit 37cdee5101
3 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/app/apps/advanced_chat/app_generator.py
View File

@ -216,5 +216,5 @@ class AdvancedChatAppGenerator(MessageBasedAppGenerator):
else: else:
logger.exception(e) logger.exception(e)
raise e raise e
finally: # finally:
db.session.remove() # db.session.remove()

2
api/core/app/apps/workflow/app_generator.py
View File

@ -168,5 +168,3 @@ class WorkflowAppGenerator(BaseAppGenerator):
else: else:
logger.exception(e) logger.exception(e)
raise e raise e
finally:
db.session.remove()

4
api/core/workflow/workflow_engine_manager.py
View File

@ -354,12 +354,12 @@ class WorkflowEngineManager:
if workflow_run_state.workflow.type == WorkflowType.CHAT.value and node.node_type == NodeType.END: 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] 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:
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: if not node_run_result.outputs:
node_run_result.outputs = {} node_run_result.outputs = {}
node_run_result.outputs['text'] = workflow_nodes_and_result_before_end.result.outputs.get('text') 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: if not node_run_result.outputs:
node_run_result.outputs = {} node_run_result.outputs = {}