kaiyuan/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-06-02 00:57:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(models): add baidu ernie-5.1

Browse Source
This commit is contained in:
zhayujie
2026-05-10 18:39:38 +08:00
parent fb341b869b
commit 907825601d
15 changed files with 53 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
models/qianfan/qianfan_bot.py
View File

@@ -14,12 +14,13 @@ from models.session_manager import SessionManager
from .qianfan_session import QianfanSession
DEFAULT_API_BASE = "https://qianfan.baidubce.com/v2"
DEFAULT_MODEL = const.ERNIE_5
DEFAULT_MODEL = const.ERNIE_5_1
DEFAULT_VISION_MODEL = const.ERNIE_45_TURBO_VL
# Qianfan models that natively understand images. Other ERNIE variants
# are text-only and must not receive image payloads.
_VISION_CAPABLE_MODELS = {
const.ERNIE_5_1,
const.ERNIE_5,
const.ERNIE_X1_1,
const.ERNIE_45_TURBO_VL,

2
models/qianfan/qianfan_session.py
View File

@@ -3,7 +3,7 @@ from common.log import logger
class QianfanSession(Session):
def __init__(self, session_id, system_prompt=None, model="ernie-5.0"):
def __init__(self, session_id, system_prompt=None, model="ernie-5.1"):
super().__init__(session_id, system_prompt)
self.model = model
self.reset()