zhayujie
GitHub
1 year ago
3 changed files with 3 additions and 0 deletions
+ 1
- 0
bot/chatgpt/chat_gpt_bot.py
View File
| @@ -16,6 +16,7 @@ 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') | |||||
| proxy = conf().get('proxy') | proxy = conf().get('proxy') | ||||
| if proxy: | if proxy: | ||||
| openai.proxy = proxy | openai.proxy = proxy | ||||
+ 1
- 0
bot/openai/open_ai_bot.py
View File
| @@ -12,6 +12,7 @@ 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') | |||||
| def reply(self, query, context=None): | def reply(self, query, context=None): | ||||
+ 1
- 0
config-template.json
View File
| @@ -1,5 +1,6 @@ | |||||
| { | { | ||||
| "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] ", | ||||
Loading…