AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-16 11:15:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(workflow): detached session issues

Browse Source
This commit is contained in:
takatost 2024-09-03 18:23:25 +08:00
parent 6bee121ebe
commit 78fa1f6868
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -438,14 +438,18 @@ class AdvancedChatAppGenerateTaskPipeline(BasedGenerateTaskPipeline, WorkflowCyc
if self._task_state.metadata else None if self._task_state.metadata else None
db.session.commit() db.session.commit()
db.session.refresh(self._message)
db.session.close() db.session.close()
elif isinstance(event, QueueAnnotationReplyEvent): elif isinstance(event, QueueAnnotationReplyEvent):
self._handle_annotation_reply(event) self._handle_annotation_reply(event)
self._refetch_message()
self._message.message_metadata = json.dumps(jsonable_encoder(self._task_state.metadata)) \ self._message.message_metadata = json.dumps(jsonable_encoder(self._task_state.metadata)) \
if self._task_state.metadata else None if self._task_state.metadata else None
db.session.commit() db.session.commit()
db.session.refresh(self._message)
db.session.close() db.session.close()
elif isinstance(event, QueueTextChunkEvent): elif isinstance(event, QueueTextChunkEvent):
delta_text = event.text delta_text = event.text

3
api/core/app/apps/workflow/app_runner.py
View File

@ -17,6 +17,7 @@ from core.workflow.enums import SystemVariableKey
from core.workflow.workflow_entry import WorkflowEntry from core.workflow.workflow_entry import WorkflowEntry
from extensions.ext_database import db from extensions.ext_database import db
from models.model import App, EndUser from models.model import App, EndUser
from models.workflow import WorkflowType
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -107,7 +108,7 @@ class WorkflowAppRunner(WorkflowBasedAppRunner):
tenant_id=workflow.tenant_id, tenant_id=workflow.tenant_id,
app_id=workflow.app_id, app_id=workflow.app_id,
workflow_id=workflow.id, workflow_id=workflow.id,
workflow_type=workflow.type, workflow_type=WorkflowType.value_of(workflow.type),
graph=graph, graph=graph,
graph_config=workflow.graph_dict, graph_config=workflow.graph_dict,
user_id=self.application_generate_entity.user_id, user_id=self.application_generate_entity.user_id,