Преглед на файлове

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

master
zhayujie преди 1 година
родител
ревизия
ffec4df563
променени са 1 файла, в които са добавени 11 реда и са изтрити 0 реда
  1. +11
    -0
      bot/openai/open_ai_bot.py

+ 11
- 0
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 + "?"

Loading…
Отказ
Запис