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

refactor: convert isinstance chains to match/case (part 6) (#36705)

Browse Source 
Signed-off-by: EvanYao826 <155432245+EvanYao826@users.noreply.github.com>
This commit is contained in:
Evan 2026-05-27 12:09:01 +08:00 committed by GitHub
parent 4384d8910e
commit d342ff1a1e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 38 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

74
api/tasks/app_generate/workflow_execute_task.py
View File

@ -102,12 +102,13 @@ class AppExecutionParams(BaseModel):
workflow_run_id: str | None = None, workflow_run_id: str | None = None,
): ):
user_params: _Account | _EndUser user_params: _Account | _EndUser
if isinstance(user, Account): match user:
user_params = _Account(user_id=user.id) case Account():
elif isinstance(user, EndUser): user_params = _Account(user_id=user.id)
user_params = _EndUser(end_user_id=user.id) case EndUser():
else: user_params = _EndUser(end_user_id=user.id)
raise AssertionError("this statement should be unreachable.") case _:
raise AssertionError("this statement should be unreachable.")
return cls( return cls(
app_id=app_model.id, app_id=app_model.id,
workflow_id=workflow.id, workflow_id=workflow.id,
@ -365,36 +366,37 @@ def _resume_app_execution(payload: dict[str, Any]) -> None:
state_owner_user_id=workflow.created_by, state_owner_user_id=workflow.created_by,
) )
if isinstance(generate_entity, AdvancedChatAppGenerateEntity): match generate_entity:
assert conversation is not None case AdvancedChatAppGenerateEntity():
assert message is not None assert conversation is not None
_resume_advanced_chat( assert message is not None
app_model=app_model, _resume_advanced_chat(
workflow=workflow, app_model=app_model,
user=user, workflow=workflow,
conversation=conversation, user=user,
message=message, conversation=conversation,
generate_entity=generate_entity, message=message,
graph_runtime_state=graph_runtime_state, generate_entity=generate_entity,
session_factory=session_factory, graph_runtime_state=graph_runtime_state,
pause_state_config=pause_config, session_factory=session_factory,
workflow_run_id=workflow_run_id, pause_state_config=pause_config,
workflow_run=workflow_run, workflow_run_id=workflow_run_id,
) workflow_run=workflow_run,
elif isinstance(generate_entity, WorkflowAppGenerateEntity): )
_resume_workflow( case WorkflowAppGenerateEntity():
app_model=app_model, _resume_workflow(
workflow=workflow, app_model=app_model,
user=user, workflow=workflow,
generate_entity=generate_entity, user=user,
graph_runtime_state=graph_runtime_state, generate_entity=generate_entity,
session_factory=session_factory, graph_runtime_state=graph_runtime_state,
pause_state_config=pause_config, session_factory=session_factory,
workflow_run_id=workflow_run_id, pause_state_config=pause_config,
workflow_run=workflow_run, workflow_run_id=workflow_run_id,
workflow_run_repo=workflow_run_repo, workflow_run=workflow_run,
pause_entity=pause_entity, workflow_run_repo=workflow_run_repo,
) pause_entity=pause_entity,
)
def _resume_advanced_chat( def _resume_advanced_chat(