Merge pull request #76 from smokeyScraper/bot_commands

[refactor]: migrate to cog-based command architecture

Author: chandansgowda

backend/app/core/orchestration/agent_coordinator.py

@@ -72,7 +72,6 @@ async def _handle_clear_memory_request(self, message_data: Dict[str, Any]):
         """Handle requests to clear thread memory"""
         try:
             memory_thread_id = message_data.get("memory_thread_id")
-            user_id = message_data.get("user_id")
             cleanup_reason = message_data.get("cleanup_reason", "manual")
 
             if not memory_thread_id:
@@ -81,16 +80,6 @@ async def _handle_clear_memory_request(self, message_data: Dict[str, Any]):
 
             logger.info(f"Clearing memory for thread {memory_thread_id}, reason: {cleanup_reason}")
 
-            # Get current state before clearing
-            current_state_data = await self.devrel_agent.get_thread_state(memory_thread_id)
-
-            if current_state_data:
-                current_state = AgentState(**current_state_data)
-
-                # Store to database before clearing
-                await store_summary_to_database(current_state)
-                logger.info(f"Stored final summary to database for user {user_id}")
-
             # Clear from InMemorySaver
             success = await self.devrel_agent.clear_thread_memory(memory_thread_id, force_clear=True)
 

backend/bots/discord/discord_bot.py

@@ -43,10 +43,10 @@ async def on_message(self, message):
         if message.author == self.user:
             return
 
-        # Skip if message is a command (starts with !)
-        # TODO: Add support for commands
-        if message.content.startswith('!'):
-            await self.process_commands(message)
+        # if message is a command (starts with !)
+        ctx = await self.get_context(message)
+        if ctx.command is not None:
+            await self.invoke(ctx)
             return
 
         try:
@@ -184,65 +184,3 @@ async def _handle_agent_response(self, response_data: Dict[str, Any]):
 
         except Exception as e:
             logger.error(f"Error handling agent response: {str(e)}")
-
-    @commands.command(name="reset")
-    async def reset_thread(self, ctx):
-        """Reset user's DevRel thread and memory"""
-        user_id = str(ctx.author.id)
-
-        # Send clear memory request to agent coordinator
-        cleanup_message = {
-            "type": "clear_thread_memory",
-            "memory_thread_id": user_id,
-            "user_id": user_id,
-            "cleanup_reason": "manual_reset"
-        }
-        await self.queue_manager.enqueue(cleanup_message, QueuePriority.HIGH)
-
-        # Clean up Discord thread tracking
-        if user_id in self.active_threads:
-            del self.active_threads[user_id]
-
-        await ctx.send("Your DevRel thread and memory have been reset. Next message will create a new thread.")
-
-    @commands.command(name="help_devrel")
-    async def help_devrel(self, ctx):
-        """Show DevRel bot help"""
-        embed = discord.Embed(
-            title="DevRel Assistant Help",
-            description="I'm here to help you with Devr.AI related questions!",
-        )
-
-        embed.add_field(
-            name="What I can do:",
-            value="""
-            • Answer FAQs about Devr.AI
-            • Help with getting started
-            • Search the web for information
-            • Provide technical support
-            • Guide you through onboarding
-            """,
-            inline=False
-        )
-
-        embed.add_field(
-            name="Example questions:",
-            value="""
-            • "What is Devr.AI?"
-            • "How do I contribute?"
-            • "Search for latest AI news"
-            • "How to get started with LangGraph?"
-            """,
-            inline=False
-        )
-
-        embed.add_field(
-            name="Commands:",
-            value="""
-            • `!reset` - Reset your chat thread and memory
-            • `!help_devrel` - Show this help message
-            """,
-            inline=False
-        )
-
-        await ctx.send(embed=embed)

backend/bots/discord/discord_cogs.py

@@ -0,0 +1,40 @@
+from discord.ext import commands
+import discord
+from app.core.orchestration.queue_manager import AsyncQueueManager, QueuePriority
+from bots.discord.discord_bot import DiscordBot
+
+class DevRelCommands(commands.Cog):
+    def __init__(self, bot: DiscordBot, queue_manager: AsyncQueueManager):
+        self.bot = bot
+        self.queue = queue_manager
+
+    @commands.command(name="reset")
+    async def reset_thread(self, ctx: commands.Context):
+        """Reset your DevRel thread and memory."""
+        user_id = str(ctx.author.id)
+        cleanup = {
+            "type": "clear_thread_memory",
+            "memory_thread_id": user_id,
+            "user_id": user_id,
+            "cleanup_reason": "manual_reset"
+        }
+        await self.queue.enqueue(cleanup, QueuePriority.HIGH)
+        self.bot.active_threads.pop(user_id, None)
+        await ctx.send("Your DevRel thread & memory have been reset! Send another message to start fresh.")
+
+    @commands.command(name="help_devrel")
+    async def help_devrel(self, ctx: commands.Context):
+        """Show DevRel assistant help."""
+        embed = discord.Embed(
+            title="DevRel Assistant Help",
+            description="I can help you with Devr.AI related questions!"
+        )
+        embed.add_field(
+            name="Commands",
+            value=(
+                "• `!reset` – Reset your DevRel thread and memory\n"
+                "• `!help_devrel` – Show this help message"
+            ),
+            inline=False
+        )
+        await ctx.send(embed=embed)

backend/main.py

@@ -7,6 +7,7 @@
 from app.core.orchestration.queue_manager import AsyncQueueManager
 from app.core.orchestration.agent_coordinator import AgentCoordinator
 from bots.discord.discord_bot import DiscordBot
+from bots.discord.discord_cogs import DevRelCommands
 from app.db.weaviate.weaviate_client import get_client
 
 # Configure logging
@@ -29,6 +30,7 @@ def __init__(self):
         self.queue_manager = AsyncQueueManager()
         self.agent_coordinator = AgentCoordinator(self.queue_manager)
         self.discord_bot = DiscordBot(self.queue_manager)
+        self.discord_bot.add_cog(DevRelCommands(self.discord_bot, self.queue_manager))
         self.running = False
 
     async def start(self):