fix: fix test

krrishdholakia · krrishdholakia · commit 92e841e311aa · 2025-09-18T23:37:38.000-07:00
diff --git a/tests/local_testing/test_get_optional_params_embeddings.py b/tests/local_testing/test_get_optional_params_embeddings.py
@@ -68,4 +68,4 @@ def test_bedrock_embed_v2_with_drop_params():
         custom_llm_provider=custom_llm_provider,
     )
     print(f"received optional_params: {optional_params}")
-    assert optional_params == {"dimensions": 512}
+    assert optional_params == {"dimensions": 512, "embeddingTypes": ["binary"]}

Original file line number	Diff line number	Diff line change
`@@ -68,4 +68,4 @@ def test_bedrock_embed_v2_with_drop_params():`
`68`	`68`	`custom_llm_provider=custom_llm_provider,`
`69`	`69`	`)`
`70`	`70`	`print(f"received optional_params: {optional_params}")`
`71`		`- assert optional_params == {"dimensions": 512}`
	`71`	`+ assert optional_params == {"dimensions": 512, "embeddingTypes": ["binary"]}`