offerpai/offerpai_python_ai
6
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

修改模型供应商

Browse Source
This commit is contained in:
zk
2026-04-14 18:37:47 +08:00
parent 6609fedaa1
commit 1acc4e89c9
2 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/ai/models.py
+6 -4
View File
@@ -16,6 +16,8 @@ from app.config import settings
# 供应商连接配置
_VOLCENGINE = (lambda: settings.volcengine_api_key, lambda: settings.volcengine_base_url)
_CARDIAC = (lambda: settings.cardiacBrder_api_key, lambda: settings.cardiacBrder_base_url)
_JIEKOU = (lambda: settings.jiekou_api_key, lambda: settings.jiekou_base_url)
class LLM(Enum):
@@ -30,10 +32,10 @@ class LLM(Enum):
DOUBAO_SEED_PRO = ("doubao-seed-2-0-pro-260215", *_VOLCENGINE)
# 心缘
GPT_4O = ("gpt-4o", *_CARDIAC)
GPT_4O_MINI = ("gpt-4o-mini", *_CARDIAC)
CLAUDE_SONNET_4 = ("claude-sonnet-4-20250514", *_CARDIAC)
GEMINI_FLASH = ("gemini-2.5-flash", *_CARDIAC)
GPT_4O = ("gpt-4o", *_JIEKOU)
GPT_4O_MINI = ("gpt-4o-mini", *_JIEKOU)
CLAUDE_SONNET_4 = ("claude-sonnet-4-20250514", *_JIEKOU)
GEMINI_FLASH = ("gemini-2.5-flash", *_JIEKOU)
def __init__(self, model_name: str, api_key_fn, base_url_fn):
self.model_name = model_name