Public/kotaemon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: comfort CI

Browse Source
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
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
libs/ktem/ktem/index/file/graph/lightrag_pipelines.py
View File

@ -30,9 +30,8 @@ from .visualize import create_knowledge_graph, visualize_graph
try: try:
from lightrag import LightRAG, QueryParam from lightrag import LightRAG, QueryParam
# newer verisons of LightRAG needs to be initialized before using # newer versions of LightRAG needs to be initialized before using
from lightrag.kg.shared_storage import initialize_pipeline_status from lightrag.kg.shared_storage import initialize_pipeline_status
from lightrag.operate import ( from lightrag.operate import (
_find_most_related_edges_from_entities, _find_most_related_edges_from_entities,
_find_most_related_text_unit_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, 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(graphrag_func.initialize_storages())
asyncio.run(initialize_pipeline_status()) asyncio.run(initialize_pipeline_status())