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