From ffec4df5635a0c892f6666e64b47a7203564bc10 Mon Sep 17 00:00:00 2001
From: zhayujie <yjzha1996@163.com>
Date: Fri, 3 Feb 2023 01:16:34 +0800
Subject: [PATCH] feat: auto append a question mark when there is no ending
 punctuation

---
 bot/openai/open_ai_bot.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/bot/openai/open_ai_bot.py b/bot/openai/open_ai_bot.py
index a391ac9..dc554d7 100644
--- a/bot/openai/open_ai_bot.py
+++ b/bot/openai/open_ai_bot.py
@@ -12,6 +12,10 @@ class OpenAIBot(Bot):
         openai.api_key = conf().get('open_ai_api_key')
 
     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':
             return self.reply_text(query)
         elif context.get('type', None) == 'IMAGE_CREATE':
@@ -82,3 +86,10 @@ class OpenAIBot(Bot):
             logger.exception(e)
             return None
         return image_url
+
+    def append_question_mark(self, query):
+        end_symbols = [".", "。", "?", "？", "!", "！"]
+        for symbol in end_symbols:
+            if query.endswith(symbol):
+                return query
+        return query + "?"