gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

summary

Browse source
This commit is contained in:
hzywhite 2025-09-02 06:21:08 +08:00
parent 36c81039b1
commit 745aa085db
2 changed files with 18 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
lightrag/api/routers/document_routes.py
View file

@ -1250,7 +1250,10 @@ async def pipeline_enqueue_file(
try: try:
await rag.apipeline_enqueue_documents( await rag.apipeline_enqueue_documents(
content, file_paths=file_path.name, track_id=track_id, scheme_name=scheme_name content,
file_paths=file_path.name,
track_id=track_id,
scheme_name=scheme_name,
) )
logger.info( logger.info(
@ -1334,7 +1337,9 @@ async def pipeline_enqueue_file(
logger.error(f"Error deleting file {file_path}: {str(e)}") logger.error(f"Error deleting file {file_path}: {str(e)}")
async def pipeline_index_file(rag: LightRAG, file_path: Path, track_id: str = None, scheme_name: str = None): async def pipeline_index_file(
rag: LightRAG, file_path: Path, track_id: str = None, scheme_name: str = None
):
"""Index a file with track_id """Index a file with track_id
Args: Args:
@ -1376,7 +1381,9 @@ async def pipeline_index_files(
# Process files sequentially with track_id # Process files sequentially with track_id
for file_path in sorted_file_paths: for file_path in sorted_file_paths:
success, _ = await pipeline_enqueue_file(rag, file_path, track_id, scheme_name) success, _ = await pipeline_enqueue_file(
rag, file_path, track_id, scheme_name
)
if success: if success:
enqueued = True enqueued = True
@ -1421,13 +1428,11 @@ async def pipeline_index_files_raganything(
# Process files sequentially with track_id # Process files sequentially with track_id
for file_path in sorted_file_paths: for file_path in sorted_file_paths:
success = ( success = await rag_anything.process_document_complete_lightrag_api(
await rag_anything.process_document_complete_lightrag_api( file_path=str(file_path),
file_path=str(file_path), output_dir="./output",
output_dir="./output", parse_method="auto",
parse_method="auto", scheme_name=scheme_name,
scheme_name=scheme_name,
)
) )
if success: if success:
pass pass

8
lightrag/lightrag.py
View file

@ -1,6 +1,5 @@
from __future__ import annotations from __future__ import annotations
from re import A
import traceback import traceback
import asyncio import asyncio
import configparser import configparser
@ -930,9 +929,7 @@ class LightRAG:
return track_id return track_id
def move_file_to_enqueue(self, file_path): def move_file_to_enqueue(self, file_path):
try: try:
enqueued_dir = file_path.parent / "__enqueued__" enqueued_dir = file_path.parent / "__enqueued__"
enqueued_dir.mkdir(exist_ok=True) enqueued_dir.mkdir(exist_ok=True)
@ -955,8 +952,9 @@ class LightRAG:
) )
# Don't affect the main function's success status # Don't affect the main function's success status
def get_unique_filename_in_enqueued(
def get_unique_filename_in_enqueued(self, target_dir: Path, original_name: str) -> str: self, target_dir: Path, original_name: str
) -> str:
from pathlib import Path from pathlib import Path
import time import time