diff --git a/src/llama_stack_client/lib/cli/datasets/datasets.py b/src/llama_stack_client/lib/cli/datasets/datasets.py index a48294de..c01b875a 100644 --- a/src/llama_stack_client/lib/cli/datasets/datasets.py +++ b/src/llama_stack_client/lib/cli/datasets/datasets.py @@ -7,6 +7,7 @@ from .list import list_datasets from .register import register +from .unregister import unregister @click.group() @@ -18,3 +19,4 @@ def datasets(): # Register subcommands datasets.add_command(list_datasets) datasets.add_command(register) +datasets.add_command(unregister) diff --git a/src/llama_stack_client/lib/cli/datasets/unregister.py b/src/llama_stack_client/lib/cli/datasets/unregister.py new file mode 100644 index 00000000..8ca7cceb --- /dev/null +++ b/src/llama_stack_client/lib/cli/datasets/unregister.py @@ -0,0 +1,20 @@ +# Copyright (c) Meta Platforms, Inc. and affiliates. +# All rights reserved. +# +# This source code is licensed under the terms described in the LICENSE file in +# the root directory of this source tree. +import click + +from ..common.utils import handle_client_errors + + +@click.command("unregister") +@click.help_option("-h", "--help") +@click.argument("dataset-id", required=True) +@click.pass_context +@handle_client_errors("unregister dataset") +def unregister(ctx, dataset_id: str): + """Remove a dataset""" + client = ctx.obj["client"] + client.datasets.unregister(dataset_id=dataset_id) + click.echo(f"Dataset '{dataset_id}' unregistered successfully")