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

fix: error 'history_messages' with LightRAG latest version (#719) bump:patch

Browse Source 
* fix: Error: 'history_messages' with LightRAG

* added comment
This commit is contained in:
Benno Löffler 2025-04-01 06:31:44 +02:00 committed by GitHub
parent 2ffe374c2f
commit 911b20caf0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -29,6 +29,10 @@ 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
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,
@ -235,6 +239,11 @@ def build_graphrag(working_dir, llm_func, embedding_func):
llm_model_func=llm_func, llm_model_func=llm_func,
embedding_func=embedding_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 return graphrag_func