Add test and documentation

Author: nsorros

README.md

@@ -64,7 +64,37 @@ Commands:
 To create an endpoint:
 
 ```
-hugie endpoint create examples/development.json
+Usage: hugie endpoint create [OPTIONS] [DATA]
+
+  Create an endpoint
+
+  Args:     data (str): Path to JSON data to create the endpoint
+
+Arguments:
+  [DATA]  Path JSON data to create the endpoint
+
+Options:
+  --account-id TEXT      ID of the account (for private endpoints)
+  --name TEXT            Name of the endpoint  [default: hf-endpoint]
+  --type TEXT            Type of endpoint, one of ['public', 'protected',
+                         'private']
+  --accelerator TEXT     Accelerator to use. One of ['CPU','GPU']  [default:
+                         cpu]
+  --instance-type TEXT   [default: c6i]
+  --instance-size TEXT   [default: small]
+  --min-replica INTEGER  Minimum number of replicas  [default: 1]
+  --max-replica INTEGER  Maximum number of replicas  [default: 1]
+  --framework TEXT       Framework to use  [default: custom]
+  --repository TEXT      Name of the hf model repository  [default: t5-small]
+  --revision TEXT        Revision of the hf model repository  [default: main]
+  --task TEXT            Task of the model  [default: text-generation]
+  --image TEXT           Image to use from huggingface or tgi  [default:
+                         huggingface]
+  --vendor TEXT          Vendor to use. One of ['aws','gcp']  [default: aws]
+  --region TEXT          Vendor specific region, e.g. 'us-east-1'  [default:
+                         us-east-1]
+  --json                 Prints the full output in JSON.
+  --help                 Show this message and exit.
 ```
 
 ## List

docs/endpoint.md

@@ -30,20 +30,37 @@ hugie endpoint create examples/development.json
 
 Command reference:
 ```
- Usage: hugie endpoint create [OPTIONS] DATA
-
- Create an endpoint
- Args:     data (str): Path to JSON data to create the endpoint
-
-╭─ Arguments ──────────────────────────────────────────────────────────────────╮
-│ *    data      TEXT  Path JSON data to create the endpoint [default: None]   │
-│                      [required]                                              │
-╰──────────────────────────────────────────────────────────────────────────────╯
-╭─ Options ────────────────────────────────────────────────────────────────────╮
-│ --json          Prints the full output in JSON.                              │
-│ --help          Show this message and exit.                                  │
-╰──────────────────────────────────────────────────────────────────────────────╯
-
+Usage: hugie endpoint create [OPTIONS] [DATA]
+
+  Create an endpoint
+
+  Args:     data (str): Path to JSON data to create the endpoint
+
+Arguments:
+  [DATA]  Path JSON data to create the endpoint
+
+Options:
+  --account-id TEXT      ID of the account (for private endpoints)
+  --name TEXT            Name of the endpoint  [default: hf-endpoint]
+  --type TEXT            Type of endpoint, one of ['public', 'protected',
+                         'private']
+  --accelerator TEXT     Accelerator to use. One of ['CPU','GPU']  [default:
+                         cpu]
+  --instance-type TEXT   [default: c6i]
+  --instance-size TEXT   [default: small]
+  --min-replica INTEGER  Minimum number of replicas  [default: 1]
+  --max-replica INTEGER  Maximum number of replicas  [default: 1]
+  --framework TEXT       Framework to use  [default: custom]
+  --repository TEXT      Name of the hf model repository  [default: t5-small]
+  --revision TEXT        Revision of the hf model repository  [default: main]
+  --task TEXT            Task of the model  [default: text-generation]
+  --image TEXT           Image to use from huggingface or tgi  [default:
+                         huggingface]
+  --vendor TEXT          Vendor to use. One of ['aws','gcp']  [default: aws]
+  --region TEXT          Vendor specific region, e.g. 'us-east-1'  [default:
+                         us-east-1]
+  --json                 Prints the full output in JSON.
+  --help                 Show this message and exit.
 ```
 
 ## list/ls

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "hugie"
-version = "0.3.2"
+version = "0.4.0"
 description = "CLI for managing Hugging Face Inference Endpoints"
 authors = ["Matthew Upson <[email protected]>",
             "Nick Sorros <[email protected]>"]

tests/test_endpoint.py

@@ -70,6 +70,38 @@ def raise_for_status():
     assert result.exit_code == 0
 
 
+def test_create_args(monkeypatch):
+    class MockCreateResponse:
+        def json():
+            pass
+
+        def raise_for_status():
+            pass
+
+        status_code = 200
+
+    monkeypatch.setattr("requests.post", lambda url, **kwargs: MockCreateResponse)
+    result = runner.invoke(
+        app,
+        [
+            "create",
+            "--name",
+            "test",
+            "--instance-type",
+            "local",
+            "--instance-size",
+            "medium",
+            "--repository",
+            "bert",
+            "--task",
+            "text-generation",
+            "--accelerator",
+            "gpu",
+        ],
+    )
+    assert result.exit_code == 0
+
+
 def test_update(monkeypatch, data_path):
     class MockUpdateResponse:
         def json():