Support downloading configuration files from external URLs.

README.md

@@ -92,6 +92,8 @@ services:
   uni-api:
     container_name: uni-api
     image: yym68686/uni-api:latest
+    environment:
+      - CONFIG_URL=http://file_url/api.yaml
     ports:
       - 8001:8000
     volumes:
@@ -120,6 +122,7 @@ set -eu
 docker pull yym68686/uni-api:latest
 docker rm -f uni-api
 docker run --user root -p 8001:8000 -dit --name uni-api \
+-e CONFIG_URL=http://file_url/api.yaml \
 -v ./api.yaml:/home/api.yaml \
 yym68686/uni-api:latest
 docker logs -f uni-api

docker-compose.yml

@@ -2,6 +2,8 @@ services:
   uni-api:
     container_name: uni-api
     image: yym68686/uni-api:latest
+    environment:
+      - CONFIG_URL=http://file_url/api.yaml
     ports:
       - 8001:8000
     volumes:

main.py

@@ -7,7 +7,7 @@ from contextlib import asynccontextmanager
 from fastapi import FastAPI, HTTPException, Depends
 
 from models import RequestModel
-from utils import config, api_keys_db, api_list, error_handling_wrapper, get_all_models, verify_api_key, post_all_models
+from utils import config, api_keys_db, api_list, error_handling_wrapper, get_all_models, verify_api_key, post_all_models, update_config
 from request import get_payload
 from response import fetch_response, fetch_response_stream
 
@@ -21,6 +21,21 @@ async def lifespan(app: FastAPI):
     # 启动时的代码
     timeout = httpx.Timeout(connect=15.0, read=10.0, write=30.0, pool=30.0)
     app.state.client = httpx.AsyncClient(timeout=timeout)
+    import os
+    import json
+    # 新增： 从环境变量获取配置URL并拉取配置
+    config_url = os.environ.get('CONFIG_URL')
+    if config_url:
+        try:
+            response = await app.state.client.get(config_url)
+            response.raise_for_status()
+            config_data = json.loads(response.text)
+            # 更新配置
+            global config, api_keys_db, api_list
+            config, api_keys_db, api_list = update_config(config_data)
+        except Exception as e:
+            logger.error(f"Error fetching or parsing config from {config_url}: {str(e)}")
+
     yield
     # 关闭时的代码
     await app.state.client.aclose()

utils.py

@@ -5,25 +5,28 @@ from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
 
 from log_config import logger
 
+def update_config(config_data):
+    for index, provider in enumerate(config_data['providers']):
+        model_dict = {}
+        for model in provider['model']:
+            if type(model) == str:
+                model_dict[model] = model
+            if type(model) == dict:
+                model_dict.update({new: old for old, new in model.items()})
+                model_dict.update({old: old for old, new in model.items()})
+        provider['model'] = model_dict
+        config_data['providers'][index] = provider
+    api_keys_db = config_data['api_keys']
+    api_list = [item["api"] for item in api_keys_db]
+    # logger.info(json.dumps(conf, indent=4, ensure_ascii=False))
+    return config_data, api_keys_db, api_list
+
 # 读取YAML配置文件
 def load_config():
     try:
         with open('./api.yaml', 'r') as f:
             conf = yaml.safe_load(f)
-            for index, provider in enumerate(conf['providers']):
-                model_dict = {}
-                for model in provider['model']:
-                    if type(model) == str:
-                        model_dict[model] = model
-                    if type(model) == dict:
-                        model_dict.update({new: old for old, new in model.items()})
-                        model_dict.update({old: old for old, new in model.items()})
-                provider['model'] = model_dict
-                conf['providers'][index] = provider
-            api_keys_db = conf['api_keys']
-            api_list = [item["api"] for item in api_keys_db]
-            # logger.info(json.dumps(conf, indent=4, ensure_ascii=False))
-            return conf, api_keys_db, api_list
+            return update_config(conf)
     except FileNotFoundError:
         logger.error("配置文件 'api.yaml' 未找到。请确保文件存在于正确的位置。")
         return [], [], []