Fix content embedding container name (#1358)

AlonsoGuevara · web-flow · commit 83bd5cefe539 · 2024-11-05T15:56:32.000-06:00
diff --git a/.semversioner/next-release/patch-20241105205747256773.json b/.semversioner/next-release/patch-20241105205747256773.json
@@ -0,0 +1,4 @@
+{
+  "type": "patch",
+  "description": "Fix var name for embedding"
+}
diff --git a/graphrag/api/query.py b/graphrag/api/query.py
@@ -365,7 +365,7 @@ async def drift_search(
 
     full_content_embedding_store = _get_embedding_store(
         config_args=vector_store_args,  # type: ignore
-        container_suffix="community-full-content",
+        container_suffix="community-full_content",
     )
 
     _entities = read_indexer_entities(nodes, entities, community_level)
@@ -460,7 +460,7 @@ def _patch_vector_store(
 
         full_content_embedding_store = LanceDBVectorStore(
             db_uri=config.embeddings.vector_store["db_uri"],
-            collection_name=f"{collection_name}-community-full-content",
+            collection_name=f"{collection_name}-community-full_content",
             overwrite=config.embeddings.vector_store["overwrite"],
         )
         full_content_embedding_store.connect(

-Original file line number
+Diff line change
@@ @@ -0,0 +1,4 @@ @@
 +{
 +  "type": "patch",
 +  "description": "Fix var name for embedding"
 +}
Original file line number	Diff line number	Diff line change
`@@ -365,7 +365,7 @@ async def drift_search(`
`365`	`365`
`366`	`366`	`full_content_embedding_store = _get_embedding_store(`
`367`	`367`	`config_args=vector_store_args, # type: ignore`
`368`		`- container_suffix="community-full-content",`
	`368`	`+ container_suffix="community-full_content",`
`369`	`369`	`)`
`370`	`370`
`371`	`371`	`_entities = read_indexer_entities(nodes, entities, community_level)`
`@@ -460,7 +460,7 @@ def _patch_vector_store(`
`460`	`460`
`461`	`461`	`full_content_embedding_store = LanceDBVectorStore(`
`462`	`462`	`db_uri=config.embeddings.vector_store["db_uri"],`
`463`		`- collection_name=f"{collection_name}-community-full-content",`
	`463`	`+ collection_name=f"{collection_name}-community-full_content",`
`464`	`464`	`overwrite=config.embeddings.vector_store["overwrite"],`
`465`	`465`	`)`
`466`	`466`	`full_content_embedding_store.connect(`