71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
"""一键配置 BaoDan API Key 到 docker-compose.baodan.yml。
|
||
|
||
使用方式：
|
||
python scripts/setup_bodan_api_keys.py
|
||
|
||
会提示输入三个 API Key，自动更新 docker-compose 配置。
|
||
"""
|
||
import re
|
||
import sys
|
||
|
||
COMPOSE_FILE = "docker-compose.baodan.yml"
|
||
|
||
|
||
def update_env(compose_path: str, key: str, value: str):
|
||
"""更新 docker-compose 中的环境变量。"""
|
||
with open(compose_path, "r", encoding="utf-8") as f:
|
||
content = f.read()
|
||
|
||
# 匹配 KEY: "value" 或 KEY: "placeholder-..."
|
||
pattern = rf'({re.escape(key)}:\s*")[^"]*(")'
|
||
replacement = rf'\g<1>{value}\2'
|
||
|
||
new_content, count = re.subn(pattern, replacement, content, count=1)
|
||
if count == 0:
|
||
print(f" [WARN] 未找到 {key}，跳过")
|
||
return False
|
||
|
||
with open(compose_path, "w", encoding="utf-8") as f:
|
||
f.write(new_content)
|
||
return True
|
||
|
||
|
||
def main():
|
||
print("=" * 50)
|
||
print(" BaoDan API Key 配置工具")
|
||
print("=" * 50)
|
||
print()
|
||
print("请在后台创建应用后，输入对应的 API Key。")
|
||
print("如果暂时没有，可以直接回车跳过（用占位值）。")
|
||
print()
|
||
|
||
keys = {
|
||
"BAODAN_CHAT_API_KEY": "对话应用 API Key（用于智能问答）",
|
||
"BAODAN_WORKFLOW_API_KEY": "Workflow 应用 API Key（用于产品推荐）",
|
||
"BAODAN_KB_API_KEY": "知识库 API Key（用于文档管理）",
|
||
}
|
||
|
||
compose_path = COMPOSE_FILE
|
||
updated = 0
|
||
|
||
for env_key, desc in keys.items():
|
||
value = input(f" {desc}\n > {env_key}: ").strip()
|
||
if value:
|
||
if update_env(compose_path, env_key, value):
|
||
print(f" ✓ {env_key} 已更新")
|
||
updated += 1
|
||
else:
|
||
print(f" - 跳过 {env_key}")
|
||
|
||
print()
|
||
if updated > 0:
|
||
print(f"完成！已更新 {updated} 个 API Key。")
|
||
print("请运行以下命令重启服务：")
|
||
print(" docker compose -f docker-compose.baodan.yml restart baodan-api baodan-worker")
|
||
else:
|
||
print("未更新任何配置。")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|