kaiyuan/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-06-02 00:57:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: decouple openai session

Browse Source
This commit is contained in:
lanvent
2023-03-26 23:09:05 +08:00
parent 38c901a1c5
commit ea95ab9062
5 changed files with 122 additions and 92 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
bot/chatgpt/chat_gpt_bot.py
View File

@@ -21,11 +21,12 @@ class ChatGPTBot(Bot,OpenAIImage):
if conf().get('open_ai_api_base'):
openai.api_base = conf().get('open_ai_api_base')
proxy = conf().get('proxy')
self.sessions = SessionManager(ChatGPTSession, model= conf().get("model") or "gpt-3.5-turbo")
if proxy:
openai.proxy = proxy
if conf().get('rate_limit_chatgpt'):
self.tb4chatgpt = TokenBucket(conf().get('rate_limit_chatgpt', 20))
self.sessions = SessionManager(ChatGPTSession, model= conf().get("model") or "gpt-3.5-turbo")
def reply(self, query, context=None):
# acquire reply content