Chat model stream readability nit (#10469)

pull/10478/head
Bagatur 1 year ago committed by GitHub
parent d45b042d3e
commit 6598178343
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -222,7 +222,8 @@ class ChatKonko(ChatOpenAI):
stream: Optional[bool] = None,
**kwargs: Any,
) -> ChatResult:
if stream if stream is not None else self.streaming:
should_stream = stream if stream is not None else self.streaming
if should_stream:
generation: Optional[ChatGenerationChunk] = None
for chunk in self._stream(
messages=messages, stop=stop, run_manager=run_manager, **kwargs

@ -318,7 +318,8 @@ class ChatLiteLLM(BaseChatModel):
stream: Optional[bool] = None,
**kwargs: Any,
) -> ChatResult:
if stream if stream is not None else self.streaming:
should_stream = stream if stream is not None else self.streaming
if should_stream:
generation: Optional[ChatGenerationChunk] = None
for chunk in self._stream(
messages=messages, stop=stop, run_manager=run_manager, **kwargs
@ -418,7 +419,8 @@ class ChatLiteLLM(BaseChatModel):
stream: Optional[bool] = None,
**kwargs: Any,
) -> ChatResult:
if stream if stream is not None else self.streaming:
should_stream = stream if stream is not None else self.streaming
if should_stream:
generation: Optional[ChatGenerationChunk] = None
async for chunk in self._astream(
messages=messages, stop=stop, run_manager=run_manager, **kwargs

@ -328,7 +328,8 @@ class ChatOpenAI(BaseChatModel):
stream: Optional[bool] = None,
**kwargs: Any,
) -> ChatResult:
if stream if stream is not None else self.streaming:
should_stream = stream if stream is not None else self.streaming
if should_stream:
generation: Optional[ChatGenerationChunk] = None
for chunk in self._stream(
messages=messages, stop=stop, run_manager=run_manager, **kwargs
@ -408,7 +409,8 @@ class ChatOpenAI(BaseChatModel):
stream: Optional[bool] = None,
**kwargs: Any,
) -> ChatResult:
if stream if stream is not None else self.streaming:
should_stream = stream if stream is not None else self.streaming
if should_stream:
generation: Optional[ChatGenerationChunk] = None
async for chunk in self._astream(
messages=messages, stop=stop, run_manager=run_manager, **kwargs

Loading…
Cancel
Save