feat(api): share chat completion chunk model between chat and agent.chat

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: 9c2e548d86a376bc5f6c458de6944504
+config_hash: 6ed9ee8d3f0d6392816bfaf9dc4894a6

api.md

@@ -1,7 +1,7 @@
 # Shared Types
 
 ```python
-from gradientai.types import APILinks, APIMeta, ChatCompletionTokenLogprob
+from gradientai.types import APILinks, APIMeta, ChatCompletionChunk, ChatCompletionTokenLogprob
 ```
 
 # Agents
@@ -65,7 +65,7 @@ Methods:
 Types:
 
 ```python
-from gradientai.types.agents.chat import AgentChatCompletionChunk, CompletionCreateResponse
+from gradientai.types.agents.chat import CompletionCreateResponse
 ```
 
 Methods:
@@ -260,7 +260,7 @@ Methods:
 Types:
 
 ```python
-from gradientai.types.chat import ChatCompletionChunk, CompletionCreateResponse
+from gradientai.types.chat import 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.shared.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[AgentChatCompletionChunk]:
+    ) -> Stream[ChatCompletionChunk]:
         """
         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[AgentChatCompletionChunk]:
+    ) -> CompletionCreateResponse | Stream[ChatCompletionChunk]:
         """
         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[AgentChatCompletionChunk]:
+    ) -> CompletionCreateResponse | Stream[ChatCompletionChunk]:
         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[AgentChatCompletionChunk],
+            stream_cls=Stream[ChatCompletionChunk],
         )
 
 
@@ -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[AgentChatCompletionChunk]:
+    ) -> AsyncStream[ChatCompletionChunk]:
         """
         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[AgentChatCompletionChunk]:
+    ) -> CompletionCreateResponse | AsyncStream[ChatCompletionChunk]:
         """
         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[AgentChatCompletionChunk]:
+    ) -> CompletionCreateResponse | AsyncStream[ChatCompletionChunk]:
         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[AgentChatCompletionChunk],
+            stream_cls=AsyncStream[ChatCompletionChunk],
         )
 
 

src/gradientai/resources/chat/completions.py

@@ -20,7 +20,7 @@
 from ..._streaming import Stream, AsyncStream
 from ...types.chat import completion_create_params
 from ..._base_client import make_request_options
-from ...types.chat.chat_completion_chunk import ChatCompletionChunk
+from ...types.shared.chat_completion_chunk import ChatCompletionChunk
 from ...types.chat.completion_create_response import CompletionCreateResponse
 
 __all__ = ["CompletionsResource", "AsyncCompletionsResource"]

src/gradientai/types/__init__.py

@@ -2,7 +2,12 @@
 
 from __future__ import annotations
 
-from .shared import APIMeta as APIMeta, APILinks as APILinks, ChatCompletionTokenLogprob as ChatCompletionTokenLogprob
+from .shared import (
+    APIMeta as APIMeta,
+    APILinks as APILinks,
+    ChatCompletionChunk as ChatCompletionChunk,
+    ChatCompletionTokenLogprob as ChatCompletionTokenLogprob,
+)
 from .api_agent import APIAgent as APIAgent
 from .api_model import APIModel as APIModel
 from .api_agreement import APIAgreement as APIAgreement

src/gradientai/types/agents/chat/__init__.py

@@ -4,4 +4,3 @@
 
 from .completion_create_params import CompletionCreateParams as CompletionCreateParams
 from .completion_create_response import CompletionCreateResponse as CompletionCreateResponse
-from .agent_chat_completion_chunk import AgentChatCompletionChunk as AgentChatCompletionChunk

src/gradientai/types/agents/chat/agent_chat_completion_chunk.py

@@ -2,6 +2,5 @@
 
 from __future__ import annotations
 
-from .chat_completion_chunk import ChatCompletionChunk as ChatCompletionChunk
 from .completion_create_params import CompletionCreateParams as CompletionCreateParams
 from .completion_create_response import CompletionCreateResponse as CompletionCreateResponse

src/gradientai/types/chat/__init__.py

@@ -2,4 +2,5 @@
 
 from .api_meta import APIMeta as APIMeta
 from .api_links import APILinks as APILinks
+from .chat_completion_chunk import ChatCompletionChunk as ChatCompletionChunk
 from .chat_completion_token_logprob import ChatCompletionTokenLogprob as ChatCompletionTokenLogprob

src/gradientai/types/shared/__init__.py

@@ -4,7 +4,7 @@
 from typing_extensions import Literal
 
 from ..._models import BaseModel
-from ..shared.chat_completion_token_logprob import ChatCompletionTokenLogprob
+from .chat_completion_token_logprob import ChatCompletionTokenLogprob
 
 __all__ = ["ChatCompletionChunk", "Choice", "ChoiceDelta", "ChoiceLogprobs", "Usage"]