gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update utils_api.py

Browse Source
This commit is contained in:
Saifeddine ALOUI
2025-03-05 15:36:17 +01:00
committed by GitHub
parent c62422eade
commit 39c24f4a59
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
lightrag/api/utils_api.py
View File

@@ -17,9 +17,7 @@ from starlette.status import HTTP_403_FORBIDDEN
# Load environment variables # Load environment variables
load_dotenv(override=True) load_dotenv(override=True)
global_args={ global_args = {"main_args": None}
"main_args":None
}
class OllamaServerInfos: class OllamaServerInfos:
@@ -344,13 +342,13 @@ def parse_args(is_uvicorn_mode: bool = False) -> argparse.Namespace:
# Inject chunk configuration # Inject chunk configuration
args.chunk_size = get_env_value("CHUNK_SIZE", 1200, int) args.chunk_size = get_env_value("CHUNK_SIZE", 1200, int)
args.chunk_overlap_size = get_env_value("CHUNK_OVERLAP_SIZE", 100, int) args.chunk_overlap_size = get_env_value("CHUNK_OVERLAP_SIZE", 100, int)
# Select Document loading tool # Select Document loading tool
args.document_loading_tool = get_env_value("DOCUMENT_LOADING_TOOL", "DOCLING") args.document_loading_tool = get_env_value("DOCUMENT_LOADING_TOOL", "DOCLING")
ollama_server_infos.LIGHTRAG_MODEL = args.simulated_model_name ollama_server_infos.LIGHTRAG_MODEL = args.simulated_model_name
global_args["main_args"]= args global_args["main_args"] = args
return args return args