Update haystack/components/generators/chat/hugging_face_api.py

Co-authored-by: Stefano Fiorucci <[email protected]>
deepset-ai · Jan 17, 2025 · 833e019 · 833e019
1 parent 8e02206
commit 833e019
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/haystack/components/generators/chat/hugging_face_api.py b/haystack/components/generators/chat/hugging_face_api.py
@@ -275,14 +275,12 @@ def _run_streaming(self, messages: List[Dict[str, str]], generation_kwargs: Dict
 
             finish_reason = choice.finish_reason
 
-            meta = {
-                "finish_reason": finish_reason if finish_reason else None,
-                "received_at": datetime.now().isoformat(),
-            }
-            chunks_meta.append(meta)
+            meta = {}
+            if finish_reason:
+                meta["finish_reason"] = finish_reason
 
             if first_chunk_time is None:
-                first_chunk_time = meta["received_at"]
+                first_chunk_time = datetime.now().isoformat()
 
             stream_chunk = StreamingChunk(text, meta)
             self.streaming_callback(stream_chunk)  # type: ignore # streaming_callback is not None (verified in the run method)