添加模型

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
+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) _JIAYU = (lambda: settings.jiayu_api_key, lambda: settings.jiayu_base_url)
_JIEKOU = (lambda: settings.jiekou_api_key, lambda: settings.jiekou_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): class LLM(Enum):
@@ -47,6 +48,15 @@ class LLM(Enum):
JIAYU_MINIMAX_M2_5 = ("minimax-m2.5", *_JIAYU) 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): def __init__(self, model_name: str, api_key_fn, base_url_fn):
self.model_name = model_name self.model_name = model_name
self._api_key_fn = api_key_fn self._api_key_fn = api_key_fn
+2 -2
View File
@@ -14,14 +14,14 @@ from app.tool.json_helper import parse_llm_json
# 诊断链(StrOutputParser 拿原始文本,再手动解析 JSON,避免 markdown 代码块导致解析失败) # 诊断链(StrOutputParser 拿原始文本,再手动解析 JSON,避免 markdown 代码块导致解析失败)
_diagnose_chain = ( _diagnose_chain = (
ChatPromptTemplate.from_messages([("system", DIAGNOSE_MODULE_PROMPT), ("human", "请开始诊断。")]) ChatPromptTemplate.from_messages([("system", DIAGNOSE_MODULE_PROMPT), ("human", "请开始诊断。")])
| LLM.JIAYU_CLAUDE_SONNET_4_5.create(temperature=0) | LLM.ZM_GPT_5_5.create(temperature=0)
| StrOutputParser() | StrOutputParser()
) )
# 汇总评价链(纯文本输出) # 汇总评价链(纯文本输出)
_summary_chain = ( _summary_chain = (
ChatPromptTemplate.from_messages([("system", SUMMARY_PROMPT), ("human", "请生成整体评价。")]) ChatPromptTemplate.from_messages([("system", SUMMARY_PROMPT), ("human", "请生成整体评价。")])
| LLM.JIAYU_CLAUDE_SONNET_4_5.create(temperature=0.3) | LLM.ZM_GPT_5_5.create(temperature=0.3)
| StrOutputParser() | StrOutputParser()
) )
+7
View File
@@ -46,6 +46,13 @@ class Settings(BaseSettings):
jiayu_api_key: str = "Aa123123" jiayu_api_key: str = "Aa123123"
jiayu_base_url: str = "http://192.168.31.133:8800/v1" jiayu_base_url: str = "http://192.168.31.133:8800/v1"
#zm
zm_api_key: str = "3-chatgpt-is-free-free-free"
zm_base_url: str = "http://re.94xy.cn/v1"
# JWT # JWT
jwt_secret: str = "Aa123123" jwt_secret: str = "Aa123123"
token_expire_seconds: int = 5184000 token_expire_seconds: int = 5184000