diff --git a/.semversioner/next-release/patch-20241105205747256773.json b/.semversioner/next-release/patch-20241105205747256773.json new file mode 100644 index 00000000..b6743076 --- /dev/null +++ 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 index 679b55c0..1daad9ae 100644 --- 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(