yujunjun/graphrag
1
0
Fork 0
mirror of https://github.com/microsoft/graphrag.git synced 2025-06-26 23:19:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix content embedding container name (#1358)

Browse Source
This commit is contained in:
Alonso Guevara 2024-11-05 15:56:32 -06:00 committed by GitHub
parent 1557ce34f9
commit 83bd5cefe5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
.semversioner/next-release/patch-20241105205747256773.json Normal file
View File

@ -0,0 +1,4 @@
{
"type": "patch",
"description": "Fix var name for embedding"
}

4
graphrag/api/query.py
View File

@ -365,7 +365,7 @@ async def drift_search(
full_content_embedding_store = _get_embedding_store( full_content_embedding_store = _get_embedding_store(
config_args=vector_store_args, # type: ignore 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) _entities = read_indexer_entities(nodes, entities, community_level)
@ -460,7 +460,7 @@ def _patch_vector_store(
full_content_embedding_store = LanceDBVectorStore( full_content_embedding_store = LanceDBVectorStore(
db_uri=config.embeddings.vector_store["db_uri"], 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"], overwrite=config.embeddings.vector_store["overwrite"],
) )
full_content_embedding_store.connect( full_content_embedding_store.connect(