15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 01:33:05 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix: Update tool_type assignment in ToolSerializer to use tool_data value

This commit is contained in:
CaptainB 2025-12-04 21:30:10 +08:00
parent cec3381954
commit b10ce2296b
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
apps/tools/serializers/tool.py
View File

@ -797,6 +797,7 @@ class ToolSerializer(serializers.Serializer):
res = requests.get(download_url, timeout=5)
tool_data = RestrictedUnpickler(io.BytesIO(res.content)).load().tool
tool_id = uuid.uuid7()
print(instance.get('tool_type'))
tool = Tool(
id=tool_id,
name=instance.get('name'),
@ -808,7 +809,7 @@ class ToolSerializer(serializers.Serializer):
input_field_list=tool_data.get('input_field_list', []),
init_field_list=tool_data.get('init_field_list', []),
scope=ToolScope.WORKSPACE,
tool_type=instance.get('tool_type', ToolType.CUSTOM),
tool_type=tool_data.get('tool_type', ToolType.CUSTOM),
folder_id=instance.get('folder_id', self.data.get('workspace_id')),
template_id=self.data.get('tool_id'),
label=instance.get('label'),