feat: tool output schema

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

View File

@ -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"]]

View File

@ -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

View File

@ -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 [],
)