shen/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-01-19 09:41:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update model list

Browse Source
This commit is contained in:
6vision
2024-06-05 00:44:08 +08:00
parent 405372d1a7
commit 7a2c15d912
4 changed files with 51 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
bot/baidu/baidu_wenxin.py
View File

@@ -1,6 +1,8 @@
# encoding:utf-8
import requests, json
import requests
import json
from common import const
from bot.bot import Bot
from bot.session_manager import SessionManager
from bridge.context import ContextType
@@ -16,9 +18,12 @@ class BaiduWenxinBot(Bot):
def __init__(self):
super().__init__()
wenxin_model = conf().get("baidu_wenxin_model") or "eb-instant"
if conf().get("model") and conf().get("model") == "wenxin-4":
wenxin_model = conf().get("baidu_wenxin_model") or "completions"
if conf().get("model") and conf().get("model") == const.WEN_XIN:
wenxin_model = "completions"
if conf().get("model") and conf().get("model") == const.WEN_XIN_4:
wenxin_model = "completions_pro"
self.sessions = SessionManager(BaiduWenxinSession, model=wenxin_model)
def reply(self, query, context=None):