fix: GitHubClient must be used as async context manager

Author: echobt

src/swe_forge/cli/benchmark.py

@@ -253,42 +253,41 @@ async def _mine_tasks(
     verbose: bool,
 ) -> list[SweTask]:
     """Mine tasks using the SWE pipeline."""
-    gh_client = GitHubClient(token=github_token)
-
-    config = SwePipelineConfig(
-        max_tasks=num_tasks,
-        once=True,
-        difficulty_filter=difficulty_filter,
-    )
-
-    tasks: list[SweTask] = []
-
-    with Progress(
-        SpinnerColumn(),
-        TextColumn("[progress.description]{task.description}"),
-        BarColumn(),
-        TaskProgressColumn(),
-        TimeElapsedColumn(),
-        console=console,
-    ) as progress:
-        task_id = progress.add_task("Mining tasks...", total=num_tasks)
-
-        async with SwePipeline(gh_client, config=config) as pipeline:
-            from swe_forge.swe.pipeline import SwePipelineEventType
-
-            async for event in pipeline.run_with_progress():
-                if event.event_type == SwePipelineEventType.TASK_EXTRACTED:
-                    task = event.data.get("task")
-                    if task and isinstance(task, SweTask):
-                        tasks.append(task)
-                        progress.update(
-                            task_id, advance=1, description=f"Mined {len(tasks)} tasks"
-                        )
-
-                    if len(tasks) >= num_tasks:
-                        break
+    async with GitHubClient(token=github_token) as gh_client:
+        config = SwePipelineConfig(
+            max_tasks=num_tasks,
+            once=True,
+            difficulty_filter=difficulty_filter,
+        )
 
-    return tasks[:num_tasks]
+        tasks: list[SweTask] = []
+
+        with Progress(
+            SpinnerColumn(),
+            TextColumn("[progress.description]{task.description}"),
+            BarColumn(),
+            TaskProgressColumn(),
+            TimeElapsedColumn(),
+            console=console,
+        ) as progress:
+            task_id = progress.add_task("Mining tasks...", total=num_tasks)
+
+            async with SwePipeline(gh_client, config=config) as pipeline:
+                from swe_forge.swe.pipeline import SwePipelineEventType
+
+                async for event in pipeline.run_with_progress():
+                    if event.event_type == SwePipelineEventType.TASK_EXTRACTED:
+                        task = event.data.get("task")
+                        if task and isinstance(task, SweTask):
+                            tasks.append(task)
+                            progress.update(
+                                task_id, advance=1, description=f"Mined {len(tasks)} tasks"
+                            )
+
+                        if len(tasks) >= num_tasks:
+                            break
+
+        return tasks[:num_tasks]
 
 
 async def _run_harness(

src/swe_forge/cli/mine.py

@@ -303,39 +303,39 @@ class PipelineResult:
         tasks: list = field(default_factory=list)
         benchmark_metrics: object = None
 
-    gh_client = GitHubClient(token=token)
-    gh_archive_client = GhArchiveClient(token=token) if not repo_filter else None
-
-    tasks: list[SweTask] = []
-    metrics = None
-
-    with Progress(
-        SpinnerColumn(),
-        TextColumn("[progress.description]{task.description}"),
-        BarColumn(),
-        TaskProgressColumn(),
-        TimeElapsedColumn(),
-        console=console,
-    ) as progress:
-        task_id = progress.add_task("Mining tasks...", total=config.max_tasks)
-
-        async with SwePipeline(
-            gh_client, gh_archive_client=gh_archive_client, config=config
-        ) as pipeline:
-            async for event in pipeline.run_with_progress():
-                if event.event_type == SwePipelineEventType.TASK_EXTRACTED:
-                    task = event.data.get("task")
-                    if task and isinstance(task, SweTask):
-                        tasks.append(task)
-                        progress.update(
-                            task_id, advance=1, description=f"Mined {len(tasks)} tasks"
-                        )
-
-                elif event.event_type == SwePipelineEventType.PIPELINE_COMPLETED:
-                    metrics = event.data.get("metrics")
-                    progress.update(task_id, completed=len(tasks))
-
-    return PipelineResult(tasks=tasks, benchmark_metrics=metrics)
+    async with GitHubClient(token=token) as gh_client:
+        gh_archive_client = GhArchiveClient(token=token) if not repo_filter else None
+
+        tasks: list[SweTask] = []
+        metrics = None
+
+        with Progress(
+            SpinnerColumn(),
+            TextColumn("[progress.description]{task.description}"),
+            BarColumn(),
+            TaskProgressColumn(),
+            TimeElapsedColumn(),
+            console=console,
+        ) as progress:
+            task_id = progress.add_task("Mining tasks...", total=config.max_tasks)
+
+            async with SwePipeline(
+                gh_client, gh_archive_client=gh_archive_client, config=config
+            ) as pipeline:
+                async for event in pipeline.run_with_progress():
+                    if event.event_type == SwePipelineEventType.TASK_EXTRACTED:
+                        task = event.data.get("task")
+                        if task and isinstance(task, SweTask):
+                            tasks.append(task)
+                            progress.update(
+                                task_id, advance=1, description=f"Mined {len(tasks)} tasks"
+                            )
+
+                    elif event.event_type == SwePipelineEventType.PIPELINE_COMPLETED:
+                        metrics = event.data.get("metrics")
+                        progress.update(task_id, completed=len(tasks))
+
+        return PipelineResult(tasks=tasks, benchmark_metrics=metrics)
 
 
 @app.command("complete")