github/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-03-18 12:40:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #465 from B1gM8c/master

Browse Source 
支持自定义openai_api_base
This commit is contained in:
zhayujie
2023-03-15 00:24:04 +08:00
committed by GitHub
parent 9e3a5395c7 2c8da59b47
commit 1dc39af423
3 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
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
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
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] ",