phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-22 22:42:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

rename UserDefinedToolDef to ToolDef

Browse source
This commit is contained in:
Dinesh Yeduguru 2025-01-07 09:14:26 -08:00
parent db0b2a60c1
commit e3775eb6f6
8 changed files with 180 additions and 322 deletions
Download patch file Download diff file Expand all files Collapse all files

26
llama_stack/apis/tools/tools.py
View file

@ -48,30 +48,16 @@ class Tool(Resource):
@json_schema_type
class UserDefinedToolDef(BaseModel):
type: Literal["user_defined"] = "user_defined"
class ToolDef(BaseModel):
name: str
description: str
parameters: List[ToolParameter]
metadata: Dict[str, Any]
description: Optional[str] = None
parameters: Optional[List[ToolParameter]] = None
metadata: Optional[Dict[str, Any]] = None
tool_prompt_format: Optional[ToolPromptFormat] = Field(
default=ToolPromptFormat.json
)
@json_schema_type
class BuiltInToolDef(BaseModel):
type: Literal["built_in"] = "built_in"
built_in_type: BuiltinTool
metadata: Optional[Dict[str, Any]] = None
ToolDef = register_schema(
Annotated[Union[UserDefinedToolDef, BuiltInToolDef], Field(discriminator="type")],
name="ToolDef",
)
@json_schema_type
class MCPToolGroupDef(BaseModel):
"""
@ -100,7 +86,7 @@ ToolGroupDef = register_schema(
@json_schema_type
class ToolGroupInput(BaseModel):
tool_group_id: str
tool_group: ToolGroupDef
tool_group_def: ToolGroupDef
provider_id: Optional[str] = None
@ -127,7 +113,7 @@ class ToolGroups(Protocol):
async def register_tool_group(
self,
tool_group_id: str,
tool_group: ToolGroupDef,
tool_group_def: ToolGroupDef,
provider_id: Optional[str] = None,
) -> None:
"""Register a tool group"""