feat(api): make kwargs match the env vars

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 170
 openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/digitalocean%2Fgradient-9aca3802735e1375125412aa28ac36bf2175144b8218610a73d2e7f775694dff.yml
 openapi_spec_hash: e29d14e3e4679fcf22b3e760e49931b1
-config_hash: 136e1973eb6297e6308a165594bd00a3
+config_hash: 99e3cd5dde0beb796f4547410869f726

src/gradient/_client.py

@@ -105,9 +105,9 @@ def __init__(
         """Construct a new synchronous Gradient client instance.
 
         This automatically infers the following arguments from their corresponding environment variables if they are not provided:
-        - `api_key` from `DIGITALOCEAN_ACCESS_TOKEN`
-        - `inference_key` from `GRADIENT_MODEL_ACCESS_KEY`
-        - `agent_key` from `GRADIENT_AGENT_ACCESS_KEY`
+        - `access_token` from `DIGITALOCEAN_ACCESS_TOKEN`
+        - `model_access_key` from `GRADIENT_MODEL_ACCESS_KEY`
+        - `agent_access_key` from `GRADIENT_AGENT_ACCESS_KEY`
         """
         if api_key is None:
             api_key = os.environ.get("DIGITALOCEAN_ACCESS_TOKEN")
@@ -229,10 +229,10 @@ def qs(self) -> Querystring:
     @property
     @override
     def auth_headers(self) -> dict[str, str]:
-        api_key = self.api_key
-        if api_key is None:
+        access_token = self.access_token
+        if access_token is None:
             return {}
-        return {"Authorization": f"Bearer {api_key}"}
+        return {"Authorization": f"Bearer {access_token}"}
 
     @property
     @override
@@ -245,15 +245,15 @@ def default_headers(self) -> dict[str, str | Omit]:
 
     @override
     def _validate_headers(self, headers: Headers, custom_headers: Headers) -> None:
-        if (self.api_key or self.agent_key or self.inference_key) and headers.get(
-            "Authorization"
-        ):
+        if (
+            self.access_token or self.agent_access_key or self.model_access_key
+        ) and headers.get("Authorization"):
             return
         if isinstance(custom_headers.get("Authorization"), Omit):
             return
 
         raise TypeError(
-            '"Could not resolve authentication method. Expected api_key, agent_key, or inference_key to be set. Or for the `Authorization` headers to be explicitly omitted"'
+            '"Could not resolve authentication method. Expected access_token, agent_access_key, or model_access_key to be set. Or for the `Authorization` headers to be explicitly omitted"'
         )
 
     def copy(
@@ -397,9 +397,9 @@ def __init__(
         """Construct a new async AsyncGradient client instance.
 
         This automatically infers the following arguments from their corresponding environment variables if they are not provided:
-        - `api_key` from `DIGITALOCEAN_ACCESS_TOKEN`
-        - `inference_key` from `GRADIENT_MODEL_ACCESS_KEY`
-        - `agent_key` from `GRADIENT_AGENT_ACCESS_KEY`
+        - `access_token` from `DIGITALOCEAN_ACCESS_TOKEN`
+        - `model_access_key` from `GRADIENT_MODEL_ACCESS_KEY`
+        - `agent_access_key` from `GRADIENT_AGENT_ACCESS_KEY`
         """
         if api_key is None:
             api_key = os.environ.get("DIGITALOCEAN_ACCESS_TOKEN")
@@ -521,10 +521,10 @@ def qs(self) -> Querystring:
     @property
     @override
     def auth_headers(self) -> dict[str, str]:
-        api_key = self.api_key
-        if api_key is None:
+        access_token = self.access_token
+        if access_token is None:
             return {}
-        return {"Authorization": f"Bearer {api_key}"}
+        return {"Authorization": f"Bearer {access_token}"}
 
     @property
     @override
@@ -537,15 +537,15 @@ def default_headers(self) -> dict[str, str | Omit]:
 
     @override
     def _validate_headers(self, headers: Headers, custom_headers: Headers) -> None:
-        if (self.api_key or self.agent_key or self.inference_key) and headers.get(
-            "Authorization"
-        ):
+        if (
+            self.access_token or self.agent_access_key or self.model_access_key
+        ) and headers.get("Authorization"):
             return
         if isinstance(custom_headers.get("Authorization"), Omit):
             return
 
         raise TypeError(
-            '"Could not resolve authentication method. Expected api_key, agent_key, or inference_key to be set. Or for the `Authorization` headers to be explicitly omitted"'
+            '"Could not resolve authentication method. Expected access_token, agent_access_key, or model_access_key to be set. Or for the `Authorization` headers to be explicitly omitted"'
         )
 
     def copy(

tests/conftest.py

@@ -45,10 +45,9 @@ def pytest_collection_modifyitems(items: list[pytest.Function]) -> None:
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
 
-api_key = "My API Key"
-inference_key = "My Inference Key"
-agent_key = "My Agent Key"
-agent_endpoint = "https://inference.do-ai.run"
+access_token = "My Access Token"
+model_access_key = "My Model Access Key"
+agent_access_key = "My Agent Access Key"
 
 
 @pytest.fixture(scope="session")
@@ -59,10 +58,9 @@ def client(request: FixtureRequest) -> Iterator[Gradient]:
 
     with Gradient(
         base_url=base_url,
-        api_key=api_key,
-        inference_key=inference_key,
-        agent_key=agent_key,
-        agent_endpoint=agent_endpoint,
+        access_token=access_token,
+        model_access_key=model_access_key,
+        agent_access_key=agent_access_key,
         _strict_response_validation=strict,
     ) as client:
         yield client
@@ -90,9 +88,9 @@ async def async_client(request: FixtureRequest) -> AsyncIterator[AsyncGradient]:
 
     async with AsyncGradient(
         base_url=base_url,
-        api_key=api_key,
-        inference_key=inference_key,
-        agent_key=agent_key,
+        access_token=access_token,
+        model_access_key=model_access_key,
+        agent_access_key=agent_access_key,
         _strict_response_validation=strict,
         http_client=http_client,
     ) as client: