test: update Bedrock tests with ComponentInfo (#343)

* test: update Bedrock tests with ComponentInfo

* update transformers

Author: anakin87

pyproject.toml

@@ -52,7 +52,7 @@ fmt-check = "ruff check {args} && ruff format --check {args}"
 [tool.hatch.envs.test]
 extra-dependencies = [
   "colorama",                                         # Pipeline checkpoints experiment
-  "transformers[torch,sentencepiece]>=4.51.1,<4.52",  # Pipeline checkpoints experiment
+  "transformers[torch,sentencepiece]>=4.52.4,<4.53",  # Pipeline checkpoints experiment
   "arrow>=1.3.0",                                     # Multimodal experiment - ChatPromptBuilder
   "pypdfium2",                                        # Multimodal experiment - PDFToImageContent
   "pillow",                                           # Multimodal experiment - ImageFileToImageContent, PDFToImageContent

test/components/generators/chat/amazon_bedrock/test_chat_generator_utils.py

@@ -1,7 +1,7 @@
 import pytest
 import base64
 
-from haystack.dataclasses import StreamingChunk
+from haystack.dataclasses import ComponentInfo, StreamingChunk
 from haystack.tools import Tool
 from haystack_integrations.components.generators.amazon_bedrock.chat.utils import (
     _format_tools,
@@ -15,7 +15,6 @@
 )
 from haystack_experimental.dataclasses.chat_message import ChatMessage, ChatRole, ToolCall, ImageContent
 
-
 # NOTE: original module and tests
 def weather(city: str):
     """Get weather for a given city."""
@@ -346,6 +345,9 @@ def test_process_streaming_response_one_tool_call(self, mock_boto3_session):
         Test that process_streaming_response correctly handles streaming events and accumulates responses
         """
         model = "anthropic.claude-3-5-sonnet-20240620-v1:0"
+        type_ = (
+            "haystack_integrations.components.generators.amazon_bedrock.chat.chat_generator.AmazonBedrockChatGenerator"
+        )
         streaming_chunks = []
 
         def test_callback(chunk: StreamingChunk):
@@ -386,7 +388,11 @@ def test_callback(chunk: StreamingChunk):
             },
         ]
 
-        replies = _parse_streaming_response(events, test_callback, model)
+        component_info = ComponentInfo(
+            type=type_,
+        )
+
+        replies = _parse_streaming_response(events, test_callback, model, component_info)
         # Pop completion_start_time since it will always change
         replies[0].meta.pop("completion_start_time")
         expected_messages = [
@@ -421,12 +427,19 @@ def test_callback(chunk: StreamingChunk):
             }
         ]
 
+        for chunk in streaming_chunks:
+            assert chunk.component_info.type == type_
+            assert chunk.component_info.name is None  # not in a pipeline
+
         # Verify final replies
         assert len(replies) == 1
         assert replies == expected_messages
 
     def test_parse_streaming_response_with_two_tool_calls(self, mock_boto3_session):
         model = "anthropic.claude-3-5-sonnet-20240620-v1:0"
+        type_ = (
+            "haystack_integrations.components.generators.amazon_bedrock.chat.chat_generator.AmazonBedrockChatGenerator"
+        )
         streaming_chunks = []
 
         def test_callback(chunk: StreamingChunk):
@@ -475,7 +488,11 @@ def test_callback(chunk: StreamingChunk):
             },
         ]
 
-        replies = _parse_streaming_response(events, test_callback, model)
+        component_info = ComponentInfo(
+            type=type_,
+        )
+
+        replies = _parse_streaming_response(events, test_callback, model, component_info)
         # Pop completion_start_time since it will always change
         replies[0].meta.pop("completion_start_time")
         expected_messages = [