outman/dify
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: tool output schema

Browse Source
This commit is contained in:
Yeuoly 2024-09-29 20:58:07 +08:00
parent 7c83d5ce76
commit 0dd05d7b6d
No known key found for this signature in database
GPG Key ID: A66E7E320FB19F61
3 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
api/core/tools/entities/api_entities.py
View File

@ -16,6 +16,7 @@ class ToolApiEntity(BaseModel):
description: I18nObject description: I18nObject
parameters: Optional[list[ToolParameter]] = None parameters: Optional[list[ToolParameter]] = None
labels: list[str] = Field(default_factory=list) labels: list[str] = Field(default_factory=list)
output_schema: Optional[dict] = None
ToolProviderTypeApiLiteral = Optional[Literal["builtin", "api", "workflow"]] ToolProviderTypeApiLiteral = Optional[Literal["builtin", "api", "workflow"]]

2
api/core/tools/entities/tool_entities.py
View File

@ -298,7 +298,7 @@ class ToolEntity(BaseModel):
identity: ToolIdentity identity: ToolIdentity
parameters: list[ToolParameter] = Field(default_factory=list) parameters: list[ToolParameter] = Field(default_factory=list)
description: Optional[ToolDescription] = None description: Optional[ToolDescription] = None
# TODO: output schema output_schema: Optional[dict] = None
has_runtime_parameters: bool = Field(default=False, description="Whether the tool has runtime parameters") has_runtime_parameters: bool = Field(default=False, description="Whether the tool has runtime parameters")
# pydantic configs # pydantic configs

1
api/services/tools/tools_transform_service.py
View File

@ -262,6 +262,7 @@ class ToolTransformService:
name=tool.entity.identity.name, name=tool.entity.identity.name,
label=tool.entity.identity.label, label=tool.entity.identity.label,
description=tool.entity.description.human if tool.entity.description else I18nObject(en_US=""), description=tool.entity.description.human if tool.entity.description else I18nObject(en_US=""),
output_schema=tool.entity.output_schema,
parameters=current_parameters, parameters=current_parameters,
labels=labels or [], labels=labels or [],
) )