Skip to content

Commit 8920408

Browse files
committed
chore(client): update stop params in stream/parse to match chat
1 parent daf31d5 commit 8920408

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

src/writerai/resources/chat.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -440,7 +440,7 @@ def parse(
440440
logprobs: bool | NotGiven = NOT_GIVEN,
441441
max_tokens: int | NotGiven = NOT_GIVEN,
442442
n: int | NotGiven = NOT_GIVEN,
443-
stop: Union[List[str], str] | NotGiven = NOT_GIVEN,
443+
stop: Union[SequenceNotStr[str], str] | NotGiven = NOT_GIVEN,
444444
temperature: float | NotGiven = NOT_GIVEN,
445445
tool_choice: chat_chat_params.ToolChoice | NotGiven = NOT_GIVEN,
446446
tools: Iterable[ToolParam] | NotGiven = NOT_GIVEN,
@@ -549,7 +549,7 @@ def stream(
549549
logprobs: bool | NotGiven = NOT_GIVEN,
550550
max_tokens: int | NotGiven = NOT_GIVEN,
551551
n: int | NotGiven = NOT_GIVEN,
552-
stop: Union[List[str], str] | NotGiven = NOT_GIVEN,
552+
stop: Union[SequenceNotStr[str], str] | NotGiven = NOT_GIVEN,
553553
stream_options: chat_chat_params.StreamOptions | NotGiven = NOT_GIVEN,
554554
temperature: float | NotGiven = NOT_GIVEN,
555555
tool_choice: chat_chat_params.ToolChoice | NotGiven = NOT_GIVEN,
@@ -1014,7 +1014,7 @@ async def parse(
10141014
logprobs: bool | NotGiven = NOT_GIVEN,
10151015
max_tokens: int | NotGiven = NOT_GIVEN,
10161016
n: int | NotGiven = NOT_GIVEN,
1017-
stop: Union[List[str], str] | NotGiven = NOT_GIVEN,
1017+
stop: Union[SequenceNotStr[str], str] | NotGiven = NOT_GIVEN,
10181018
temperature: float | NotGiven = NOT_GIVEN,
10191019
tool_choice: chat_chat_params.ToolChoice | NotGiven = NOT_GIVEN,
10201020
tools: Iterable[ToolParam] | NotGiven = NOT_GIVEN,
@@ -1123,7 +1123,7 @@ def stream(
11231123
logprobs: bool | NotGiven = NOT_GIVEN,
11241124
max_tokens: int | NotGiven = NOT_GIVEN,
11251125
n: int | NotGiven = NOT_GIVEN,
1126-
stop: Union[List[str], str] | NotGiven = NOT_GIVEN,
1126+
stop: Union[SequenceNotStr[str], str] | NotGiven = NOT_GIVEN,
11271127
stream_options: chat_chat_params.StreamOptions | NotGiven = NOT_GIVEN,
11281128
temperature: float | NotGiven = NOT_GIVEN,
11291129
tool_choice: chat_chat_params.ToolChoice | NotGiven = NOT_GIVEN,

0 commit comments

Comments
 (0)