ensure compatibility with huggingface_hub==0.26.0 (#8464)

deepset-ai · Oct 18, 2024 · dfd339c · dfd339c
1 parent 939698b
commit dfd339c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/releasenotes/notes/compatibility-hh-hub-0.26.0-a5b61f7bf99a97b7.yaml b/releasenotes/notes/compatibility-hh-hub-0.26.0-a5b61f7bf99a97b7.yaml
@@ -0,0 +1,4 @@
+---
+fixes:
+  - |
+    Adjusted a test on `HuggingFaceAPIGenerator` to ensure compatibility with the `huggingface_hub==0.26.0`.
diff --git a/test/components/generators/test_hugging_face_api.py b/test/components/generators/test_hugging_face_api.py
@@ -250,7 +250,9 @@ def mock_iter(self):
                 index=1,
                 generated_text=None,
                 token=TextGenerationOutputToken(id=1, text="Ok bye", logprob=0.0, special=False),
-                details=TextGenerationStreamOutputStreamDetails(finish_reason="length", generated_tokens=5, seed=None),
+                details=TextGenerationStreamOutputStreamDetails(
+                    finish_reason="length", generated_tokens=5, seed=None, input_length=10
+                ),
             )
 
         mock_response = Mock(**{"__iter__": mock_iter})