github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-24 15:43:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Chat model stream readability nit (#10469)

Browse Source
This commit is contained in:
Bagatur 2023-09-11 18:05:24 -07:00 committed by GitHub
parent d45b042d3e
commit 6598178343
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
libs/langchain/langchain/chat_models/konko.py
View File

@ -222,7 +222,8 @@ class ChatKonko(ChatOpenAI):
stream: Optional[bool] = None, stream: Optional[bool] = None,
**kwargs: Any, **kwargs: Any,
) -> ChatResult: ) -> 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 generation: Optional[ChatGenerationChunk] = None
for chunk in self._stream( for chunk in self._stream(
messages=messages, stop=stop, run_manager=run_manager, **kwargs messages=messages, stop=stop, run_manager=run_manager, **kwargs

6
libs/langchain/langchain/chat_models/litellm.py
View File

@ -318,7 +318,8 @@ class ChatLiteLLM(BaseChatModel):
stream: Optional[bool] = None, stream: Optional[bool] = None,
**kwargs: Any, **kwargs: Any,
) -> ChatResult: ) -> 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 generation: Optional[ChatGenerationChunk] = None
for chunk in self._stream( for chunk in self._stream(
messages=messages, stop=stop, run_manager=run_manager, **kwargs messages=messages, stop=stop, run_manager=run_manager, **kwargs
@ -418,7 +419,8 @@ class ChatLiteLLM(BaseChatModel):
stream: Optional[bool] = None, stream: Optional[bool] = None,
**kwargs: Any, **kwargs: Any,
) -> ChatResult: ) -> 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 generation: Optional[ChatGenerationChunk] = None
async for chunk in self._astream( async for chunk in self._astream(
messages=messages, stop=stop, run_manager=run_manager, **kwargs messages=messages, stop=stop, run_manager=run_manager, **kwargs

6
libs/langchain/langchain/chat_models/openai.py
View File

@ -328,7 +328,8 @@ class ChatOpenAI(BaseChatModel):
stream: Optional[bool] = None, stream: Optional[bool] = None,
**kwargs: Any, **kwargs: Any,
) -> ChatResult: ) -> 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 generation: Optional[ChatGenerationChunk] = None
for chunk in self._stream( for chunk in self._stream(
messages=messages, stop=stop, run_manager=run_manager, **kwargs messages=messages, stop=stop, run_manager=run_manager, **kwargs
@ -408,7 +409,8 @@ class ChatOpenAI(BaseChatModel):
stream: Optional[bool] = None, stream: Optional[bool] = None,
**kwargs: Any, **kwargs: Any,
) -> ChatResult: ) -> 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 generation: Optional[ChatGenerationChunk] = None
async for chunk in self._astream( async for chunk in self._astream(
messages=messages, stop=stop, run_manager=run_manager, **kwargs messages=messages, stop=stop, run_manager=run_manager, **kwargs