phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 10:54:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

make default tool prompt format none in agent config (#863)

Browse source 
# What does this PR do?

Previously the tests hard coded the tool prompt format to be json which
will cause it to fail when using 3.2/3.3 family of models. This change
make the default to be none for the agent config and just remove the
specification in the tests.


## Test Plan
LLAMA_STACK_BASE_URL=http://localhost:8321 pytest -v
tests/client-sdk/agents/test_agents.py
This commit is contained in:
Dinesh Yeduguru 2025-01-23 14:44:59 -08:00 • committed by GitHub
parent 94ffaf468c
commit a78f1fc70d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 1 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/apis/agents/agents.py
View file

@ -155,9 +155,7 @@ class AgentConfigCommon(BaseModel):
toolgroups: Optional[List[AgentToolGroup]] = Field(default_factory=list) toolgroups: Optional[List[AgentToolGroup]] = Field(default_factory=list)
client_tools: Optional[List[ToolDef]] = Field(default_factory=list) client_tools: Optional[List[ToolDef]] = Field(default_factory=list)
tool_choice: Optional[ToolChoice] = Field(default=ToolChoice.auto) tool_choice: Optional[ToolChoice] = Field(default=ToolChoice.auto)
tool_prompt_format: Optional[ToolPromptFormat] = Field( tool_prompt_format: Optional[ToolPromptFormat] = Field(default=None)
default=ToolPromptFormat.json
)
max_infer_iters: int = 10 max_infer_iters: int = 10

1
tests/client-sdk/agents/test_agents.py
View file

@ -98,7 +98,6 @@ def agent_config(llama_stack_client, text_model_id):
}, },
toolgroups=[], toolgroups=[],
tool_choice="auto", tool_choice="auto",
tool_prompt_format="json",
input_shields=available_shields, input_shields=available_shields,
output_shields=available_shields, output_shields=available_shields,
enable_session_persistence=False, enable_session_persistence=False,