feat: allow passing single endpoint

Author: umarbutler

CHANGELOG.md

@@ -1,6 +1,10 @@
 ## Changelog 🔄
 All notable changes to `isaacus-sagemaker` will be documented here. This project adheres to [Keep a Changelog](https://keepachangelog.com/en/1.1.0/) and [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.1.4] - 2025-10-28
+### Added
+- Allowed for passing of a single endpoint as the `endpoints` argument to both `IsaacusSageMakerRuntimeHTTPClient` and `AsyncIsaacusSageMakerRuntimeHTTPClient`.
+
 ## [0.1.3] - 2025-10-27
 ### Changed
 - Updated README to link to Isaacus' AWS Marketplace listings.
@@ -24,6 +28,7 @@ All notable changes to `isaacus-sagemaker` will be documented here. This project
 ### Added
 - Initial release of `isaacus-sagemaker`, providing synchronous and asynchronous HTTP clients for interacting with Isaacus models deployed on AWS SageMaker via the SageMaker Runtime InvokeEndpoint API.
 
+[0.1.4]: https://github.com/isaacus-dev/isaacus-sagemaker-python/compare/v0.1.3...v0.1.4
 [0.1.3]: https://github.com/isaacus-dev/isaacus-sagemaker-python/compare/v0.1.2...v0.1.3
 [0.1.2]: https://github.com/isaacus-dev/isaacus-sagemaker-python/compare/v0.1.1...v0.1.2
 [0.1.1]: https://github.com/isaacus-dev/isaacus-sagemaker-python/compare/v0.1.0...v0.1.1

README.md

@@ -92,7 +92,7 @@ class IsaacusSageMakerRuntimeHTTPClient(httpx.Client):
     This client extends `httpx.Client`.
 
     Arguments:
-        `endpoints` (`Sequence[IsaacusSageMakerRuntimeEndpoint]`): A sequence of SageMaker endpoints to route requests to.
+        `endpoints` (`Sequence[IsaacusSageMakerRuntimeEndpoint] | IsaacusSageMakerRuntimeEndpoint`): A sequence of SageMaker endpoints to route requests to or a single endpoint.
         `region` (`str`, optional): The AWS region where the SageMaker endpoints are deployed. Overriden by any region specified at the endpoint-level. Defaults to `None`, in which case the AWS SDK's default region resolution is used.
         `profile` (`str`, optional): The AWS profile to use when accessing the SageMaker endpoints. Overriden by any profile specified at the endpoint-level. Defaults to `None`, in which case the AWS SDK's default profile resolution is used.
         `boto_session_kwargs` (`Dict[str, Any]`, optional): Additional keyword arguments to pass to the `boto3.Session` constructor when creating the AWS session. Defaults to `None`.
@@ -109,7 +109,7 @@ class AsyncIsaacusSageMakerRuntimeHTTPClient(httpx.AsyncClient):
     This client extends `httpx.AsyncClient`.
 
     Arguments:
-        `endpoints` (`Sequence[IsaacusSageMakerRuntimeEndpoint]`): A sequence of SageMaker endpoints to route requests to.
+        `endpoints` (`Sequence[IsaacusSageMakerRuntimeEndpoint] | IsaacusSageMakerRuntimeEndpoint`): A sequence of SageMaker endpoints to route requests to or a single endpoint.
         `region` (`str`, optional): The AWS region where the SageMaker endpoints are deployed. Overriden by any region specified at the endpoint-level. Defaults to `None`, in which case the AWS SDK's default region resolution is used.
         `profile` (`str`, optional): The AWS profile to use when accessing the SageMaker endpoints. Overriden by any profile specified at the endpoint-level. Defaults to `None`, in which case the AWS SDK's default profile resolution is used.
         `boto_session_kwargs` (`Dict[str, Any]`, optional): Additional keyword arguments to pass to the `boto3.Session` constructor when creating the AWS session. Defaults to `None`.

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "isaacus-sagemaker"
-version = "0.1.3"
+version = "0.1.4"
 authors = [
     {name="Isaacus", email="support@isaacus.com"},
 ]

src/isaacus_sagemaker/runtime_client.py

@@ -200,7 +200,7 @@ class IsaacusSageMakerRuntimeHTTPClient(httpx.Client):
     This client extends `httpx.AsyncClient`.
 
     Arguments:
-        `endpoints` (`Sequence[IsaacusSageMakerRuntimeEndpoint]`): A sequence of SageMaker endpoints to route requests to.
+        `endpoints` (`Sequence[IsaacusSageMakerRuntimeEndpoint] | IsaacusSageMakerRuntimeEndpoint`): A sequence of SageMaker endpoints to route requests to or a single endpoint.
         `region` (`str`, optional): The AWS region where the SageMaker endpoints are deployed. Overriden by any region specified at the endpoint-level. Defaults to `None`, in which case the AWS SDK's default region resolution is used.
         `profile` (`str`, optional): The AWS profile to use when accessing the SageMaker endpoints. Overriden by any profile specified at the endpoint-level. Defaults to `None`, in which case the AWS SDK's default profile resolution is used.
         `boto_session_kwargs` (`Dict[str, Any]`, optional): Additional keyword arguments to pass to the `boto3.Session` constructor when creating the AWS session. Defaults to `None`.
@@ -210,13 +210,16 @@ class IsaacusSageMakerRuntimeHTTPClient(httpx.Client):
     def __init__(
         self,
         *,
-        endpoints: Sequence[IsaacusSageMakerRuntimeEndpoint],
+        endpoints: Union[Sequence[IsaacusSageMakerRuntimeEndpoint], IsaacusSageMakerRuntimeEndpoint],
         region: Optional[str] = None,
         profile: Optional[str] = None,
         boto_session_kwargs: Optional[Dict[str, Any]] = None,
         **httpx_kwargs: Any,
     ) -> None:
         super().__init__(**httpx_kwargs)
+        
+        if isinstance(endpoints, IsaacusSageMakerRuntimeEndpoint):
+            endpoints = [endpoints]
 
         self._router = Router(endpoints)
 
@@ -261,7 +264,7 @@ class AsyncIsaacusSageMakerRuntimeHTTPClient(httpx.AsyncClient):
     This client extends `httpx.Client`.
 
     Arguments:
-        `endpoints` (`Sequence[IsaacusSageMakerRuntimeEndpoint]`): A sequence of SageMaker endpoints to route requests to.
+        `endpoints` (`Sequence[IsaacusSageMakerRuntimeEndpoint] | IsaacusSageMakerRuntimeEndpoint`): A sequence of SageMaker endpoints to route requests to or a single endpoint.
         `region` (`str`, optional): The AWS region where the SageMaker endpoints are deployed. Overriden by any region specified at the endpoint-level. Defaults to `None`, in which case the AWS SDK's default region resolution is used.
         `profile` (`str`, optional): The AWS profile to use when accessing the SageMaker endpoints. Overriden by any profile specified at the endpoint-level. Defaults to `None`, in which case the AWS SDK's default profile resolution is used.
         `boto_session_kwargs` (`Dict[str, Any]`, optional): Additional keyword arguments to pass to the `boto3.Session` constructor when creating the AWS session. Defaults to `None`.
@@ -271,13 +274,16 @@ class AsyncIsaacusSageMakerRuntimeHTTPClient(httpx.AsyncClient):
     def __init__(
         self,
         *,
-        endpoints: Sequence[IsaacusSageMakerRuntimeEndpoint],
+        endpoints: Union[Sequence[IsaacusSageMakerRuntimeEndpoint], IsaacusSageMakerRuntimeEndpoint],
         region: Optional[str] = None,
         profile: Optional[str] = None,
         boto_session_kwargs: Optional[Dict[str, Any]] = None,
         **httpx_kwargs: Any,
     ) -> None:
         super().__init__(**httpx_kwargs)
+        
+        if isinstance(endpoints, IsaacusSageMakerRuntimeEndpoint):
+            endpoints = [endpoints]
 
         self._router = Router(endpoints)