mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-03 09:53:45 +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"]
|
uri = mcp_with_schemas["server_url"]
|
||||||
|
|
||||||
try:
|
try:
|
||||||
llama_stack_client.toolgroups.unregister(toolgroup_id=test_toolgroup_id)
|
llama_stack_client.toolgroups.unregister_toolgroup(toolgroup_id=test_toolgroup_id)
|
||||||
except Exception:
|
except Exception:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
llama_stack_client.toolgroups.register(
|
llama_stack_client.toolgroups.register_tool_group(
|
||||||
toolgroup_id=test_toolgroup_id,
|
toolgroup_id=test_toolgroup_id,
|
||||||
provider_id="model-context-protocol",
|
provider_id="model-context-protocol",
|
||||||
mcp_endpoint=dict(uri=uri),
|
mcp_endpoint=dict(uri=uri),
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue