39 lines
794 B
Python
39 lines
794 B
Python
"""
|
|
简单配置管理
|
|
"""
|
|
import os
|
|
from typing import Literal
|
|
from dotenv import load_dotenv
|
|
|
|
# 加载 .env 文件
|
|
load_dotenv()
|
|
|
|
# 模型配置
|
|
MODEL_CONFIG = {
|
|
"gpt-3.5-turbo": {
|
|
"input_cost_per_1k": 0.0005,
|
|
"output_cost_per_1k": 0.0015,
|
|
"max_tokens": 4096,
|
|
},
|
|
"gpt-4o-mini": {
|
|
"input_cost_per_1k": 0.00015,
|
|
"output_cost_per_1k": 0.0006,
|
|
"max_tokens": 128000,
|
|
},
|
|
"gpt-4o": {
|
|
"input_cost_per_1k": 0.005,
|
|
"output_cost_per_1k": 0.015,
|
|
"max_tokens": 128000,
|
|
},
|
|
}
|
|
|
|
# 路由阈值
|
|
ROUTING_THRESHOLDS = {
|
|
"simple": 100, # < 100 tokens -> gpt-3.5-turbo
|
|
"medium": 500, # < 500 tokens -> gpt-4o-mini
|
|
# >= 500 tokens -> gpt-4o
|
|
}
|
|
|
|
# API Key
|
|
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")
|