youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-02 16:38:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: test

Browse Source
This commit is contained in:
Yeuoly
2024-04-01 17:07:09 +08:00
parent 51f225e567
commit 75e95e09d3
2 changed files with 47 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/workflow/nodes/tool/entities.py
View File

@ -16,8 +16,8 @@ class ToolEntity(BaseModel):
class ToolNodeData(BaseNodeData, ToolEntity):
class ToolInput(BaseModel):
type: Literal['mixed', 'variable', 'constant']
value: Union[ToolParameterValue, list[str]]
type: Literal['mixed', 'variable', 'constant']
@validator('type', pre=True, always=True)
def check_type(cls, value, values):
@ -29,7 +29,7 @@ class ToolNodeData(BaseNodeData, ToolEntity):
raise ValueError('value must be a list')
elif typ == 'constant' and not isinstance(value, ToolParameterValue):
raise ValueError('value must be a string, int, float, or bool')
return value
return typ
"""
Tool Node Schema