aljaz/chatgpt-telegram-bot
1
1
Fork 0
mirror of https://github.com/aljazceru/chatgpt-telegram-bot.git synced 2025-12-23 23:55:05 +01:00
Code Issues Packages Projects Releases Wiki Activity

allow content passed by system role to be customized on reset

Browse Source
This commit is contained in:
Gerardo Parrello
2023-03-19 21:18:15 +01:00
parent b7d07f5023
commit 7d62305444
2 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
bot/openai_helper.py
View File

@@ -124,11 +124,13 @@ class OpenAIHelper:
result = await openai.Audio.atranscribe("whisper-1", audio)
return result.text
def reset_chat_history(self, chat_id):
def reset_chat_history(self, chat_id, content):
"""
Resets the conversation history.
"""
self.conversations[chat_id] = [{"role": "system", "content": self.config['assistant_prompt']}]
if content == '':
content = self.config['assistant_prompt']
self.conversations[chat_id] = [{"role": "system", "content": content}]
def __max_age_reached(self, chat_id) -> bool:
"""

3
bot/telegram_bot.py
View File

@@ -95,7 +95,8 @@ class ChatGPT3TelegramBot:
logging.info(f'Resetting the conversation for user {update.message.from_user.name}...')
chat_id = update.effective_chat.id
self.openai.reset_chat_history(chat_id=chat_id)
reset_content = update.message.text.replace('/reset', '').strip()
self.openai.reset_chat_history(chat_id=chat_id, content=reset_content)
await context.bot.send_message(chat_id=chat_id, text='Done!')
async def image(self, update: Update, context: ContextTypes.DEFAULT_TYPE):