Merge pull request #101 from stainless-sdks/chandrudo/update-kbaas

Chandrudo/update kbaas

Author: chandrudo

.stats.yml

@@ -1,4 +1,4 @@
-configured_endpoints: 188
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/digitalocean%2Fgradient-f07d74847e620dfa26d8df40ea4680814af9bba381b3a57a7b6ed76ad49d85f8.yml
-openapi_spec_hash: e3553dc2abf2afd4368b736bcc32a289
-config_hash: b28984dd49d4baf1d68572efe83ac103
+configured_endpoints: 189
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/digitalocean%2Fgradient-ca993fd0bc66f703323a773c92da75207266f7f9d8c54ddac2fbd271a3cdaf86.yml
+openapi_spec_hash: 35d7edb04aab2ab28bc7e5851a54b4e3
+config_hash: fad48c8ac796b240fe3b90181586d1a4

api.md

@@ -1027,3 +1027,15 @@ Methods:
 - <code title="get /v2/nfs/snapshots/{nfs_snapshot_id}">client.nfs.snapshots.<a href="./src/gradient/resources/nfs/snapshots.py">retrieve</a>(nfs_snapshot_id, \*\*<a href="src/gradient/types/nfs/snapshot_retrieve_params.py">params</a>) -> <a href="./src/gradient/types/nfs/snapshot_retrieve_response.py">SnapshotRetrieveResponse</a></code>
 - <code title="get /v2/nfs/snapshots">client.nfs.snapshots.<a href="./src/gradient/resources/nfs/snapshots.py">list</a>(\*\*<a href="src/gradient/types/nfs/snapshot_list_params.py">params</a>) -> <a href="./src/gradient/types/nfs/snapshot_list_response.py">SnapshotListResponse</a></code>
 - <code title="delete /v2/nfs/snapshots/{nfs_snapshot_id}">client.nfs.snapshots.<a href="./src/gradient/resources/nfs/snapshots.py">delete</a>(nfs_snapshot_id, \*\*<a href="src/gradient/types/nfs/snapshot_delete_params.py">params</a>) -> None</code>
+
+# Retrieve
+
+Types:
+
+```python
+from gradient.types import RetrieveDocumentsResponse
+```
+
+Methods:
+
+- <code title="post /{knowledgeBaseId}/retrieve">client.retrieve.<a href="./src/gradient/resources/retrieve.py">documents</a>(knowledge_base_id, \*\*<a href="src/gradient/types/retrieve_documents_params.py">params</a>) -> <a href="./src/gradient/types/retrieve_documents_response.py">RetrieveDocumentsResponse</a></code>

src/gradient/_client.py

@@ -39,6 +39,7 @@
         images,
         models,
         regions,
+        retrieve,
         databases,
         inference,
         gpu_droplets,
@@ -47,6 +48,7 @@
     from .resources.images import ImagesResource, AsyncImagesResource
     from .resources.nfs.nfs import NfsResource, AsyncNfsResource
     from .resources.regions import RegionsResource, AsyncRegionsResource
+    from .resources.retrieve import RetrieveResource, AsyncRetrieveResource
     from .resources.chat.chat import ChatResource, AsyncChatResource
     from .resources.gpu_droplets import (
         GPUDropletsResource,
@@ -80,6 +82,7 @@ class Gradient(SyncAPIClient):
     agent_access_key: str | None
     _agent_endpoint: str | None
     inference_endpoint: str | None
+    kbass_endpoint: str | None
 
     def __init__(
         self,
@@ -89,6 +92,7 @@ def __init__(
         agent_access_key: str | None = None,
         agent_endpoint: str | None = None,
         inference_endpoint: str | None = None,
+        kbass_endpoint: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: float | Timeout | None | NotGiven = not_given,
         max_retries: int = DEFAULT_MAX_RETRIES,
@@ -119,6 +123,7 @@ def __init__(
         - `agent_access_key` from `GRADIENT_AGENT_ACCESS_KEY`
         - `agent_endpoint` from `GRADIENT_AGENT_ENDPOINT`
         - `inference_endpoint` from `GRADIENT_INFERENCE_ENDPOINT`
+        - `kbass_endpoint` from `GRADIENT_KBASS_ENDPOINT`
         """
         if access_token is None:
             access_token = os.environ.get("DIGITALOCEAN_ACCESS_TOKEN")
@@ -143,6 +148,10 @@ def __init__(
             )
         self.inference_endpoint = inference_endpoint
 
+        if kbass_endpoint is None:
+            kbass_endpoint = os.environ.get("GRADIENT_KBASS_ENDPOINT") or "kbaas.do-ai.run"
+        self.kbass_endpoint = kbass_endpoint
+
         if base_url is None:
             base_url = os.environ.get("GRADIENT_BASE_URL")
         self._base_url_overridden = base_url is not None
@@ -237,6 +246,12 @@ def nfs(self) -> NfsResource:
 
         return NfsResource(self)
 
+    @cached_property
+    def retrieve(self) -> RetrieveResource:
+        from .resources.retrieve import RetrieveResource
+
+        return RetrieveResource(self)
+
     @cached_property
     def with_raw_response(self) -> GradientWithRawResponse:
         return GradientWithRawResponse(self)
@@ -326,6 +341,7 @@ def copy(
         agent_access_key: str | None = None,
         agent_endpoint: str | None = None,
         inference_endpoint: str | None = None,
+        kbass_endpoint: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: float | Timeout | None | NotGiven = not_given,
         http_client: httpx.Client | None = None,
@@ -370,6 +386,7 @@ def copy(
             agent_access_key=agent_access_key or self.agent_access_key,
             agent_endpoint=agent_endpoint or self._agent_endpoint,
             inference_endpoint=inference_endpoint or self.inference_endpoint,
+            kbass_endpoint=kbass_endpoint or self.kbass_endpoint,
             base_url=base_url or self.base_url,
             timeout=self.timeout if isinstance(timeout, NotGiven) else timeout,
             http_client=http_client,
@@ -436,6 +453,7 @@ class AsyncGradient(AsyncAPIClient):
     agent_access_key: str | None
     _agent_endpoint: str | None
     inference_endpoint: str | None
+    kbass_endpoint: str | None
 
     def __init__(
         self,
@@ -445,6 +463,7 @@ def __init__(
         agent_access_key: str | None = None,
         agent_endpoint: str | None = None,
         inference_endpoint: str | None = None,
+        kbass_endpoint: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: float | Timeout | None | NotGiven = not_given,
         max_retries: int = DEFAULT_MAX_RETRIES,
@@ -475,6 +494,7 @@ def __init__(
         - `agent_access_key` from `GRADIENT_AGENT_ACCESS_KEY`
         - `agent_endpoint` from `GRADIENT_AGENT_ENDPOINT`
         - `inference_endpoint` from `GRADIENT_INFERENCE_ENDPOINT`
+        - `kbass_endpoint` from `GRADIENT_KBASS_ENDPOINT`
         """
         if access_token is None:
             access_token = os.environ.get("DIGITALOCEAN_ACCESS_TOKEN")
@@ -499,6 +519,10 @@ def __init__(
             )
         self.inference_endpoint = inference_endpoint
 
+        if kbass_endpoint is None:
+            kbass_endpoint = os.environ.get("GRADIENT_KBASS_ENDPOINT") or "kbaas.do-ai.run"
+        self.kbass_endpoint = kbass_endpoint
+
         if base_url is None:
             base_url = os.environ.get("GRADIENT_BASE_URL")
         self._base_url_overridden = base_url is not None
@@ -593,6 +617,12 @@ def nfs(self) -> AsyncNfsResource:
 
         return AsyncNfsResource(self)
 
+    @cached_property
+    def retrieve(self) -> AsyncRetrieveResource:
+        from .resources.retrieve import AsyncRetrieveResource
+
+        return AsyncRetrieveResource(self)
+
     @cached_property
     def with_raw_response(self) -> AsyncGradientWithRawResponse:
         return AsyncGradientWithRawResponse(self)
@@ -682,6 +712,7 @@ def copy(
         model_access_key: str | None = None,
         agent_access_key: str | None = None,
         inference_endpoint: str | None = None,
+        kbass_endpoint: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: float | Timeout | None | NotGiven = not_given,
         http_client: httpx.AsyncClient | None = None,
@@ -726,6 +757,7 @@ def copy(
             agent_access_key=agent_access_key or self.agent_access_key,
             agent_endpoint=agent_endpoint or self._agent_endpoint,
             inference_endpoint=inference_endpoint or self.inference_endpoint,
+            kbass_endpoint=kbass_endpoint or self.kbass_endpoint,
             base_url=base_url or self.base_url,
             timeout=self.timeout if isinstance(timeout, NotGiven) else timeout,
             http_client=http_client,
@@ -851,6 +883,12 @@ def nfs(self) -> nfs.NfsResourceWithRawResponse:
 
         return NfsResourceWithRawResponse(self._client.nfs)
 
+    @cached_property
+    def retrieve(self) -> retrieve.RetrieveResourceWithRawResponse:
+        from .resources.retrieve import RetrieveResourceWithRawResponse
+
+        return RetrieveResourceWithRawResponse(self._client.retrieve)
+
 
 class AsyncGradientWithRawResponse:
     _client: AsyncGradient
@@ -922,6 +960,12 @@ def nfs(self) -> nfs.AsyncNfsResourceWithRawResponse:
 
         return AsyncNfsResourceWithRawResponse(self._client.nfs)
 
+    @cached_property
+    def retrieve(self) -> retrieve.AsyncRetrieveResourceWithRawResponse:
+        from .resources.retrieve import AsyncRetrieveResourceWithRawResponse
+
+        return AsyncRetrieveResourceWithRawResponse(self._client.retrieve)
+
 
 class GradientWithStreamedResponse:
     _client: Gradient
@@ -993,6 +1037,12 @@ def nfs(self) -> nfs.NfsResourceWithStreamingResponse:
 
         return NfsResourceWithStreamingResponse(self._client.nfs)
 
+    @cached_property
+    def retrieve(self) -> retrieve.RetrieveResourceWithStreamingResponse:
+        from .resources.retrieve import RetrieveResourceWithStreamingResponse
+
+        return RetrieveResourceWithStreamingResponse(self._client.retrieve)
+
 
 class AsyncGradientWithStreamedResponse:
     _client: AsyncGradient
@@ -1070,6 +1120,12 @@ def nfs(self) -> nfs.AsyncNfsResourceWithStreamingResponse:
 
         return AsyncNfsResourceWithStreamingResponse(self._client.nfs)
 
+    @cached_property
+    def retrieve(self) -> retrieve.AsyncRetrieveResourceWithStreamingResponse:
+        from .resources.retrieve import AsyncRetrieveResourceWithStreamingResponse
+
+        return AsyncRetrieveResourceWithStreamingResponse(self._client.retrieve)
+
 
 Client = Gradient
 

src/gradient/resources/__init__.py

@@ -48,6 +48,14 @@
     RegionsResourceWithStreamingResponse,
     AsyncRegionsResourceWithStreamingResponse,
 )
+from .retrieve import (
+    RetrieveResource,
+    AsyncRetrieveResource,
+    RetrieveResourceWithRawResponse,
+    AsyncRetrieveResourceWithRawResponse,
+    RetrieveResourceWithStreamingResponse,
+    AsyncRetrieveResourceWithStreamingResponse,
+)
 from .databases import (
     DatabasesResource,
     AsyncDatabasesResource,
@@ -142,4 +150,10 @@
     "AsyncNfsResourceWithRawResponse",
     "NfsResourceWithStreamingResponse",
     "AsyncNfsResourceWithStreamingResponse",
+    "RetrieveResource",
+    "AsyncRetrieveResource",
+    "RetrieveResourceWithRawResponse",
+    "AsyncRetrieveResourceWithRawResponse",
+    "RetrieveResourceWithStreamingResponse",
+    "AsyncRetrieveResourceWithStreamingResponse",
 ]

src/gradient/resources/agents/chat/completions.py

@@ -472,9 +472,7 @@ def create(
         headers = {"Authorization": f"Bearer {self._client.agent_access_key}", **headers}
 
         return self._post(
-            "/chat/completions?agent=true"
-            if self._client._base_url_overridden
-            else f"{self._client.agent_endpoint}/api/v1/chat/completions?agent=true",
+            "/chat/completions?agent=true",
             body=maybe_transform(
                 {
                     "messages": messages,
@@ -958,9 +956,7 @@ async def create(
         headers = {"Authorization": f"Bearer {self._client.agent_access_key}", **headers}
 
         return await self._post(
-            "/chat/completions?agent=true"
-            if self._client._base_url_overridden
-            else f"{self._client.agent_endpoint}/api/v1/chat/completions?agent=true",
+            "/chat/completions?agent=true",
             body=await async_maybe_transform(
                 {
                     "messages": messages,

src/gradient/resources/chat/completions.py

@@ -473,9 +473,7 @@ def create(
         }
 
         return self._post(
-            "/chat/completions"
-            if self._client._base_url_overridden
-            else f"{self._client.inference_endpoint}/v1/chat/completions",
+            "/chat/completions",
             body=maybe_transform(
                 {
                     "messages": messages,
@@ -963,9 +961,7 @@ async def create(
         }
 
         return await self._post(
-            "/chat/completions"
-            if self._client._base_url_overridden
-            else f"{self._client.inference_endpoint}/v1/chat/completions",
+            "/chat/completions",
             body=await async_maybe_transform(
                 {
                     "messages": messages,

src/gradient/resources/images.py

@@ -317,9 +317,7 @@ def generate(
         }
 
         return self._post(
-            "/images/generations"
-            if self._client._base_url_overridden
-            else f"{self._client.inference_endpoint}/v1/images/generations",
+            "/images/generations",
             body=maybe_transform(
                 {
                     "prompt": prompt,
@@ -638,9 +636,7 @@ async def generate(
             **headers,
         }
         return await self._post(
-            "/images/generations"
-            if self._client._base_url_overridden
-            else f"{self._client.inference_endpoint}/v1/images/generations",
+            "/images/generations",
             body=await async_maybe_transform(
                 {
                     "prompt": prompt,