c63a5cfe-b235-4fbe-8bbb-82a9e02a482a-python 0.1.0a3__py3-none-any.whl → 0.1.0a4__py3-none-any.whl

{digitalocean_genai_sdk → gradientai}/_base_client.py

@@ -389,7 +389,7 @@ class BaseClient(Generic[_HttpxClientT, _DefaultStreamT]):
 
         if max_retries is None:  # pyright: ignore[reportUnnecessaryComparison]
             raise TypeError(
-                "max_retries cannot be None. If you want to disable retries, pass `0`; if you want unlimited retries, pass `math.inf` or a very high number; if you want the default behavior, pass `digitalocean_genai_sdk.DEFAULT_MAX_RETRIES`"
+                "max_retries cannot be None. If you want to disable retries, pass `0`; if you want unlimited retries, pass `math.inf` or a very high number; if you want the default behavior, pass `gradientai.DEFAULT_MAX_RETRIES`"
             )
 
     def _enforce_trailing_slash(self, url: URL) -> URL:
@@ -1071,7 +1071,14 @@ class SyncAPIClient(BaseClient[httpx.Client, Stream[Any]]):
     ) -> ResponseT:
         origin = get_origin(cast_to) or cast_to
 
-        if inspect.isclass(origin) and issubclass(origin, BaseAPIResponse):
+        if (
+            inspect.isclass(origin)
+            and issubclass(origin, BaseAPIResponse)
+            # we only want to actually return the custom BaseAPIResponse class if we're
+            # returning the raw response, or if we're not streaming SSE, as if we're streaming
+            # SSE then `cast_to` doesn't actively reflect the type we need to parse into
+            and (not stream or bool(response.request.headers.get(RAW_RESPONSE_HEADER)))
+        ):
             if not issubclass(origin, APIResponse):
                 raise TypeError(f"API Response types must subclass {APIResponse}; Received {origin}")
 
@@ -1282,6 +1289,24 @@ class _DefaultAsyncHttpxClient(httpx.AsyncClient):
         super().__init__(**kwargs)
 
 
+try:
+    import httpx_aiohttp
+except ImportError:
+
+    class _DefaultAioHttpClient(httpx.AsyncClient):
+        def __init__(self, **_kwargs: Any) -> None:
+            raise RuntimeError("To use the aiohttp client you must have installed the package with the `aiohttp` extra")
+else:
+
+    class _DefaultAioHttpClient(httpx_aiohttp.HttpxAiohttpClient):  # type: ignore
+        def __init__(self, **kwargs: Any) -> None:
+            kwargs.setdefault("timeout", DEFAULT_TIMEOUT)
+            kwargs.setdefault("limits", DEFAULT_CONNECTION_LIMITS)
+            kwargs.setdefault("follow_redirects", True)
+
+            super().__init__(**kwargs)
+
+
 if TYPE_CHECKING:
     DefaultAsyncHttpxClient = httpx.AsyncClient
     """An alias to `httpx.AsyncClient` that provides the same defaults that this SDK
@@ -1290,8 +1315,12 @@ if TYPE_CHECKING:
     This is useful because overriding the `http_client` with your own instance of
     `httpx.AsyncClient` will result in httpx's defaults being used, not ours.
     """
+
+    DefaultAioHttpClient = httpx.AsyncClient
+    """An alias to `httpx.AsyncClient` that changes the default HTTP transport to `aiohttp`."""
 else:
     DefaultAsyncHttpxClient = _DefaultAsyncHttpxClient
+    DefaultAioHttpClient = _DefaultAioHttpClient
 
 
 class AsyncHttpxClientWrapper(DefaultAsyncHttpxClient):
@@ -1574,7 +1603,14 @@ class AsyncAPIClient(BaseClient[httpx.AsyncClient, AsyncStream[Any]]):
     ) -> ResponseT:
         origin = get_origin(cast_to) or cast_to
 
-        if inspect.isclass(origin) and issubclass(origin, BaseAPIResponse):
+        if (
+            inspect.isclass(origin)
+            and issubclass(origin, BaseAPIResponse)
+            # we only want to actually return the custom BaseAPIResponse class if we're
+            # returning the raw response, or if we're not streaming SSE, as if we're streaming
+            # SSE then `cast_to` doesn't actively reflect the type we need to parse into
+            and (not stream or bool(response.request.headers.get(RAW_RESPONSE_HEADER)))
+        ):
             if not issubclass(origin, AsyncAPIResponse):
                 raise TypeError(f"API Response types must subclass {AsyncAPIResponse}; Received {origin}")
 

{digitalocean_genai_sdk → gradientai}/_client.py

@@ -23,7 +23,7 @@ from ._utils import is_given, get_async_library
 from ._compat import cached_property
 from ._version import __version__
 from ._streaming import Stream as Stream, AsyncStream as AsyncStream
-from ._exceptions import APIStatusError, DigitaloceanGenaiSDKError
+from ._exceptions import APIStatusError, GradientAIError
 from ._base_client import (
     DEFAULT_MAX_RETRIES,
     SyncAPIClient,
@@ -31,26 +31,13 @@ from ._base_client import (
 )
 
 if TYPE_CHECKING:
-    from .resources import (
-        auth,
-        chat,
-        agents,
-        models,
-        regions,
-        api_keys,
-        providers,
-        embeddings,
-        indexing_jobs,
-        knowledge_bases,
-    )
-    from .resources.chat import ChatResource, AsyncChatResource
+    from .resources import chat, agents, models, regions, inference, providers, indexing_jobs, knowledge_bases
     from .resources.models import ModelsResource, AsyncModelsResource
-    from .resources.regions import RegionsResource, AsyncRegionsResource
-    from .resources.auth.auth import AuthResource, AsyncAuthResource
-    from .resources.embeddings import EmbeddingsResource, AsyncEmbeddingsResource
+    from .resources.chat.chat import ChatResource, AsyncChatResource
     from .resources.agents.agents import AgentsResource, AsyncAgentsResource
     from .resources.indexing_jobs import IndexingJobsResource, AsyncIndexingJobsResource
-    from .resources.api_keys.api_keys import APIKeysResource, AsyncAPIKeysResource
+    from .resources.regions.regions import RegionsResource, AsyncRegionsResource
+    from .resources.inference.inference import InferenceResource, AsyncInferenceResource
     from .resources.providers.providers import ProvidersResource, AsyncProvidersResource
     from .resources.knowledge_bases.knowledge_bases import KnowledgeBasesResource, AsyncKnowledgeBasesResource
 
@@ -59,14 +46,14 @@ __all__ = [
     "Transport",
     "ProxiesTypes",
     "RequestOptions",
-    "DigitaloceanGenaiSDK",
-    "AsyncDigitaloceanGenaiSDK",
+    "GradientAI",
+    "AsyncGradientAI",
     "Client",
     "AsyncClient",
 ]
 
 
-class DigitaloceanGenaiSDK(SyncAPIClient):
+class GradientAI(SyncAPIClient):
     # client options
     api_key: str
 
@@ -93,20 +80,21 @@ class DigitaloceanGenaiSDK(SyncAPIClient):
         # part of our public interface in the future.
         _strict_response_validation: bool = False,
     ) -> None:
-        """Construct a new synchronous DigitaloceanGenaiSDK client instance.
+        """Construct a new synchronous GradientAI client instance.
 
-        This automatically infers the `api_key` argument from the `DIGITALOCEAN_GENAI_SDK_API_KEY` environment variable if it is not provided.
+        This automatically infers the `api_key` argument from the `GRADIENTAI_API_KEY` environment variable if it is not provided.
         """
         if api_key is None:
-            api_key = os.environ.get("DIGITALOCEAN_GENAI_SDK_API_KEY")
+            api_key = os.environ.get("GRADIENTAI_API_KEY")
         if api_key is None:
-            raise DigitaloceanGenaiSDKError(
-                "The api_key client option must be set either by passing api_key to the client or by setting the DIGITALOCEAN_GENAI_SDK_API_KEY environment variable"
+            raise GradientAIError(
+                "The api_key client option must be set either by passing api_key to the client or by setting the GRADIENTAI_API_KEY environment variable"
             )
         self.api_key = api_key
 
         if base_url is None:
-            base_url = os.environ.get("DIGITALOCEAN_GENAI_SDK_BASE_URL")
+            base_url = os.environ.get("GRADIENT_AI_BASE_URL")
+        self._base_url_overridden = base_url is not None
         if base_url is None:
             base_url = f"https://api.digitalocean.com/"
 
@@ -133,12 +121,6 @@ class DigitaloceanGenaiSDK(SyncAPIClient):
 
         return ProvidersResource(self)
 
-    @cached_property
-    def auth(self) -> AuthResource:
-        from .resources.auth import AuthResource
-
-        return AuthResource(self)
-
     @cached_property
     def regions(self) -> RegionsResource:
         from .resources.regions import RegionsResource
@@ -157,12 +139,6 @@ class DigitaloceanGenaiSDK(SyncAPIClient):
 
         return KnowledgeBasesResource(self)
 
-    @cached_property
-    def api_keys(self) -> APIKeysResource:
-        from .resources.api_keys import APIKeysResource
-
-        return APIKeysResource(self)
-
     @cached_property
     def chat(self) -> ChatResource:
         from .resources.chat import ChatResource
@@ -170,10 +146,10 @@ class DigitaloceanGenaiSDK(SyncAPIClient):
         return ChatResource(self)
 
     @cached_property
-    def embeddings(self) -> EmbeddingsResource:
-        from .resources.embeddings import EmbeddingsResource
+    def inference(self) -> InferenceResource:
+        from .resources.inference import InferenceResource
 
-        return EmbeddingsResource(self)
+        return InferenceResource(self)
 
     @cached_property
     def models(self) -> ModelsResource:
@@ -182,12 +158,12 @@ class DigitaloceanGenaiSDK(SyncAPIClient):
         return ModelsResource(self)
 
     @cached_property
-    def with_raw_response(self) -> DigitaloceanGenaiSDKWithRawResponse:
-        return DigitaloceanGenaiSDKWithRawResponse(self)
+    def with_raw_response(self) -> GradientAIWithRawResponse:
+        return GradientAIWithRawResponse(self)
 
     @cached_property
-    def with_streaming_response(self) -> DigitaloceanGenaiSDKWithStreamedResponse:
-        return DigitaloceanGenaiSDKWithStreamedResponse(self)
+    def with_streaming_response(self) -> GradientAIWithStreamedResponse:
+        return GradientAIWithStreamedResponse(self)
 
     @property
     @override
@@ -245,7 +221,7 @@ class DigitaloceanGenaiSDK(SyncAPIClient):
             params = set_default_query
 
         http_client = http_client or self._client
-        return self.__class__(
+        client = self.__class__(
             api_key=api_key or self.api_key,
             base_url=base_url or self.base_url,
             timeout=self.timeout if isinstance(timeout, NotGiven) else timeout,
@@ -255,6 +231,8 @@ class DigitaloceanGenaiSDK(SyncAPIClient):
             default_query=params,
             **_extra_kwargs,
         )
+        client._base_url_overridden = self._base_url_overridden or base_url is not None
+        return client
 
     # Alias for `copy` for nicer inline usage, e.g.
     # client.with_options(timeout=10).foo.create(...)
@@ -294,7 +272,7 @@ class DigitaloceanGenaiSDK(SyncAPIClient):
         return APIStatusError(err_msg, response=response, body=body)
 
 
-class AsyncDigitaloceanGenaiSDK(AsyncAPIClient):
+class AsyncGradientAI(AsyncAPIClient):
     # client options
     api_key: str
 
@@ -321,20 +299,21 @@ class AsyncDigitaloceanGenaiSDK(AsyncAPIClient):
         # part of our public interface in the future.
         _strict_response_validation: bool = False,
     ) -> None:
-        """Construct a new async AsyncDigitaloceanGenaiSDK client instance.
+        """Construct a new async AsyncGradientAI client instance.
 
-        This automatically infers the `api_key` argument from the `DIGITALOCEAN_GENAI_SDK_API_KEY` environment variable if it is not provided.
+        This automatically infers the `api_key` argument from the `GRADIENTAI_API_KEY` environment variable if it is not provided.
         """
         if api_key is None:
-            api_key = os.environ.get("DIGITALOCEAN_GENAI_SDK_API_KEY")
+            api_key = os.environ.get("GRADIENTAI_API_KEY")
         if api_key is None:
-            raise DigitaloceanGenaiSDKError(
-                "The api_key client option must be set either by passing api_key to the client or by setting the DIGITALOCEAN_GENAI_SDK_API_KEY environment variable"
+            raise GradientAIError(
+                "The api_key client option must be set either by passing api_key to the client or by setting the GRADIENTAI_API_KEY environment variable"
             )
         self.api_key = api_key
 
         if base_url is None:
-            base_url = os.environ.get("DIGITALOCEAN_GENAI_SDK_BASE_URL")
+            base_url = os.environ.get("GRADIENT_AI_BASE_URL")
+        self._base_url_overridden = base_url is not None
         if base_url is None:
             base_url = f"https://api.digitalocean.com/"
 
@@ -361,12 +340,6 @@ class AsyncDigitaloceanGenaiSDK(AsyncAPIClient):
 
         return AsyncProvidersResource(self)
 
-    @cached_property
-    def auth(self) -> AsyncAuthResource:
-        from .resources.auth import AsyncAuthResource
-
-        return AsyncAuthResource(self)
-
     @cached_property
     def regions(self) -> AsyncRegionsResource:
         from .resources.regions import AsyncRegionsResource
@@ -385,12 +358,6 @@ class AsyncDigitaloceanGenaiSDK(AsyncAPIClient):
 
         return AsyncKnowledgeBasesResource(self)
 
-    @cached_property
-    def api_keys(self) -> AsyncAPIKeysResource:
-        from .resources.api_keys import AsyncAPIKeysResource
-
-        return AsyncAPIKeysResource(self)
-
     @cached_property
     def chat(self) -> AsyncChatResource:
         from .resources.chat import AsyncChatResource
@@ -398,10 +365,10 @@ class AsyncDigitaloceanGenaiSDK(AsyncAPIClient):
         return AsyncChatResource(self)
 
     @cached_property
-    def embeddings(self) -> AsyncEmbeddingsResource:
-        from .resources.embeddings import AsyncEmbeddingsResource
+    def inference(self) -> AsyncInferenceResource:
+        from .resources.inference import AsyncInferenceResource
 
-        return AsyncEmbeddingsResource(self)
+        return AsyncInferenceResource(self)
 
     @cached_property
     def models(self) -> AsyncModelsResource:
@@ -410,12 +377,12 @@ class AsyncDigitaloceanGenaiSDK(AsyncAPIClient):
         return AsyncModelsResource(self)
 
     @cached_property
-    def with_raw_response(self) -> AsyncDigitaloceanGenaiSDKWithRawResponse:
-        return AsyncDigitaloceanGenaiSDKWithRawResponse(self)
+    def with_raw_response(self) -> AsyncGradientAIWithRawResponse:
+        return AsyncGradientAIWithRawResponse(self)
 
     @cached_property
-    def with_streaming_response(self) -> AsyncDigitaloceanGenaiSDKWithStreamedResponse:
-        return AsyncDigitaloceanGenaiSDKWithStreamedResponse(self)
+    def with_streaming_response(self) -> AsyncGradientAIWithStreamedResponse:
+        return AsyncGradientAIWithStreamedResponse(self)
 
     @property
     @override
@@ -473,7 +440,7 @@ class AsyncDigitaloceanGenaiSDK(AsyncAPIClient):
             params = set_default_query
 
         http_client = http_client or self._client
-        return self.__class__(
+        client = self.__class__(
             api_key=api_key or self.api_key,
             base_url=base_url or self.base_url,
             timeout=self.timeout if isinstance(timeout, NotGiven) else timeout,
@@ -483,6 +450,8 @@ class AsyncDigitaloceanGenaiSDK(AsyncAPIClient):
             default_query=params,
             **_extra_kwargs,
         )
+        client._base_url_overridden = self._base_url_overridden or base_url is not None
+        return client
 
     # Alias for `copy` for nicer inline usage, e.g.
     # client.with_options(timeout=10).foo.create(...)
@@ -522,10 +491,10 @@ class AsyncDigitaloceanGenaiSDK(AsyncAPIClient):
         return APIStatusError(err_msg, response=response, body=body)
 
 
-class DigitaloceanGenaiSDKWithRawResponse:
-    _client: DigitaloceanGenaiSDK
+class GradientAIWithRawResponse:
+    _client: GradientAI
 
-    def __init__(self, client: DigitaloceanGenaiSDK) -> None:
+    def __init__(self, client: GradientAI) -> None:
         self._client = client
 
     @cached_property
@@ -540,12 +509,6 @@ class DigitaloceanGenaiSDKWithRawResponse:
 
         return ProvidersResourceWithRawResponse(self._client.providers)
 
-    @cached_property
-    def auth(self) -> auth.AuthResourceWithRawResponse:
-        from .resources.auth import AuthResourceWithRawResponse
-
-        return AuthResourceWithRawResponse(self._client.auth)
-
     @cached_property
     def regions(self) -> regions.RegionsResourceWithRawResponse:
         from .resources.regions import RegionsResourceWithRawResponse
@@ -564,12 +527,6 @@ class DigitaloceanGenaiSDKWithRawResponse:
 
         return KnowledgeBasesResourceWithRawResponse(self._client.knowledge_bases)
 
-    @cached_property
-    def api_keys(self) -> api_keys.APIKeysResourceWithRawResponse:
-        from .resources.api_keys import APIKeysResourceWithRawResponse
-
-        return APIKeysResourceWithRawResponse(self._client.api_keys)
-
     @cached_property
     def chat(self) -> chat.ChatResourceWithRawResponse:
         from .resources.chat import ChatResourceWithRawResponse
@@ -577,10 +534,10 @@ class DigitaloceanGenaiSDKWithRawResponse:
         return ChatResourceWithRawResponse(self._client.chat)
 
     @cached_property
-    def embeddings(self) -> embeddings.EmbeddingsResourceWithRawResponse:
-        from .resources.embeddings import EmbeddingsResourceWithRawResponse
+    def inference(self) -> inference.InferenceResourceWithRawResponse:
+        from .resources.inference import InferenceResourceWithRawResponse
 
-        return EmbeddingsResourceWithRawResponse(self._client.embeddings)
+        return InferenceResourceWithRawResponse(self._client.inference)
 
     @cached_property
     def models(self) -> models.ModelsResourceWithRawResponse:
@@ -589,10 +546,10 @@ class DigitaloceanGenaiSDKWithRawResponse:
         return ModelsResourceWithRawResponse(self._client.models)
 
 
-class AsyncDigitaloceanGenaiSDKWithRawResponse:
-    _client: AsyncDigitaloceanGenaiSDK
+class AsyncGradientAIWithRawResponse:
+    _client: AsyncGradientAI
 
-    def __init__(self, client: AsyncDigitaloceanGenaiSDK) -> None:
+    def __init__(self, client: AsyncGradientAI) -> None:
         self._client = client
 
     @cached_property
@@ -607,12 +564,6 @@ class AsyncDigitaloceanGenaiSDKWithRawResponse:
 
         return AsyncProvidersResourceWithRawResponse(self._client.providers)
 
-    @cached_property
-    def auth(self) -> auth.AsyncAuthResourceWithRawResponse:
-        from .resources.auth import AsyncAuthResourceWithRawResponse
-
-        return AsyncAuthResourceWithRawResponse(self._client.auth)
-
     @cached_property
     def regions(self) -> regions.AsyncRegionsResourceWithRawResponse:
         from .resources.regions import AsyncRegionsResourceWithRawResponse
@@ -631,12 +582,6 @@ class AsyncDigitaloceanGenaiSDKWithRawResponse:
 
         return AsyncKnowledgeBasesResourceWithRawResponse(self._client.knowledge_bases)
 
-    @cached_property
-    def api_keys(self) -> api_keys.AsyncAPIKeysResourceWithRawResponse:
-        from .resources.api_keys import AsyncAPIKeysResourceWithRawResponse
-
-        return AsyncAPIKeysResourceWithRawResponse(self._client.api_keys)
-
     @cached_property
     def chat(self) -> chat.AsyncChatResourceWithRawResponse:
         from .resources.chat import AsyncChatResourceWithRawResponse
@@ -644,10 +589,10 @@ class AsyncDigitaloceanGenaiSDKWithRawResponse:
         return AsyncChatResourceWithRawResponse(self._client.chat)
 
     @cached_property
-    def embeddings(self) -> embeddings.AsyncEmbeddingsResourceWithRawResponse:
-        from .resources.embeddings import AsyncEmbeddingsResourceWithRawResponse
+    def inference(self) -> inference.AsyncInferenceResourceWithRawResponse:
+        from .resources.inference import AsyncInferenceResourceWithRawResponse
 
-        return AsyncEmbeddingsResourceWithRawResponse(self._client.embeddings)
+        return AsyncInferenceResourceWithRawResponse(self._client.inference)
 
     @cached_property
     def models(self) -> models.AsyncModelsResourceWithRawResponse:
@@ -656,10 +601,10 @@ class AsyncDigitaloceanGenaiSDKWithRawResponse:
         return AsyncModelsResourceWithRawResponse(self._client.models)
 
 
-class DigitaloceanGenaiSDKWithStreamedResponse:
-    _client: DigitaloceanGenaiSDK
+class GradientAIWithStreamedResponse:
+    _client: GradientAI
 
-    def __init__(self, client: DigitaloceanGenaiSDK) -> None:
+    def __init__(self, client: GradientAI) -> None:
         self._client = client
 
     @cached_property
@@ -674,12 +619,6 @@ class DigitaloceanGenaiSDKWithStreamedResponse:
 
         return ProvidersResourceWithStreamingResponse(self._client.providers)
 
-    @cached_property
-    def auth(self) -> auth.AuthResourceWithStreamingResponse:
-        from .resources.auth import AuthResourceWithStreamingResponse
-
-        return AuthResourceWithStreamingResponse(self._client.auth)
-
     @cached_property
     def regions(self) -> regions.RegionsResourceWithStreamingResponse:
         from .resources.regions import RegionsResourceWithStreamingResponse
@@ -698,12 +637,6 @@ class DigitaloceanGenaiSDKWithStreamedResponse:
 
         return KnowledgeBasesResourceWithStreamingResponse(self._client.knowledge_bases)
 
-    @cached_property
-    def api_keys(self) -> api_keys.APIKeysResourceWithStreamingResponse:
-        from .resources.api_keys import APIKeysResourceWithStreamingResponse
-
-        return APIKeysResourceWithStreamingResponse(self._client.api_keys)
-
     @cached_property
     def chat(self) -> chat.ChatResourceWithStreamingResponse:
         from .resources.chat import ChatResourceWithStreamingResponse
@@ -711,10 +644,10 @@ class DigitaloceanGenaiSDKWithStreamedResponse:
         return ChatResourceWithStreamingResponse(self._client.chat)
 
     @cached_property
-    def embeddings(self) -> embeddings.EmbeddingsResourceWithStreamingResponse:
-        from .resources.embeddings import EmbeddingsResourceWithStreamingResponse
+    def inference(self) -> inference.InferenceResourceWithStreamingResponse:
+        from .resources.inference import InferenceResourceWithStreamingResponse
 
-        return EmbeddingsResourceWithStreamingResponse(self._client.embeddings)
+        return InferenceResourceWithStreamingResponse(self._client.inference)
 
     @cached_property
     def models(self) -> models.ModelsResourceWithStreamingResponse:
@@ -723,10 +656,10 @@ class DigitaloceanGenaiSDKWithStreamedResponse:
         return ModelsResourceWithStreamingResponse(self._client.models)
 
 
-class AsyncDigitaloceanGenaiSDKWithStreamedResponse:
-    _client: AsyncDigitaloceanGenaiSDK
+class AsyncGradientAIWithStreamedResponse:
+    _client: AsyncGradientAI
 
-    def __init__(self, client: AsyncDigitaloceanGenaiSDK) -> None:
+    def __init__(self, client: AsyncGradientAI) -> None:
         self._client = client
 
     @cached_property
@@ -741,12 +674,6 @@ class AsyncDigitaloceanGenaiSDKWithStreamedResponse:
 
         return AsyncProvidersResourceWithStreamingResponse(self._client.providers)
 
-    @cached_property
-    def auth(self) -> auth.AsyncAuthResourceWithStreamingResponse:
-        from .resources.auth import AsyncAuthResourceWithStreamingResponse
-
-        return AsyncAuthResourceWithStreamingResponse(self._client.auth)
-
     @cached_property
     def regions(self) -> regions.AsyncRegionsResourceWithStreamingResponse:
         from .resources.regions import AsyncRegionsResourceWithStreamingResponse
@@ -765,12 +692,6 @@ class AsyncDigitaloceanGenaiSDKWithStreamedResponse:
 
         return AsyncKnowledgeBasesResourceWithStreamingResponse(self._client.knowledge_bases)
 
-    @cached_property
-    def api_keys(self) -> api_keys.AsyncAPIKeysResourceWithStreamingResponse:
-        from .resources.api_keys import AsyncAPIKeysResourceWithStreamingResponse
-
-        return AsyncAPIKeysResourceWithStreamingResponse(self._client.api_keys)
-
     @cached_property
     def chat(self) -> chat.AsyncChatResourceWithStreamingResponse:
         from .resources.chat import AsyncChatResourceWithStreamingResponse
@@ -778,10 +699,10 @@ class AsyncDigitaloceanGenaiSDKWithStreamedResponse:
         return AsyncChatResourceWithStreamingResponse(self._client.chat)
 
     @cached_property
-    def embeddings(self) -> embeddings.AsyncEmbeddingsResourceWithStreamingResponse:
-        from .resources.embeddings import AsyncEmbeddingsResourceWithStreamingResponse
+    def inference(self) -> inference.AsyncInferenceResourceWithStreamingResponse:
+        from .resources.inference import AsyncInferenceResourceWithStreamingResponse
 
-        return AsyncEmbeddingsResourceWithStreamingResponse(self._client.embeddings)
+        return AsyncInferenceResourceWithStreamingResponse(self._client.inference)
 
     @cached_property
     def models(self) -> models.AsyncModelsResourceWithStreamingResponse:
@@ -790,6 +711,6 @@ class AsyncDigitaloceanGenaiSDKWithStreamedResponse:
         return AsyncModelsResourceWithStreamingResponse(self._client.models)
 
 
-Client = DigitaloceanGenaiSDK
+Client = GradientAI
 
-AsyncClient = AsyncDigitaloceanGenaiSDK
+AsyncClient = AsyncGradientAI

{digitalocean_genai_sdk → gradientai}/_exceptions.py

@@ -18,11 +18,11 @@ __all__ = [
 ]
 
 
-class DigitaloceanGenaiSDKError(Exception):
+class GradientAIError(Exception):
     pass
 
 
-class APIError(DigitaloceanGenaiSDKError):
+class APIError(GradientAIError):
     message: str
     request: httpx.Request
 

{digitalocean_genai_sdk → gradientai}/_resource.py

@@ -8,13 +8,13 @@ from typing import TYPE_CHECKING
 import anyio
 
 if TYPE_CHECKING:
-    from ._client import DigitaloceanGenaiSDK, AsyncDigitaloceanGenaiSDK
+    from ._client import GradientAI, AsyncGradientAI
 
 
 class SyncAPIResource:
-    _client: DigitaloceanGenaiSDK
+    _client: GradientAI
 
-    def __init__(self, client: DigitaloceanGenaiSDK) -> None:
+    def __init__(self, client: GradientAI) -> None:
         self._client = client
         self._get = client.get
         self._post = client.post
@@ -28,9 +28,9 @@ class SyncAPIResource:
 
 
 class AsyncAPIResource:
-    _client: AsyncDigitaloceanGenaiSDK
+    _client: AsyncGradientAI
 
-    def __init__(self, client: AsyncDigitaloceanGenaiSDK) -> None:
+    def __init__(self, client: AsyncGradientAI) -> None:
         self._client = client
         self._get = client.get
         self._post = client.post

{digitalocean_genai_sdk → gradientai}/_response.py

@@ -29,7 +29,7 @@ from ._utils import is_given, extract_type_arg, is_annotated_type, is_type_alias
 from ._models import BaseModel, is_basemodel
 from ._constants import RAW_RESPONSE_HEADER, OVERRIDE_CAST_TO_HEADER
 from ._streaming import Stream, AsyncStream, is_stream_class_type, extract_stream_chunk_type
-from ._exceptions import DigitaloceanGenaiSDKError, APIResponseValidationError
+from ._exceptions import GradientAIError, APIResponseValidationError
 
 if TYPE_CHECKING:
     from ._models import FinalRequestOptions
@@ -218,7 +218,7 @@ class BaseAPIResponse(Generic[R]):
             and issubclass(origin, pydantic.BaseModel)
         ):
             raise TypeError(
-                "Pydantic models must subclass our base model type, e.g. `from digitalocean_genai_sdk import BaseModel`"
+                "Pydantic models must subclass our base model type, e.g. `from gradientai import BaseModel`"
             )
 
         if (
@@ -285,7 +285,7 @@ class APIResponse(BaseAPIResponse[R]):
         the `to` argument, e.g.
 
         ```py
-        from digitalocean_genai_sdk import BaseModel
+        from gradientai import BaseModel
 
 
         class MyModel(BaseModel):
@@ -387,7 +387,7 @@ class AsyncAPIResponse(BaseAPIResponse[R]):
         the `to` argument, e.g.
 
         ```py
-        from digitalocean_genai_sdk import BaseModel
+        from gradientai import BaseModel
 
 
         class MyModel(BaseModel):
@@ -558,11 +558,11 @@ class AsyncStreamedBinaryAPIResponse(AsyncAPIResponse[bytes]):
 class MissingStreamClassError(TypeError):
     def __init__(self) -> None:
         super().__init__(
-            "The `stream` argument was set to `True` but the `stream_cls` argument was not given. See `digitalocean_genai_sdk._streaming` for reference",
+            "The `stream` argument was set to `True` but the `stream_cls` argument was not given. See `gradientai._streaming` for reference",
         )
 
 
-class StreamAlreadyConsumed(DigitaloceanGenaiSDKError):
+class StreamAlreadyConsumed(GradientAIError):
     """
     Attempted to read or stream content, but the content has already
     been streamed.