mirror of
https://github.com/hwchase17/langchain.git
synced 2025-07-11 15:35:09 +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)
|
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
|
||||||
|
Loading…
Reference in New Issue
Block a user