From 8cb1d4bedaafabeea2fecedb4702c366c2b3aae1 Mon Sep 17 00:00:00 2001
From: Vladimir Blagojevic <dovlex@gmail.com>
Date: Wed, 6 Mar 2024 12:05:29 +0100
Subject: [PATCH] Replace single quotes with double

---
 .../tests/test_chat_generator.py              | 26 +++++++++----------
 1 file changed, 13 insertions(+), 13 deletions(-)

diff --git a/integrations/amazon_bedrock/tests/test_chat_generator.py b/integrations/amazon_bedrock/tests/test_chat_generator.py
index 52449a85d..c46459a7f 100644
--- a/integrations/amazon_bedrock/tests/test_chat_generator.py
+++ b/integrations/amazon_bedrock/tests/test_chat_generator.py
@@ -146,10 +146,10 @@ def test_prepare_body_with_default_params(self) -> None:
         layer = AnthropicClaudeChatAdapter(generation_kwargs={})
         prompt = "Hello, how are you?"
         expected_body = {
-            'anthropic_version': 'bedrock-2023-05-31',
-            'max_tokens': 512,
-            'messages': [{'content': [{'text': 'Hello, how are you?', 'type': 'text'}],
-                          'role': 'user'}]}
+            "anthropic_version": "bedrock-2023-05-31",
+            "max_tokens": 512,
+            "messages": [{"content": [{"text": "Hello, how are you?", "type": "text"}],
+                          "role": "user"}]}
 
         body = layer.prepare_body([ChatMessage.from_user(prompt)])
 
@@ -158,15 +158,15 @@ def test_prepare_body_with_default_params(self) -> None:
     def test_prepare_body_with_custom_inference_params(self) -> None:
         layer = AnthropicClaudeChatAdapter(generation_kwargs={"temperature": 0.7, "top_p": 0.8, "top_k": 4})
         prompt = "Hello, how are you?"
-        expected_body = {'anthropic_version': 'bedrock-2023-05-31',
-                         'max_tokens': 512,
-                         'max_tokens_to_sample': 69,
-                         'messages': [{'content': [{'text': 'Hello, how are you?', 'type': 'text'}],
-                                       'role': 'user'}],
-                         'stop_sequences': ['CUSTOM_STOP'],
-                         'temperature': 0.7,
-                         'top_k': 5,
-                         'top_p': 0.8}
+        expected_body = {"anthropic_version": "bedrock-2023-05-31",
+                         "max_tokens": 512,
+                         "max_tokens_to_sample": 69,
+                         "messages": [{"content": [{"text": "Hello, how are you?", "type": "text"}],
+                                       "role": "user"}],
+                         "stop_sequences": ["CUSTOM_STOP"],
+                         "temperature": 0.7,
+                         "top_k": 5,
+                         "top_p": 0.8}
 
         body = layer.prepare_body(
             [ChatMessage.from_user(prompt)], top_p=0.8, top_k=5, max_tokens_to_sample=69, stop_sequences=["CUSTOM_STOP"]