yangdx commited on
Commit
6c89115
·
1 Parent(s): 0cfee1e

Fix linting

Browse files
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