kaiyuan/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-07-21 22:27:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: auto append a question mark when there is no ending punctuation

Browse Source
This commit is contained in:
zhayujie
2023-02-03 01:16:34 +08:00
parent 4c42136d0c
commit ffec4df563
1 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
bot/openai/open_ai_bot.py
View File

@@ -12,6 +12,10 @@ class OpenAIBot(Bot):
openai.api_key = conf().get('open_ai_api_key') openai.api_key = conf().get('open_ai_api_key')
def reply(self, query, context=None): def reply(self, query, context=None):
# auto append question mark
query = self.append_question_mark(query)
# acquire reply content
if not context or not context.get('type') or context.get('type') == 'TEXT': if not context or not context.get('type') or context.get('type') == 'TEXT':
return self.reply_text(query) return self.reply_text(query)
elif context.get('type', None) == 'IMAGE_CREATE': elif context.get('type', None) == 'IMAGE_CREATE':
@@ -82,3 +86,10 @@ class OpenAIBot(Bot):
logger.exception(e) logger.exception(e)
return None return None
return image_url return image_url
def append_question_mark(self, query):
end_symbols = [".", "。", "?", "？", "!", "！"]
for symbol in end_symbols:
if query.endswith(symbol):
return query
return query + "?"