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

formatting code

Browse Source
This commit is contained in:
lanvent
2023-03-12 01:22:49 +08:00
parent 9ae7b7773e
commit 9e07703eb1
3 changed files with 83 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
bot/chatgpt/chat_gpt_bot.py
View File

@@ -13,28 +13,28 @@ class ChatGPTBot(Bot):
def __init__(self):
openai.api_key = conf().get('open_ai_api_key')
proxy = conf().get('proxy')
self.sessions=SessionManager()
self.sessions = SessionManager()
if proxy:
openai.proxy = proxy
def reply(self, query, context=None):
# acquire reply content
if context['type']=='TEXT':
if context['type'] == 'TEXT':
logger.info("[OPEN_AI] query={}".format(query))
session_id = context['session_id']
reply=None
reply = None
if query == '#清除记忆':
self.sessions.clear_session(session_id)
reply={'type':'INFO', 'content':'记忆已清除'}
reply = {'type': 'INFO', 'content': '记忆已清除'}
elif query == '#清除所有':
self.sessions.clear_all_session()
reply={'type':'INFO', 'content':'所有人记忆已清除'}
reply = {'type': 'INFO', 'content': '所有人记忆已清除'}
elif query == '#更新配置':
load_config()
reply={'type':'INFO', 'content':'配置已更新'}
reply = {'type': 'INFO', 'content': '配置已更新'}
elif query == '#DEBUG':
logger.setLevel('DEBUG')
reply={'type':'INFO', 'content':'DEBUG模式已开启'}
reply = {'type': 'INFO', 'content': 'DEBUG模式已开启'}
if reply:
return reply
session = self.sessions.build_session_query(query, session_id)
@@ -46,28 +46,28 @@ class ChatGPTBot(Bot):
reply_content = self.reply_text(session, session_id, 0)
logger.debug("[OPEN_AI] new_query={}, session_id={}, reply_cont={}".format(session, session_id, reply_content["content"]))
if reply_content['completion_tokens']==0 and len(reply_content['content'])>0:
reply={'type':'ERROR', 'content':reply_content['content']}
if reply_content['completion_tokens'] == 0 and len(reply_content['content']) > 0:
reply = {'type': 'ERROR', 'content': reply_content['content']}
elif reply_content["completion_tokens"] > 0:
self.sessions.save_session(reply_content["content"], session_id, reply_content["total_tokens"])
reply={'type':'TEXT', 'content':reply_content["content"]}
else:
reply={'type':'ERROR', 'content':reply_content['content']}
reply = {'type': 'ERROR', 'content': reply_content['content']}
logger.debug("[OPEN_AI] reply {} used 0 tokens.".format(reply_content))
return reply
elif context['type'] == 'IMAGE_CREATE':
ok, retstring=self.create_img(query, 0)
reply=None
ok, retstring = self.create_img(query, 0)
reply = None
if ok:
reply = {'type':'IMAGE', 'content':retstring}
reply = {'type': 'IMAGE', 'content': retstring}
else:
reply = {'type':'ERROR', 'content':retstring}
reply = {'type': 'ERROR', 'content': retstring}
return reply
else:
reply= {'type':'ERROR', 'content':'Bot不支持处理{}类型的消息'.format(context['type'])}
def reply_text(self, session, session_id, retry_count=0) ->dict:
def reply_text(self, session, session_id, retry_count=0) -> dict:
'''
call openai's ChatCompletion to get the answer
:param session: a conversation session
@@ -86,8 +86,8 @@ class ChatGPTBot(Bot):
presence_penalty=0.0, # [-2,2]之间，该值越大则更倾向于产生不同的内容
)
# logger.info("[ChatGPT] reply={}, total_tokens={}".format(response.choices[0]['message']['content'], response["usage"]["total_tokens"]))
return {"total_tokens": response["usage"]["total_tokens"],
"completion_tokens": response["usage"]["completion_tokens"],
return {"total_tokens": response["usage"]["total_tokens"],
"completion_tokens": response["usage"]["completion_tokens"],
"content": response.choices[0]['message']['content']}
except openai.error.RateLimitError as e:
# rate limit exception
@@ -102,11 +102,11 @@ class ChatGPTBot(Bot):
# api connection exception
logger.warn(e)
logger.warn("[OPEN_AI] APIConnection failed")
return {"completion_tokens": 0, "content":"我连接不到你的网络"}
return {"completion_tokens": 0, "content": "我连接不到你的网络"}
except openai.error.Timeout as e:
logger.warn(e)
logger.warn("[OPEN_AI] Timeout")
return {"completion_tokens": 0, "content":"我没有收到你的消息"}
return {"completion_tokens": 0, "content": "我没有收到你的消息"}
except Exception as e:
# unknown exception
logger.exception(e)
@@ -123,7 +123,7 @@ class ChatGPTBot(Bot):
)
image_url = response['data'][0]['url']
logger.info("[OPEN_AI] image_url={}".format(image_url))
return True,image_url
return True, image_url
except openai.error.RateLimitError as e:
logger.warn(e)
if retry_count < 1:
@@ -131,15 +131,17 @@ class ChatGPTBot(Bot):
logger.warn("[OPEN_AI] ImgCreate RateLimit exceed, 第{}次重试".format(retry_count+1))
return self.create_img(query, retry_count+1)
else:
return False,"提问太快啦，请休息一下再问我吧"
return False, "提问太快啦，请休息一下再问我吧"
except Exception as e:
logger.exception(e)
return False,str(e)
return False, str(e)
class SessionManager(object):
def __init__(self):
self.sessions = {}
def build_session_query(self,query, session_id):
def build_session_query(self, query, session_id):
'''
build query with conversation history
e.g. [
@@ -167,7 +169,7 @@ class SessionManager(object):
if not max_tokens:
# default 3000
max_tokens = 1000
max_tokens=int(max_tokens)
max_tokens = int(max_tokens)
session = self.sessions.get(session_id)
if session:
@@ -177,7 +179,7 @@ class SessionManager(object):
# discard exceed limit conversation
self.discard_exceed_conversation(session, max_tokens, total_tokens)
def discard_exceed_conversation(self, session, max_tokens, total_tokens):
dec_tokens = int(total_tokens)
# logger.info("prompt tokens used={},max_tokens={}".format(used_tokens,max_tokens))
@@ -187,10 +189,10 @@ class SessionManager(object):
session.pop(1)
session.pop(1)
else:
break
break
dec_tokens = dec_tokens - max_tokens
def clear_session(self,session_id):
def clear_session(self, session_id):
self.sessions[session_id] = []
def clear_all_session(self):