opensource
/
dify
mirror of https://github.com/langgenius/dify.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix generate bug

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:
logger.exception(e)
raise e
finally:
db.session.remove()
# finally:
# db.session.remove()

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

@ -168,5 +168,3 @@ class WorkflowAppGenerator(BaseAppGenerator):
else:
logger.exception(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:
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 = {}