More Pyright

Author: AlonsoGuevara

graphrag/config/defaults.py

@@ -97,6 +97,13 @@
     overwrite: true\
 """
 
+VECTOR_STORE_DICT = {
+    "type": VectorStoreType.LanceDB.value,
+    "db_uri": str((Path(STORAGE_BASE_DIR) / "lancedb")),
+    "container_name": "default",
+    "overwrite": True,
+}
+
 # Local Search
 LOCAL_SEARCH_TEXT_UNIT_PROP = 0.5
 LOCAL_SEARCH_COMMUNITY_PROP = 0.1

graphrag/config/models/text_embedding_config.py

@@ -25,8 +25,8 @@ class TextEmbeddingConfig(LLMConfig):
         default=defs.EMBEDDING_TARGET,
     )
     skip: list[str] = Field(description="The specific embeddings to skip.", default=[])
-    vector_store: dict | str | None = Field(
-        description="The vector storage configuration", default=defs.VECTOR_STORE
+    vector_store: dict | None = Field(
+        description="The vector storage configuration", default=defs.VECTOR_STORE_DICT
     )
     strategy: dict | None = Field(
         description="The override strategy to use.", default=None