Selaa lähdekoodia

feat: tool output schema

Yeuoly 8 kuukautta sitten
vanhempi
commit
0dd05d7b6d

+ 1 - 0
api/core/tools/entities/api_entities.py

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

+ 1 - 1
api/core/tools/entities/tool_entities.py

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

+ 1 - 0
api/services/tools/tools_transform_service.py

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