diff --git a/libs/ktem/ktem/index/file/graph/lightrag_pipelines.py b/libs/ktem/ktem/index/file/graph/lightrag_pipelines.py index 694ad40..21dab31 100644 --- a/libs/ktem/ktem/index/file/graph/lightrag_pipelines.py +++ b/libs/ktem/ktem/index/file/graph/lightrag_pipelines.py @@ -29,10 +29,9 @@ 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 + # newer versions 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, @@ -240,7 +239,7 @@ def build_graphrag(working_dir, llm_func, embedding_func): embedding_func=embedding_func, ) - # newer verisons of LightRAG needs to be initialized before using + # newer versions of LightRAG needs to be initialized before using asyncio.run(graphrag_func.initialize_storages()) asyncio.run(initialize_pipeline_status())