fix: enable datasets works

Author: gyliu513

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

@@ -21,7 +21,7 @@ def list_datasets(ctx):
     console = Console()
     headers = ["identifier", "provider_id", "metadata", "type", "purpose"]
 
-    datasets_list_response = client.datasets.list()
+    datasets_list_response = client.beta.datasets.list()
     if datasets_list_response:
         table = Table()
         for header in headers:

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

@@ -68,7 +68,7 @@ def register(
         if not url:
             raise click.BadParameter("URL is required when dataset path is not specified")
 
-    response = client.datasets.register(
+    response = client.beta.datasets.register(
         dataset_id=dataset_id,
         source={"uri": url},
         metadata=metadata,

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

@@ -16,5 +16,5 @@
 def unregister(ctx, dataset_id: str):
     """Remove a dataset"""
     client = ctx.obj["client"]
-    client.datasets.unregister(dataset_id=dataset_id)
+    client.beta.datasets.unregister(dataset_id=dataset_id)
     click.echo(f"Dataset '{dataset_id}' unregistered successfully")

src/llama_stack_client/lib/cli/eval/run_benchmark.py

@@ -100,7 +100,7 @@ def run_benchmark(
         scoring_functions = benchmark.scoring_functions
         dataset_id = benchmark.dataset_id
 
-        results = client.datasets.iterrows(dataset_id=dataset_id, limit=-1 if num_examples is None else num_examples)
+        results = client.beta.datasets.iterrows(dataset_id=dataset_id, limit=-1 if num_examples is None else num_examples)
 
         output_res = {}
 

src/llama_stack_client/lib/cli/eval/run_scoring.py

@@ -77,15 +77,15 @@ def run_scoring(
     output_res = {}
 
     if dataset_id is not None:
-        dataset = client.datasets.retrieve(dataset_id=dataset_id)
+        dataset = client.beta.datasets.retrieve(dataset_id=dataset_id)
         if not dataset:
             click.BadParameter(
                 f"Dataset {dataset_id} not found. Please register using llama-stack-client datasets register"
             )
 
         # TODO: this will eventually be replaced with jobs polling from server vis score_bath
         # For now, get all datasets rows via datasets API
-        results = client.datasets.iterrows(dataset_id=dataset_id, limit=-1 if num_examples is None else num_examples)
+        results = client.beta.datasets.iterrows(dataset_id=dataset_id, limit=-1 if num_examples is None else num_examples)
         rows = results.rows
 
     if dataset_path is not None: