From 2e21b97eb2b891bbe640ee9d6a24947077a151dc Mon Sep 17 00:00:00 2001
From: Lu Peng <118394507+lu-ohai@users.noreply.github.com>
Date: Tue, 14 Jan 2025 14:41:34 -0500
Subject: [PATCH] Update test_oci_model_deployment_endpoint.py

---
 .../embeddings/test_oci_model_deployment_endpoint.py        | 6 +++++-
 1 file changed, 5 insertions(+), 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
index a6fe7fb9a3b9d..7f10b9218e83a 100644
--- 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,
     )