feat: tool output schema

pull/9184/head
Yeuoly 2 years ago
parent 7c83d5ce76
commit 0dd05d7b6d
No known key found for this signature in database
GPG Key ID: A66E7E320FB19F61

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

@ -298,7 +298,7 @@ class ToolEntity(BaseModel):
identity: ToolIdentity
parameters: list[ToolParameter] = Field(default_factory=list)
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")
# pydantic configs

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

Loading…
Cancel
Save