Skip to content

Commit 38cd91c

Browse files
authored
feat: Add llama-stack-client datasets unregister command (#222)
# What does this PR do? [Provide a short summary of what this PR does and why. Link to relevant issues if applicable.] [//]: # (If resolving an issue, uncomment and update the line below) [//]: # (Closes #[issue-number]) ## Test Plan [Describe the tests you ran to verify your changes with result summaries. *Provide clear instructions so the plan can be easily re-executed.*] [//]: # (## Documentation) [//]: # (- [ ] Added a Changelog entry if the change is significant) Signed-off-by: Josh Salomon <jsalomon@redhat.com>
1 parent 87cc44d commit 38cd91c

File tree

2 files changed

+22
-0
lines changed

2 files changed

+22
-0
lines changed

src/llama_stack_client/lib/cli/datasets/datasets.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77

88
from .list import list_datasets
99
from .register import register
10+
from .unregister import unregister
1011

1112

1213
@click.group()
@@ -18,3 +19,4 @@ def datasets():
1819
# Register subcommands
1920
datasets.add_command(list_datasets)
2021
datasets.add_command(register)
22+
datasets.add_command(unregister)
Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
# Copyright (c) Meta Platforms, Inc. and affiliates.
2+
# All rights reserved.
3+
#
4+
# This source code is licensed under the terms described in the LICENSE file in
5+
# the root directory of this source tree.
6+
import click
7+
8+
from ..common.utils import handle_client_errors
9+
10+
11+
@click.command("unregister")
12+
@click.help_option("-h", "--help")
13+
@click.argument("dataset-id", required=True)
14+
@click.pass_context
15+
@handle_client_errors("unregister dataset")
16+
def unregister(ctx, dataset_id: str):
17+
"""Remove a dataset"""
18+
client = ctx.obj["client"]
19+
client.datasets.unregister(dataset_id=dataset_id)
20+
click.echo(f"Dataset '{dataset_id}' unregistered successfully")

0 commit comments

Comments
 (0)