yangdx
commited on
Commit
·
6c89115
1
Parent(s):
0cfee1e
Fix linting
Browse files- lightrag/api/lightrag_server.py +2 -1
- run_with_gunicorn.py +3 -2
lightrag/api/lightrag_server.py
CHANGED
@@ -510,13 +510,14 @@ def check_and_install_dependencies():
|
|
510 |
"fastapi",
|
511 |
# Add other required packages here
|
512 |
]
|
513 |
-
|
514 |
for package in required_packages:
|
515 |
if not pm.is_installed(package):
|
516 |
print(f"Installing {package}...")
|
517 |
pm.install(package)
|
518 |
print(f"{package} installed successfully")
|
519 |
|
|
|
520 |
def main():
|
521 |
# Check if running under Gunicorn
|
522 |
if "GUNICORN_CMD_ARGS" in os.environ:
|
|
|
510 |
"fastapi",
|
511 |
# Add other required packages here
|
512 |
]
|
513 |
+
|
514 |
for package in required_packages:
|
515 |
if not pm.is_installed(package):
|
516 |
print(f"Installing {package}...")
|
517 |
pm.install(package)
|
518 |
print(f"{package} installed successfully")
|
519 |
|
520 |
+
|
521 |
def main():
|
522 |
# Check if running under Gunicorn
|
523 |
if "GUNICORN_CMD_ARGS" in os.environ:
|
run_with_gunicorn.py
CHANGED
@@ -10,6 +10,7 @@ import pipmaster as pm
|
|
10 |
from lightrag.api.utils_api import parse_args, display_splash_screen
|
11 |
from lightrag.kg.shared_storage import initialize_share_data, finalize_share_data
|
12 |
|
|
|
13 |
def check_and_install_dependencies():
|
14 |
"""Check and install required dependencies"""
|
15 |
required_packages = [
|
@@ -17,7 +18,7 @@ def check_and_install_dependencies():
|
|
17 |
"tiktoken",
|
18 |
# Add other required packages here
|
19 |
]
|
20 |
-
|
21 |
for package in required_packages:
|
22 |
if not pm.is_installed(package):
|
23 |
print(f"Installing {package}...")
|
@@ -42,7 +43,7 @@ def signal_handler(sig, frame):
|
|
42 |
def main():
|
43 |
# Check and install dependencies
|
44 |
check_and_install_dependencies()
|
45 |
-
|
46 |
# Register signal handlers for graceful shutdown
|
47 |
signal.signal(signal.SIGINT, signal_handler) # Ctrl+C
|
48 |
signal.signal(signal.SIGTERM, signal_handler) # kill command
|
|
|
10 |
from lightrag.api.utils_api import parse_args, display_splash_screen
|
11 |
from lightrag.kg.shared_storage import initialize_share_data, finalize_share_data
|
12 |
|
13 |
+
|
14 |
def check_and_install_dependencies():
|
15 |
"""Check and install required dependencies"""
|
16 |
required_packages = [
|
|
|
18 |
"tiktoken",
|
19 |
# Add other required packages here
|
20 |
]
|
21 |
+
|
22 |
for package in required_packages:
|
23 |
if not pm.is_installed(package):
|
24 |
print(f"Installing {package}...")
|
|
|
43 |
def main():
|
44 |
# Check and install dependencies
|
45 |
check_and_install_dependencies()
|
46 |
+
|
47 |
# Register signal handlers for graceful shutdown
|
48 |
signal.signal(signal.SIGINT, signal_handler) # Ctrl+C
|
49 |
signal.signal(signal.SIGTERM, signal_handler) # kill command
|