fix: comfort CI

This commit is contained in:
Tadashi 2025-04-01 11:33:48 +07:00
parent 911b20caf0
commit a3e2e20735
No known key found for this signature in database
GPG Key ID: 399380A00CC9028D

View File

@ -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())