release: 完成了mvp,具有基本的模型训练,语料生成和推理功能

This commit is contained in:
carry
2025-03-26 18:08:15 +08:00
parent 22438d65d3
commit f882b82e57
7 changed files with 1816 additions and 0 deletions

15
config/llm.py Normal file
View 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")
}
}