work in progress on /verifyrun

src/discord-cluster-manager/cogs/github_cog.py

@@ -35,6 +35,7 @@ async def run_github(
         interaction: discord.Interaction,
         script: discord.Attachment,
         gpu_type: app_commands.Choice[str],
+        use_followup: bool = False
     ):
         if not script.filename.endswith(".py") and not script.filename.endswith(".cu"):
             await interaction.response.send_message(
@@ -43,10 +44,15 @@ async def run_github(
             return
 
         thread = await self.bot.create_thread(interaction, gpu_type.name, "GitHub Job")
+        message = f"Created thread {thread.mention} for your GitHub job"
+
+        if use_followup:
+            if not interaction.response.is_done():
+                await interaction.response.defer()
+            await interaction.followup.send(message)
+        else:
+            await interaction.response.send_message(message)
 
-        await interaction.response.send_message(
-            f"Created thread {thread.mention} for your GitHub job"
-        )
         await thread.send(f"Processing `{script.filename}` with {gpu_type.name}...")
 
         try:

src/discord-cluster-manager/cogs/modal_cog.py

@@ -29,6 +29,7 @@ async def run_modal(
         interaction: discord.Interaction,
         script: discord.Attachment,
         gpu_type: app_commands.Choice[str],
+        use_followup: bool = False
     ):
         if not script.filename.endswith(".py") and not script.filename.endswith(".cu"):
             await interaction.response.send_message(
@@ -37,10 +38,15 @@ async def run_modal(
             return
 
         thread = await self.bot.create_thread(interaction, gpu_type.name, "Modal Job")
+        message = f"Created thread {thread.mention} for your Modal job"
+
+        if use_followup:
+            if not interaction.response.is_done():
+                await interaction.response.defer()
+            await interaction.followup.send(message)
+        else:
+            await interaction.response.send_message(message)
 
-        await interaction.response.send_message(
-            f"Created thread {thread.mention} for your Modal job"
-        )
         await thread.send(f"Processing `{script.filename}` with {gpu_type.name}...")
 
         try:

src/discord-cluster-manager/cogs/verify_run_cog.py

@@ -3,9 +3,20 @@
 from discord.ext import commands
 import re
 from utils import setup_logging
+from unittest.mock import AsyncMock
 
 logger = setup_logging()
 
+def create_mock_attachment(filename: str, content: str):
+    "Create an AsyncMock to simulate discord.Attachment"
+
+    mock_attachment = AsyncMock(spec=discord.Attachment)
+    mock_attachment.filename = filename
+    mock_attachment.content_type = 'text/plain'
+    # Simulate the read method
+    mock_attachment.read = AsyncMock(return_value=content.encode('utf-8'))
+    return mock_attachment
+
 class VerifyRunCog(commands.Cog):
     """
     A Discord cog for verifying the success of trainingruns.
@@ -100,4 +111,44 @@ async def verify_run(self, interaction: discord.Interaction):
         elif any("Running on Modal..." in content for content in message_contents):
             await self.verify_modal_run(interaction, message_contents)
         else:
-            await interaction.response.send_message("❌ Could not determine run type!")
+            await interaction.response.send_message("❌ Could not determine run type!")
+
+    @app_commands.command(name='verifyrun2')
+    async def verify_run2(self, interaction: discord.Interaction):
+        """Verify runs on on Modal, GitHub Nvidia, and GitHub AMD."""
+
+        try:
+            # Get instances of the other cogs
+            modal_cog = self.bot.get_cog('ModalCog')
+            github_cog = self.bot.get_cog('GitHubCog')
+
+            if not all([modal_cog, github_cog]):
+                await interaction.followup.send("❌ Required cogs not found!")
+                return
+
+            script_content = "print('Hello, world!')"
+            script_file = create_mock_attachment("test_script.py", script_content)
+
+            t4 = app_commands.Choice(name="NVIDIA T4", value="t4")
+            nvidia = app_commands.Choice(name="NVIDIA", value="nvidia")
+            amd = app_commands.Choice(name="AMD", value="amd")
+
+            modal_command = modal_cog.run_modal
+            await modal_command.callback(modal_cog, interaction, script_file, t4, use_followup=True)
+
+            github_command = github_cog.run_github
+            await github_command.callback(github_cog, interaction, script_file, nvidia, use_followup=True)
+            await github_command.callback(github_cog, interaction, script_file, amd, use_followup=True)
+
+            await interaction.followup.send(
+               "✅ Started all verification runs:\n"
+               "- Modal run\n"
+               "- GitHub Nvidia run\n"
+               "- GitHub AMD run"
+            )
+
+        except Exception as e:
+            logger.error(f"Error starting verification runs: {e}", exc_info=True)
+            await interaction.followup.send(
+                f"❌ Error starting verification runs: {str(e)}"
+            )