Fix streaming_callback serialization in AmazonBedrockChatGenerator

FloRul · Apr 24, 2024 · 499d2dc · 499d2dc
1 parent 00354a6
commit 499d2dc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...ock/src/haystack_integrations/components/generators/amazon_bedrock/chat/chat_generator.py b/...ock/src/haystack_integrations/components/generators/amazon_bedrock/chat/chat_generator.py
@@ -210,6 +210,7 @@ def to_dict(self) -> Dict[str, Any]:
         :returns:
             Dictionary with serialized data.
         """
+        callback_name = serialize_callable(self.streaming_callback) if self.streaming_callback else None
         return default_to_dict(
             self,
             aws_access_key_id=self.aws_access_key_id.to_dict() if self.aws_access_key_id else None,
@@ -220,7 +221,7 @@ def to_dict(self) -> Dict[str, Any]:
             model=self.model,
             stop_words=self.stop_words,
             generation_kwargs=self.model_adapter.generation_kwargs,
-            streaming_callback=serialize_callable(self.streaming_callback),
+            streaming_callback=callback_name,
         )
 
     @classmethod