community[patch]: fix ollama astream (#16070)

Update ollama.py
2025-09-11 16:01:33 +00:00 · 2024-01-18 01:42:41 +08:00
parent bc0cb1148a
commit d0e101e4e0
1 changed files with 1 additions and 1 deletions
--- a/libs/community/langchain_community/llms/ollama.py
+++ b/libs/community/langchain_community/llms/ollama.py
@@ -468,7 +468,7 @@ class Ollama(BaseLLM, _OllamaCommon):
        run_manager: Optional[AsyncCallbackManagerForLLMRun] = None,
        **kwargs: Any,
    ) -> AsyncIterator[GenerationChunk]:
-        async for stream_resp in self._acreate_stream(prompt, stop, **kwargs):
+        async for stream_resp in self._acreate_generate_stream(prompt, stop, **kwargs):
            if stream_resp:
                chunk = _stream_response_to_generation_chunk(stream_resp)
                yield chunk