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

{digitalocean_genai_sdk → gradientai}/resources/indexing_jobs.py

@@ -34,7 +34,7 @@ class IndexingJobsResource(SyncAPIResource):
         This property can be used as a prefix for any HTTP method call to return
         the raw response object instead of the parsed content.
 
-        For more information, see https://www.github.com/digitalocean/genai-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
         """
         return IndexingJobsResourceWithRawResponse(self)
 
@@ -43,7 +43,7 @@ class IndexingJobsResource(SyncAPIResource):
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/digitalocean/genai-python#with_streaming_response
+        For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
         """
         return IndexingJobsResourceWithStreamingResponse(self)
 
@@ -73,7 +73,9 @@ class IndexingJobsResource(SyncAPIResource):
           timeout: Override the client-level default timeout for this request, in seconds
         """
         return self._post(
-            "/v2/genai/indexing_jobs",
+            "/v2/gen-ai/indexing_jobs"
+            if self._client._base_url_overridden
+            else "https://api.digitalocean.com/v2/gen-ai/indexing_jobs",
             body=maybe_transform(
                 {
                     "data_source_uuids": data_source_uuids,
@@ -114,7 +116,9 @@ class IndexingJobsResource(SyncAPIResource):
         if not uuid:
             raise ValueError(f"Expected a non-empty value for `uuid` but received {uuid!r}")
         return self._get(
-            f"/v2/genai/indexing_jobs/{uuid}",
+            f"/v2/gen-ai/indexing_jobs/{uuid}"
+            if self._client._base_url_overridden
+            else f"https://api.digitalocean.com/v2/gen-ai/indexing_jobs/{uuid}",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
@@ -151,7 +155,9 @@ class IndexingJobsResource(SyncAPIResource):
           timeout: Override the client-level default timeout for this request, in seconds
         """
         return self._get(
-            "/v2/genai/indexing_jobs",
+            "/v2/gen-ai/indexing_jobs"
+            if self._client._base_url_overridden
+            else "https://api.digitalocean.com/v2/gen-ai/indexing_jobs",
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -195,7 +201,9 @@ class IndexingJobsResource(SyncAPIResource):
         if not indexing_job_uuid:
             raise ValueError(f"Expected a non-empty value for `indexing_job_uuid` but received {indexing_job_uuid!r}")
         return self._get(
-            f"/v2/genai/indexing_jobs/{indexing_job_uuid}/data_sources",
+            f"/v2/gen-ai/indexing_jobs/{indexing_job_uuid}/data_sources"
+            if self._client._base_url_overridden
+            else f"https://api.digitalocean.com/v2/gen-ai/indexing_jobs/{indexing_job_uuid}/data_sources",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
@@ -232,7 +240,9 @@ class IndexingJobsResource(SyncAPIResource):
         if not path_uuid:
             raise ValueError(f"Expected a non-empty value for `path_uuid` but received {path_uuid!r}")
         return self._put(
-            f"/v2/genai/indexing_jobs/{path_uuid}/cancel",
+            f"/v2/gen-ai/indexing_jobs/{path_uuid}/cancel"
+            if self._client._base_url_overridden
+            else f"https://api.digitalocean.com/v2/gen-ai/indexing_jobs/{path_uuid}/cancel",
             body=maybe_transform(
                 {"body_uuid": body_uuid}, indexing_job_update_cancel_params.IndexingJobUpdateCancelParams
             ),
@@ -250,7 +260,7 @@ class AsyncIndexingJobsResource(AsyncAPIResource):
         This property can be used as a prefix for any HTTP method call to return
         the raw response object instead of the parsed content.
 
-        For more information, see https://www.github.com/digitalocean/genai-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
         """
         return AsyncIndexingJobsResourceWithRawResponse(self)
 
@@ -259,7 +269,7 @@ class AsyncIndexingJobsResource(AsyncAPIResource):
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/digitalocean/genai-python#with_streaming_response
+        For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
         """
         return AsyncIndexingJobsResourceWithStreamingResponse(self)
 
@@ -289,7 +299,9 @@ class AsyncIndexingJobsResource(AsyncAPIResource):
           timeout: Override the client-level default timeout for this request, in seconds
         """
         return await self._post(
-            "/v2/genai/indexing_jobs",
+            "/v2/gen-ai/indexing_jobs"
+            if self._client._base_url_overridden
+            else "https://api.digitalocean.com/v2/gen-ai/indexing_jobs",
             body=await async_maybe_transform(
                 {
                     "data_source_uuids": data_source_uuids,
@@ -330,7 +342,9 @@ class AsyncIndexingJobsResource(AsyncAPIResource):
         if not uuid:
             raise ValueError(f"Expected a non-empty value for `uuid` but received {uuid!r}")
         return await self._get(
-            f"/v2/genai/indexing_jobs/{uuid}",
+            f"/v2/gen-ai/indexing_jobs/{uuid}"
+            if self._client._base_url_overridden
+            else f"https://api.digitalocean.com/v2/gen-ai/indexing_jobs/{uuid}",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
@@ -367,7 +381,9 @@ class AsyncIndexingJobsResource(AsyncAPIResource):
           timeout: Override the client-level default timeout for this request, in seconds
         """
         return await self._get(
-            "/v2/genai/indexing_jobs",
+            "/v2/gen-ai/indexing_jobs"
+            if self._client._base_url_overridden
+            else "https://api.digitalocean.com/v2/gen-ai/indexing_jobs",
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -411,7 +427,9 @@ class AsyncIndexingJobsResource(AsyncAPIResource):
         if not indexing_job_uuid:
             raise ValueError(f"Expected a non-empty value for `indexing_job_uuid` but received {indexing_job_uuid!r}")
         return await self._get(
-            f"/v2/genai/indexing_jobs/{indexing_job_uuid}/data_sources",
+            f"/v2/gen-ai/indexing_jobs/{indexing_job_uuid}/data_sources"
+            if self._client._base_url_overridden
+            else f"https://api.digitalocean.com/v2/gen-ai/indexing_jobs/{indexing_job_uuid}/data_sources",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
@@ -448,7 +466,9 @@ class AsyncIndexingJobsResource(AsyncAPIResource):
         if not path_uuid:
             raise ValueError(f"Expected a non-empty value for `path_uuid` but received {path_uuid!r}")
         return await self._put(
-            f"/v2/genai/indexing_jobs/{path_uuid}/cancel",
+            f"/v2/gen-ai/indexing_jobs/{path_uuid}/cancel"
+            if self._client._base_url_overridden
+            else f"https://api.digitalocean.com/v2/gen-ai/indexing_jobs/{path_uuid}/cancel",
             body=await async_maybe_transform(
                 {"body_uuid": body_uuid}, indexing_job_update_cancel_params.IndexingJobUpdateCancelParams
             ),

gradientai/resources/inference/__init__.py

@@ -0,0 +1,47 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from .models import (
+    ModelsResource,
+    AsyncModelsResource,
+    ModelsResourceWithRawResponse,
+    AsyncModelsResourceWithRawResponse,
+    ModelsResourceWithStreamingResponse,
+    AsyncModelsResourceWithStreamingResponse,
+)
+from .api_keys import (
+    APIKeysResource,
+    AsyncAPIKeysResource,
+    APIKeysResourceWithRawResponse,
+    AsyncAPIKeysResourceWithRawResponse,
+    APIKeysResourceWithStreamingResponse,
+    AsyncAPIKeysResourceWithStreamingResponse,
+)
+from .inference import (
+    InferenceResource,
+    AsyncInferenceResource,
+    InferenceResourceWithRawResponse,
+    AsyncInferenceResourceWithRawResponse,
+    InferenceResourceWithStreamingResponse,
+    AsyncInferenceResourceWithStreamingResponse,
+)
+
+__all__ = [
+    "APIKeysResource",
+    "AsyncAPIKeysResource",
+    "APIKeysResourceWithRawResponse",
+    "AsyncAPIKeysResourceWithRawResponse",
+    "APIKeysResourceWithStreamingResponse",
+    "AsyncAPIKeysResourceWithStreamingResponse",
+    "ModelsResource",
+    "AsyncModelsResource",
+    "ModelsResourceWithRawResponse",
+    "AsyncModelsResourceWithRawResponse",
+    "ModelsResourceWithStreamingResponse",
+    "AsyncModelsResourceWithStreamingResponse",
+    "InferenceResource",
+    "AsyncInferenceResource",
+    "InferenceResourceWithRawResponse",
+    "AsyncInferenceResourceWithRawResponse",
+    "InferenceResourceWithStreamingResponse",
+    "AsyncInferenceResourceWithStreamingResponse",
+]

digitalocean_genai_sdk/resources/api_keys/api_keys_.py → gradientai/resources/inference/api_keys.py

@@ -15,12 +15,12 @@ from ..._response import (
     async_to_streamed_response_wrapper,
 )
 from ..._base_client import make_request_options
-from ...types.api_keys import api_key_list_params, api_key_create_params, api_key_update_params
-from ...types.api_keys.api_key_list_response import APIKeyListResponse
-from ...types.api_keys.api_key_create_response import APIKeyCreateResponse
-from ...types.api_keys.api_key_delete_response import APIKeyDeleteResponse
-from ...types.api_keys.api_key_update_response import APIKeyUpdateResponse
-from ...types.api_keys.api_key_update_regenerate_response import APIKeyUpdateRegenerateResponse
+from ...types.inference import api_key_list_params, api_key_create_params, api_key_update_params
+from ...types.inference.api_key_list_response import APIKeyListResponse
+from ...types.inference.api_key_create_response import APIKeyCreateResponse
+from ...types.inference.api_key_delete_response import APIKeyDeleteResponse
+from ...types.inference.api_key_update_response import APIKeyUpdateResponse
+from ...types.inference.api_key_update_regenerate_response import APIKeyUpdateRegenerateResponse
 
 __all__ = ["APIKeysResource", "AsyncAPIKeysResource"]
 
@@ -32,7 +32,7 @@ class APIKeysResource(SyncAPIResource):
         This property can be used as a prefix for any HTTP method call to return
         the raw response object instead of the parsed content.
 
-        For more information, see https://www.github.com/digitalocean/genai-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
         """
         return APIKeysResourceWithRawResponse(self)
 
@@ -41,7 +41,7 @@ class APIKeysResource(SyncAPIResource):
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/digitalocean/genai-python#with_streaming_response
+        For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
         """
         return APIKeysResourceWithStreamingResponse(self)
 
@@ -69,7 +69,9 @@ class APIKeysResource(SyncAPIResource):
           timeout: Override the client-level default timeout for this request, in seconds
         """
         return self._post(
-            "/v2/genai/models/api_keys",
+            "/v2/gen-ai/models/api_keys"
+            if self._client._base_url_overridden
+            else "https://api.digitalocean.com/v2/gen-ai/models/api_keys",
             body=maybe_transform({"name": name}, api_key_create_params.APIKeyCreateParams),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
@@ -106,7 +108,9 @@ class APIKeysResource(SyncAPIResource):
         if not path_api_key_uuid:
             raise ValueError(f"Expected a non-empty value for `path_api_key_uuid` but received {path_api_key_uuid!r}")
         return self._put(
-            f"/v2/genai/models/api_keys/{path_api_key_uuid}",
+            f"/v2/gen-ai/models/api_keys/{path_api_key_uuid}"
+            if self._client._base_url_overridden
+            else f"https://api.digitalocean.com/v2/gen-ai/models/api_keys/{path_api_key_uuid}",
             body=maybe_transform(
                 {
                     "body_api_key_uuid": body_api_key_uuid,
@@ -149,7 +153,9 @@ class APIKeysResource(SyncAPIResource):
           timeout: Override the client-level default timeout for this request, in seconds
         """
         return self._get(
-            "/v2/genai/models/api_keys",
+            "/v2/gen-ai/models/api_keys"
+            if self._client._base_url_overridden
+            else "https://api.digitalocean.com/v2/gen-ai/models/api_keys",
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -193,7 +199,9 @@ class APIKeysResource(SyncAPIResource):
         if not api_key_uuid:
             raise ValueError(f"Expected a non-empty value for `api_key_uuid` but received {api_key_uuid!r}")
         return self._delete(
-            f"/v2/genai/models/api_keys/{api_key_uuid}",
+            f"/v2/gen-ai/models/api_keys/{api_key_uuid}"
+            if self._client._base_url_overridden
+            else f"https://api.digitalocean.com/v2/gen-ai/models/api_keys/{api_key_uuid}",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
@@ -227,7 +235,9 @@ class APIKeysResource(SyncAPIResource):
         if not api_key_uuid:
             raise ValueError(f"Expected a non-empty value for `api_key_uuid` but received {api_key_uuid!r}")
         return self._put(
-            f"/v2/genai/models/api_keys/{api_key_uuid}/regenerate",
+            f"/v2/gen-ai/models/api_keys/{api_key_uuid}/regenerate"
+            if self._client._base_url_overridden
+            else f"https://api.digitalocean.com/v2/gen-ai/models/api_keys/{api_key_uuid}/regenerate",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
@@ -242,7 +252,7 @@ class AsyncAPIKeysResource(AsyncAPIResource):
         This property can be used as a prefix for any HTTP method call to return
         the raw response object instead of the parsed content.
 
-        For more information, see https://www.github.com/digitalocean/genai-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
         """
         return AsyncAPIKeysResourceWithRawResponse(self)
 
@@ -251,7 +261,7 @@ class AsyncAPIKeysResource(AsyncAPIResource):
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/digitalocean/genai-python#with_streaming_response
+        For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
         """
         return AsyncAPIKeysResourceWithStreamingResponse(self)
 
@@ -279,7 +289,9 @@ class AsyncAPIKeysResource(AsyncAPIResource):
           timeout: Override the client-level default timeout for this request, in seconds
         """
         return await self._post(
-            "/v2/genai/models/api_keys",
+            "/v2/gen-ai/models/api_keys"
+            if self._client._base_url_overridden
+            else "https://api.digitalocean.com/v2/gen-ai/models/api_keys",
             body=await async_maybe_transform({"name": name}, api_key_create_params.APIKeyCreateParams),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
@@ -316,7 +328,9 @@ class AsyncAPIKeysResource(AsyncAPIResource):
         if not path_api_key_uuid:
             raise ValueError(f"Expected a non-empty value for `path_api_key_uuid` but received {path_api_key_uuid!r}")
         return await self._put(
-            f"/v2/genai/models/api_keys/{path_api_key_uuid}",
+            f"/v2/gen-ai/models/api_keys/{path_api_key_uuid}"
+            if self._client._base_url_overridden
+            else f"https://api.digitalocean.com/v2/gen-ai/models/api_keys/{path_api_key_uuid}",
             body=await async_maybe_transform(
                 {
                     "body_api_key_uuid": body_api_key_uuid,
@@ -359,7 +373,9 @@ class AsyncAPIKeysResource(AsyncAPIResource):
           timeout: Override the client-level default timeout for this request, in seconds
         """
         return await self._get(
-            "/v2/genai/models/api_keys",
+            "/v2/gen-ai/models/api_keys"
+            if self._client._base_url_overridden
+            else "https://api.digitalocean.com/v2/gen-ai/models/api_keys",
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -403,7 +419,9 @@ class AsyncAPIKeysResource(AsyncAPIResource):
         if not api_key_uuid:
             raise ValueError(f"Expected a non-empty value for `api_key_uuid` but received {api_key_uuid!r}")
         return await self._delete(
-            f"/v2/genai/models/api_keys/{api_key_uuid}",
+            f"/v2/gen-ai/models/api_keys/{api_key_uuid}"
+            if self._client._base_url_overridden
+            else f"https://api.digitalocean.com/v2/gen-ai/models/api_keys/{api_key_uuid}",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
@@ -437,7 +455,9 @@ class AsyncAPIKeysResource(AsyncAPIResource):
         if not api_key_uuid:
             raise ValueError(f"Expected a non-empty value for `api_key_uuid` but received {api_key_uuid!r}")
         return await self._put(
-            f"/v2/genai/models/api_keys/{api_key_uuid}/regenerate",
+            f"/v2/gen-ai/models/api_keys/{api_key_uuid}/regenerate"
+            if self._client._base_url_overridden
+            else f"https://api.digitalocean.com/v2/gen-ai/models/api_keys/{api_key_uuid}/regenerate",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),

gradientai/resources/inference/inference.py

@@ -0,0 +1,134 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from .models import (
+    ModelsResource,
+    AsyncModelsResource,
+    ModelsResourceWithRawResponse,
+    AsyncModelsResourceWithRawResponse,
+    ModelsResourceWithStreamingResponse,
+    AsyncModelsResourceWithStreamingResponse,
+)
+from .api_keys import (
+    APIKeysResource,
+    AsyncAPIKeysResource,
+    APIKeysResourceWithRawResponse,
+    AsyncAPIKeysResourceWithRawResponse,
+    APIKeysResourceWithStreamingResponse,
+    AsyncAPIKeysResourceWithStreamingResponse,
+)
+from ..._compat import cached_property
+from ..._resource import SyncAPIResource, AsyncAPIResource
+
+__all__ = ["InferenceResource", "AsyncInferenceResource"]
+
+
+class InferenceResource(SyncAPIResource):
+    @cached_property
+    def api_keys(self) -> APIKeysResource:
+        return APIKeysResource(self._client)
+
+    @cached_property
+    def models(self) -> ModelsResource:
+        return ModelsResource(self._client)
+
+    @cached_property
+    def with_raw_response(self) -> InferenceResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
+        """
+        return InferenceResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> InferenceResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
+        """
+        return InferenceResourceWithStreamingResponse(self)
+
+
+class AsyncInferenceResource(AsyncAPIResource):
+    @cached_property
+    def api_keys(self) -> AsyncAPIKeysResource:
+        return AsyncAPIKeysResource(self._client)
+
+    @cached_property
+    def models(self) -> AsyncModelsResource:
+        return AsyncModelsResource(self._client)
+
+    @cached_property
+    def with_raw_response(self) -> AsyncInferenceResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
+        """
+        return AsyncInferenceResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncInferenceResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
+        """
+        return AsyncInferenceResourceWithStreamingResponse(self)
+
+
+class InferenceResourceWithRawResponse:
+    def __init__(self, inference: InferenceResource) -> None:
+        self._inference = inference
+
+    @cached_property
+    def api_keys(self) -> APIKeysResourceWithRawResponse:
+        return APIKeysResourceWithRawResponse(self._inference.api_keys)
+
+    @cached_property
+    def models(self) -> ModelsResourceWithRawResponse:
+        return ModelsResourceWithRawResponse(self._inference.models)
+
+
+class AsyncInferenceResourceWithRawResponse:
+    def __init__(self, inference: AsyncInferenceResource) -> None:
+        self._inference = inference
+
+    @cached_property
+    def api_keys(self) -> AsyncAPIKeysResourceWithRawResponse:
+        return AsyncAPIKeysResourceWithRawResponse(self._inference.api_keys)
+
+    @cached_property
+    def models(self) -> AsyncModelsResourceWithRawResponse:
+        return AsyncModelsResourceWithRawResponse(self._inference.models)
+
+
+class InferenceResourceWithStreamingResponse:
+    def __init__(self, inference: InferenceResource) -> None:
+        self._inference = inference
+
+    @cached_property
+    def api_keys(self) -> APIKeysResourceWithStreamingResponse:
+        return APIKeysResourceWithStreamingResponse(self._inference.api_keys)
+
+    @cached_property
+    def models(self) -> ModelsResourceWithStreamingResponse:
+        return ModelsResourceWithStreamingResponse(self._inference.models)
+
+
+class AsyncInferenceResourceWithStreamingResponse:
+    def __init__(self, inference: AsyncInferenceResource) -> None:
+        self._inference = inference
+
+    @cached_property
+    def api_keys(self) -> AsyncAPIKeysResourceWithStreamingResponse:
+        return AsyncAPIKeysResourceWithStreamingResponse(self._inference.api_keys)
+
+    @cached_property
+    def models(self) -> AsyncModelsResourceWithStreamingResponse:
+        return AsyncModelsResourceWithStreamingResponse(self._inference.models)

{digitalocean_genai_sdk/resources → gradientai/resources/inference}/models.py

@@ -4,18 +4,18 @@ from __future__ import annotations
 
 import httpx
 
-from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
-from .._compat import cached_property
-from .._resource import SyncAPIResource, AsyncAPIResource
-from .._response import (
+from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from ..._compat import cached_property
+from ..._resource import SyncAPIResource, AsyncAPIResource
+from ..._response import (
     to_raw_response_wrapper,
     to_streamed_response_wrapper,
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ..types.model import Model
-from .._base_client import make_request_options
-from ..types.model_list_response import ModelListResponse
+from ..._base_client import make_request_options
+from ...types.inference.model import Model
+from ...types.inference.model_list_response import ModelListResponse
 
 __all__ = ["ModelsResource", "AsyncModelsResource"]
 
@@ -27,7 +27,7 @@ class ModelsResource(SyncAPIResource):
         This property can be used as a prefix for any HTTP method call to return
         the raw response object instead of the parsed content.
 
-        For more information, see https://www.github.com/digitalocean/genai-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
         """
         return ModelsResourceWithRawResponse(self)
 
@@ -36,7 +36,7 @@ class ModelsResource(SyncAPIResource):
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/digitalocean/genai-python#with_streaming_response
+        For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
         """
         return ModelsResourceWithStreamingResponse(self)
 
@@ -67,7 +67,9 @@ class ModelsResource(SyncAPIResource):
         if not model:
             raise ValueError(f"Expected a non-empty value for `model` but received {model!r}")
         return self._get(
-            f"/models/{model}",
+            f"/models/{model}"
+            if self._client._base_url_overridden
+            else f"https://inference.do-ai.run/v1/models/{model}",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
@@ -89,7 +91,7 @@ class ModelsResource(SyncAPIResource):
         one such as the owner and availability.
         """
         return self._get(
-            "/models",
+            "/models" if self._client._base_url_overridden else "https://inference.do-ai.run/v1/models",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
@@ -104,7 +106,7 @@ class AsyncModelsResource(AsyncAPIResource):
         This property can be used as a prefix for any HTTP method call to return
         the raw response object instead of the parsed content.
 
-        For more information, see https://www.github.com/digitalocean/genai-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
         """
         return AsyncModelsResourceWithRawResponse(self)
 
@@ -113,7 +115,7 @@ class AsyncModelsResource(AsyncAPIResource):
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/digitalocean/genai-python#with_streaming_response
+        For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
         """
         return AsyncModelsResourceWithStreamingResponse(self)
 
@@ -144,7 +146,9 @@ class AsyncModelsResource(AsyncAPIResource):
         if not model:
             raise ValueError(f"Expected a non-empty value for `model` but received {model!r}")
         return await self._get(
-            f"/models/{model}",
+            f"/models/{model}"
+            if self._client._base_url_overridden
+            else f"https://inference.do-ai.run/v1/models/{model}",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
@@ -166,7 +170,7 @@ class AsyncModelsResource(AsyncAPIResource):
         one such as the owner and availability.
         """
         return await self._get(
-            "/models",
+            "/models" if self._client._base_url_overridden else "https://inference.do-ai.run/v1/models",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),