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) invocation_params = self._invocation_params(stop, **kwargs)
if self.streaming: if self.streaming:
completion = "" 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 completion += chunk.text
return completion return completion
else: else:
@ -333,7 +335,7 @@ class HuggingFaceEndpoint(LLM):
if self.streaming: if self.streaming:
completion = "" completion = ""
async for chunk in self._astream( async for chunk in self._astream(
prompt, stop, run_manager, **invocation_params prompt, run_manager=run_manager, **invocation_params
): ):
completion += chunk.text completion += chunk.text
return completion return completion