precommit

Author: yanxi0830

docs/cli_reference.md

@@ -123,4 +123,4 @@ $ llama-stack-client shields list
 +==============+==========+================+=============+
 | llama_guard  | {}       | meta-reference | llama_guard |
 +--------------+----------+----------------+-------------+
-```
+```

src/llama_stack_client/lib/.keep

@@ -1,4 +1,4 @@
 File generated from our OpenAPI spec by Stainless.
 
 This directory can be used to store custom files to expand the SDK.
-It is ignored by Stainless code generation and its content (other than this keep file) won't be touched.
+It is ignored by Stainless code generation and its content (other than this keep file) won't be touched.

src/llama_stack_client/lib/cli/configure.py

@@ -5,12 +5,14 @@
 # the root directory of this source tree.
 
 import os
+
 import click
 import yaml
-from llama_stack_client.lib.cli.constants import get_config_file_path, LLAMA_STACK_CLIENT_CONFIG_DIR
 from prompt_toolkit import prompt
 from prompt_toolkit.validation import Validator
 
+from llama_stack_client.lib.cli.constants import get_config_file_path, LLAMA_STACK_CLIENT_CONFIG_DIR
+
 
 def get_config():
     config_file = get_config_file_path()

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

@@ -4,12 +4,12 @@
 # This source code is licensed under the terms described in the LICENSE file in
 # the root directory of this source tree.
 
+import json
+from typing import Optional
+
 import click
 import yaml
-from typing import Optional
-import json
 
-from llama_models.llama3.api.datatypes import URL
 from .list import list_datasets
 
 
@@ -41,14 +41,14 @@ def register(
 
     try:
         dataset_schema = json.loads(schema)
-    except json.JSONDecodeError:
-        raise click.BadParameter("Schema must be valid JSON")
+    except json.JSONDecodeError as err:
+        raise click.BadParameter("Schema must be valid JSON") from err
 
     if metadata:
         try:
             metadata = json.loads(metadata)
-        except json.JSONDecodeError:
-            raise click.BadParameter("Metadata must be valid JSON")
+        except json.JSONDecodeError as err:
+            raise click.BadParameter("Metadata must be valid JSON") from err
 
     response = client.datasets.register(
         dataset_id=dataset_id,

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

@@ -5,8 +5,8 @@
 # the root directory of this source tree.
 
 import click
-from rich.table import Table
 from rich.console import Console
+from rich.table import Table
 
 
 @click.command("list")

src/llama_stack_client/lib/cli/eval_tasks/eval_tasks.py

@@ -5,11 +5,12 @@
 # the root directory of this source tree.
 
 
-import click
 import json
-import yaml
 from typing import Optional
 
+import click
+import yaml
+
 from .list import list_eval_tasks
 
 
@@ -42,8 +43,8 @@ def register(
     if metadata:
         try:
             metadata = json.loads(metadata)
-        except json.JSONDecodeError:
-            raise click.BadParameter("Metadata must be valid JSON")
+        except json.JSONDecodeError as err:
+            raise click.BadParameter("Metadata must be valid JSON") from err
 
     response = client.eval_tasks.register(
         eval_task_id=eval_task_id,

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

@@ -5,8 +5,8 @@
 # the root directory of this source tree.
 
 import click
-from rich.table import Table
 from rich.console import Console
+from rich.table import Table
 
 
 @click.command("list")

src/llama_stack_client/lib/cli/llama_stack_client.py

@@ -6,10 +6,11 @@
 
 import click
 import yaml
+
 from llama_stack_client import LlamaStackClient
+from .configure import configure
 
 from .constants import get_config_file_path
-from .configure import configure
 from .datasets import datasets
 from .eval_tasks import eval_tasks
 from .memory_banks import memory_banks

src/llama_stack_client/lib/cli/memory_banks/memory_banks.py

@@ -4,11 +4,12 @@
 # This source code is licensed under the terms described in the LICENSE file in
 # the root directory of this source tree.
 
-import click
 from typing import Optional
+
+import click
 import yaml
-from rich.table import Table
 from rich.console import Console
+from rich.table import Table
 
 
 @click.group()

src/llama_stack_client/lib/cli/models/__init__.py

@@ -5,3 +5,5 @@
 # the root directory of this source tree.
 
 from .models import models
+
+__all__ = ["models"]