diff --git a/libs/ktem/ktem/index/file/graph/lightrag_pipelines.py b/libs/ktem/ktem/index/file/graph/lightrag_pipelines.py index 2f8bc88..694ad40 100644 --- a/libs/ktem/ktem/index/file/graph/lightrag_pipelines.py +++ b/libs/ktem/ktem/index/file/graph/lightrag_pipelines.py @@ -29,6 +29,10 @@ from .visualize import create_knowledge_graph, visualize_graph try: from lightrag import LightRAG, QueryParam + + # newer verisons of LightRAG needs to be initialized before using + from lightrag.kg.shared_storage import initialize_pipeline_status + from lightrag.operate import ( _find_most_related_edges_from_entities, _find_most_related_text_unit_from_entities, @@ -235,6 +239,11 @@ def build_graphrag(working_dir, llm_func, embedding_func): llm_model_func=llm_func, embedding_func=embedding_func, ) + + # newer verisons of LightRAG needs to be initialized before using + asyncio.run(graphrag_func.initialize_storages()) + asyncio.run(initialize_pipeline_status()) + return graphrag_func