feat(api): manual updates (#127)

Author: stainless-app[bot]

.stats.yml

@@ -1,2 +1,2 @@
 configured_endpoints: 21
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/writerai%2Fwriter-6b363dd34169cab18f5ec3bcf6586aecd4799f79a80c90bf54e5a12f91d9e7c2.yml
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/writerai%2Fwriter-82683f2fd5f8778a27960ebabda40d6dc4640bdfb77ac4ec7f173b8bf8076d3c.yml

README.md

@@ -32,11 +32,11 @@ client = Writer(
     api_key=os.environ.get("WRITER_API_KEY"),  # This is the default and can be omitted
 )
 
-chat_completion = client.chat.chat(
+chat = client.chat.chat(
     messages=[{"role": "user"}],
     model="palmyra-x-004",
 )
-print(chat_completion.id)
+print(chat.id)
 ```
 
 While you can provide an `api_key` keyword argument,
@@ -59,11 +59,11 @@ client = AsyncWriter(
 
 
 async def main() -> None:
-    chat_completion = await client.chat.chat(
+    chat = await client.chat.chat(
         messages=[{"role": "user"}],
         model="palmyra-x-004",
     )
-    print(chat_completion.id)
+    print(chat.id)
 
 
 asyncio.run(main())

api.md

@@ -1,23 +1,3 @@
-# Shared Types
-
-```python
-from writerai.types import (
-    ErrorMessage,
-    ErrorObject,
-    FunctionDefinition,
-    FunctionParams,
-    GraphData,
-    Logprobs,
-    LogprobsToken,
-    Source,
-    ToolCall,
-    ToolCallStreaming,
-    ToolChoiceJsonObject,
-    ToolChoiceString,
-    ToolParam,
-)
-```
-
 # Applications
 
 Types:
@@ -35,26 +15,19 @@ Methods:
 Types:
 
 ```python
-from writerai.types import (
-    ChatCompletion,
-    ChatCompletionChoice,
-    ChatCompletionChunk,
-    ChatCompletionMessage,
-    ChatCompletionParams,
-    ChatCompletionUsage,
-)
+from writerai.types import Chat, ChatCompletionChunk
 ```
 
 Methods:
 
-- <code title="post /v1/chat">client.chat.<a href="./src/writerai/resources/chat.py">chat</a>(\*\*<a href="src/writerai/types/chat_chat_params.py">params</a>) -> <a href="./src/writerai/types/chat_completion.py">ChatCompletion</a></code>
+- <code title="post /v1/chat">client.chat.<a href="./src/writerai/resources/chat.py">chat</a>(\*\*<a href="src/writerai/types/chat_chat_params.py">params</a>) -> <a href="./src/writerai/types/chat.py">Chat</a></code>
 
 # Completions
 
 Types:
 
 ```python
-from writerai.types import Completion, CompletionChunk, CompletionParams
+from writerai.types import Completion, StreamingData
 ```
 
 Methods:

src/writerai/_streaming.py

@@ -55,9 +55,6 @@ def __stream__(self) -> Iterator[_T]:
         iterator = self._iter_events()
 
         for sse in iterator:
-            if sse.data.startswith("[DONE]"):
-                break
-
             if sse.event is None:
                 yield process_data(data=sse.json(), cast_to=cast_to, response=response)
 
@@ -138,9 +135,6 @@ async def __stream__(self) -> AsyncIterator[_T]:
         iterator = self._iter_events()
 
         async for sse in iterator:
-            if sse.data.startswith("[DONE]"):
-                break
-
             if sse.event is None:
                 yield process_data(data=sse.json(), cast_to=cast_to, response=response)
 

src/writerai/resources/chat.py

@@ -23,10 +23,9 @@
     async_to_streamed_response_wrapper,
 )
 from .._streaming import Stream, AsyncStream
+from ..types.chat import Chat
 from .._base_client import make_request_options
-from ..types.chat_completion import ChatCompletion
 from ..types.chat_completion_chunk import ChatCompletionChunk
-from ..types.shared_params.tool_param import ToolParam
 
 __all__ = ["ChatResource", "AsyncChatResource"]
 
@@ -65,15 +64,15 @@ def chat(
         stream_options: chat_chat_params.StreamOptions | NotGiven = NOT_GIVEN,
         temperature: float | NotGiven = NOT_GIVEN,
         tool_choice: chat_chat_params.ToolChoice | NotGiven = NOT_GIVEN,
-        tools: Iterable[ToolParam] | NotGiven = NOT_GIVEN,
+        tools: Iterable[chat_chat_params.Tool] | NotGiven = NOT_GIVEN,
         top_p: float | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> ChatCompletion:
+    ) -> Chat:
         """Generate a chat completion based on the provided messages.
 
         The response shown
@@ -148,7 +147,7 @@ def chat(
         stream_options: chat_chat_params.StreamOptions | NotGiven = NOT_GIVEN,
         temperature: float | NotGiven = NOT_GIVEN,
         tool_choice: chat_chat_params.ToolChoice | NotGiven = NOT_GIVEN,
-        tools: Iterable[ToolParam] | NotGiven = NOT_GIVEN,
+        tools: Iterable[chat_chat_params.Tool] | NotGiven = NOT_GIVEN,
         top_p: float | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -231,15 +230,15 @@ def chat(
         stream_options: chat_chat_params.StreamOptions | NotGiven = NOT_GIVEN,
         temperature: float | NotGiven = NOT_GIVEN,
         tool_choice: chat_chat_params.ToolChoice | NotGiven = NOT_GIVEN,
-        tools: Iterable[ToolParam] | NotGiven = NOT_GIVEN,
+        tools: Iterable[chat_chat_params.Tool] | NotGiven = NOT_GIVEN,
         top_p: float | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> ChatCompletion | Stream[ChatCompletionChunk]:
+    ) -> Chat | Stream[ChatCompletionChunk]:
         """Generate a chat completion based on the provided messages.
 
         The response shown
@@ -314,15 +313,15 @@ def chat(
         stream_options: chat_chat_params.StreamOptions | NotGiven = NOT_GIVEN,
         temperature: float | NotGiven = NOT_GIVEN,
         tool_choice: chat_chat_params.ToolChoice | NotGiven = NOT_GIVEN,
-        tools: Iterable[ToolParam] | NotGiven = NOT_GIVEN,
+        tools: Iterable[chat_chat_params.Tool] | NotGiven = NOT_GIVEN,
         top_p: float | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> ChatCompletion | Stream[ChatCompletionChunk]:
+    ) -> Chat | Stream[ChatCompletionChunk]:
         return self._post(
             "/v1/chat",
             body=maybe_transform(
@@ -345,7 +344,7 @@ def chat(
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=ChatCompletion,
+            cast_to=Chat,
             stream=stream or False,
             stream_cls=Stream[ChatCompletionChunk],
         )
@@ -385,15 +384,15 @@ async def chat(
         stream_options: chat_chat_params.StreamOptions | NotGiven = NOT_GIVEN,
         temperature: float | NotGiven = NOT_GIVEN,
         tool_choice: chat_chat_params.ToolChoice | NotGiven = NOT_GIVEN,
-        tools: Iterable[ToolParam] | NotGiven = NOT_GIVEN,
+        tools: Iterable[chat_chat_params.Tool] | NotGiven = NOT_GIVEN,
         top_p: float | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> ChatCompletion:
+    ) -> Chat:
         """Generate a chat completion based on the provided messages.
 
         The response shown
@@ -468,7 +467,7 @@ async def chat(
         stream_options: chat_chat_params.StreamOptions | NotGiven = NOT_GIVEN,
         temperature: float | NotGiven = NOT_GIVEN,
         tool_choice: chat_chat_params.ToolChoice | NotGiven = NOT_GIVEN,
-        tools: Iterable[ToolParam] | NotGiven = NOT_GIVEN,
+        tools: Iterable[chat_chat_params.Tool] | NotGiven = NOT_GIVEN,
         top_p: float | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -551,15 +550,15 @@ async def chat(
         stream_options: chat_chat_params.StreamOptions | NotGiven = NOT_GIVEN,
         temperature: float | NotGiven = NOT_GIVEN,
         tool_choice: chat_chat_params.ToolChoice | NotGiven = NOT_GIVEN,
-        tools: Iterable[ToolParam] | NotGiven = NOT_GIVEN,
+        tools: Iterable[chat_chat_params.Tool] | NotGiven = NOT_GIVEN,
         top_p: float | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> ChatCompletion | AsyncStream[ChatCompletionChunk]:
+    ) -> Chat | AsyncStream[ChatCompletionChunk]:
         """Generate a chat completion based on the provided messages.
 
         The response shown
@@ -634,15 +633,15 @@ async def chat(
         stream_options: chat_chat_params.StreamOptions | NotGiven = NOT_GIVEN,
         temperature: float | NotGiven = NOT_GIVEN,
         tool_choice: chat_chat_params.ToolChoice | NotGiven = NOT_GIVEN,
-        tools: Iterable[ToolParam] | NotGiven = NOT_GIVEN,
+        tools: Iterable[chat_chat_params.Tool] | NotGiven = NOT_GIVEN,
         top_p: float | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> ChatCompletion | AsyncStream[ChatCompletionChunk]:
+    ) -> Chat | AsyncStream[ChatCompletionChunk]:
         return await self._post(
             "/v1/chat",
             body=await async_maybe_transform(
@@ -665,7 +664,7 @@ async def chat(
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=ChatCompletion,
+            cast_to=Chat,
             stream=stream or False,
             stream_cls=AsyncStream[ChatCompletionChunk],
         )

src/writerai/resources/completions.py

@@ -25,7 +25,7 @@
 from .._streaming import Stream, AsyncStream
 from .._base_client import make_request_options
 from ..types.completion import Completion
-from ..types.completion_chunk import CompletionChunk
+from ..types.streaming_data import StreamingData
 
 __all__ = ["CompletionsResource", "AsyncCompletionsResource"]
 
@@ -130,7 +130,7 @@ def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> Stream[CompletionChunk]:
+    ) -> Stream[StreamingData]:
         """
         Text generation
 
@@ -191,7 +191,7 @@ def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> Completion | Stream[CompletionChunk]:
+    ) -> Completion | Stream[StreamingData]:
         """
         Text generation
 
@@ -252,7 +252,7 @@ def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> Completion | Stream[CompletionChunk]:
+    ) -> Completion | Stream[StreamingData]:
         return self._post(
             "/v1/completions",
             body=maybe_transform(
@@ -274,7 +274,7 @@ def create(
             ),
             cast_to=Completion,
             stream=stream or False,
-            stream_cls=Stream[CompletionChunk],
+            stream_cls=Stream[StreamingData],
         )
 
 
@@ -378,7 +378,7 @@ async def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> AsyncStream[CompletionChunk]:
+    ) -> AsyncStream[StreamingData]:
         """
         Text generation
 
@@ -439,7 +439,7 @@ async def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> Completion | AsyncStream[CompletionChunk]:
+    ) -> Completion | AsyncStream[StreamingData]:
         """
         Text generation
 
@@ -500,7 +500,7 @@ async def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> Completion | AsyncStream[CompletionChunk]:
+    ) -> Completion | AsyncStream[StreamingData]:
         return await self._post(
             "/v1/completions",
             body=await async_maybe_transform(
@@ -522,7 +522,7 @@ async def create(
             ),
             cast_to=Completion,
             stream=stream or False,
-            stream_cls=AsyncStream[CompletionChunk],
+            stream_cls=AsyncStream[StreamingData],
         )
 
 

src/writerai/types/__init__.py

@@ -2,28 +2,13 @@
 
 from __future__ import annotations
 
+from .chat import Chat as Chat
 from .file import File as File
 from .graph import Graph as Graph
-from .shared import (
-    Source as Source,
-    Logprobs as Logprobs,
-    ToolCall as ToolCall,
-    GraphData as GraphData,
-    ToolParam as ToolParam,
-    ErrorObject as ErrorObject,
-    ErrorMessage as ErrorMessage,
-    LogprobsToken as LogprobsToken,
-    FunctionParams as FunctionParams,
-    ToolChoiceString as ToolChoiceString,
-    ToolCallStreaming as ToolCallStreaming,
-    FunctionDefinition as FunctionDefinition,
-    ToolChoiceJsonObject as ToolChoiceJsonObject,
-)
 from .question import Question as Question
 from .completion import Completion as Completion
-from .chat_completion import ChatCompletion as ChatCompletion
+from .streaming_data import StreamingData as StreamingData
 from .chat_chat_params import ChatChatParams as ChatChatParams
-from .completion_chunk import CompletionChunk as CompletionChunk
 from .file_list_params import FileListParams as FileListParams
 from .file_retry_params import FileRetryParams as FileRetryParams
 from .graph_list_params import GraphListParams as GraphListParams
@@ -34,14 +19,11 @@
 from .model_list_response import ModelListResponse as ModelListResponse
 from .file_delete_response import FileDeleteResponse as FileDeleteResponse
 from .chat_completion_chunk import ChatCompletionChunk as ChatCompletionChunk
-from .chat_completion_usage import ChatCompletionUsage as ChatCompletionUsage
 from .graph_create_response import GraphCreateResponse as GraphCreateResponse
 from .graph_delete_response import GraphDeleteResponse as GraphDeleteResponse
 from .graph_question_params import GraphQuestionParams as GraphQuestionParams
 from .graph_update_response import GraphUpdateResponse as GraphUpdateResponse
 from .tool_parse_pdf_params import ToolParsePdfParams as ToolParsePdfParams
-from .chat_completion_choice import ChatCompletionChoice as ChatCompletionChoice
-from .chat_completion_message import ChatCompletionMessage as ChatCompletionMessage
 from .question_response_chunk import QuestionResponseChunk as QuestionResponseChunk
 from .tool_parse_pdf_response import ToolParsePdfResponse as ToolParsePdfResponse
 from .completion_create_params import CompletionCreateParams as CompletionCreateParams