Browse Source

feat: add wenxin 4.0 model

master
zhayujie 1 year ago
parent
commit
49446d4872
3 changed files with 7 additions and 6 deletions
  1. +4
    -1
      bot/baidu/baidu_wenxin.py
  2. +1
    -1
      bridge/bridge.py
  3. +2
    -4
      common/const.py

+ 4
- 1
bot/baidu/baidu_wenxin.py View File

@@ -16,7 +16,10 @@ class BaiduWenxinBot(Bot):

def __init__(self):
super().__init__()
self.sessions = SessionManager(BaiduWenxinSession, model=conf().get("baidu_wenxin_model") or "eb-instant")
wenxin_model = conf().get("baidu_wenxin_model") or "eb-instant"
if conf().get("model") and conf().get("model") == "wenxin-4":
wenxin_model = "completions_pro"
self.sessions = SessionManager(BaiduWenxinSession, model=wenxin_model)

def reply(self, query, context=None):
# acquire reply content


+ 1
- 1
bridge/bridge.py View File

@@ -23,7 +23,7 @@ class Bridge(object):
self.btype["chat"] = const.OPEN_AI
if conf().get("use_azure_chatgpt", False):
self.btype["chat"] = const.CHATGPTONAZURE
if model_type in ["wenxin"]:
if model_type in ["wenxin", "wenxin-4"]:
self.btype["chat"] = const.BAIDU
if model_type in ["xunfei"]:
self.btype["chat"] = const.XUNFEI


+ 2
- 4
common/const.py View File

@@ -5,8 +5,6 @@ BAIDU = "baidu"
XUNFEI = "xunfei"
CHATGPTONAZURE = "chatGPTOnAzure"
LINKAI = "linkai"

VERSION = "1.3.0"

CLAUDEAI = "claude"
MODEL_LIST = ["gpt-3.5-turbo", "gpt-3.5-turbo-16k", "gpt-4", "wenxin", "xunfei","claude"]

MODEL_LIST = ["gpt-3.5-turbo", "gpt-3.5-turbo-16k", "gpt-4", "wenxin", "wenxin-4", "xunfei", "claude"]

Loading…
Cancel
Save