浏览代码

feat: add LITELLM_PROXY_HOST to configure address litellm listens on

Jun Siang Cheah 1 年之前
父节点
当前提交
9e9306fd2b
共有 2 个文件被更改,包括 3 次插入1 次删除
  1. 2 1
      backend/apps/litellm/main.py
  2. 1 0
      backend/config.py

+ 2 - 1
backend/apps/litellm/main.py

@@ -28,6 +28,7 @@ from config import (
     MODEL_FILTER_LIST,
     MODEL_FILTER_LIST,
     DATA_DIR,
     DATA_DIR,
     LITELLM_PROXY_PORT,
     LITELLM_PROXY_PORT,
+    LITELLM_PROXY_HOST,
 )
 )
 
 
 from litellm.utils import get_llm_provider
 from litellm.utils import get_llm_provider
@@ -95,7 +96,7 @@ async def run_background_process(command):
 async def start_litellm_background():
 async def start_litellm_background():
     log.info("start_litellm_background")
     log.info("start_litellm_background")
     # Command to run in the background
     # Command to run in the background
-    command = f"litellm --port {LITELLM_PROXY_PORT} --telemetry False --config ./data/litellm/config.yaml"
+    command = f"litellm --port {LITELLM_PROXY_PORT} --host {LITELLM_PROXY_HOST} --telemetry False --config ./data/litellm/config.yaml"
 
 
     await run_background_process(command)
     await run_background_process(command)
 
 

+ 1 - 0
backend/config.py

@@ -507,3 +507,4 @@ AUDIO_OPENAI_API_KEY = os.getenv("AUDIO_OPENAI_API_KEY", OPENAI_API_KEY)
 LITELLM_PROXY_PORT = int(os.getenv("LITELLM_PROXY_PORT", "14365"))
 LITELLM_PROXY_PORT = int(os.getenv("LITELLM_PROXY_PORT", "14365"))
 if LITELLM_PROXY_PORT < 0 or LITELLM_PROXY_PORT > 65535:
 if LITELLM_PROXY_PORT < 0 or LITELLM_PROXY_PORT > 65535:
     raise ValueError("Invalid port number for LITELLM_PROXY_PORT")
     raise ValueError("Invalid port number for LITELLM_PROXY_PORT")
+LITELLM_PROXY_HOST = os.getenv("LITELLM_PROXY_HOST", "127.0.0.1")