test validation

Author: Gaudy Blanco

graphrag/config/models/vector_store_config.py

@@ -95,6 +95,12 @@ def _validate_embeddings_schema(self) -> None:
             if name not in all_embeddings:
                 msg = f"vector_store.embeddings_schema contains an invalid embedding schema name: {name}. Please rerun `graphrag init` and select the correct embedding schema names."
                 raise ValueError(msg)
+            
+        if self.type == VectorStoreType.CosmosDB:
+            for id_field in self.embeddings_schema.values():
+                if id_field.id_field != "id":
+                    msg = "When using CosmosDB, the id_field in embeddings_schema must be 'id'. Please rerun `graphrag init` and set the id_field to 'id'."
+                    raise ValueError(msg)
 
     @model_validator(mode="after")
     def _validate_model(self):

tests/integration/vector_stores/test_cosmosdb.py

@@ -110,7 +110,7 @@ def test_vector_store_customization():
     vector_store = CosmosDBVectorStore(
         vector_store_schema_config=VectorStoreSchemaConfig(
             index_name="text-embeddings",
-            id_field="id",
+            id_field="id_custom",
             text_field="text_custom",
             vector_field="vector_custom",
             attributes_field="attributes_custom",