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

Fix linting

Browse source
This commit is contained in:
yangdx 2025-03-02 00:13:11 +08:00
parent f76cf98dbd
commit 8d6960f280
2 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
lightrag/api/lightrag_server.py
View file

@ -510,13 +510,14 @@ def check_and_install_dependencies():
"fastapi", "fastapi",
# Add other required packages here # Add other required packages here
] ]
for package in required_packages: for package in required_packages:
if not pm.is_installed(package): if not pm.is_installed(package):
print(f"Installing {package}...") print(f"Installing {package}...")
pm.install(package) pm.install(package)
print(f"{package} installed successfully") print(f"{package} installed successfully")
def main(): def main():
# Check if running under Gunicorn # Check if running under Gunicorn
if "GUNICORN_CMD_ARGS" in os.environ: if "GUNICORN_CMD_ARGS" in os.environ:

5
run_with_gunicorn.py
View file

@ -10,6 +10,7 @@ import pipmaster as pm
from lightrag.api.utils_api import parse_args, display_splash_screen from lightrag.api.utils_api import parse_args, display_splash_screen
from lightrag.kg.shared_storage import initialize_share_data, finalize_share_data from lightrag.kg.shared_storage import initialize_share_data, finalize_share_data
def check_and_install_dependencies(): def check_and_install_dependencies():
"""Check and install required dependencies""" """Check and install required dependencies"""
required_packages = [ required_packages = [
@ -17,7 +18,7 @@ def check_and_install_dependencies():
"tiktoken", "tiktoken",
# Add other required packages here # Add other required packages here
] ]
for package in required_packages: for package in required_packages:
if not pm.is_installed(package): if not pm.is_installed(package):
print(f"Installing {package}...") print(f"Installing {package}...")
@ -42,7 +43,7 @@ def signal_handler(sig, frame):
def main(): def main():
# Check and install dependencies # Check and install dependencies
check_and_install_dependencies() check_and_install_dependencies()
# Register signal handlers for graceful shutdown # Register signal handlers for graceful shutdown
signal.signal(signal.SIGINT, signal_handler) # Ctrl+C signal.signal(signal.SIGINT, signal_handler) # Ctrl+C
signal.signal(signal.SIGTERM, signal_handler) # kill command signal.signal(signal.SIGTERM, signal_handler) # kill command