move common code into generic RunProgressReporter

Author: ngc92

src/discord-cluster-manager/api/utils.py

@@ -186,7 +186,7 @@ async def _run_submission(
                 submission.code,
                 submission.file_name,
                 gpu,
-                RunProgressReporterAPI(),
+                RunProgressReporterAPI(f"{gpu.name} on {gpu.runner}"),
                 req.task,
                 mode,
                 None,
@@ -200,7 +200,7 @@ async def _run_submission(
                     submission.code,
                     submission.file_name,
                     gpu,
-                    RunProgressReporterAPI(),
+                    RunProgressReporterAPI(f"{gpu.name} on {gpu.runner} (secret)"),
                     req.task,
                     SubmissionMode.PRIVATE,
                     req.secret_seed,

src/discord-cluster-manager/report.py

@@ -346,18 +346,35 @@ async def _update_message(self):
 
 
 class RunProgressReporter:
+    def __init__(self, title: str):
+        self.title = title
+        self.lines = []
+
     async def push(self, content: str | list[str]):
-        raise NotImplementedError()
+        if isinstance(content, str):
+            self.lines.append(f"> {content}")
+        else:
+            for line in content:
+                self.lines.append(f"> {line}")
+        await self._update_message()
 
     async def update(self, new_content: str):
-        raise NotImplementedError()
+        self.lines[-1] = f"> {new_content}"
+        await self._update_message()
 
     async def update_title(self, new_title):
-        raise NotImplementedError()
+        self.title = new_title
+        await self._update_message()
+
+    def get_message(self):
+        return str.join("\n", [f"**{self.title}**"] + self.lines)
 
     async def generate_report(self, title: str, runs: dict[str, EvalResult]):
         raise NotImplementedError()
 
+    async def _update_message(self):
+        raise NotImplementedError()
+
 
 class RunProgressReporterDiscord(RunProgressReporter):
     def __init__(
@@ -366,33 +383,13 @@ def __init__(
         interaction: discord.Interaction,
         title: str,
     ):
-        self.title = title
-        self.lines = []
+        super().__init__(title=title)
         self.root = root
         self.interaction = interaction
 
-    async def push(self, content: str | list[str]):
-        if isinstance(content, str):
-            self.lines.append(f"> {content}")
-        else:
-            for line in content:
-                self.lines.append(f"> {line}")
-        await self._update_message()
-
-    async def update(self, new_content: str):
-        self.lines[-1] = f"> {new_content}"
-        await self._update_message()
-
-    async def update_title(self, new_title):
-        self.title = new_title
-        await self._update_message()
-
     async def _update_message(self):
         await self.root._update_message()
 
-    def get_message(self):
-        return str.join("\n", [f"**{self.title}**"] + self.lines)
-
     async def generate_report(self, title: str, runs: dict[str, EvalResult]):
         thread = await self.interaction.channel.create_thread(
             name=title,
@@ -405,17 +402,10 @@ async def generate_report(self, title: str, runs: dict[str, EvalResult]):
 
 
 class RunProgressReporterAPI(RunProgressReporter):
-    def __init__(self):
-        self.title = ""
-        self.lines = []
+    def __init__(self, title: str):
+        super().__init__(title=title)
 
-    async def push(self, content: str | list[str]):
-        pass
-
-    async def update(self, new_content: str):
-        pass
-
-    async def update_title(self, new_title):
+    async def _update_message(self):
         pass
 
     async def generate_report(self, title: str, runs: dict[str, EvalResult]):