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

{digitalocean_genai_sdk → do_gradientai}/_client.py

@@ -13,6 +13,7 @@ from ._qs import Querystring
 from ._types import (
     NOT_GIVEN,
     Omit,
+    Headers,
     Timeout,
     NotGiven,
     Transport,
@@ -23,7 +24,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
 from ._base_client import (
     DEFAULT_MAX_RETRIES,
     SyncAPIClient,
@@ -31,49 +32,37 @@ 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, knowledge_bases, model_providers
     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.providers.providers import ProvidersResource, AsyncProvidersResource
+    from .resources.inference.inference import InferenceResource, AsyncInferenceResource
     from .resources.knowledge_bases.knowledge_bases import KnowledgeBasesResource, AsyncKnowledgeBasesResource
+    from .resources.model_providers.model_providers import ModelProvidersResource, AsyncModelProvidersResource
 
 __all__ = [
     "Timeout",
     "Transport",
     "ProxiesTypes",
     "RequestOptions",
-    "DigitaloceanGenaiSDK",
-    "AsyncDigitaloceanGenaiSDK",
+    "GradientAI",
+    "AsyncGradientAI",
     "Client",
     "AsyncClient",
 ]
 
 
-class DigitaloceanGenaiSDK(SyncAPIClient):
+class GradientAI(SyncAPIClient):
     # client options
-    api_key: str
+    api_key: str | None
+    inference_key: str | None
 
     def __init__(
         self,
         *,
         api_key: str | None = None,
+        inference_key: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: Union[float, Timeout, None, NotGiven] = NOT_GIVEN,
         max_retries: int = DEFAULT_MAX_RETRIES,
@@ -93,20 +82,23 @@ 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 following arguments from their corresponding environment variables if they are not provided:
+        - `api_key` from `GRADIENTAI_API_KEY`
+        - `inference_key` from `GRADIENTAI_INFERENCE_KEY`
         """
         if api_key is None:
-            api_key = os.environ.get("DIGITALOCEAN_GENAI_SDK_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"
-            )
+            api_key = os.environ.get("GRADIENTAI_API_KEY")
         self.api_key = api_key
 
+        if inference_key is None:
+            inference_key = os.environ.get("GRADIENTAI_INFERENCE_KEY")
+        self.inference_key = inference_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/"
 
@@ -128,16 +120,10 @@ class DigitaloceanGenaiSDK(SyncAPIClient):
         return AgentsResource(self)
 
     @cached_property
-    def providers(self) -> ProvidersResource:
-        from .resources.providers import ProvidersResource
-
-        return ProvidersResource(self)
-
-    @cached_property
-    def auth(self) -> AuthResource:
-        from .resources.auth import AuthResource
+    def model_providers(self) -> ModelProvidersResource:
+        from .resources.model_providers import ModelProvidersResource
 
-        return AuthResource(self)
+        return ModelProvidersResource(self)
 
     @cached_property
     def regions(self) -> RegionsResource:
@@ -145,24 +131,12 @@ class DigitaloceanGenaiSDK(SyncAPIClient):
 
         return RegionsResource(self)
 
-    @cached_property
-    def indexing_jobs(self) -> IndexingJobsResource:
-        from .resources.indexing_jobs import IndexingJobsResource
-
-        return IndexingJobsResource(self)
-
     @cached_property
     def knowledge_bases(self) -> KnowledgeBasesResource:
         from .resources.knowledge_bases import KnowledgeBasesResource
 
         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 +144,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 +156,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
@@ -198,6 +172,8 @@ class DigitaloceanGenaiSDK(SyncAPIClient):
     @override
     def auth_headers(self) -> dict[str, str]:
         api_key = self.api_key
+        if api_key is None:
+            return {}
         return {"Authorization": f"Bearer {api_key}"}
 
     @property
@@ -209,10 +185,22 @@ class DigitaloceanGenaiSDK(SyncAPIClient):
             **self._custom_headers,
         }
 
+    @override
+    def _validate_headers(self, headers: Headers, custom_headers: Headers) -> None:
+        if self.api_key and headers.get("Authorization"):
+            return
+        if isinstance(custom_headers.get("Authorization"), Omit):
+            return
+
+        raise TypeError(
+            '"Could not resolve authentication method. Expected the api_key to be set. Or for the `Authorization` headers to be explicitly omitted"'
+        )
+
     def copy(
         self,
         *,
         api_key: str | None = None,
+        inference_key: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: float | Timeout | None | NotGiven = NOT_GIVEN,
         http_client: httpx.Client | None = None,
@@ -245,8 +233,9 @@ 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,
+            inference_key=inference_key or self.inference_key,
             base_url=base_url or self.base_url,
             timeout=self.timeout if isinstance(timeout, NotGiven) else timeout,
             http_client=http_client,
@@ -255,6 +244,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,14 +285,16 @@ class DigitaloceanGenaiSDK(SyncAPIClient):
         return APIStatusError(err_msg, response=response, body=body)
 
 
-class AsyncDigitaloceanGenaiSDK(AsyncAPIClient):
+class AsyncGradientAI(AsyncAPIClient):
     # client options
-    api_key: str
+    api_key: str | None
+    inference_key: str | None
 
     def __init__(
         self,
         *,
         api_key: str | None = None,
+        inference_key: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: Union[float, Timeout, None, NotGiven] = NOT_GIVEN,
         max_retries: int = DEFAULT_MAX_RETRIES,
@@ -321,20 +314,23 @@ 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 following arguments from their corresponding environment variables if they are not provided:
+        - `api_key` from `GRADIENTAI_API_KEY`
+        - `inference_key` from `GRADIENTAI_INFERENCE_KEY`
         """
         if api_key is None:
-            api_key = os.environ.get("DIGITALOCEAN_GENAI_SDK_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"
-            )
+            api_key = os.environ.get("GRADIENTAI_API_KEY")
         self.api_key = api_key
 
+        if inference_key is None:
+            inference_key = os.environ.get("GRADIENTAI_INFERENCE_KEY")
+        self.inference_key = inference_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/"
 
@@ -356,16 +352,10 @@ class AsyncDigitaloceanGenaiSDK(AsyncAPIClient):
         return AsyncAgentsResource(self)
 
     @cached_property
-    def providers(self) -> AsyncProvidersResource:
-        from .resources.providers import AsyncProvidersResource
+    def model_providers(self) -> AsyncModelProvidersResource:
+        from .resources.model_providers import AsyncModelProvidersResource
 
-        return AsyncProvidersResource(self)
-
-    @cached_property
-    def auth(self) -> AsyncAuthResource:
-        from .resources.auth import AsyncAuthResource
-
-        return AsyncAuthResource(self)
+        return AsyncModelProvidersResource(self)
 
     @cached_property
     def regions(self) -> AsyncRegionsResource:
@@ -373,24 +363,12 @@ class AsyncDigitaloceanGenaiSDK(AsyncAPIClient):
 
         return AsyncRegionsResource(self)
 
-    @cached_property
-    def indexing_jobs(self) -> AsyncIndexingJobsResource:
-        from .resources.indexing_jobs import AsyncIndexingJobsResource
-
-        return AsyncIndexingJobsResource(self)
-
     @cached_property
     def knowledge_bases(self) -> AsyncKnowledgeBasesResource:
         from .resources.knowledge_bases import AsyncKnowledgeBasesResource
 
         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 +376,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 +388,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
@@ -426,6 +404,8 @@ class AsyncDigitaloceanGenaiSDK(AsyncAPIClient):
     @override
     def auth_headers(self) -> dict[str, str]:
         api_key = self.api_key
+        if api_key is None:
+            return {}
         return {"Authorization": f"Bearer {api_key}"}
 
     @property
@@ -437,10 +417,22 @@ class AsyncDigitaloceanGenaiSDK(AsyncAPIClient):
             **self._custom_headers,
         }
 
+    @override
+    def _validate_headers(self, headers: Headers, custom_headers: Headers) -> None:
+        if self.api_key and headers.get("Authorization"):
+            return
+        if isinstance(custom_headers.get("Authorization"), Omit):
+            return
+
+        raise TypeError(
+            '"Could not resolve authentication method. Expected the api_key to be set. Or for the `Authorization` headers to be explicitly omitted"'
+        )
+
     def copy(
         self,
         *,
         api_key: str | None = None,
+        inference_key: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: float | Timeout | None | NotGiven = NOT_GIVEN,
         http_client: httpx.AsyncClient | None = None,
@@ -473,8 +465,9 @@ 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,
+            inference_key=inference_key or self.inference_key,
             base_url=base_url or self.base_url,
             timeout=self.timeout if isinstance(timeout, NotGiven) else timeout,
             http_client=http_client,
@@ -483,6 +476,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 +517,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
@@ -535,16 +530,10 @@ class DigitaloceanGenaiSDKWithRawResponse:
         return AgentsResourceWithRawResponse(self._client.agents)
 
     @cached_property
-    def providers(self) -> providers.ProvidersResourceWithRawResponse:
-        from .resources.providers import ProvidersResourceWithRawResponse
-
-        return ProvidersResourceWithRawResponse(self._client.providers)
-
-    @cached_property
-    def auth(self) -> auth.AuthResourceWithRawResponse:
-        from .resources.auth import AuthResourceWithRawResponse
+    def model_providers(self) -> model_providers.ModelProvidersResourceWithRawResponse:
+        from .resources.model_providers import ModelProvidersResourceWithRawResponse
 
-        return AuthResourceWithRawResponse(self._client.auth)
+        return ModelProvidersResourceWithRawResponse(self._client.model_providers)
 
     @cached_property
     def regions(self) -> regions.RegionsResourceWithRawResponse:
@@ -552,24 +541,12 @@ class DigitaloceanGenaiSDKWithRawResponse:
 
         return RegionsResourceWithRawResponse(self._client.regions)
 
-    @cached_property
-    def indexing_jobs(self) -> indexing_jobs.IndexingJobsResourceWithRawResponse:
-        from .resources.indexing_jobs import IndexingJobsResourceWithRawResponse
-
-        return IndexingJobsResourceWithRawResponse(self._client.indexing_jobs)
-
     @cached_property
     def knowledge_bases(self) -> knowledge_bases.KnowledgeBasesResourceWithRawResponse:
         from .resources.knowledge_bases import KnowledgeBasesResourceWithRawResponse
 
         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 +554,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 +566,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
@@ -602,16 +579,10 @@ class AsyncDigitaloceanGenaiSDKWithRawResponse:
         return AsyncAgentsResourceWithRawResponse(self._client.agents)
 
     @cached_property
-    def providers(self) -> providers.AsyncProvidersResourceWithRawResponse:
-        from .resources.providers import AsyncProvidersResourceWithRawResponse
-
-        return AsyncProvidersResourceWithRawResponse(self._client.providers)
-
-    @cached_property
-    def auth(self) -> auth.AsyncAuthResourceWithRawResponse:
-        from .resources.auth import AsyncAuthResourceWithRawResponse
+    def model_providers(self) -> model_providers.AsyncModelProvidersResourceWithRawResponse:
+        from .resources.model_providers import AsyncModelProvidersResourceWithRawResponse
 
-        return AsyncAuthResourceWithRawResponse(self._client.auth)
+        return AsyncModelProvidersResourceWithRawResponse(self._client.model_providers)
 
     @cached_property
     def regions(self) -> regions.AsyncRegionsResourceWithRawResponse:
@@ -619,24 +590,12 @@ class AsyncDigitaloceanGenaiSDKWithRawResponse:
 
         return AsyncRegionsResourceWithRawResponse(self._client.regions)
 
-    @cached_property
-    def indexing_jobs(self) -> indexing_jobs.AsyncIndexingJobsResourceWithRawResponse:
-        from .resources.indexing_jobs import AsyncIndexingJobsResourceWithRawResponse
-
-        return AsyncIndexingJobsResourceWithRawResponse(self._client.indexing_jobs)
-
     @cached_property
     def knowledge_bases(self) -> knowledge_bases.AsyncKnowledgeBasesResourceWithRawResponse:
         from .resources.knowledge_bases import AsyncKnowledgeBasesResourceWithRawResponse
 
         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 +603,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 +615,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
@@ -669,16 +628,10 @@ class DigitaloceanGenaiSDKWithStreamedResponse:
         return AgentsResourceWithStreamingResponse(self._client.agents)
 
     @cached_property
-    def providers(self) -> providers.ProvidersResourceWithStreamingResponse:
-        from .resources.providers import ProvidersResourceWithStreamingResponse
+    def model_providers(self) -> model_providers.ModelProvidersResourceWithStreamingResponse:
+        from .resources.model_providers import ModelProvidersResourceWithStreamingResponse
 
-        return ProvidersResourceWithStreamingResponse(self._client.providers)
-
-    @cached_property
-    def auth(self) -> auth.AuthResourceWithStreamingResponse:
-        from .resources.auth import AuthResourceWithStreamingResponse
-
-        return AuthResourceWithStreamingResponse(self._client.auth)
+        return ModelProvidersResourceWithStreamingResponse(self._client.model_providers)
 
     @cached_property
     def regions(self) -> regions.RegionsResourceWithStreamingResponse:
@@ -686,24 +639,12 @@ class DigitaloceanGenaiSDKWithStreamedResponse:
 
         return RegionsResourceWithStreamingResponse(self._client.regions)
 
-    @cached_property
-    def indexing_jobs(self) -> indexing_jobs.IndexingJobsResourceWithStreamingResponse:
-        from .resources.indexing_jobs import IndexingJobsResourceWithStreamingResponse
-
-        return IndexingJobsResourceWithStreamingResponse(self._client.indexing_jobs)
-
     @cached_property
     def knowledge_bases(self) -> knowledge_bases.KnowledgeBasesResourceWithStreamingResponse:
         from .resources.knowledge_bases import KnowledgeBasesResourceWithStreamingResponse
 
         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 +652,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 +664,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
@@ -736,16 +677,10 @@ class AsyncDigitaloceanGenaiSDKWithStreamedResponse:
         return AsyncAgentsResourceWithStreamingResponse(self._client.agents)
 
     @cached_property
-    def providers(self) -> providers.AsyncProvidersResourceWithStreamingResponse:
-        from .resources.providers import AsyncProvidersResourceWithStreamingResponse
-
-        return AsyncProvidersResourceWithStreamingResponse(self._client.providers)
-
-    @cached_property
-    def auth(self) -> auth.AsyncAuthResourceWithStreamingResponse:
-        from .resources.auth import AsyncAuthResourceWithStreamingResponse
+    def model_providers(self) -> model_providers.AsyncModelProvidersResourceWithStreamingResponse:
+        from .resources.model_providers import AsyncModelProvidersResourceWithStreamingResponse
 
-        return AsyncAuthResourceWithStreamingResponse(self._client.auth)
+        return AsyncModelProvidersResourceWithStreamingResponse(self._client.model_providers)
 
     @cached_property
     def regions(self) -> regions.AsyncRegionsResourceWithStreamingResponse:
@@ -753,24 +688,12 @@ class AsyncDigitaloceanGenaiSDKWithStreamedResponse:
 
         return AsyncRegionsResourceWithStreamingResponse(self._client.regions)
 
-    @cached_property
-    def indexing_jobs(self) -> indexing_jobs.AsyncIndexingJobsResourceWithStreamingResponse:
-        from .resources.indexing_jobs import AsyncIndexingJobsResourceWithStreamingResponse
-
-        return AsyncIndexingJobsResourceWithStreamingResponse(self._client.indexing_jobs)
-
     @cached_property
     def knowledge_bases(self) -> knowledge_bases.AsyncKnowledgeBasesResourceWithStreamingResponse:
         from .resources.knowledge_bases import AsyncKnowledgeBasesResourceWithStreamingResponse
 
         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 +701,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 +713,6 @@ class AsyncDigitaloceanGenaiSDKWithStreamedResponse:
         return AsyncModelsResourceWithStreamingResponse(self._client.models)
 
 
-Client = DigitaloceanGenaiSDK
+Client = GradientAI
 
-AsyncClient = AsyncDigitaloceanGenaiSDK
+AsyncClient = AsyncGradientAI