Files
chatgpt-on-wechat/bridge/bridge.py
2023-03-15 23:27:51 +08:00

24 lines
787 B
Python

from bot import bot_factory
from voice import voice_factory
from config import conf
from common import const
class Bridge(object):
def __init__(self):
pass
def fetch_reply_content(self, 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)
def fetch_text_to_voice(self, text):
return voice_factory.create_voice("baidu").textToVoice(text)