github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-11 15:35:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
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
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libs/partners/huggingface/langchain_huggingface/llms/huggingface_endpoint.py
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