[Cosmos] Remove GeoSpatial index limitation (#35790)

* Update README.md

* add samples, add tests

* Update index_management_async.py

Author: simorenoh

sdk/cosmos/azure-cosmos/README.md

@@ -164,7 +164,6 @@ Streamable queries like `SELECT * FROM WHERE` *do* support continuation tokens.
 ### Control Plane Limitations:
 
 * Get CollectionSizeUsage, DatabaseUsage, and DocumentUsage metrics
-* Create Geospatial Index
 * Get the connection string
 * Get the minimum RU/s of a container
 

sdk/cosmos/azure-cosmos/samples/index_management.py

@@ -633,6 +633,54 @@ def perform_multi_orderby_query(db):
         print("Entity doesn't exist")
 
 
+def use_geospatial_indexing_policy(db):
+    try:
+        delete_container_if_exists(db, CONTAINER_ID)
+
+        # Create a container with vector embedding policy and vector indexes
+        indexing_policy = {
+            'includedPaths': [
+                {'path': '/"Location"/?',
+                    'indexes': [
+                        {
+                            'kind': 'Spatial',
+                            'dataType': 'Point'
+                        }]
+                 },
+                {
+                    'path': '/'
+                }
+            ]
+        }
+
+        created_container = db.create_container(
+            id=CONTAINER_ID,
+            partition_key=PARTITION_KEY,
+            indexing_policy=indexing_policy
+        )
+        properties = created_container.read()
+        print(created_container)
+
+        print("\n" + "-" * 25 + "\n9. Container created with geospatial indexes")
+        print_dictionary_items(properties["indexingPolicy"])
+
+        # Create some items
+        doc9 = created_container.create_item(body={"id": "loc1", 'Location': {'type': 'Point', 'coordinates': [20.0, 20.0]}})
+        doc9 = created_container.create_item(body={"id": "loc2", 'Location': {'type': 'Point', 'coordinates': [100.0, 100.0]}})
+
+        # Run ST_DISTANCE queries using the geospatial index
+        query = "SELECT * FROM root WHERE (ST_DISTANCE(root.Location, {type: 'Point', coordinates: [20.1, 20]}) < 20000)"
+        query_documents_with_custom_query(created_container, query)
+
+        # Cleanup
+        db.delete_container(created_container)
+        print("\n")
+    except exceptions.CosmosResourceExistsError:
+        print("Entity already exists")
+    except exceptions.CosmosResourceNotFoundError:
+        print("Entity doesn't exist")
+
+
 def use_vector_embedding_policy(db):
     try:
         delete_container_if_exists(db, CONTAINER_ID)
@@ -663,7 +711,7 @@ def use_vector_embedding_policy(db):
         properties = created_container.read()
         print(created_container)
 
-        print("\n" + "-" * 25 + "\n9. Container created with vector embedding policy and vector indexes")
+        print("\n" + "-" * 25 + "\n10. Container created with vector embedding policy and vector indexes")
         print_dictionary_items(properties["indexingPolicy"])
         print_dictionary_items(properties["vectorEmbeddingPolicy"])
 
@@ -728,7 +776,10 @@ def run_sample():
         # 8. Perform Multi Orderby queries using composite indexes
         perform_multi_orderby_query(created_db)
 
-        # 9. Create and use a vector embedding policy
+        # 9. Create and use a geospatial indexing policy
+        use_geospatial_indexing_policy(created_db)
+
+        # 10. Create and use a vector embedding policy
         use_vector_embedding_policy(created_db)
 
     except exceptions.AzureError as e:

sdk/cosmos/azure-cosmos/samples/index_management_async.py

@@ -628,6 +628,54 @@ async def perform_multi_orderby_query(db):
         print("Entity doesn't exist")
 
 
+async def use_geospatial_indexing_policy(db):
+    try:
+        await delete_container_if_exists(db, CONTAINER_ID)
+
+        # Create a container with vector embedding policy and vector indexes
+        indexing_policy = {
+            'includedPaths': [
+                {'path': '/"Location"/?',
+                    'indexes': [
+                        {
+                            'kind': 'Spatial',
+                            'dataType': 'Point'
+                        }]
+                 },
+                {
+                    'path': '/'
+                }
+            ]
+        }
+
+        created_container = await db.create_container(
+            id=CONTAINER_ID,
+            partition_key=PARTITION_KEY,
+            indexing_policy=indexing_policy
+        )
+        properties = await created_container.read()
+        print(created_container)
+
+        print("\n" + "-" * 25 + "\n9. Container created with geospatial indexes")
+        print_dictionary_items(properties["indexingPolicy"])
+
+        # Create some items
+        doc9 = await created_container.create_item(body={"id": "loc1", 'Location': {'type': 'Point', 'coordinates': [20.0, 20.0]}})
+        doc9 = await created_container.create_item(body={"id": "loc2", 'Location': {'type': 'Point', 'coordinates': [100.0, 100.0]}})
+
+        # Run ST_DISTANCE queries using the geospatial index
+        query = "SELECT * FROM root WHERE (ST_DISTANCE(root.Location, {type: 'Point', coordinates: [20.1, 20]}) < 20000)"
+        await query_documents_with_custom_query(created_container, query)
+
+        # Cleanup
+        await db.delete_container(created_container)
+        print("\n")
+    except exceptions.CosmosResourceExistsError:
+        print("Entity already exists")
+    except exceptions.CosmosResourceNotFoundError:
+        print("Entity doesn't exist")
+
+
 async def use_vector_embedding_policy(db):
     try:
         await delete_container_if_exists(db, CONTAINER_ID)
@@ -658,7 +706,7 @@ async def use_vector_embedding_policy(db):
         properties = await created_container.read()
         print(created_container)
 
-        print("\n" + "-" * 25 + "\n9. Container created with vector embedding policy and vector indexes")
+        print("\n" + "-" * 25 + "\n10. Container created with vector embedding policy and vector indexes")
         print_dictionary_items(properties["indexingPolicy"])
         print_dictionary_items(properties["vectorEmbeddingPolicy"])
 
@@ -723,7 +771,10 @@ async def run_sample():
             # 8. Perform Multi Orderby queries using composite indexes
             await perform_multi_orderby_query(created_db)
 
-            # 9. Create and use a vector embedding policy
+            # 9. Create and use a geospatial indexing policy
+            await use_geospatial_indexing_policy(created_db)
+
+            # 10. Create and use a vector embedding policy
             await use_vector_embedding_policy(created_db)
 
     except exceptions.AzureError as e:

sdk/cosmos/azure-cosmos/test/test_crud.py

@@ -948,7 +948,7 @@ def test_document_upsert(self):
             before_create_documents_count,
             'number of documents should remain same')
 
-    def _test_spatial_index(self):
+    def test_geospatial_index(self):
         db = self.databaseForTest
         # partial policy specified
         collection = db.create_container(

sdk/cosmos/azure-cosmos/test/test_crud_async.py

@@ -881,8 +881,7 @@ async def test_document_upsert_async(self):
         document_list = [document async for document in created_collection.read_all_items()]
         assert len(document_list) == before_create_documents_count
 
-    async def _test_spatial_index(self):
-
+    async def test_geospatial_index_async(self):
         db = self.database_for_test
         # partial policy specified
         collection = await db.create_container(