youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-20 18:57:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(api): reuse advanced chat refresh session

Browse Source
This commit is contained in:
Yanli 盐粒
2026-03-17 20:18:21 +08:00
parent 1dce81c604
commit 7639d8e43f
1 changed files with 10 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -691,21 +691,19 @@ class AdvancedChatAppGenerator(MessageBasedAppGenerator):
@overload
def _refresh_model(session: object, model: Workflow) -> Workflow: ...
def _refresh_model(session: Session, model: Workflow) -> Workflow: ...
@overload
def _refresh_model(session: object, model: Message) -> Message: ...
def _refresh_model(session: Session, model: Message) -> Message: ...
def _refresh_model(session: object, model: Workflow | Message) -> Workflow | Message:
_ = session
with Session(bind=db.engine, expire_on_commit=False) as db_session:
if isinstance(model, Workflow):
detached_workflow = db_session.get(Workflow, model.id)
assert detached_workflow is not None
return detached_workflow
def _refresh_model(session: Session, model: Workflow | Message) -> Workflow | Message:
if isinstance(model, Workflow):
detached_workflow = session.get(Workflow, model.id)
assert detached_workflow is not None
return detached_workflow
detached_message = db_session.get(Message, model.id)
assert detached_message is not None
return detached_message
detached_message = session.get(Message, model.id)
assert detached_message is not None
return detached_message