opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-09-08 11:04:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: convert file-transfer-method-tools if/elif to match/case (#30001) (#34783)

Browse Source
This commit is contained in:
dataCenter430 2026-04-08 17:42:13 -07:00 committed by GitHub
parent 1898a3f8a5
commit fd2843b0fb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
api/core/tools/workflow_as_tool/tool.py
View File

@ -305,14 +305,15 @@ class WorkflowTool(Tool):
"transfer_method": file.transfer_method.value, "transfer_method": file.transfer_method.value,
"type": file.type.value, "type": file.type.value,
} }
if file.transfer_method == FileTransferMethod.TOOL_FILE: match file.transfer_method:
file_dict["tool_file_id"] = resolve_file_record_id(file.reference) case FileTransferMethod.TOOL_FILE:
elif file.transfer_method == FileTransferMethod.LOCAL_FILE: file_dict["tool_file_id"] = resolve_file_record_id(file.reference)
file_dict["upload_file_id"] = resolve_file_record_id(file.reference) case FileTransferMethod.LOCAL_FILE:
elif file.transfer_method == FileTransferMethod.DATASOURCE_FILE: file_dict["upload_file_id"] = resolve_file_record_id(file.reference)
file_dict["datasource_file_id"] = resolve_file_record_id(file.reference) case FileTransferMethod.DATASOURCE_FILE:
elif file.transfer_method == FileTransferMethod.REMOTE_URL: file_dict["datasource_file_id"] = resolve_file_record_id(file.reference)
file_dict["url"] = file.generate_url() case FileTransferMethod.REMOTE_URL:
file_dict["url"] = file.generate_url()
files.append(file_dict) files.append(file_dict)
except Exception: except Exception:
@ -357,8 +358,11 @@ class WorkflowTool(Tool):
def _update_file_mapping(self, file_dict: dict): def _update_file_mapping(self, file_dict: dict):
file_id = resolve_file_record_id(file_dict.get("reference") or file_dict.get("related_id")) file_id = resolve_file_record_id(file_dict.get("reference") or file_dict.get("related_id"))
transfer_method = FileTransferMethod.value_of(file_dict.get("transfer_method")) transfer_method = FileTransferMethod.value_of(file_dict.get("transfer_method"))
if transfer_method == FileTransferMethod.TOOL_FILE: match transfer_method:
file_dict["tool_file_id"] = file_id case FileTransferMethod.TOOL_FILE:
elif transfer_method == FileTransferMethod.LOCAL_FILE: file_dict["tool_file_id"] = file_id
file_dict["upload_file_id"] = file_id case FileTransferMethod.LOCAL_FILE:
file_dict["upload_file_id"] = file_id
case FileTransferMethod.REMOTE_URL | FileTransferMethod.DATASOURCE_FILE:
pass
return file_dict return file_dict