@@ -16,7 +16,8 @@ else: | |||||
class ChatGPTBot(Bot): | class ChatGPTBot(Bot): | ||||
def __init__(self): | def __init__(self): | ||||
openai.api_key = conf().get('open_ai_api_key') | openai.api_key = conf().get('open_ai_api_key') | ||||
openai.api_base = conf().get('open_ai_api_base') | |||||
if conf().get('open_ai_api_base'): | |||||
openai.api_base = conf().get('open_ai_api_base') | |||||
proxy = conf().get('proxy') | proxy = conf().get('proxy') | ||||
if proxy: | if proxy: | ||||
openai.proxy = proxy | openai.proxy = proxy | ||||
@@ -12,8 +12,11 @@ user_session = dict() | |||||
class OpenAIBot(Bot): | class OpenAIBot(Bot): | ||||
def __init__(self): | def __init__(self): | ||||
openai.api_key = conf().get('open_ai_api_key') | openai.api_key = conf().get('open_ai_api_key') | ||||
openai.api_base = conf().get('open_ai_api_base') | |||||
if conf().get('open_ai_api_base'): | |||||
openai.api_base = conf().get('open_ai_api_base') | |||||
proxy = conf().get('proxy') | |||||
if proxy: | |||||
openai.proxy = proxy | |||||
def reply(self, query, context=None): | def reply(self, query, context=None): | ||||
# acquire reply content | # acquire reply content | ||||
@@ -1,6 +1,5 @@ | |||||
{ | { | ||||
"open_ai_api_key": "YOUR API KEY", | "open_ai_api_key": "YOUR API KEY", | ||||
"open_ai_api_base":"", | |||||
"proxy": "", | "proxy": "", | ||||
"single_chat_prefix": ["bot", "@bot"], | "single_chat_prefix": ["bot", "@bot"], | ||||
"single_chat_reply_prefix": "[bot] ", | "single_chat_reply_prefix": "[bot] ", | ||||