|
|
| """基于环境变量生成 Toolify 配置文件。"""
|
|
|
| import os
|
| import sys
|
| from pathlib import Path
|
| from typing import Any, List
|
|
|
| import yaml
|
|
|
|
|
| def _str_to_bool(value: str | None, default: bool) -> bool:
|
| """将字符串环境变量解析为布尔值。"""
|
| if value is None:
|
| return default
|
| normalized = value.strip().lower()
|
| if normalized in {"1", "true", "yes", "y", "on"}:
|
| return True
|
| if normalized in {"0", "false", "no", "n", "off"}:
|
| return False
|
| return default
|
|
|
|
|
| def _parse_list(env_value: str | None) -> List[str]:
|
| """将逗号分隔的环境变量解析为字符串列表。"""
|
| if not env_value:
|
| return []
|
| return [item.strip() for item in env_value.split(",") if item.strip()]
|
|
|
|
|
| def _load_config_from_env() -> Any:
|
| """尝试从直接提供的 YAML 或 JSON 环境变量加载配置。"""
|
| direct_yaml = os.getenv("TOOLIFY_CONFIG_YAML")
|
| if direct_yaml:
|
| return yaml.safe_load(direct_yaml)
|
|
|
| direct_json = os.getenv("TOOLIFY_CONFIG_JSON")
|
| if direct_json:
|
| return yaml.safe_load(direct_json)
|
|
|
| return None
|
|
|
|
|
| def _build_default_config() -> dict:
|
| """根据基础环境变量构造最小可用配置。"""
|
| openai_api_key = os.getenv("OPENAI_API_KEY")
|
| allowed_keys = _parse_list(os.getenv("CLIENT_ALLOWED_KEYS"))
|
|
|
| if not openai_api_key:
|
| raise RuntimeError("缺少 OPENAI_API_KEY 环境变量，无法生成配置。")
|
|
|
| if not allowed_keys:
|
| raise RuntimeError("缺少 CLIENT_ALLOWED_KEYS 环境变量，无法生成配置。")
|
|
|
| base_url = os.getenv("OPENAI_BASE_URL", "https://api.openai.com/v1").rstrip("/")
|
| models = _parse_list(os.getenv("OPENAI_MODELS", "gpt-4o-mini,gpt-4o"))
|
| if not models:
|
| raise RuntimeError("OPENAI_MODELS 环境变量解析为空，请至少指定一个模型。")
|
|
|
| server_port = int(os.getenv("SERVER_PORT", os.getenv("PORT", "8000")))
|
| server_host = os.getenv("SERVER_HOST", "0.0.0.0")
|
| server_timeout = int(os.getenv("SERVER_TIMEOUT", "180"))
|
|
|
| features_log_level = os.getenv("FEATURE_LOG_LEVEL", "INFO").upper()
|
|
|
| prompt_template = os.getenv("FEATURE_PROMPT_TEMPLATE")
|
| if prompt_template:
|
| prompt_template = prompt_template.replace("\\n", "\n")
|
|
|
| return {
|
| "server": {
|
| "port": server_port,
|
| "host": server_host,
|
| "timeout": server_timeout,
|
| },
|
| "upstream_services": [
|
| {
|
| "name": os.getenv("OPENAI_NAME", "openai"),
|
| "base_url": base_url,
|
| "api_key": openai_api_key,
|
| "description": os.getenv("OPENAI_DESCRIPTION", "OpenAI Service"),
|
| "is_default": True,
|
| "models": models,
|
| }
|
| ],
|
| "client_authentication": {
|
| "allowed_keys": allowed_keys,
|
| },
|
| "features": {
|
| "enable_function_calling": _str_to_bool(os.getenv("FEATURE_ENABLE_FUNCTION_CALLING"), True),
|
| "log_level": features_log_level if features_log_level else "INFO",
|
| "convert_developer_to_system": _str_to_bool(os.getenv("FEATURE_CONVERT_DEVELOPER_TO_SYSTEM"), True),
|
| "prompt_template": prompt_template,
|
| "key_passthrough": _str_to_bool(os.getenv("FEATURE_KEY_PASSTHROUGH"), False),
|
| "model_passthrough": _str_to_bool(os.getenv("FEATURE_MODEL_PASSTHROUGH"), False),
|
| },
|
| }
|
|
|
|
|
| def main() -> None:
|
| """入口函数：生成配置文件。"""
|
| config_path = Path(os.getenv("CONFIG_PATH", "/app/config.yaml"))
|
|
|
| if config_path.exists():
|
| print(f"配置文件已存在，跳过生成: {config_path}")
|
| return
|
|
|
| config_obj = _load_config_from_env()
|
| if config_obj is None:
|
| config_obj = _build_default_config()
|
|
|
| if not isinstance(config_obj, dict):
|
| raise ValueError("环境变量提供的配置必须是字典结构。")
|
|
|
| config_path.parent.mkdir(parents=True, exist_ok=True)
|
| config_yaml = yaml.safe_dump(config_obj, sort_keys=False, allow_unicode=True)
|
| config_path.write_text(config_yaml, encoding="utf-8")
|
| print(f"已生成配置文件: {config_path}")
|
|
|
|
|
| if __name__ == "__main__":
|
| try:
|
| main()
|
| except Exception as exc:
|
| print(f"生成配置文件时出现错误: {exc}", file=sys.stderr)
|
| sys.exit(1)
|
|
|
