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

Add items and title to ToolParameter/ToolParamDefinition

Browse source
This commit is contained in:
Tami Takamiya 2025-07-31 14:03:04 -04:00
parent 6b855af96f
commit 30768cc8c6
No known key found for this signature in database
GPG key ID: B23C22B8CB1E70C7
5 changed files with 18 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/utils/inference/openai_compat.py
View file

@ -805,6 +805,10 @@ def convert_tooldef_to_openai_tool(tool: ToolDefinition) -> dict:
properties[param_name].update(description=param.description)
if param.default:
properties[param_name].update(default=param.default)
if param.items:
properties[param_name].update(items=param.items)
if param.title:
properties[param_name].update(title=param.title)
if param.required:
required.append(param_name)

4
llama_stack/providers/utils/tools/mcp.py
View file

@ -120,6 +120,10 @@ async def list_mcp_tools(endpoint: str, headers: dict[str, str]) -> ListToolDefs
name=param_name,
parameter_type=param_schema.get("type", "string"),
description=param_schema.get("description", ""),
required="default" not in param_schema,
items=param_schema.get("items", None),
title=param_schema.get("title", None),
default=param_schema.get("default", None),
)
)
tools.append(