feat(api): manual updates

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
+configured_endpoints: 77
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/digitalocean%2Fgradientai-391afaae764eb758523b67805cb47ae3bc319dc119d83414afdd66f123ceaf5c.yml
 openapi_spec_hash: c773d792724f5647ae25a5ae4ccec208
-config_hash: 558ec54e9e056494abf623ff424c104e
+config_hash: 6edaff3557194ba8897d14f7ca74589c

README.md

@@ -73,7 +73,7 @@ client = AsyncGradientAI(
 
 
 async def main() -> None:
-    completion = await client.agents.chat.completions.create(
+    completion = await client.chat.completions.create(
         messages=[
             {
                 "role": "user",
@@ -115,7 +115,7 @@ async def main() -> None:
         api_key=os.environ.get("GRADIENTAI_API_KEY"),  # This is the default and can be omitted
         http_client=DefaultAioHttpClient(),
     ) as client:
-        completion = await client.agents.chat.completions.create(
+        completion = await client.chat.completions.create(
             messages=[
                 {
                     "role": "user",
@@ -139,7 +139,7 @@ from gradientai import GradientAI
 
 client = GradientAI()
 
-stream = client.agents.chat.completions.create(
+stream = client.chat.completions.create(
     messages=[
         {
             "role": "user",
@@ -160,7 +160,7 @@ from gradientai import AsyncGradientAI
 
 client = AsyncGradientAI()
 
-stream = await client.agents.chat.completions.create(
+stream = await client.chat.completions.create(
     messages=[
         {
             "role": "user",
@@ -192,7 +192,7 @@ from gradientai import GradientAI
 
 client = GradientAI()
 
-completion = client.agents.chat.completions.create(
+completion = client.chat.completions.create(
     messages=[
         {
             "content": "string",
@@ -221,7 +221,7 @@ from gradientai import GradientAI
 client = GradientAI()
 
 try:
-    client.agents.chat.completions.create(
+    client.chat.completions.create(
         messages=[
             {
                 "role": "user",
@@ -272,7 +272,7 @@ client = GradientAI(
 )
 
 # Or, configure per-request:
-client.with_options(max_retries=5).agents.chat.completions.create(
+client.with_options(max_retries=5).chat.completions.create(
     messages=[
         {
             "role": "user",
@@ -303,7 +303,7 @@ client = GradientAI(
 )
 
 # Override per-request:
-client.with_options(timeout=5.0).agents.chat.completions.create(
+client.with_options(timeout=5.0).chat.completions.create(
     messages=[
         {
             "role": "user",
@@ -352,7 +352,7 @@ The "raw" Response object can be accessed by prefixing `.with_raw_response.` to
 from gradientai import GradientAI
 
 client = GradientAI()
-response = client.agents.chat.completions.with_raw_response.create(
+response = client.chat.completions.with_raw_response.create(
     messages=[{
         "role": "user",
         "content": "What is the capital of France?",
@@ -361,7 +361,7 @@ response = client.agents.chat.completions.with_raw_response.create(
 )
 print(response.headers.get('X-My-Header'))
 
-completion = response.parse()  # get the object that `agents.chat.completions.create()` would have returned
+completion = response.parse()  # get the object that `chat.completions.create()` would have returned
 print(completion.choices)
 ```
 
@@ -376,7 +376,7 @@ The above interface eagerly reads the full response body when you make the reque
 To stream the response body, use `.with_streaming_response` instead, which requires a context manager and only reads the response body once you call `.read()`, `.text()`, `.json()`, `.iter_bytes()`, `.iter_text()`, `.iter_lines()` or `.parse()`. In the async client, these are async methods.
 
 ```python
-with client.agents.chat.completions.with_streaming_response.create(
+with client.chat.completions.with_streaming_response.create(
     messages=[
         {
             "role": "user",

api.md

@@ -70,7 +70,7 @@ from gradientai.types.agents.chat import CompletionCreateResponse
 
 Methods:
 
-- <code title="post /chat/completions">client.agents.chat.completions.<a href="./src/gradientai/resources/agents/chat/completions.py">create</a>(\*\*<a href="src/gradientai/types/agents/chat/completion_create_params.py">params</a>) -> <a href="./src/gradientai/types/agents/chat/completion_create_response.py">CompletionCreateResponse</a></code>
+- <code title="post /chat/completions?agent=true">client.agents.chat.completions.<a href="./src/gradientai/resources/agents/chat/completions.py">create</a>(\*\*<a href="src/gradientai/types/agents/chat/completion_create_params.py">params</a>) -> <a href="./src/gradientai/types/agents/chat/completion_create_response.py">CompletionCreateResponse</a></code>
 
 ## EvaluationMetrics
 

src/gradientai/resources/agents/chat/completions.py

@@ -414,9 +414,9 @@ def create(
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> CompletionCreateResponse | Stream[ChatCompletionChunk]:
         return self._post(
-            "/chat/completions"
+            "/chat/completions?agent=true"
             if self._client._base_url_overridden
-            else "https://inference.do-ai.run/v1/chat/completions",
+            else "https://inference.do-ai.run/v1/chat/completions?agent=true",
             body=maybe_transform(
                 {
                     "messages": messages,
@@ -838,9 +838,9 @@ async def create(
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> CompletionCreateResponse | AsyncStream[ChatCompletionChunk]:
         return await self._post(
-            "/chat/completions"
+            "/chat/completions?agent=true"
             if self._client._base_url_overridden
-            else "https://inference.do-ai.run/v1/chat/completions",
+            else "https://inference.do-ai.run/v1/chat/completions?agent=true",
             body=await async_maybe_transform(
                 {
                     "messages": messages,

tests/test_client.py

@@ -879,7 +879,7 @@ def test_retrying_timeout_errors_doesnt_leak(self, respx_mock: MockRouter, clien
         respx_mock.post("/chat/completions").mock(side_effect=httpx.TimeoutException("Test timeout error"))
 
         with pytest.raises(APITimeoutError):
-            client.agents.chat.completions.with_streaming_response.create(
+            client.chat.completions.with_streaming_response.create(
                 messages=[
                     {
                         "content": "string",
@@ -897,7 +897,7 @@ def test_retrying_status_errors_doesnt_leak(self, respx_mock: MockRouter, client
         respx_mock.post("/chat/completions").mock(return_value=httpx.Response(500))
 
         with pytest.raises(APIStatusError):
-            client.agents.chat.completions.with_streaming_response.create(
+            client.chat.completions.with_streaming_response.create(
                 messages=[
                     {
                         "content": "string",
@@ -934,7 +934,7 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
 
         respx_mock.post("/chat/completions").mock(side_effect=retry_handler)
 
-        response = client.agents.chat.completions.with_raw_response.create(
+        response = client.chat.completions.with_raw_response.create(
             messages=[
                 {
                     "content": "string",
@@ -966,7 +966,7 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
 
         respx_mock.post("/chat/completions").mock(side_effect=retry_handler)
 
-        response = client.agents.chat.completions.with_raw_response.create(
+        response = client.chat.completions.with_raw_response.create(
             messages=[
                 {
                     "content": "string",
@@ -998,7 +998,7 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
 
         respx_mock.post("/chat/completions").mock(side_effect=retry_handler)
 
-        response = client.agents.chat.completions.with_raw_response.create(
+        response = client.chat.completions.with_raw_response.create(
             messages=[
                 {
                     "content": "string",
@@ -1888,7 +1888,7 @@ async def test_retrying_timeout_errors_doesnt_leak(
         respx_mock.post("/chat/completions").mock(side_effect=httpx.TimeoutException("Test timeout error"))
 
         with pytest.raises(APITimeoutError):
-            await async_client.agents.chat.completions.with_streaming_response.create(
+            await async_client.chat.completions.with_streaming_response.create(
                 messages=[
                     {
                         "content": "string",
@@ -1908,7 +1908,7 @@ async def test_retrying_status_errors_doesnt_leak(
         respx_mock.post("/chat/completions").mock(return_value=httpx.Response(500))
 
         with pytest.raises(APIStatusError):
-            await async_client.agents.chat.completions.with_streaming_response.create(
+            await async_client.chat.completions.with_streaming_response.create(
                 messages=[
                     {
                         "content": "string",
@@ -1946,7 +1946,7 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
 
         respx_mock.post("/chat/completions").mock(side_effect=retry_handler)
 
-        response = await client.agents.chat.completions.with_raw_response.create(
+        response = await client.chat.completions.with_raw_response.create(
             messages=[
                 {
                     "content": "string",
@@ -1979,7 +1979,7 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
 
         respx_mock.post("/chat/completions").mock(side_effect=retry_handler)
 
-        response = await client.agents.chat.completions.with_raw_response.create(
+        response = await client.chat.completions.with_raw_response.create(
             messages=[
                 {
                     "content": "string",
@@ -2012,7 +2012,7 @@ def retry_handler(_request: httpx.Request) -> httpx.Response:
 
         respx_mock.post("/chat/completions").mock(side_effect=retry_handler)
 
-        response = await client.agents.chat.completions.with_raw_response.create(
+        response = await client.chat.completions.with_raw_response.create(
             messages=[
                 {
                     "content": "string",