pdf / init_config.py
megatrump's picture
使用测试分支的代码, 能够支持密钥隐藏的功能
cafe2bc
raw
history blame contribute delete
687 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",
}
}
],
"ENABLED_SERVICES": [
"OpenAI",
],
"HIDDEN_GRADIO_DETAILS": True,
}
with open("config.json", "w") as f:
json.dump(config, f)