mirror of
https://github.com/zhayujie/bot-on-anything.git
synced 2026-02-02 09:38:26 +08:00
26 lines
565 B
Python
26 lines
565 B
Python
"""
|
|
channel factory
|
|
"""
|
|
|
|
from common import const
|
|
|
|
def create_bot(model_type):
|
|
"""
|
|
create a channel instance
|
|
:param channel_type: channel type code
|
|
:return: channel instance
|
|
"""
|
|
|
|
if model_type == const.OPEN_AI:
|
|
# OpenAI 官方对话模型API (gpt-3.0)
|
|
from model.openai.open_ai_model import OpenAIModel
|
|
return OpenAIModel()
|
|
|
|
elif model_type == const.CHATGPT:
|
|
# ChatGPT API (gpt-3.5-turbo)
|
|
from model.chatgpt.chatgpt_model import ChatGPTModel
|
|
return ChatGPTModel()
|
|
|
|
raise RuntimeError
|
|
|