diff --git a/libs/langchain/langchain/chat_models/konko.py b/libs/langchain/langchain/chat_models/konko.py index b7b9bc658..e27ee4205 100644 --- a/libs/langchain/langchain/chat_models/konko.py +++ b/libs/langchain/langchain/chat_models/konko.py @@ -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 diff --git a/libs/langchain/langchain/chat_models/litellm.py b/libs/langchain/langchain/chat_models/litellm.py index 9d263872f..f9ecf6707 100644 --- a/libs/langchain/langchain/chat_models/litellm.py +++ b/libs/langchain/langchain/chat_models/litellm.py @@ -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 diff --git a/libs/langchain/langchain/chat_models/openai.py b/libs/langchain/langchain/chat_models/openai.py index 5d944852d..47f29eaf2 100644 --- a/libs/langchain/langchain/chat_models/openai.py +++ b/libs/langchain/langchain/chat_models/openai.py @@ -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