Merge pull request #53 from gizatechxyz/fix/endpoints-retrieval

Fix jobs and endpoints retrieval using RootModel

docs/README.md

@@ -1,5 +1,5 @@
 ---
-description: Giza CLI 0.14.0
+description: Giza CLI 0.14.1
 ---
 
 # Giza CLI

docs/examples/full_transpilation.md

@@ -21,7 +21,7 @@ pip install -r requirements.txt
 Or:
 
 ```bash
-pip install giza-cli==0.14.0 onnx==1.14.1 torch==2.1.0 torchvision==0.16.0
+pip install giza-cli==0.14.1 onnx==1.14.1 torch==2.1.0 torchvision==0.16.0
 ```
 
 We will use the libraries for the following purposes:

examples/mnist/mnist_pytorch.ipynb

@@ -41,7 +41,7 @@
     "Or:\n",
     "\n",
     "```bash\n",
-    "pip install giza-cli==0.14.0 onnx==1.14.1 torch==2.1.0 torchvision==0.16.0\n",
+    "pip install giza-cli==0.14.1 onnx==1.14.1 torch==2.1.0 torchvision==0.16.0\n",
     "```\n",
     "\n",
     "We will use the libraries for the following purposes:\n",

examples/mnist/requirements.txt

@@ -1,4 +1,4 @@
-giza-cli==0.14.0
+giza-cli==0.14.1
 onnx==1.14.1
 tf2onnx==1.15.1
 torch==2.1.0

giza/__init__.py

@@ -1,5 +1,5 @@
 import os
 
-__version__ = "0.14.0"
+__version__ = "0.14.1"
 # Until DNS is fixed
 API_HOST = os.environ.get("GIZA_API_HOST", "https://api.gizatech.xyz")

giza/schemas/endpoints.py

@@ -1,6 +1,6 @@
 from typing import Optional
 
-from pydantic import BaseModel, ConfigDict
+from pydantic import BaseModel, ConfigDict, RootModel
 
 from giza.utils.enums import Framework, ServiceSize
 
@@ -33,5 +33,5 @@ class Endpoint(BaseModel):
     model_config["protected_namespaces"] = ()
 
 
-class EndpointsList(BaseModel):
+class EndpointsList(RootModel):
     root: list[Endpoint]

giza/schemas/jobs.py

@@ -1,7 +1,7 @@
 import datetime
 from typing import Optional
 
-from pydantic import BaseModel, ConfigDict
+from pydantic import BaseModel, ConfigDict, RootModel
 
 from giza.utils.enums import Framework, JobKind, JobSize, JobStatus
 
@@ -29,5 +29,5 @@ class JobCreate(BaseModel):
     model_config["protected_namespaces"] = ()
 
 
-class JobList(BaseModel):
+class JobList(RootModel):
     root: list[Job]

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "giza-cli"
-version = "0.14.0"
+version = "0.14.1"
 description = "CLI for interacting with Giza"
 authors = ["Gonzalo Mellizo-Soto <[email protected]>"]
 readme = "README.md"

tests/commands/test_endpoints.py

@@ -144,6 +144,33 @@ def test_list_deployments():
     assert "giza-deployment-2" in result.stdout
 
 
+def test_create_deployments_empty():
+    deployments_list = EndpointsList(root=[])
+    with patch.object(
+        EndpointsClient, "list", return_value=deployments_list
+    ) as mock_list, patch.object(
+        EndpointsClient,
+        "create",
+        return_value=Endpoint(
+            id=1,
+            status="COMPLETED",
+            uri="https://giza-api.com/deployments/1",
+            size="S",
+            service_name="giza-deployment-1",
+            model_id=1,
+            version_id=1,
+            is_active=True,
+        ),
+    ):
+        result = invoke_cli_runner(
+            ["endpoints", "deploy", "--model-id", "1", "--version-id", "1"],
+        )
+    mock_list.assert_called_once()
+    assert result.exit_code == 0
+    assert "Endpoint is successful" in result.stdout
+    assert "https://giza-api.com/deployments/1" in result.stdout
+
+
 def test_list_deployments_http_error():
     with patch.object(EndpointsClient, "list", side_effect=HTTPError):
         result = invoke_cli_runner(