release: 完成了mvp,具有基本的模型训练,语料生成和推理功能
This commit is contained in:
15
config/llm.py
Normal file
15
config/llm.py
Normal file
@@ -0,0 +1,15 @@
|
||||
import os
|
||||
from dotenv import load_dotenv
|
||||
from typing import Dict, Any
|
||||
|
||||
def load_config() -> Dict[str, Any]:
|
||||
"""从.env文件加载配置"""
|
||||
load_dotenv()
|
||||
|
||||
return {
|
||||
"openai": {
|
||||
"api_key": os.getenv("OPENAI_API_KEY"),
|
||||
"base_url": os.getenv("OPENAI_BASE_URL"),
|
||||
"model_id": os.getenv("OPENAI_MODEL_ID")
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user