diff --git a/lightrag/api/config.py b/lightrag/api/config.py index d7620353..1bbdb1c9 100644 --- a/lightrag/api/config.py +++ b/lightrag/api/config.py @@ -12,6 +12,7 @@ from dotenv import load_dotenv # the OS environment variables take precedence over the .env file load_dotenv(dotenv_path=".env", override=False) + class OllamaServerInfos: # Constants for emulated Ollama model information LIGHTRAG_NAME = "lightrag" @@ -302,7 +303,7 @@ def parse_args() -> argparse.Namespace: # Select Document loading tool (DOCLING, DEFAULT) args.document_loading_engine = get_env_value("DOCUMENT_LOADING_ENGINE", "DEFAULT") - + # Add environment variables that were previously read directly args.cors_origins = get_env_value("CORS_ORIGINS", "*") args.summary_language = get_env_value("SUMMARY_LANGUAGE", "en") diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py index 84390afc..4d1eaf6d 100644 --- a/lightrag/api/lightrag_server.py +++ b/lightrag/api/lightrag_server.py @@ -321,9 +321,7 @@ def create_app(args): # namespace_prefix=args.namespace_prefix, auto_manage_storages_states=False, max_parallel_insert=args.max_parallel_insert, - addon_params={ - "language": args.summary_language - }, + addon_params={"language": args.summary_language}, ) else: # azure_openai rag = LightRAG( @@ -354,9 +352,7 @@ def create_app(args): # namespace_prefix=args.namespace_prefix, auto_manage_storages_states=False, max_parallel_insert=args.max_parallel_insert, - addon_params={ - "language": args.summary_language - }, + addon_params={"language": args.summary_language}, ) # Add routes diff --git a/lightrag/api/run_with_gunicorn.py b/lightrag/api/run_with_gunicorn.py index 95b12ab3..cf902a8a 100644 --- a/lightrag/api/run_with_gunicorn.py +++ b/lightrag/api/run_with_gunicorn.py @@ -120,12 +120,22 @@ def main(): # Set configuration variables in gunicorn_config, prioritizing command line arguments gunicorn_config.workers = ( - global_args.workers if global_args.workers else int(os.getenv("WORKERS", 1)) + global_args.workers + if global_args.workers + else int(os.getenv("WORKERS", 1)) ) # Bind configuration prioritizes command line arguments - host = global_args.host if global_args.host != "0.0.0.0" else os.getenv("HOST", "0.0.0.0") - port = global_args.port if global_args.port != 9621 else int(os.getenv("PORT", 9621)) + host = ( + global_args.host + if global_args.host != "0.0.0.0" + else os.getenv("HOST", "0.0.0.0") + ) + port = ( + global_args.port + if global_args.port != 9621 + else int(os.getenv("PORT", 9621)) + ) gunicorn_config.bind = f"{host}:{port}" # Log level configuration prioritizes command line arguments @@ -137,7 +147,9 @@ def main(): # Timeout configuration prioritizes command line arguments gunicorn_config.timeout = ( - global_args.timeout if global_args.timeout * 2 else int(os.getenv("TIMEOUT", 150 * 2)) + global_args.timeout + if global_args.timeout * 2 + else int(os.getenv("TIMEOUT", 150 * 2)) ) # Keepalive configuration @@ -157,7 +169,9 @@ def main(): else os.getenv("SSL_CERTFILE") ) gunicorn_config.keyfile = ( - global_args.ssl_keyfile if global_args.ssl_keyfile else os.getenv("SSL_KEYFILE") + global_args.ssl_keyfile + if global_args.ssl_keyfile + else os.getenv("SSL_KEYFILE") ) # Set configuration options from the module