Skip to content

Commit d8df0ea

Browse files
Gaudy BlancoGaudy Blanco
authored andcommitted
fix cosmosdb bug
1 parent f6d923e commit d8df0ea

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

graphrag/vector_stores/cosmosdb.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ def _database_exists(self) -> bool:
8383

8484
def _create_container(self) -> None:
8585
"""Create the container if it doesn't exist."""
86-
partition_key = PartitionKey(path="/id", kind="Hash")
86+
partition_key = PartitionKey(path=f"/{self.id_field}", kind="Hash")
8787

8888
# Define the container vector policy
8989
vector_embedding_policy = {

0 commit comments

Comments
 (0)