aljaz/chatgpt-telegram-bot
1
1
Fork 0
mirror of https://github.com/aljazceru/chatgpt-telegram-bot.git synced 2026-01-27 16:56:00 +01:00
Code Issues Packages Projects Releases Wiki Activity

reset conversation and help menu

Browse Source
This commit is contained in:
ned
2022-12-04 21:18:53 +01:00
parent 1fc8f18040
commit aad0c8c39b
2 changed files with 18 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
chat_gpt3_bot.py
View File

@@ -89,3 +89,7 @@ class ChatGPT3Bot:
return {'message': message}
except:
raise ValueError("Error: " + response.text)
def reset(self):
self.conversation_id = None
self.parent_id = self.generate_uuid()

19
telegram_bot.py
View File

@@ -10,11 +10,21 @@ class ChatGPT3TelegramBot:
self.config = config
self.gpt3_bot = gpt3_bot
# Help menu
async def help(self, update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
await update.message.reply_text("/start - Start the bot\n/reset - Reset conversation\n/help - Help menu")
# Start the bot
async def start(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
logging.info('Bot started')
await context.bot.send_message(chat_id=update.effective_chat.id, text="I'm a Chat-GPT3 Bot, please talk to me!")
# Reset the conversation
async def reset(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
logging.info('Resetting the conversation...')
self.gpt3_bot.reset()
await context.bot.send_message(chat_id=update.effective_chat.id, text="Done!")
# React to messages
async def prompt(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
logging.info('New message received')
@@ -26,10 +36,9 @@ class ChatGPT3TelegramBot:
def run(self):
application = ApplicationBuilder().token(self.config['telegram_bot_token']).build()
start_handler = CommandHandler('start', self.start)
prompt_handler = MessageHandler(filters.TEXT & (~filters.COMMAND), self.prompt)
application.add_handler(start_handler)
application.add_handler(prompt_handler)
application.add_handler(CommandHandler('start', self.start))
application.add_handler(CommandHandler('reset', self.reset))
application.add_handler(CommandHandler('help', self.help))
application.add_handler(MessageHandler(filters.TEXT & (~filters.COMMAND), self.prompt))
application.run_polling()