feat(api): manual updates

add streaming to chat completions

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: e1b3d85ba9ae21d729a914c789422ba7
+config_hash: 0bc3af28d4abd9be8bcc81f615bc832d

api.md

@@ -65,7 +65,7 @@ Methods:
 Types:
 
 ```python
-from gradientai.types.agents.chat import ChatCompletionChunk, CompletionCreateResponse
+from gradientai.types.agents.chat import AgentChatCompletionChunk, CompletionCreateResponse
 ```
 
 Methods:
@@ -396,7 +396,7 @@ Methods:
 Types:
 
 ```python
-from gradientai.types.chat import CompletionCreateResponse
+from gradientai.types.chat import ChatCompletionChunk, CompletionCreateResponse
 ```
 
 Methods:

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

@@ -20,8 +20,8 @@
 from ...._streaming import Stream, AsyncStream
 from ...._base_client import make_request_options
 from ....types.agents.chat import completion_create_params
-from ....types.agents.chat.chat_completion_chunk import ChatCompletionChunk
 from ....types.agents.chat.completion_create_response import CompletionCreateResponse
+from ....types.agents.chat.agent_chat_completion_chunk import AgentChatCompletionChunk
 
 __all__ = ["CompletionsResource", "AsyncCompletionsResource"]
 
@@ -186,7 +186,7 @@ def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> Stream[ChatCompletionChunk]:
+    ) -> Stream[AgentChatCompletionChunk]:
         """
         Creates a model response for the given chat conversation.
 
@@ -299,7 +299,7 @@ def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> CompletionCreateResponse | Stream[ChatCompletionChunk]:
+    ) -> CompletionCreateResponse | Stream[AgentChatCompletionChunk]:
         """
         Creates a model response for the given chat conversation.
 
@@ -412,7 +412,7 @@ def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> CompletionCreateResponse | Stream[ChatCompletionChunk]:
+    ) -> CompletionCreateResponse | Stream[AgentChatCompletionChunk]:
         return self._post(
             "/chat/completions"
             if self._client._base_url_overridden
@@ -446,7 +446,7 @@ def create(
             ),
             cast_to=CompletionCreateResponse,
             stream=stream or False,
-            stream_cls=Stream[ChatCompletionChunk],
+            stream_cls=Stream[AgentChatCompletionChunk],
         )
 
 
@@ -610,7 +610,7 @@ async def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> AsyncStream[ChatCompletionChunk]:
+    ) -> AsyncStream[AgentChatCompletionChunk]:
         """
         Creates a model response for the given chat conversation.
 
@@ -723,7 +723,7 @@ async def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> CompletionCreateResponse | AsyncStream[ChatCompletionChunk]:
+    ) -> CompletionCreateResponse | AsyncStream[AgentChatCompletionChunk]:
         """
         Creates a model response for the given chat conversation.
 
@@ -836,7 +836,7 @@ async def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> CompletionCreateResponse | AsyncStream[ChatCompletionChunk]:
+    ) -> CompletionCreateResponse | AsyncStream[AgentChatCompletionChunk]:
         return await self._post(
             "/chat/completions"
             if self._client._base_url_overridden
@@ -870,7 +870,7 @@ async def create(
             ),
             cast_to=CompletionCreateResponse,
             stream=stream or False,
-            stream_cls=AsyncStream[ChatCompletionChunk],
+            stream_cls=AsyncStream[AgentChatCompletionChunk],
         )