scripts/discordbot.py

@@ -18,60 +18,89 @@ intents = discord.Intents.default()
 intents.message_content = True
 bot = commands.Bot(command_prefix="!", intents=intents)
 
-async def get_chat_history(channel, limit=50):
-    messages = []
-    async for message in channel.history(limit=limit):
-        # Skip bot's own messages
-        if message.author == bot.user:
-            continue
-        messages.append({"role": "user", "content": message.content})
-    return list(reversed(messages))  # Return in chronological order
+# Add a dictionary to store conversation histories
+conversation_histories = {}
+DEFAULT_HISTORY_LIMIT = 50
+MAX_HISTORY_LIMIT = 200
 
-async def get_ai_response(messages):
+async def get_ai_response(prompt, channel_id=None, include_history=False, history_limit=DEFAULT_HISTORY_LIMIT):
     try:
+        messages = []
+        
+        # If history is requested and exists for this channel, include it
+        if include_history and channel_id in conversation_histories:
+            messages = conversation_histories[channel_id]
+        
+        # Add the current prompt
+        messages.append({"role": "user", "content": prompt})
+        
         response = client.chat.completions.create(
             model=os.getenv('MODEL_NAME') or "us.anthropic.claude-3-5-sonnet-20241022-v2:0",
             messages=messages,
             temperature=0.7,
             max_tokens=500
         )
+        
+        # Store the conversation history if channel_id is provided
+        if channel_id:
+            if channel_id not in conversation_histories:
+                conversation_histories[channel_id] = []
+            conversation_histories[channel_id].extend([
+                {"role": "user", "content": prompt},
+                {"role": "assistant", "content": response.choices[0].message.content}
+            ])
+            
+            # Limit history based on specified or default limit
+            if len(conversation_histories[channel_id]) > history_limit * 2:  # multiply by 2 because each exchange has 2 messages
+                conversation_histories[channel_id] = conversation_histories[channel_id][-(history_limit * 2):]
+        
         return response.choices[0].message.content
     except Exception as e:
         print(f"Error getting AI response: {e}")
         return "Sorry, I encountered an error while processing your request."
 
-@bot.event 
+@bot.event
 async def on_message(message):
-    # Ignore messages from the bot itself
     if message.author == bot.user:
         return
 
-    # Respond to DMs or when mentioned in a server
-    if isinstance(message.channel, discord.DMChannel) or bot.user in message.mentions:
-        # For mentions, remove the bot mention from the message
-        if bot.user in message.mentions:
-            prompt = message.content.replace(f'<@{bot.user.id}>', '').strip()
-        else:
-            prompt = message.content.strip()
+    if bot.user in message.mentions:
+        prompt = message.content.replace(f'<@{bot.user.id}>', '').strip()
         
-        # If there's no prompt
         if not prompt:
             await message.channel.send("Hello! How can I help you?")
             return
 
-        # Show typing indicator while processing
+        # Check if the message includes a request for history
+        include_history = False
+        history_limit = DEFAULT_HISTORY_LIMIT
+        
+        # Check for "with history", "with X history", or "X lines of chat" patterns
+        import re
+        if "with history" in prompt.lower() or re.search(r"\d+\s*lines of chat", prompt.lower()):
+            include_history = True
+            # Check for specific history limit
+            if match := re.search(r"with (\d+) history", prompt.lower()):
+                requested_limit = int(match.group(1))
+                history_limit = min(requested_limit, MAX_HISTORY_LIMIT)
+                prompt = re.sub(r"with \d+ history", "", prompt, flags=re.IGNORECASE)
+            elif match := re.search(r"(\d+)\s*lines of chat", prompt.lower()):
+                requested_limit = int(match.group(1))
+                history_limit = min(requested_limit, MAX_HISTORY_LIMIT)
+                prompt = re.sub(r"\d+\s*lines of chat", "", prompt, flags=re.IGNORECASE)
+            else:
+                prompt = prompt.lower().replace("with history", "")
+                
+        prompt = prompt.strip()
+
         async with message.channel.typing():
-            # Get chat history
-            chat_history = await get_chat_history(message.channel)
+            response = await get_ai_response(
+                prompt,
+                channel_id=str(message.channel.id),
+                include_history=include_history,
+                history_limit=history_limit
+            )
             
-            # Add current message to history
-            chat_history.append({"role": "user", "content": prompt})
-            
-            # Get response from OpenWebUI with chat history context
-            response = await get_ai_response(chat_history)
-            
-            # Send the response
-            # Split long messages if they exceed Discord's character limit
             if len(response) > 2000:
                 chunks = [response[i:i+2000] for i in range(0, len(response), 2000)]
                 for chunk in chunks:
@@ -81,6 +110,15 @@ async def on_message(message):
 
     await bot.process_commands(message)
 
+@bot.command(name='clearhistory')
+async def clear_history(ctx):
+    channel_id = str(ctx.channel.id)
+    if channel_id in conversation_histories:
+        conversation_histories[channel_id] = []
+        await ctx.send("Conversation history has been cleared.")
+    else:
+        await ctx.send("No conversation history exists for this channel.")
+
 def main():
     # Get the Discord token from environment variables
     discord_token = os.getenv('DISCORD_TOKEN')