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

Fix linting

Browse Source
This commit is contained in:
yangdx
2025-04-04 21:34:13 +08:00
parent 27d478385d
commit caac0479d3
3 changed files with 23 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lightrag/api/config.py
View File

@@ -12,6 +12,7 @@ from dotenv import load_dotenv
# the OS environment variables take precedence over the .env file # the OS environment variables take precedence over the .env file
load_dotenv(dotenv_path=".env", override=False) load_dotenv(dotenv_path=".env", override=False)
class OllamaServerInfos: class OllamaServerInfos:
# Constants for emulated Ollama model information # Constants for emulated Ollama model information
LIGHTRAG_NAME = "lightrag" LIGHTRAG_NAME = "lightrag"
@@ -302,7 +303,7 @@ def parse_args() -> argparse.Namespace:
# Select Document loading tool (DOCLING, DEFAULT) # Select Document loading tool (DOCLING, DEFAULT)
args.document_loading_engine = get_env_value("DOCUMENT_LOADING_ENGINE", "DEFAULT") args.document_loading_engine = get_env_value("DOCUMENT_LOADING_ENGINE", "DEFAULT")
# Add environment variables that were previously read directly # Add environment variables that were previously read directly
args.cors_origins = get_env_value("CORS_ORIGINS", "*") args.cors_origins = get_env_value("CORS_ORIGINS", "*")
args.summary_language = get_env_value("SUMMARY_LANGUAGE", "en") args.summary_language = get_env_value("SUMMARY_LANGUAGE", "en")

8
lightrag/api/lightrag_server.py
View File

@@ -321,9 +321,7 @@ def create_app(args):
# namespace_prefix=args.namespace_prefix, # namespace_prefix=args.namespace_prefix,
auto_manage_storages_states=False, auto_manage_storages_states=False,
max_parallel_insert=args.max_parallel_insert, max_parallel_insert=args.max_parallel_insert,
addon_params={ addon_params={"language": args.summary_language},
"language": args.summary_language
},
) )
else: # azure_openai else: # azure_openai
rag = LightRAG( rag = LightRAG(
@@ -354,9 +352,7 @@ def create_app(args):
# namespace_prefix=args.namespace_prefix, # namespace_prefix=args.namespace_prefix,
auto_manage_storages_states=False, auto_manage_storages_states=False,
max_parallel_insert=args.max_parallel_insert, max_parallel_insert=args.max_parallel_insert,
addon_params={ addon_params={"language": args.summary_language},
"language": args.summary_language
},
) )
# Add routes # Add routes

24
lightrag/api/run_with_gunicorn.py
View File

@@ -120,12 +120,22 @@ def main():
# Set configuration variables in gunicorn_config, prioritizing command line arguments # Set configuration variables in gunicorn_config, prioritizing command line arguments
gunicorn_config.workers = ( 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 # 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") host = (
port = global_args.port if global_args.port != 9621 else int(os.getenv("PORT", 9621)) 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}" gunicorn_config.bind = f"{host}:{port}"
# Log level configuration prioritizes command line arguments # Log level configuration prioritizes command line arguments
@@ -137,7 +147,9 @@ def main():
# Timeout configuration prioritizes command line arguments # Timeout configuration prioritizes command line arguments
gunicorn_config.timeout = ( 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 # Keepalive configuration
@@ -157,7 +169,9 @@ def main():
else os.getenv("SSL_CERTFILE") else os.getenv("SSL_CERTFILE")
) )
gunicorn_config.keyfile = ( 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 # Set configuration options from the module