From c62422eadee4ac9f666c55fa04706ce52812fd32 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Wed, 5 Mar 2025 15:33:54 +0100 Subject: [PATCH] Update utils_api.py --- lightrag/api/utils_api.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/lightrag/api/utils_api.py b/lightrag/api/utils_api.py index ed1250d4..39b2950f 100644 --- a/lightrag/api/utils_api.py +++ b/lightrag/api/utils_api.py @@ -17,6 +17,10 @@ from starlette.status import HTTP_403_FORBIDDEN # Load environment variables load_dotenv(override=True) +global_args={ + "main_args":None +} + class OllamaServerInfos: # Constants for emulated Ollama model information @@ -340,9 +344,13 @@ def parse_args(is_uvicorn_mode: bool = False) -> argparse.Namespace: # Inject chunk configuration args.chunk_size = get_env_value("CHUNK_SIZE", 1200, int) args.chunk_overlap_size = get_env_value("CHUNK_OVERLAP_SIZE", 100, int) + + # Select Document loading tool + args.document_loading_tool = get_env_value("DOCUMENT_LOADING_TOOL", "DOCLING") ollama_server_infos.LIGHTRAG_MODEL = args.simulated_model_name + global_args["main_args"]= args return args