Skip to content

Commit 2e7821f

Browse files
committed
lint fix
1 parent 06e41ae commit 2e7821f

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

libs/aws/tests/unit_tests/chat_models/test_bedrock_converse.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1113,27 +1113,27 @@ def test__lc_content_to_bedrock_mime_types() -> None:
11131113
file_data = base64.b64encode(b"file_test_data").decode("utf-8")
11141114

11151115
# Create content with one of each type
1116-
content = [
1116+
content: List[Dict[str, Any]] = [
11171117
{
11181118
"type": "video",
11191119
"source": {
11201120
"type": "base64",
1121-
"mediaType": "video/mp4",
1121+
"media_type": "video/mp4",
11221122
"data": video_data,
11231123
},
11241124
},
11251125
{
11261126
"type": "image",
11271127
"source": {
11281128
"type": "base64",
1129-
"mediaType": "image/jpeg",
1129+
"media_type": "image/jpeg",
11301130
"data": image_data,
11311131
},
11321132
},
11331133
{
11341134
"type": "file",
1135-
"sourceType": "base64",
1136-
"mimeType": "application/pdf",
1135+
"source_type": "base64",
1136+
"mime_type": "application/pdf",
11371137
"data": file_data,
11381138
"name": "test_document.pdf",
11391139
},
@@ -1635,4 +1635,4 @@ def side_effect(service_name: str, **kwargs: Any) -> mock.Mock:
16351635
)
16361636

16371637
# The streaming should be disabled for models with no streaming support
1638-
assert chat_model.disable_streaming is True
1638+
assert chat_model.disable_streaming is True

0 commit comments

Comments
 (0)