fixed integration tests and added back usage_type filtering since PR got merged togethercomputer/together-dedicated-endpoints#92

Author: atihkin

poetry.lock

@@ -43,6 +43,7 @@ numpy = [
     { version = ">=1.26.0", python = ">=3.12" },
 ]
 pillow = "^11.1.0"
+black = "^25.9.0"
 
 [tool.poetry.extras]
 pyarrow = ["pyarrow"]

pyproject.toml

@@ -343,16 +343,24 @@ def delete(client: Together, endpoint_id: str) -> None:
     default=None,
     help="true (only mine), false (exclude mine), default=all",
 )
+@click.option(
+    "--usage-type",
+    type=click.Choice(["on-demand", "reserved"]),
+    help="Filter by endpoint usage type",
+)
 @click.pass_obj
 @handle_api_errors
 def list(
-    client: Together, 
-    json: bool, 
+    client: Together,
+    json: bool,
     type: Literal["dedicated", "serverless"] | None,
+    usage_type: Literal["on-demand", "reserved"] | None,
     mine: bool | None,
 ) -> None:
     """List all inference endpoints (includes both dedicated and serverless endpoints)."""
-    endpoints: List[ListEndpoint] = client.endpoints.list(type=type, mine=mine)
+    endpoints: List[ListEndpoint] = client.endpoints.list(
+        type=type, usage_type=usage_type, mine=mine
+    )
 
     if not endpoints:
         click.echo("No dedicated endpoints found", err=True)

src/together/cli/api/endpoints.py

@@ -15,13 +15,15 @@ def __init__(self, client: TogetherClient) -> None:
     def list(
         self,
         type: Optional[Literal["dedicated", "serverless"]] = None,
+        usage_type: Optional[Literal["on-demand", "reserved"]] = None,
         mine: Optional[bool] = None,
     ) -> List[ListEndpoint]:
         """
         List all endpoints, can be filtered by endpoint type and ownership.
 
         Args:
             type (str, optional): Filter endpoints by endpoint type ("dedicated" or "serverless"). Defaults to None.
+            usage_type (str, optional): Filter endpoints by usage type ("on-demand" or "reserved"). Defaults to None.
             mine (bool, optional): If True, return only endpoints owned by the caller. If False, return endpoints not owned by the caller. Defaults to None.
 
         Returns:
@@ -31,9 +33,18 @@ def list(
             client=self._client,
         )
 
-        params = {}
+        params: Dict[
+            str,
+            Union[
+                Literal["dedicated", "serverless"],
+                Literal["on-demand", "reserved"],
+                bool,
+            ],
+        ] = {}
         if type is not None:
             params["type"] = type
+        if usage_type is not None:
+            params["usage_type"] = usage_type
         if mine is not None:
             params["mine"] = mine
 
@@ -270,13 +281,15 @@ def __init__(self, client: TogetherClient) -> None:
     async def list(
         self,
         type: Optional[Literal["dedicated", "serverless"]] = None,
+        usage_type: Optional[Literal["on-demand", "reserved"]] = None,
         mine: Optional[bool] = None,
     ) -> List[ListEndpoint]:
         """
         List all endpoints, can be filtered by type and ownership.
 
         Args:
             type (str, optional): Filter endpoints by type ("dedicated" or "serverless"). Defaults to None.
+            usage_type (str, optional): Filter endpoints by usage type ("on-demand" or "reserved"). Defaults to None.
             mine (bool, optional): If True, return only endpoints owned by the caller. If False, return endpoints not owned by the caller. Defaults to None.
 
         Returns:
@@ -286,9 +299,18 @@ async def list(
             client=self._client,
         )
 
-        params = {}
+        params: Dict[
+            str,
+            Union[
+                Literal["dedicated", "serverless"],
+                Literal["on-demand", "reserved"],
+                bool,
+            ],
+        ] = {}
         if type is not None:
             params["type"] = type
+        if usage_type is not None:
+            params["usage_type"] = usage_type
         if mine is not None:
             params["mine"] = mine