root/Gitcloned-RAG-Anything
1
1
Fork 0
Code Issues Pull Requests Projects Releases Activity

summary

Browse Source
This commit is contained in:
hzywhite
2025-09-02 18:19:56 +08:00
parent 92cf72fe8a
commit 6877983a71
2 changed files with 23 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
raganything/processor.py
View File

@@ -1424,7 +1424,9 @@ class ProcessorMixin:
# Ensure LightRAG is initialized
result = await self._ensure_lightrag_initialized()
if not result["success"]:
current_doc_status = await self.lightrag.doc_status.get_by_id(doc_pre_id)
current_doc_status = await self.lightrag.doc_status.get_by_id(
doc_pre_id
)
if not current_doc_status:
await self.lightrag.doc_status.upsert(
{

33
raganything/raganything.py
View File

@@ -220,7 +220,6 @@ class RAGAnything(QueryMixin, ProcessorMixin, BatchMixin):
else:
self.logger.warning(f"Unknown config parameter: {key}")
async def _ensure_lightrag_initialized(self):
"""Ensure LightRAG instance is initialized, create if necessary"""
try:
@@ -242,14 +241,17 @@ class RAGAnything(QueryMixin, ProcessorMixin, BatchMixin):
try:
# Ensure LightRAG storages are initialized
if (
not hasattr(self.lightrag, "_storages_status")
or self.lightrag._storages_status.name != "INITIALIZED"
not hasattr(self.lightrag, "_storages_status")
or self.lightrag._storages_status.name != "INITIALIZED"
):
self.logger.info(
"Initializing storages for pre-provided LightRAG instance"
)
await self.lightrag.initialize_storages()
from lightrag.kg.shared_storage import initialize_pipeline_status
from lightrag.kg.shared_storage import (
initialize_pipeline_status,
)
await initialize_pipeline_status()
# Initialize parse cache if not already done
@@ -257,11 +259,13 @@ class RAGAnything(QueryMixin, ProcessorMixin, BatchMixin):
self.logger.info(
"Initializing parse cache for pre-provided LightRAG instance"
)
self.parse_cache = self.lightrag.key_string_value_json_storage_cls(
namespace="parse_cache",
workspace=self.lightrag.workspace,
global_config=self.lightrag.__dict__,
embedding_func=self.embedding_func,
self.parse_cache = (
self.lightrag.key_string_value_json_storage_cls(
namespace="parse_cache",
workspace=self.lightrag.workspace,
global_config=self.lightrag.__dict__,
embedding_func=self.embedding_func,
)
)
await self.parse_cache.initialize()
@@ -272,7 +276,9 @@ class RAGAnything(QueryMixin, ProcessorMixin, BatchMixin):
return {"success": True}
except Exception as e:
error_msg = f"Failed to initialize pre-provided LightRAG instance: {str(e)}"
error_msg = (
f"Failed to initialize pre-provided LightRAG instance: {str(e)}"
)
self.logger.error(error_msg, exc_info=True)
return {"success": False, "error": error_msg}
@@ -304,7 +310,7 @@ class RAGAnything(QueryMixin, ProcessorMixin, BatchMixin):
k: v
for k, v in lightrag_params.items()
if not callable(v)
and k not in ["llm_model_kwargs", "vector_db_storage_cls_kwargs"]
and k not in ["llm_model_kwargs", "vector_db_storage_cls_kwargs"]
}
self.logger.info(f"Initializing LightRAG with parameters: {log_params}")
@@ -326,7 +332,9 @@ class RAGAnything(QueryMixin, ProcessorMixin, BatchMixin):
# Initialize processors after LightRAG is ready
self._initialize_processors()
self.logger.info("LightRAG, parse cache, and multimodal processors initialized")
self.logger.info(
"LightRAG, parse cache, and multimodal processors initialized"
)
return {"success": True}
except Exception as e:
@@ -339,7 +347,6 @@ class RAGAnything(QueryMixin, ProcessorMixin, BatchMixin):
self.logger.error(error_msg, exc_info=True)
return {"success": False, "error": error_msg}
async def finalize_storages(self):
"""Finalize all storages including parse cache and LightRAG storages