feat(api): manual updates

Add GRADIENTAI_AGENT_KEY

Author: stainless-app[bot]

.stats.yml

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

src/gradientai/_client.py

@@ -56,13 +56,15 @@ class GradientAI(SyncAPIClient):
     # client options
     api_key: str | None
     inference_key: str | None
+    agent_key: str | None
     agent_domain: str | None
 
     def __init__(
         self,
         *,
         api_key: str | None = None,
         inference_key: str | None = None,
+        agent_key: str | None = None,
         agent_domain: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: Union[float, Timeout, None, NotGiven] = NOT_GIVEN,
@@ -88,6 +90,7 @@ def __init__(
         This automatically infers the following arguments from their corresponding environment variables if they are not provided:
         - `api_key` from `GRADIENTAI_API_KEY`
         - `inference_key` from `GRADIENTAI_INFERENCE_KEY`
+        - `agent_key` from `GRADIENTAI_AGENT_KEY`
         """
         if api_key is None:
             api_key = os.environ.get("GRADIENTAI_API_KEY")
@@ -97,6 +100,10 @@ def __init__(
             inference_key = os.environ.get("GRADIENTAI_INFERENCE_KEY")
         self.inference_key = inference_key
 
+        if agent_key is None:
+            agent_key = os.environ.get("GRADIENTAI_AGENT_KEY")
+        self.agent_key = agent_key
+
         self.agent_domain = agent_domain
 
         if base_url is None:
@@ -200,6 +207,7 @@ def copy(
         *,
         api_key: str | None = None,
         inference_key: str | None = None,
+        agent_key: str | None = None,
         agent_domain: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: float | Timeout | None | NotGiven = NOT_GIVEN,
@@ -236,6 +244,7 @@ def copy(
         client = self.__class__(
             api_key=api_key or self.api_key,
             inference_key=inference_key or self.inference_key,
+            agent_key=agent_key or self.agent_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,
@@ -290,13 +299,15 @@ class AsyncGradientAI(AsyncAPIClient):
     # client options
     api_key: str | None
     inference_key: str | None
+    agent_key: str | None
     agent_domain: str | None
 
     def __init__(
         self,
         *,
         api_key: str | None = None,
         inference_key: str | None = None,
+        agent_key: str | None = None,
         agent_domain: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: Union[float, Timeout, None, NotGiven] = NOT_GIVEN,
@@ -322,6 +333,7 @@ def __init__(
         This automatically infers the following arguments from their corresponding environment variables if they are not provided:
         - `api_key` from `GRADIENTAI_API_KEY`
         - `inference_key` from `GRADIENTAI_INFERENCE_KEY`
+        - `agent_key` from `GRADIENTAI_AGENT_KEY`
         """
         if api_key is None:
             api_key = os.environ.get("GRADIENTAI_API_KEY")
@@ -331,6 +343,10 @@ def __init__(
             inference_key = os.environ.get("GRADIENTAI_INFERENCE_KEY")
         self.inference_key = inference_key
 
+        if agent_key is None:
+            agent_key = os.environ.get("GRADIENTAI_AGENT_KEY")
+        self.agent_key = agent_key
+
         self.agent_domain = agent_domain
 
         if base_url is None:
@@ -434,6 +450,7 @@ def copy(
         *,
         api_key: str | None = None,
         inference_key: str | None = None,
+        agent_key: str | None = None,
         agent_domain: str | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: float | Timeout | None | NotGiven = NOT_GIVEN,
@@ -470,6 +487,7 @@ def copy(
         client = self.__class__(
             api_key=api_key or self.api_key,
             inference_key=inference_key or self.inference_key,
+            agent_key=agent_key or self.agent_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,

tests/conftest.py

@@ -47,6 +47,7 @@ def pytest_collection_modifyitems(items: list[pytest.Function]) -> None:
 
 api_key = "My API Key"
 inference_key = "My Inference Key"
+agent_key = "My Agent Key"
 
 
 @pytest.fixture(scope="session")
@@ -56,7 +57,11 @@ def client(request: FixtureRequest) -> Iterator[GradientAI]:
         raise TypeError(f"Unexpected fixture parameter type {type(strict)}, expected {bool}")
 
     with GradientAI(
-        base_url=base_url, api_key=api_key, inference_key=inference_key, _strict_response_validation=strict
+        base_url=base_url,
+        api_key=api_key,
+        inference_key=inference_key,
+        agent_key=agent_key,
+        _strict_response_validation=strict,
     ) as client:
         yield client
 
@@ -85,6 +90,7 @@ async def async_client(request: FixtureRequest) -> AsyncIterator[AsyncGradientAI
         base_url=base_url,
         api_key=api_key,
         inference_key=inference_key,
+        agent_key=agent_key,
         _strict_response_validation=strict,
         http_client=http_client,
     ) as client: