Merge branch 'main' into main

Author: areeb1501

.release-please-manifest.json

@@ -1,3 +1,3 @@
 {
-  ".": "3.3.0"
+  ".": "3.6.0"
 }

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 175
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/digitalocean%2Fgradient-40c154e2fdc4fef9ca1cf8329c29b7102bf324acfc9589571d6f3452d1ca579c.yml
-openapi_spec_hash: 83a3d092965fde776b29b61f785459f9
-config_hash: 8497af1695ff361853c745dd869dc6b9
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/digitalocean%2Fgradient-cb3bf9b21459cad24410206c27a32fd31ef6cf86711700597549dbbd0d634002.yml
+openapi_spec_hash: 6a9149a81ba15e7c5c5c1f4d77daad92
+config_hash: bad49c3bf949d5168ec3896bedff253a

CHANGELOG.md

@@ -1,5 +1,40 @@
 # Changelog
 
+## 3.6.0 (2025-10-16)
+
+Full Changelog: [v3.5.0...v3.6.0](https://github.com/digitalocean/gradient-python/compare/v3.5.0...v3.6.0)
+
+### Features
+
+* **api:** manual updates ([da88e9e](https://github.com/digitalocean/gradient-python/commit/da88e9eee0adc6152d0d8212305397483be0d686))
+
+
+### Bug Fixes
+
+* lints ([a1b1fc6](https://github.com/digitalocean/gradient-python/commit/a1b1fc6b7747c00d9bfc2b86c6262e9c123416dc))
+* test setup needs all three access keys ([01ac735](https://github.com/digitalocean/gradient-python/commit/01ac735fb965686699df82ec8763b18ceb660972))
+
+## 3.5.0 (2025-10-14)
+
+Full Changelog: [v3.4.0...v3.5.0](https://github.com/digitalocean/gradient-python/compare/v3.4.0...v3.5.0)
+
+### Features
+
+* **api:** update via SDK Studio ([#74](https://github.com/digitalocean/gradient-python/issues/74)) ([e1ab040](https://github.com/digitalocean/gradient-python/commit/e1ab0407e88f5394f5c299940a4b2fe72dbbf70e))
+
+
+### Chores
+
+* **internal:** detect missing future annotations with ruff ([0fb9f92](https://github.com/digitalocean/gradient-python/commit/0fb9f9254a0f72a721fa73823399e58eec723f1a))
+
+## 3.4.0 (2025-10-09)
+
+Full Changelog: [v3.3.0...v3.4.0](https://github.com/digitalocean/gradient-python/compare/v3.3.0...v3.4.0)
+
+### Features
+
+* **api:** manual updates ([bbd7ddc](https://github.com/digitalocean/gradient-python/commit/bbd7ddccfb3d98f39e61948365b92202b3cc9e33))
+
 ## 3.3.0 (2025-10-07)
 
 Full Changelog: [v3.2.0...v3.3.0](https://github.com/digitalocean/gradient-python/compare/v3.2.0...v3.3.0)

README.md

@@ -44,12 +44,14 @@ client = Gradient(
     ),  # This is the default and can be omitted
 )
 inference_client = Gradient(
-    inference_key=os.environ.get(
+    model_access_key=os.environ.get(
         "GRADIENT_MODEL_ACCESS_KEY"
     ),  # This is the default and can be omitted
 )
 agent_client = Gradient(
-    agent_key=os.environ.get("GRADIENT_AGENT_ACCESS_KEY"),  # This is the default and can be omitted
+    agent_access_key=os.environ.get(
+        "GRADIENT_AGENT_ACCESS_KEY"
+    ),  # This is the default and can be omitted
     agent_endpoint="https://my-agent.agents.do-ai.run",
 )
 
@@ -103,11 +105,7 @@ import os
 import asyncio
 from gradient import AsyncGradient
 
-client = AsyncGradient(
-    access_token=os.environ.get(
-        "DIGITALOCEAN_ACCESS_TOKEN"
-    ),  # This is the default and can be omitted
-)
+client = AsyncGradient()
 
 
 async def main() -> None:
@@ -149,7 +147,6 @@ from gradient import AsyncGradient
 
 async def main() -> None:
     async with AsyncGradient(
-        access_token="My Access Token",
         http_client=DefaultAioHttpClient(),
     ) as client:
         completion = await client.chat.completions.create(

examples/agent_wait_until_ready.py

@@ -0,0 +1,100 @@
+"""
+Example: Wait for Agent Deployment to Complete
+
+This example demonstrates how to use the wait_until_ready() method to wait for
+an agent to finish deploying before using it.
+"""
+
+from gradient import Gradient
+from gradient._exceptions import AgentDeploymentError, AgentDeploymentTimeoutError
+
+# Initialize the Gradient client
+client = Gradient()
+
+# Create a new agent
+agent_response = client.agents.create(
+    name="My Agent",
+    instruction="You are a helpful assistant",
+    model_uuid="<your-model-uuid>",
+    region="nyc1",
+)
+
+agent_id = agent_response.agent.uuid if agent_response.agent else None
+
+if agent_id:
+    print(f"Agent created with ID: {agent_id}")
+    print("Waiting for agent to be ready...")
+    
+    try:
+        # Wait for the agent to be deployed and ready
+        # This will poll the agent status every 5 seconds (default)
+        # and wait up to 5 minutes (default timeout=300 seconds)
+        ready_agent = client.agents.wait_until_ready(
+            agent_id,
+            poll_interval=5.0,  # Check every 5 seconds
+            timeout=300.0,      # Wait up to 5 minutes
+        )
+        
+        if ready_agent.agent and ready_agent.agent.deployment:
+            print(f"Agent is ready! Status: {ready_agent.agent.deployment.status}")
+            print(f"Agent URL: {ready_agent.agent.url}")
+        
+        # Now you can use the agent
+        # ...
+        
+    except AgentDeploymentError as e:
+        print(f"Agent deployment failed: {e}")
+        print(f"Failed status: {e.status}")
+        
+    except AgentDeploymentTimeoutError as e:
+        print(f"Agent deployment timed out: {e}")
+        print(f"Agent ID: {e.agent_id}")
+        
+    except Exception as e:
+        print(f"Unexpected error: {e}")
+
+
+# Async example
+from gradient import AsyncGradient
+
+
+async def main() -> None:
+    async_client = AsyncGradient()
+    
+    # Create a new agent
+    agent_response = await async_client.agents.create(
+        name="My Async Agent",
+        instruction="You are a helpful assistant",
+        model_uuid="<your-model-uuid>",
+        region="nyc1",
+    )
+    
+    agent_id = agent_response.agent.uuid if agent_response.agent else None
+    
+    if agent_id:
+        print(f"Agent created with ID: {agent_id}")
+        print("Waiting for agent to be ready...")
+        
+        try:
+            # Wait for the agent to be deployed and ready (async)
+            ready_agent = await async_client.agents.wait_until_ready(
+                agent_id,
+                poll_interval=5.0,
+                timeout=300.0,
+            )
+            
+            if ready_agent.agent and ready_agent.agent.deployment:
+                print(f"Agent is ready! Status: {ready_agent.agent.deployment.status}")
+                print(f"Agent URL: {ready_agent.agent.url}")
+            
+        except AgentDeploymentError as e:
+            print(f"Agent deployment failed: {e}")
+            print(f"Failed status: {e.status}")
+            
+        except AgentDeploymentTimeoutError as e:
+            print(f"Agent deployment timed out: {e}")
+            print(f"Agent ID: {e.agent_id}")
+
+
+# Uncomment to run async example
+# asyncio.run(main())

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "gradient"
-version = "3.3.0"
+version = "3.6.0"
 description = "The official Python library for the Gradient API"
 dynamic = ["readme"]
 license = "Apache-2.0"
@@ -117,11 +117,14 @@ replacement = '[\1](https://github.com/digitalocean/gradient-python/tree/main/\g
 
 [tool.pytest.ini_options]
 testpaths = ["tests"]
-addopts = "--tb=short -n auto"
+addopts = "--tb=short -n auto -m 'not smoke'"
 xfail_strict = true
 asyncio_mode = "auto"
 asyncio_default_fixture_loop_scope = "session"
 filterwarnings = ["error"]
+markers = [
+    "smoke: lightweight external integration smoke tests hitting live Gradient services",
+]
 
 [tool.pyright]
 # this enables practically every flag given by pyright.
@@ -211,6 +214,8 @@ select = [
   "B",
   # remove unused imports
   "F401",
+  # check for missing future annotations
+  "FA102",
   # bare except statements
   "E722",
   # unused arguments
@@ -233,6 +238,8 @@ unfixable = [
   "T203",
 ]
 
+extend-safe-fixes = ["FA102"]
+
 [tool.ruff.lint.flake8-tidy-imports.banned-api]
 "functools.lru_cache".msg = "This function does not retain type information for the wrapped function's arguments; The `lru_cache` function from `_utils` should be used instead"
 

pytest.ini

@@ -548,14 +548,14 @@ def _build_request(
         # TODO: report this error to httpx
         return self._client.build_request(  # pyright: ignore[reportUnknownMemberType]
             headers=headers,
-            timeout=self.timeout if isinstance(options.timeout, NotGiven) else options.timeout,
+            timeout=(self.timeout if isinstance(options.timeout, NotGiven) else options.timeout),
             method=options.method,
             url=prepared_url,
             # the `Query` type that we use is incompatible with qs'
             # `Params` type as it needs to be typed as `Mapping[str, object]`
             # so that passing a `TypedDict` doesn't cause an error.
             # https://github.com/microsoft/pyright/issues/3526#event-6715453066
-            params=self.qs.stringify(cast(Mapping[str, Any], params)) if params else None,
+            params=(self.qs.stringify(cast(Mapping[str, Any], params)) if params else None),
             **kwargs,
         )
 
@@ -1067,7 +1067,12 @@ def request(
         )
 
     def _sleep_for_retry(
-        self, *, retries_taken: int, max_retries: int, options: FinalRequestOptions, response: httpx.Response | None
+        self,
+        *,
+        retries_taken: int,
+        max_retries: int,
+        options: FinalRequestOptions,
+        response: httpx.Response | None,
     ) -> None:
         remaining_retries = max_retries - retries_taken
         if remaining_retries == 1:
@@ -1248,7 +1253,11 @@ def post(
         stream_cls: type[_StreamT] | None = None,
     ) -> ResponseT | _StreamT:
         opts = FinalRequestOptions.construct(
-            method="post", url=path, json_data=body, files=to_httpx_files(files), **options
+            method="post",
+            url=path,
+            json_data=body,
+            files=to_httpx_files(files),
+            **options,
         )
         return cast(ResponseT, self.request(cast_to, opts, stream=stream, stream_cls=stream_cls))
 
@@ -1273,7 +1282,11 @@ def put(
         options: RequestOptions = {},
     ) -> ResponseT:
         opts = FinalRequestOptions.construct(
-            method="put", url=path, json_data=body, files=to_httpx_files(files), **options
+            method="put",
+            url=path,
+            json_data=body,
+            files=to_httpx_files(files),
+            **options,
         )
         return self.request(cast_to, opts)
 
@@ -1317,6 +1330,7 @@ def __init__(self, **kwargs: Any) -> None:
     class _DefaultAioHttpClient(httpx.AsyncClient):
         def __init__(self, **_kwargs: Any) -> None:
             raise RuntimeError("To use the aiohttp client you must have installed the package with the `aiohttp` extra")
+
 else:
 
     class _DefaultAioHttpClient(httpx_aiohttp.HttpxAiohttpClient):  # type: ignore
@@ -1603,7 +1617,12 @@ async def request(
         )
 
     async def _sleep_for_retry(
-        self, *, retries_taken: int, max_retries: int, options: FinalRequestOptions, response: httpx.Response | None
+        self,
+        *,
+        retries_taken: int,
+        max_retries: int,
+        options: FinalRequestOptions,
+        response: httpx.Response | None,
     ) -> None:
         remaining_retries = max_retries - retries_taken
         if remaining_retries == 1:
@@ -1772,7 +1791,11 @@ async def post(
         stream_cls: type[_AsyncStreamT] | None = None,
     ) -> ResponseT | _AsyncStreamT:
         opts = FinalRequestOptions.construct(
-            method="post", url=path, json_data=body, files=await async_to_httpx_files(files), **options
+            method="post",
+            url=path,
+            json_data=body,
+            files=await async_to_httpx_files(files),
+            **options,
         )
         return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls)
 
@@ -1797,7 +1820,11 @@ async def put(
         options: RequestOptions = {},
     ) -> ResponseT:
         opts = FinalRequestOptions.construct(
-            method="put", url=path, json_data=body, files=await async_to_httpx_files(files), **options
+            method="put",
+            url=path,
+            json_data=body,
+            files=await async_to_httpx_files(files),
+            **options,
         )
         return await self.request(cast_to, opts)