feat(client): add agent_domain option

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 77
 openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/digitalocean%2Fgradientai-e8b3cbc80e18e4f7f277010349f25e1319156704f359911dc464cc21a0d077a6.yml
 openapi_spec_hash: c773d792724f5647ae25a5ae4ccec208
-config_hash: 9c2519464cf5de240e34bd89b9f65706
+config_hash: 80395df07aec0cf06a5aa7caa2e3f4bc

src/gradientai/_client.py

@@ -57,12 +57,14 @@ class GradientAI(SyncAPIClient):
     # client options
     api_key: str | None
     inference_key: str | None
+    agent_domain: str | None
 
     def __init__(
         self,
         *,
         api_key: str | None = None,
         inference_key: str | None = None,
+        agent_domain: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: Union[float, Timeout, None, NotGiven] = NOT_GIVEN,
         max_retries: int = DEFAULT_MAX_RETRIES,
@@ -96,6 +98,8 @@ def __init__(
             inference_key = os.environ.get("GRADIENTAI_INFERENCE_KEY")
         self.inference_key = inference_key
 
+        self.agent_domain = agent_domain
+
         if base_url is None:
             base_url = os.environ.get("GRADIENT_AI_BASE_URL")
         self._base_url_overridden = base_url is not None
@@ -201,6 +205,7 @@ def copy(
         *,
         api_key: str | None = None,
         inference_key: str | None = None,
+        agent_domain: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: float | Timeout | None | NotGiven = NOT_GIVEN,
         http_client: httpx.Client | None = None,
@@ -236,6 +241,7 @@ def copy(
         client = self.__class__(
             api_key=api_key or self.api_key,
             inference_key=inference_key or self.inference_key,
+            agent_domain=agent_domain or self.agent_domain,
             base_url=base_url or self.base_url,
             timeout=self.timeout if isinstance(timeout, NotGiven) else timeout,
             http_client=http_client,
@@ -289,12 +295,14 @@ class AsyncGradientAI(AsyncAPIClient):
     # client options
     api_key: str | None
     inference_key: str | None
+    agent_domain: str | None
 
     def __init__(
         self,
         *,
         api_key: str | None = None,
         inference_key: str | None = None,
+        agent_domain: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: Union[float, Timeout, None, NotGiven] = NOT_GIVEN,
         max_retries: int = DEFAULT_MAX_RETRIES,
@@ -328,6 +336,8 @@ def __init__(
             inference_key = os.environ.get("GRADIENTAI_INFERENCE_KEY")
         self.inference_key = inference_key
 
+        self.agent_domain = agent_domain
+
         if base_url is None:
             base_url = os.environ.get("GRADIENT_AI_BASE_URL")
         self._base_url_overridden = base_url is not None
@@ -433,6 +443,7 @@ def copy(
         *,
         api_key: str | None = None,
         inference_key: str | None = None,
+        agent_domain: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: float | Timeout | None | NotGiven = NOT_GIVEN,
         http_client: httpx.AsyncClient | None = None,
@@ -468,6 +479,7 @@ def copy(
         client = self.__class__(
             api_key=api_key or self.api_key,
             inference_key=inference_key or self.inference_key,
+            agent_domain=agent_domain or self.agent_domain,
             base_url=base_url or self.base_url,
             timeout=self.timeout if isinstance(timeout, NotGiven) else timeout,
             http_client=http_client,