github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-04 20:46:45 +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
1 changed files with 1 additions and 1 deletions
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