mirror of
https://github.com/hwchase17/langchain.git
synced 2025-07-11 07:25:29 +00:00
huggingface: fix HuggingFaceEndpoint._astream() got multiple values for argument 'stop' (#31385)
This commit is contained in:
parent
e934788ca2
commit
013ce2c47f
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user