fmt

langchain-ai · Aug 20, 2024 · e49995a · e49995a
1 parent fb13e12
commit e49995a
Showing 1 changed file with 8 additions and 24 deletions.
diff --git a/libs/aws/tests/unit_tests/chat_models/test_bedrock_converse.py b/libs/aws/tests/unit_tests/chat_models/test_bedrock_converse.py
@@ -164,30 +164,23 @@ def test__messages_to_bedrock() -> None:
         HumanMessage(
             content=[
                 {
-                    "type": "document", 
+                    "type": "document",
                     "document": {
                         "format": "pdf",
                         "name": "doc1",
-                        "source": {
-                            "bytes": b"doc1_data"
-                        }
-                    }
+                        "source": {"bytes": b"doc1_data"},
+                    },
                 }
             ]
         ),
         HumanMessage(
             content=[
                 {
                     "type": "image",
-                    "image": {
-                        "format": "jpeg",
-                        "source": {
-                            "bytes": b"image_data"
-                        }
-                    }
+                    "image": {"format": "jpeg", "source": {"bytes": b"image_data"}},
                 }
             ]
-        )
+        ),
     ]
     expected_messages = [
         {"role": "user", "content": [{"text": "hu1"}, {"text": "hu2"}]},
@@ -260,21 +253,12 @@ def test__messages_to_bedrock() -> None:
                     "document": {
                         "format": "pdf",
                         "name": "doc1",
-                        "source": {
-                            "bytes": b"doc1_data"
-                        }
+                        "source": {"bytes": b"doc1_data"},
                     }
                 },
-                {
-                    "image": {
-                        "format": "jpeg",
-                        "source": {
-                            "bytes": b"image_data"
-                        }
-                    }
-                }
+                {"image": {"format": "jpeg", "source": {"bytes": b"image_data"}}},
             ],
-        }
+        },
     ]
     expected_system = [
         {"text": "sys1"},