aboutsummaryrefslogtreecommitdiff
path: root/.venv/lib/python3.12/site-packages/openai/cli/_api/models.py
diff options
context:
space:
mode:
Diffstat (limited to '.venv/lib/python3.12/site-packages/openai/cli/_api/models.py')
-rw-r--r--.venv/lib/python3.12/site-packages/openai/cli/_api/models.py45
1 files changed, 45 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/openai/cli/_api/models.py b/.venv/lib/python3.12/site-packages/openai/cli/_api/models.py
new file mode 100644
index 00000000..017218fa
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/openai/cli/_api/models.py
@@ -0,0 +1,45 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+from argparse import ArgumentParser
+
+from .._utils import get_client, print_model
+from .._models import BaseModel
+
+if TYPE_CHECKING:
+ from argparse import _SubParsersAction
+
+
+def register(subparser: _SubParsersAction[ArgumentParser]) -> None:
+ sub = subparser.add_parser("models.list")
+ sub.set_defaults(func=CLIModels.list)
+
+ sub = subparser.add_parser("models.retrieve")
+ sub.add_argument("-i", "--id", required=True, help="The model ID")
+ sub.set_defaults(func=CLIModels.get, args_model=CLIModelIDArgs)
+
+ sub = subparser.add_parser("models.delete")
+ sub.add_argument("-i", "--id", required=True, help="The model ID")
+ sub.set_defaults(func=CLIModels.delete, args_model=CLIModelIDArgs)
+
+
+class CLIModelIDArgs(BaseModel):
+ id: str
+
+
+class CLIModels:
+ @staticmethod
+ def get(args: CLIModelIDArgs) -> None:
+ model = get_client().models.retrieve(model=args.id)
+ print_model(model)
+
+ @staticmethod
+ def delete(args: CLIModelIDArgs) -> None:
+ model = get_client().models.delete(model=args.id)
+ print_model(model)
+
+ @staticmethod
+ def list() -> None:
+ models = get_client().models.list()
+ for model in models:
+ print_model(model)