pdf / init_config.py
megatrump's picture
添加了后台运行的功能以及密钥设置
506398c
raw
history blame
600 Bytes
import os
import json
OPENAI_API_KEY: str = os.environ.get("OPENAI_API_KEY", "")
if not OPENAI_API_KEY:
print("You Should Set OPENAI_API_KEY Enviroment.")
exit(-1)
config = {
"USE_MODELSCOPE": "0",
"translators": [
{
"name": "openai",
"envs": {
# 设置环境变量
"OPENAI_BASE_URL": "https://megatrump-oneapi.hf.space/v1",
"OPENAI_API_KEY": OPENAI_API_KEY,
"OPENAI_MODEL": "gpt-4o-mini",
}
}
]
}
with open("config.json", "w") as f:
json.dump(config, f)