github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-28 17:38:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: call correct stream method in ollama (#15104)

Browse Source 
Co-authored-by: David Kristek <david@David--MacBook-Pro.local>
This commit is contained in:
David Křístek 2024-01-01 23:03:53 +01:00 committed by GitHub
parent c2c2f252c2
commit a010f29013
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
libs/community/langchain_community/llms/ollama.py
View File

@ -440,7 +440,7 @@ class Ollama(BaseLLM, _OllamaCommon):
run_manager: Optional[CallbackManagerForLLMRun] = None,
**kwargs: Any,
) -> Iterator[GenerationChunk]:
for stream_resp in self._create_stream(prompt, stop, **kwargs):
for stream_resp in self._create_generate_stream(prompt, stop, **kwargs):
if stream_resp:
chunk = _stream_response_to_generation_chunk(stream_resp)
yield chunk