Skip to content

fix(cli): align cli toolgroups register to the new arguments #231

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
May 27, 2025
Merged

fix(cli): align cli toolgroups register to the new arguments #231

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 9 additions & 5 deletions src/llama_stack_client/lib/cli/toolgroups/toolgroups.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,8 @@
from rich.table import Table

from ..common.utils import handle_client_errors
from ....types import toolgroup_register_params
from ...._types import NOT_GIVEN, NotGiven


@click.group()
Expand Down Expand Up @@ -78,28 +80,30 @@ def get_toolgroup(ctx, toolgroup_id: str):
@click.help_option("-h", "--help")
@click.argument("toolgroup_id")
@click.option("--provider-id", help="Provider ID for the toolgroup", default=None)
@click.option("--provider-toolgroup-id", help="Provider's toolgroup ID", default=None)
@click.option("--mcp-config", help="JSON mcp_config for the toolgroup", default=None)
@click.option("--mcp-endpoint", help="JSON mcp_config for the toolgroup", default=None)
@click.option("--args", help="JSON args for the toolgroup", default=None)
@click.pass_context
@handle_client_errors("register toolgroup")
def register_toolgroup(
ctx,
toolgroup_id: str,
provider_id: Optional[str],
provider_toolgroup_id: Optional[str],
mcp_config: Optional[str],
mcp_endpoint: Optional[str],
args: Optional[str],
):
"""Register a new toolgroup at distribution endpoint"""
client = ctx.obj["client"]
console = Console()

_mcp_endpoint: toolgroup_register_params.McpEndpoint | NotGiven = NOT_GIVEN
if mcp_endpoint:
_mcp_endpoint = toolgroup_register_params.McpEndpoint(uri=mcp_endpoint)

response = client.toolgroups.register(
toolgroup_id=toolgroup_id,
provider_id=provider_id,
args=args,
mcp_config=mcp_config,
mcp_endpoint=_mcp_endpoint,
)
if response:
console.print(f"[green]Successfully registered toolgroup {toolgroup_id}[/green]")
Expand Down