feat(api): update via SDK Studio

Author: stainless-app[bot]

.stats.yml

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

src/gradient/_client.py

@@ -82,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,
@@ -91,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,
@@ -121,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")
@@ -145,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
@@ -334,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,
@@ -378,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,
@@ -444,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,
@@ -453,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,
@@ -483,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")
@@ -507,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
@@ -696,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,
@@ -740,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,