mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-03 18:00:36 +00:00
updated test_tools_with_schemas
This commit is contained in:
parent
4b6bfbac8c
commit
d913756844
1 changed files with 2 additions and 2 deletions
|
|
@ -183,11 +183,11 @@ class TestMCPToolsInChatCompletion:
|
|||
uri = mcp_with_schemas["server_url"]
|
||||
|
||||
try:
|
||||
llama_stack_client.toolgroups.unregister(toolgroup_id=test_toolgroup_id)
|
||||
llama_stack_client.toolgroups.unregister_toolgroup(toolgroup_id=test_toolgroup_id)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
llama_stack_client.toolgroups.register(
|
||||
llama_stack_client.toolgroups.register_tool_group(
|
||||
toolgroup_id=test_toolgroup_id,
|
||||
provider_id="model-context-protocol",
|
||||
mcp_endpoint=dict(uri=uri),
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue