Update test_oci_model_deployment_endpoint.py

langchain-ai · Jan 14, 2025 · 2e21b97 · 2e21b97
1 parent d5f1a5e
commit 2e21b97
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/libs/community/tests/unit_tests/embeddings/test_oci_model_deployment_endpoint.py b/libs/community/tests/unit_tests/embeddings/test_oci_model_deployment_endpoint.py
@@ -17,7 +17,11 @@ def test_embedding_call(mocker: MockerFixture) -> None:
         responses.POST,
         endpoint,
         json={
-            "embeddings": expected_output,
+            "data": [
+                {
+                    "embedding": expected_output
+                }
+            ],
         },
         status=200,
     )