diff --git a/integrations/amazon_bedrock/src/haystack_integrations/components/generators/amazon_bedrock/chat/utils.py b/integrations/amazon_bedrock/src/haystack_integrations/components/generators/amazon_bedrock/chat/utils.py index f6a5ce2251..d436faf9e8 100644 --- a/integrations/amazon_bedrock/src/haystack_integrations/components/generators/amazon_bedrock/chat/utils.py +++ b/integrations/amazon_bedrock/src/haystack_integrations/components/generators/amazon_bedrock/chat/utils.py @@ -93,7 +93,7 @@ def _format_tools( ) if tools_cachepoint_config: - tool_specs.append({"cachePoint": tools_cachepoint_config}) + tool_specs.append(tools_cachepoint_config) return {"tools": tool_specs} diff --git a/integrations/amazon_bedrock/tests/test_chat_generator_utils.py b/integrations/amazon_bedrock/tests/test_chat_generator_utils.py index b9f1ba6cd0..a3f947f718 100644 --- a/integrations/amazon_bedrock/tests/test_chat_generator_utils.py +++ b/integrations/amazon_bedrock/tests/test_chat_generator_utils.py @@ -67,7 +67,7 @@ def tools(): class TestAmazonBedrockChatGeneratorUtils: def test_format_tools(self, tools): - formatted_tool = _format_tools(tools, tools_cachepoint_config={"type": "default"}) + formatted_tool = _format_tools(tools, tools_cachepoint_config={"cachePoint": {"type": "default"}}) assert formatted_tool == { "tools": [ {