update init output and make url parameter optional due to connection …

…string possibility
microsoft · Jan 14, 2025 · a712953 · a712953
1 parent 30cd48a
commit a712953
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/graphrag/config/defaults.py b/graphrag/config/defaults.py
@@ -93,16 +93,16 @@
 UPDATE_STORAGE_BASE_DIR = "update_output"
 
 VECTOR_STORE = f"""
-    type: {VectorStoreType.LanceDB.value}
+    type: {VectorStoreType.LanceDB.value} # one of [lancedb, azure_ai_search, cosmosdb]
     db_uri: '{(Path(STORAGE_BASE_DIR) / "lancedb")!s}'
-    container_name: default
+    collection_name: default
     overwrite: true\
 """
 
 VECTOR_STORE_DICT = {
     "type": VectorStoreType.LanceDB.value,
     "db_uri": str(Path(STORAGE_BASE_DIR) / "lancedb"),
-    "container_name": "default",
+    "collection_name": "default",
     "overwrite": True,
 }
 

diff --git a/graphrag/vector_stores/cosmosdb.py b/graphrag/vector_stores/cosmosdb.py
@@ -31,12 +31,12 @@ def __init__(self, **kwargs: Any) -> None:
 
     def connect(self, **kwargs: Any) -> Any:
         """Connect to CosmosDB vector storage."""
-        url = kwargs["url"]
         connection_string = kwargs.get("connection_string")
         if connection_string:
             self._cosmos_client = CosmosClient.from_connection_string(connection_string)
         else:
-            if url is None:
+            url = kwargs.get("url")
+            if not url:
                 msg = "Either connection_string or url must be provided."
                 raise ValueError(msg)
             self._cosmos_client = CosmosClient(