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

添加模型

Browse Source
This commit is contained in:
zk
2026-05-08 16:00:12 +08:00
parent eea7c33639
commit f14f07affa
3 changed files with 19 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/ai/models.py
+10
View File
@@ -18,6 +18,7 @@ _VOLCENGINE = (lambda: settings.volcengine_api_key, lambda: settings.volcengine_
_JIAYU = (lambda: settings.jiayu_api_key, lambda: settings.jiayu_base_url)
_JIEKOU = (lambda: settings.jiekou_api_key, lambda: settings.jiekou_base_url)
_ZM = (lambda: settings.zm_api_key, lambda: settings.zm_base_url)
class LLM(Enum):
@@ -47,6 +48,15 @@ class LLM(Enum):
JIAYU_MINIMAX_M2_5 = ("minimax-m2.5", *_JIAYU)
# ZM
ZM_GPT_5_5 = ("gpt-5.5", *_ZM)
ZM_GPT_5_4 = ("gpt-5.4", *_ZM)
ZM_GPT_5_4_MINI = ("gpt-5.4-mini", *_ZM)
ZM_GPT_5_2 = ("gpt-5.2", *_ZM)
ZM_GPT_4O = ("gpt-5.5", *_ZM)
def __init__(self, model_name: str, api_key_fn, base_url_fn):
self.model_name = model_name
self._api_key_fn = api_key_fn