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

feat: add config for model selection #471

Browse Source
This commit is contained in:
zhayujie
2023-03-15 23:27:51 +08:00
parent b404e2c51f
commit 3c04325aae
9 changed files with 54 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
bridge/bridge.py
View File

@@ -1,5 +1,7 @@
from bot import bot_factory
from voice import voice_factory
from config import conf
from common import const
class Bridge(object):
@@ -7,7 +9,13 @@ class Bridge(object):
pass
def fetch_reply_content(self, query, context):
return bot_factory.create_bot("chatGPT").reply(query, context)
bot_type = const.CHATGPT
model_type = conf().get("model")
if model_type in ["gpt-3.5-turbo", "gpt-4", "gpt-4-32k"]:
bot_type = const.CHATGPT
elif model_type in ["text-davinci-003"]:
bot_type = const.OPEN_AI
return bot_factory.create_bot(bot_type).reply(query, context)
def fetch_voice_to_text(self, voiceFile):
return voice_factory.create_voice("openai").voiceToText(voiceFile)