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

gradientai/resources/agents/{child_agents.py → routes.py}

@@ -15,34 +15,34 @@ from ..._response import (
     async_to_streamed_response_wrapper,
 )
 from ..._base_client import make_request_options
-from ...types.agents import child_agent_add_params, child_agent_update_params
-from ...types.agents.child_agent_add_response import ChildAgentAddResponse
-from ...types.agents.child_agent_view_response import ChildAgentViewResponse
-from ...types.agents.child_agent_delete_response import ChildAgentDeleteResponse
-from ...types.agents.child_agent_update_response import ChildAgentUpdateResponse
+from ...types.agents import route_add_params, route_update_params
+from ...types.agents.route_add_response import RouteAddResponse
+from ...types.agents.route_view_response import RouteViewResponse
+from ...types.agents.route_delete_response import RouteDeleteResponse
+from ...types.agents.route_update_response import RouteUpdateResponse
 
-__all__ = ["ChildAgentsResource", "AsyncChildAgentsResource"]
+__all__ = ["RoutesResource", "AsyncRoutesResource"]
 
 
-class ChildAgentsResource(SyncAPIResource):
+class RoutesResource(SyncAPIResource):
     @cached_property
-    def with_raw_response(self) -> ChildAgentsResourceWithRawResponse:
+    def with_raw_response(self) -> RoutesResourceWithRawResponse:
         """
         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 ChildAgentsResourceWithRawResponse(self)
+        return RoutesResourceWithRawResponse(self)
 
     @cached_property
-    def with_streaming_response(self) -> ChildAgentsResourceWithStreamingResponse:
+    def with_streaming_response(self) -> RoutesResourceWithStreamingResponse:
         """
         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 ChildAgentsResourceWithStreamingResponse(self)
+        return RoutesResourceWithStreamingResponse(self)
 
     def update(
         self,
@@ -60,7 +60,7 @@ class ChildAgentsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> ChildAgentUpdateResponse:
+    ) -> RouteUpdateResponse:
         """
         To update an agent route for an agent, send a PUT request to
         `/v2/gen-ai/agents/{parent_agent_uuid}/child_agents/{child_agent_uuid}`.
@@ -96,12 +96,12 @@ class ChildAgentsResource(SyncAPIResource):
                     "route_name": route_name,
                     "uuid": uuid,
                 },
-                child_agent_update_params.ChildAgentUpdateParams,
+                route_update_params.RouteUpdateParams,
             ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=ChildAgentUpdateResponse,
+            cast_to=RouteUpdateResponse,
         )
 
     def delete(
@@ -115,7 +115,7 @@ class ChildAgentsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> ChildAgentDeleteResponse:
+    ) -> RouteDeleteResponse:
         """
         To delete an agent route from a parent agent, send a DELETE request to
         `/v2/gen-ai/agents/{parent_agent_uuid}/child_agents/{child_agent_uuid}`.
@@ -140,7 +140,7 @@ class ChildAgentsResource(SyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=ChildAgentDeleteResponse,
+            cast_to=RouteDeleteResponse,
         )
 
     def add(
@@ -158,7 +158,7 @@ class ChildAgentsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> ChildAgentAddResponse:
+    ) -> RouteAddResponse:
         """
         To add an agent route to an agent, send a POST request to
         `/v2/gen-ai/agents/{parent_agent_uuid}/child_agents/{child_agent_uuid}`.
@@ -193,12 +193,12 @@ class ChildAgentsResource(SyncAPIResource):
                     "body_parent_agent_uuid": body_parent_agent_uuid,
                     "route_name": route_name,
                 },
-                child_agent_add_params.ChildAgentAddParams,
+                route_add_params.RouteAddParams,
             ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=ChildAgentAddResponse,
+            cast_to=RouteAddResponse,
         )
 
     def view(
@@ -211,7 +211,7 @@ class ChildAgentsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> ChildAgentViewResponse:
+    ) -> RouteViewResponse:
         """
         To view agent routes for an agent, send a GET requtest to
         `/v2/gen-ai/agents/{uuid}/child_agents`.
@@ -234,29 +234,29 @@ class ChildAgentsResource(SyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=ChildAgentViewResponse,
+            cast_to=RouteViewResponse,
         )
 
 
-class AsyncChildAgentsResource(AsyncAPIResource):
+class AsyncRoutesResource(AsyncAPIResource):
     @cached_property
-    def with_raw_response(self) -> AsyncChildAgentsResourceWithRawResponse:
+    def with_raw_response(self) -> AsyncRoutesResourceWithRawResponse:
         """
         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 AsyncChildAgentsResourceWithRawResponse(self)
+        return AsyncRoutesResourceWithRawResponse(self)
 
     @cached_property
-    def with_streaming_response(self) -> AsyncChildAgentsResourceWithStreamingResponse:
+    def with_streaming_response(self) -> AsyncRoutesResourceWithStreamingResponse:
         """
         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 AsyncChildAgentsResourceWithStreamingResponse(self)
+        return AsyncRoutesResourceWithStreamingResponse(self)
 
     async def update(
         self,
@@ -274,7 +274,7 @@ class AsyncChildAgentsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> ChildAgentUpdateResponse:
+    ) -> RouteUpdateResponse:
         """
         To update an agent route for an agent, send a PUT request to
         `/v2/gen-ai/agents/{parent_agent_uuid}/child_agents/{child_agent_uuid}`.
@@ -310,12 +310,12 @@ class AsyncChildAgentsResource(AsyncAPIResource):
                     "route_name": route_name,
                     "uuid": uuid,
                 },
-                child_agent_update_params.ChildAgentUpdateParams,
+                route_update_params.RouteUpdateParams,
             ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=ChildAgentUpdateResponse,
+            cast_to=RouteUpdateResponse,
         )
 
     async def delete(
@@ -329,7 +329,7 @@ class AsyncChildAgentsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> ChildAgentDeleteResponse:
+    ) -> RouteDeleteResponse:
         """
         To delete an agent route from a parent agent, send a DELETE request to
         `/v2/gen-ai/agents/{parent_agent_uuid}/child_agents/{child_agent_uuid}`.
@@ -354,7 +354,7 @@ class AsyncChildAgentsResource(AsyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=ChildAgentDeleteResponse,
+            cast_to=RouteDeleteResponse,
         )
 
     async def add(
@@ -372,7 +372,7 @@ class AsyncChildAgentsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> ChildAgentAddResponse:
+    ) -> RouteAddResponse:
         """
         To add an agent route to an agent, send a POST request to
         `/v2/gen-ai/agents/{parent_agent_uuid}/child_agents/{child_agent_uuid}`.
@@ -407,12 +407,12 @@ class AsyncChildAgentsResource(AsyncAPIResource):
                     "body_parent_agent_uuid": body_parent_agent_uuid,
                     "route_name": route_name,
                 },
-                child_agent_add_params.ChildAgentAddParams,
+                route_add_params.RouteAddParams,
             ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=ChildAgentAddResponse,
+            cast_to=RouteAddResponse,
         )
 
     async def view(
@@ -425,7 +425,7 @@ class AsyncChildAgentsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> ChildAgentViewResponse:
+    ) -> RouteViewResponse:
         """
         To view agent routes for an agent, send a GET requtest to
         `/v2/gen-ai/agents/{uuid}/child_agents`.
@@ -448,77 +448,77 @@ class AsyncChildAgentsResource(AsyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=ChildAgentViewResponse,
+            cast_to=RouteViewResponse,
         )
 
 
-class ChildAgentsResourceWithRawResponse:
-    def __init__(self, child_agents: ChildAgentsResource) -> None:
-        self._child_agents = child_agents
+class RoutesResourceWithRawResponse:
+    def __init__(self, routes: RoutesResource) -> None:
+        self._routes = routes
 
         self.update = to_raw_response_wrapper(
-            child_agents.update,
+            routes.update,
         )
         self.delete = to_raw_response_wrapper(
-            child_agents.delete,
+            routes.delete,
         )
         self.add = to_raw_response_wrapper(
-            child_agents.add,
+            routes.add,
         )
         self.view = to_raw_response_wrapper(
-            child_agents.view,
+            routes.view,
         )
 
 
-class AsyncChildAgentsResourceWithRawResponse:
-    def __init__(self, child_agents: AsyncChildAgentsResource) -> None:
-        self._child_agents = child_agents
+class AsyncRoutesResourceWithRawResponse:
+    def __init__(self, routes: AsyncRoutesResource) -> None:
+        self._routes = routes
 
         self.update = async_to_raw_response_wrapper(
-            child_agents.update,
+            routes.update,
         )
         self.delete = async_to_raw_response_wrapper(
-            child_agents.delete,
+            routes.delete,
         )
         self.add = async_to_raw_response_wrapper(
-            child_agents.add,
+            routes.add,
         )
         self.view = async_to_raw_response_wrapper(
-            child_agents.view,
+            routes.view,
         )
 
 
-class ChildAgentsResourceWithStreamingResponse:
-    def __init__(self, child_agents: ChildAgentsResource) -> None:
-        self._child_agents = child_agents
+class RoutesResourceWithStreamingResponse:
+    def __init__(self, routes: RoutesResource) -> None:
+        self._routes = routes
 
         self.update = to_streamed_response_wrapper(
-            child_agents.update,
+            routes.update,
         )
         self.delete = to_streamed_response_wrapper(
-            child_agents.delete,
+            routes.delete,
         )
         self.add = to_streamed_response_wrapper(
-            child_agents.add,
+            routes.add,
         )
         self.view = to_streamed_response_wrapper(
-            child_agents.view,
+            routes.view,
         )
 
 
-class AsyncChildAgentsResourceWithStreamingResponse:
-    def __init__(self, child_agents: AsyncChildAgentsResource) -> None:
-        self._child_agents = child_agents
+class AsyncRoutesResourceWithStreamingResponse:
+    def __init__(self, routes: AsyncRoutesResource) -> None:
+        self._routes = routes
 
         self.update = async_to_streamed_response_wrapper(
-            child_agents.update,
+            routes.update,
         )
         self.delete = async_to_streamed_response_wrapper(
-            child_agents.delete,
+            routes.delete,
         )
         self.add = async_to_streamed_response_wrapper(
-            child_agents.add,
+            routes.add,
         )
         self.view = async_to_streamed_response_wrapper(
-            child_agents.view,
+            routes.view,
         )

gradientai/resources/chat/completions.py

@@ -153,6 +153,15 @@ class CompletionsResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
+
+        # This method requires an inference_key to be set via client argument or environment variable
+        if not self._client.inference_key:
+            raise TypeError(
+                "Could not resolve authentication method. Expected the inference_key to be set for chat completions."
+            )
+        headers = extra_headers or {}
+        headers = {"Authorization": f"Bearer {self._client.inference_key}", **headers}
+
         return self._post(
             "/chat/completions"
             if self._client._base_url_overridden
@@ -180,7 +189,7 @@ class CompletionsResource(SyncAPIResource):
                 completion_create_params.CompletionCreateParams,
             ),
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
             cast_to=CompletionCreateResponse,
         )
@@ -316,6 +325,15 @@ class AsyncCompletionsResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
+
+        # This method requires an inference_key to be set via client argument or environment variable
+        if not hasattr(self._client, "inference_key") or not self._client.inference_key:
+            raise TypeError(
+                "Could not resolve authentication method. Expected the inference_key to be set for chat completions."
+            )
+        headers = extra_headers or {}
+        headers = {"Authorization": f"Bearer {self._client.inference_key}", **headers}
+
         return await self._post(
             "/chat/completions"
             if self._client._base_url_overridden
@@ -343,7 +361,7 @@ class AsyncCompletionsResource(AsyncAPIResource):
                 completion_create_params.CompletionCreateParams,
             ),
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
             cast_to=CompletionCreateResponse,
         )

gradientai/resources/inference/__init__.py

@@ -1,13 +1,5 @@
 # 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,
@@ -32,12 +24,6 @@ __all__ = [
     "AsyncAPIKeysResourceWithRawResponse",
     "APIKeysResourceWithStreamingResponse",
     "AsyncAPIKeysResourceWithStreamingResponse",
-    "ModelsResource",
-    "AsyncModelsResource",
-    "ModelsResourceWithRawResponse",
-    "AsyncModelsResourceWithRawResponse",
-    "ModelsResourceWithStreamingResponse",
-    "AsyncModelsResourceWithStreamingResponse",
     "InferenceResource",
     "AsyncInferenceResource",
     "InferenceResourceWithRawResponse",

gradientai/resources/inference/inference.py

@@ -2,14 +2,6 @@
 
 from __future__ import annotations
 
-from .models import (
-    ModelsResource,
-    AsyncModelsResource,
-    ModelsResourceWithRawResponse,
-    AsyncModelsResourceWithRawResponse,
-    ModelsResourceWithStreamingResponse,
-    AsyncModelsResourceWithStreamingResponse,
-)
 from .api_keys import (
     APIKeysResource,
     AsyncAPIKeysResource,
@@ -29,10 +21,6 @@ class InferenceResource(SyncAPIResource):
     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:
         """
@@ -58,10 +46,6 @@ class AsyncInferenceResource(AsyncAPIResource):
     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:
         """
@@ -90,10 +74,6 @@ class InferenceResourceWithRawResponse:
     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:
@@ -103,10 +83,6 @@ class AsyncInferenceResourceWithRawResponse:
     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:
@@ -116,10 +92,6 @@ class InferenceResourceWithStreamingResponse:
     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:
@@ -128,7 +100,3 @@ class AsyncInferenceResourceWithStreamingResponse:
     @cached_property
     def api_keys(self) -> AsyncAPIKeysResourceWithStreamingResponse:
         return AsyncAPIKeysResourceWithStreamingResponse(self._inference.api_keys)
-
-    @cached_property
-    def models(self) -> AsyncModelsResourceWithStreamingResponse:
-        return AsyncModelsResourceWithStreamingResponse(self._inference.models)

gradientai/resources/knowledge_bases/__init__.py

@@ -8,6 +8,14 @@ from .data_sources import (
     DataSourcesResourceWithStreamingResponse,
     AsyncDataSourcesResourceWithStreamingResponse,
 )
+from .indexing_jobs import (
+    IndexingJobsResource,
+    AsyncIndexingJobsResource,
+    IndexingJobsResourceWithRawResponse,
+    AsyncIndexingJobsResourceWithRawResponse,
+    IndexingJobsResourceWithStreamingResponse,
+    AsyncIndexingJobsResourceWithStreamingResponse,
+)
 from .knowledge_bases import (
     KnowledgeBasesResource,
     AsyncKnowledgeBasesResource,
@@ -24,6 +32,12 @@ __all__ = [
     "AsyncDataSourcesResourceWithRawResponse",
     "DataSourcesResourceWithStreamingResponse",
     "AsyncDataSourcesResourceWithStreamingResponse",
+    "IndexingJobsResource",
+    "AsyncIndexingJobsResource",
+    "IndexingJobsResourceWithRawResponse",
+    "AsyncIndexingJobsResourceWithRawResponse",
+    "IndexingJobsResourceWithStreamingResponse",
+    "AsyncIndexingJobsResourceWithStreamingResponse",
     "KnowledgeBasesResource",
     "AsyncKnowledgeBasesResource",
     "KnowledgeBasesResourceWithRawResponse",

gradientai/resources/{indexing_jobs.py → knowledge_bases/indexing_jobs.py}

@@ -6,23 +6,27 @@ from typing import List
 
 import httpx
 
-from ..types import indexing_job_list_params, indexing_job_create_params, indexing_job_update_cancel_params
-from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
-from .._utils import maybe_transform, async_maybe_transform
-from .._compat import cached_property
-from .._resource import SyncAPIResource, AsyncAPIResource
-from .._response import (
+from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from ..._utils import maybe_transform, async_maybe_transform
+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 .._base_client import make_request_options
-from ..types.indexing_job_list_response import IndexingJobListResponse
-from ..types.indexing_job_create_response import IndexingJobCreateResponse
-from ..types.indexing_job_retrieve_response import IndexingJobRetrieveResponse
-from ..types.indexing_job_update_cancel_response import IndexingJobUpdateCancelResponse
-from ..types.indexing_job_retrieve_data_sources_response import IndexingJobRetrieveDataSourcesResponse
+from ..._base_client import make_request_options
+from ...types.knowledge_bases import (
+    indexing_job_list_params,
+    indexing_job_create_params,
+    indexing_job_update_cancel_params,
+)
+from ...types.knowledge_bases.indexing_job_list_response import IndexingJobListResponse
+from ...types.knowledge_bases.indexing_job_create_response import IndexingJobCreateResponse
+from ...types.knowledge_bases.indexing_job_retrieve_response import IndexingJobRetrieveResponse
+from ...types.knowledge_bases.indexing_job_update_cancel_response import IndexingJobUpdateCancelResponse
+from ...types.knowledge_bases.indexing_job_retrieve_data_sources_response import IndexingJobRetrieveDataSourcesResponse
 
 __all__ = ["IndexingJobsResource", "AsyncIndexingJobsResource"]
 

gradientai/resources/knowledge_bases/knowledge_bases.py

@@ -25,6 +25,14 @@ from .data_sources import (
     DataSourcesResourceWithStreamingResponse,
     AsyncDataSourcesResourceWithStreamingResponse,
 )
+from .indexing_jobs import (
+    IndexingJobsResource,
+    AsyncIndexingJobsResource,
+    IndexingJobsResourceWithRawResponse,
+    AsyncIndexingJobsResourceWithRawResponse,
+    IndexingJobsResourceWithStreamingResponse,
+    AsyncIndexingJobsResourceWithStreamingResponse,
+)
 from ..._base_client import make_request_options
 from ...types.knowledge_base_list_response import KnowledgeBaseListResponse
 from ...types.knowledge_base_create_response import KnowledgeBaseCreateResponse
@@ -40,6 +48,10 @@ class KnowledgeBasesResource(SyncAPIResource):
     def data_sources(self) -> DataSourcesResource:
         return DataSourcesResource(self._client)
 
+    @cached_property
+    def indexing_jobs(self) -> IndexingJobsResource:
+        return IndexingJobsResource(self._client)
+
     @cached_property
     def with_raw_response(self) -> KnowledgeBasesResourceWithRawResponse:
         """
@@ -316,6 +328,10 @@ class AsyncKnowledgeBasesResource(AsyncAPIResource):
     def data_sources(self) -> AsyncDataSourcesResource:
         return AsyncDataSourcesResource(self._client)
 
+    @cached_property
+    def indexing_jobs(self) -> AsyncIndexingJobsResource:
+        return AsyncIndexingJobsResource(self._client)
+
     @cached_property
     def with_raw_response(self) -> AsyncKnowledgeBasesResourceWithRawResponse:
         """
@@ -611,6 +627,10 @@ class KnowledgeBasesResourceWithRawResponse:
     def data_sources(self) -> DataSourcesResourceWithRawResponse:
         return DataSourcesResourceWithRawResponse(self._knowledge_bases.data_sources)
 
+    @cached_property
+    def indexing_jobs(self) -> IndexingJobsResourceWithRawResponse:
+        return IndexingJobsResourceWithRawResponse(self._knowledge_bases.indexing_jobs)
+
 
 class AsyncKnowledgeBasesResourceWithRawResponse:
     def __init__(self, knowledge_bases: AsyncKnowledgeBasesResource) -> None:
@@ -636,6 +656,10 @@ class AsyncKnowledgeBasesResourceWithRawResponse:
     def data_sources(self) -> AsyncDataSourcesResourceWithRawResponse:
         return AsyncDataSourcesResourceWithRawResponse(self._knowledge_bases.data_sources)
 
+    @cached_property
+    def indexing_jobs(self) -> AsyncIndexingJobsResourceWithRawResponse:
+        return AsyncIndexingJobsResourceWithRawResponse(self._knowledge_bases.indexing_jobs)
+
 
 class KnowledgeBasesResourceWithStreamingResponse:
     def __init__(self, knowledge_bases: KnowledgeBasesResource) -> None:
@@ -661,6 +685,10 @@ class KnowledgeBasesResourceWithStreamingResponse:
     def data_sources(self) -> DataSourcesResourceWithStreamingResponse:
         return DataSourcesResourceWithStreamingResponse(self._knowledge_bases.data_sources)
 
+    @cached_property
+    def indexing_jobs(self) -> IndexingJobsResourceWithStreamingResponse:
+        return IndexingJobsResourceWithStreamingResponse(self._knowledge_bases.indexing_jobs)
+
 
 class AsyncKnowledgeBasesResourceWithStreamingResponse:
     def __init__(self, knowledge_bases: AsyncKnowledgeBasesResource) -> None:
@@ -685,3 +713,7 @@ class AsyncKnowledgeBasesResourceWithStreamingResponse:
     @cached_property
     def data_sources(self) -> AsyncDataSourcesResourceWithStreamingResponse:
         return AsyncDataSourcesResourceWithStreamingResponse(self._knowledge_bases.data_sources)
+
+    @cached_property
+    def indexing_jobs(self) -> AsyncIndexingJobsResourceWithStreamingResponse:
+        return AsyncIndexingJobsResourceWithStreamingResponse(self._knowledge_bases.indexing_jobs)

gradientai/resources/{providers → model_providers}/__init__.py

@@ -16,13 +16,13 @@ from .anthropic import (
     AnthropicResourceWithStreamingResponse,
     AsyncAnthropicResourceWithStreamingResponse,
 )
-from .providers import (
-    ProvidersResource,
-    AsyncProvidersResource,
-    ProvidersResourceWithRawResponse,
-    AsyncProvidersResourceWithRawResponse,
-    ProvidersResourceWithStreamingResponse,
-    AsyncProvidersResourceWithStreamingResponse,
+from .model_providers import (
+    ModelProvidersResource,
+    AsyncModelProvidersResource,
+    ModelProvidersResourceWithRawResponse,
+    AsyncModelProvidersResourceWithRawResponse,
+    ModelProvidersResourceWithStreamingResponse,
+    AsyncModelProvidersResourceWithStreamingResponse,
 )
 
 __all__ = [
@@ -38,10 +38,10 @@ __all__ = [
     "AsyncOpenAIResourceWithRawResponse",
     "OpenAIResourceWithStreamingResponse",
     "AsyncOpenAIResourceWithStreamingResponse",
-    "ProvidersResource",
-    "AsyncProvidersResource",
-    "ProvidersResourceWithRawResponse",
-    "AsyncProvidersResourceWithRawResponse",
-    "ProvidersResourceWithStreamingResponse",
-    "AsyncProvidersResourceWithStreamingResponse",
+    "ModelProvidersResource",
+    "AsyncModelProvidersResource",
+    "ModelProvidersResourceWithRawResponse",
+    "AsyncModelProvidersResourceWithRawResponse",
+    "ModelProvidersResourceWithStreamingResponse",
+    "AsyncModelProvidersResourceWithStreamingResponse",
 ]