From 9f575c3c32a2674512ea740fb9737d014b9a7566 Mon Sep 17 00:00:00 2001
From: Allen Kim <allekim@microsoft.com>
Date: Thu, 23 Oct 2025 11:11:59 -0700
Subject: [PATCH 1/4] Adding test for checking float16 datatype for vector
 embedding policy

---
 .../azure-cosmos/tests/test_vector_policy.py  | 24 ++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/sdk/cosmos/azure-cosmos/tests/test_vector_policy.py b/sdk/cosmos/azure-cosmos/tests/test_vector_policy.py
index 924f4d784043..3cafe082b381 100644
--- a/sdk/cosmos/azure-cosmos/tests/test_vector_policy.py
+++ b/sdk/cosmos/azure-cosmos/tests/test_vector_policy.py
@@ -35,6 +35,28 @@ def setUpClass(cls):
     def tearDownClass(cls):
         test_config.TestConfig.try_delete_database_with_id(cls.client, cls.test_db.id)
 
+    @unittest.skip
+    def test_create_valid_vector_embedding_policy(self):
+        # Using valid data types
+        data_types = ["float32", "float16", "int8", "uint8"]
+        for data_type in data_types:
+            print("Testing data type: ", data_type)
+            vector_embedding_policy = {
+                "vectorEmbeddings": [
+                    {
+                        "path": "/vector1",
+                        "dataType": data_type,
+                        "dimensions": 256,
+                        "distanceFunction": "euclidean"
+                    }]}
+            created_container = self.test_db.create_container(
+                id='vector_container_' + data_type,
+                partition_key=PartitionKey(path="/id"),
+                vector_embedding_policy=vector_embedding_policy)
+            properties = created_container.read()
+            assert properties["vectorEmbeddingPolicy"]["vectorEmbeddings"][0]["dataType"] == data_type
+            self.test_db.delete_container('vector_container_' + data_type)
+
     def test_create_vector_embedding_container(self):
         indexing_policy = {
             "vectorIndexes": [
@@ -440,7 +462,7 @@ def test_fail_create_vector_embedding_policy(self):
             "vectorEmbeddings": [
                 {
                     "path": "/vector1",
-                    "dataType": "float33",
+                    "dataType": "float16",
                     "dimensions": 256,
                     "distanceFunction": "euclidean"
                 }]}

From 867296b2a4ea7a4bd5049f244f0c794a4b96a815 Mon Sep 17 00:00:00 2001
From: Allen Kim <allekim@microsoft.com>
Date: Thu, 23 Oct 2025 11:15:36 -0700
Subject: [PATCH 2/4] Revert unnecessary change

---
 sdk/cosmos/azure-cosmos/tests/test_vector_policy.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/sdk/cosmos/azure-cosmos/tests/test_vector_policy.py b/sdk/cosmos/azure-cosmos/tests/test_vector_policy.py
index 3cafe082b381..9c0ff1636230 100644
--- a/sdk/cosmos/azure-cosmos/tests/test_vector_policy.py
+++ b/sdk/cosmos/azure-cosmos/tests/test_vector_policy.py
@@ -462,7 +462,7 @@ def test_fail_create_vector_embedding_policy(self):
             "vectorEmbeddings": [
                 {
                     "path": "/vector1",
-                    "dataType": "float16",
+                    "dataType": "float32",
                     "dimensions": 256,
                     "distanceFunction": "euclidean"
                 }]}

From 698a8dccf8d0ef098ddc271b0db5d1ec55ab975c Mon Sep 17 00:00:00 2001
From: Allen Kim <allekim@microsoft.com>
Date: Thu, 23 Oct 2025 11:16:05 -0700
Subject: [PATCH 3/4] Remove unnecessary print

---
 sdk/cosmos/azure-cosmos/tests/test_vector_policy.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/sdk/cosmos/azure-cosmos/tests/test_vector_policy.py b/sdk/cosmos/azure-cosmos/tests/test_vector_policy.py
index 9c0ff1636230..3887106e1041 100644
--- a/sdk/cosmos/azure-cosmos/tests/test_vector_policy.py
+++ b/sdk/cosmos/azure-cosmos/tests/test_vector_policy.py
@@ -40,7 +40,6 @@ def test_create_valid_vector_embedding_policy(self):
         # Using valid data types
         data_types = ["float32", "float16", "int8", "uint8"]
         for data_type in data_types:
-            print("Testing data type: ", data_type)
             vector_embedding_policy = {
                 "vectorEmbeddings": [
                     {

From 187cc6f6bc30dfdf56ee9fcbd807cdd52a91dbc4 Mon Sep 17 00:00:00 2001
From: Allen Kim <allekim@microsoft.com>
Date: Thu, 23 Oct 2025 11:17:32 -0700
Subject: [PATCH 4/4] Revert unnecessary change

---
 sdk/cosmos/azure-cosmos/tests/test_vector_policy.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/sdk/cosmos/azure-cosmos/tests/test_vector_policy.py b/sdk/cosmos/azure-cosmos/tests/test_vector_policy.py
index 3887106e1041..5b63e8b76ef3 100644
--- a/sdk/cosmos/azure-cosmos/tests/test_vector_policy.py
+++ b/sdk/cosmos/azure-cosmos/tests/test_vector_policy.py
@@ -461,7 +461,7 @@ def test_fail_create_vector_embedding_policy(self):
             "vectorEmbeddings": [
                 {
                     "path": "/vector1",
-                    "dataType": "float32",
+                    "dataType": "float33",
                     "dimensions": 256,
                     "distanceFunction": "euclidean"
                 }]}