Fix linting

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

View File

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

View File

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