huggingface: fix HuggingFaceEndpoint._astream() got multiple values for argument 'stop' (#31385)

This commit is contained in:
m27315 2025-07-06 10:18:53 -05:00 committed by GitHub
parent e934788ca2
commit 013ce2c47f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -305,7 +305,9 @@ class HuggingFaceEndpoint(LLM):
invocation_params = self._invocation_params(stop, **kwargs)
if self.streaming:
completion = ""
for chunk in self._stream(prompt, stop, run_manager, **invocation_params):
for chunk in self._stream(
prompt, run_manager=run_manager, **invocation_params
):
completion += chunk.text
return completion
else:
@ -333,7 +335,7 @@ class HuggingFaceEndpoint(LLM):
if self.streaming:
completion = ""
async for chunk in self._astream(
prompt, stop, run_manager, **invocation_params
prompt, run_manager=run_manager, **invocation_params
):
completion += chunk.text
return completion