Add retry_job and get_failed_jobs functions to Gitlab MCP (#349)

* Add a retry_job function to retry specific failed jobs from pipeline

* Add get_failed_jobs function to get all failed jobs from MR

* Use a custom model for the failed pipeline jobs

---------

Co-authored-by: Nikola Forró <[email protected]>

Author: ljavorsk

common/models.py

@@ -275,3 +275,20 @@ class CachedMRMetadata(BaseModel):
     title: str = Field(description="Merge request title")
     package: str = Field(description="Package name")
     details: str = Field(description="Operation-specific identifier (list of upstream patch URLs for backport, version for rebase)")
+
+
+# ============================================================================
+# GitLab Failed Pipeline Jobs Schemas
+# ============================================================================
+
+class FailedPipelineJob(BaseModel):
+    """Represents a failed job in a GitLab pipeline."""
+
+    id: str = Field(description="Pipeline job ID as a string")
+    name: str = Field(description="Name of the job")
+    url: str = Field(description="Full URL to the job in GitLab")
+    status: str = Field(description="Job status")
+    stage: str = Field(description="Pipeline stage the job belongs to")
+    artifacts_url: str = Field(
+        description="URL to browse job artifacts, empty string if no artifacts"
+    )

mcp_server/gitlab_tools.py

@@ -12,6 +12,7 @@
 from ogr.services.gitlab.pull_request import GitlabPullRequest
 from pydantic import Field
 
+from common.models import FailedPipelineJob
 from common.validators import AbsolutePath
 from utils import clean_stale_repositories
 
@@ -289,3 +290,80 @@ async def create_merge_request_checklist(
         return f"Successfully created checklist for merge request {merge_request_url}"
     except Exception as e:
         raise ToolError(f"Failed to create checklist for merge request: {e}")
+
+
+async def retry_pipeline_job(
+    project_url: Annotated[str, Field(description="GitLab project URL")],
+    job_id: Annotated[int, Field(description="Job ID to retry")],
+) -> str:
+    """
+    Retries a specific job in a GitLab pipeline.
+    """
+    try:
+        project = await asyncio.to_thread(
+            get_project, url=project_url, token=os.getenv("GITLAB_TOKEN")
+        )
+
+        def retry_gitlab_job():
+            job = project.gitlab_repo.jobs.get(job_id)
+            job.retry()
+            return job
+
+        job = await asyncio.to_thread(retry_gitlab_job)
+
+        logger.info(f"Successfully retried job {job_id} for project {project_url}")
+        return f"Successfully retried job {job_id}. Status: {job.status}"
+
+    except Exception as e:
+        logger.error(f"Failed to retry job {job_id} for project {project_url}: {e}")
+        raise ToolError(f"Failed to retry job: {e}") from e
+
+
+async def get_failed_pipeline_jobs_from_merge_request(
+    merge_request_url: Annotated[str, Field(description="URL of the merge request")],
+) -> list[FailedPipelineJob]:
+    """
+    Gets the failed pipeline jobs from the latest pipeline of a merge request.
+    Returns a list of failed pipeline jobs with their details.
+    """
+    try:
+        mr = await _get_merge_request_from_url(merge_request_url)
+
+        def get_latest_pipeline_jobs():
+            # Use head_pipeline to get the latest pipeline for this MR
+            if not hasattr(mr._raw_pr, "head_pipeline") or not mr._raw_pr.head_pipeline:
+                return []
+
+            pipeline_id = mr._raw_pr.head_pipeline["id"]
+            pipeline = mr.target_project.gitlab_repo.pipelines.get(pipeline_id)
+            jobs = pipeline.jobs.list(get_all=True)
+
+            namespace = mr.target_project.namespace
+            repo = mr.target_project.repo
+            failed_jobs = [
+                FailedPipelineJob(
+                    id=str(job.id),
+                    name=job.name,
+                    url=f"https://gitlab.com/{namespace}/{repo}/-/jobs/{job.id}",
+                    status=job.status,
+                    stage=job.stage,
+                    artifacts_url=(
+                        f"https://gitlab.com/{namespace}/{repo}/-/jobs/{job.id}/artifacts/browse"
+                        if hasattr(job, "artifacts_file") and job.artifacts_file
+                        else ""
+                    ),
+                )
+                for job in jobs
+                if job.status == "failed"
+            ]
+
+            return failed_jobs
+
+        failed_jobs = await asyncio.to_thread(get_latest_pipeline_jobs)
+
+        logger.info(f"Found {len(failed_jobs)} failed jobs in latest pipeline for MR {merge_request_url}")
+        return failed_jobs
+
+    except Exception as e:
+        logger.error(f"Failed to get failed jobs from MR {merge_request_url}: {e}")
+        raise ToolError(f"Failed to get failed jobs from merge request: {e}") from e

mcp_server/tests/unit/test_gitlab_tools.py

@@ -20,6 +20,8 @@
     push_to_remote_repository,
     add_merge_request_labels,
     add_blocking_merge_request_comment,
+    retry_pipeline_job,
+    get_failed_pipeline_jobs_from_merge_request,
 )
 from test_utils import mock_git_repo_basepath
 
@@ -295,3 +297,115 @@ async def test_create_merge_request_checklist():
     )
 
     assert result == f"Successfully created checklist for merge request {merge_request_url}"
+
+
+@pytest.mark.asyncio
+async def test_retry_pipeline_job():
+    project_url = "https://gitlab.com/redhat/rhel/rpms/bash"
+    job_id = 12345678
+
+    flexmock(GitlabService).should_receive("get_project_from_url").with_args(
+        url=project_url
+    ).and_return(
+        flexmock(
+            gitlab_repo=flexmock(
+                jobs=flexmock().should_receive("get").with_args(job_id).and_return(
+                    flexmock(id=job_id, status="pending").should_receive("retry").once().mock()
+                ).mock()
+            )
+        )
+    )
+
+    result = await retry_pipeline_job(project_url=project_url, job_id=job_id)
+
+    assert result == f"Successfully retried job {job_id}. Status: pending"
+
+
+@pytest.mark.asyncio
+async def test_retry_pipeline_job_invalid_project():
+    project_url = "https://gitlab.com/nonexistent/project"
+    job_id = 12345678
+
+    flexmock(GitlabService).should_receive("get_project_from_url").with_args(
+        url=project_url
+    ).and_raise(Exception("Project not found"))
+
+    with pytest.raises(Exception) as exc_info:
+        await retry_pipeline_job(project_url=project_url, job_id=job_id)
+
+    assert "Failed to retry job" in str(exc_info.value)
+
+
+@pytest.mark.asyncio
+async def test_get_failed_pipeline_jobs_from_merge_request():
+    merge_request_url = "https://gitlab.com/redhat/centos-stream/rpms/bash/-/merge_requests/123"
+    pipeline_id = 789
+
+    flexmock(GitlabService).should_receive("get_project_from_url").with_args(
+        url="https://gitlab.com/redhat/centos-stream/rpms/bash"
+    ).and_return(
+        flexmock().should_receive("get_pr").with_args(123).and_return(
+            flexmock(
+                _raw_pr=flexmock(head_pipeline={"id": pipeline_id}),
+                target_project=flexmock(
+                    namespace="redhat/centos-stream/rpms",
+                    repo="bash",
+                    gitlab_repo=flexmock(
+                        pipelines=flexmock().should_receive("get").with_args(pipeline_id).and_return(
+                            flexmock(
+                                jobs=flexmock().should_receive("list").with_args(get_all=True).and_return([
+                                    flexmock(id=11111, name="check-tickets", status="failed", stage="build", artifacts_file={"filename": "debug.log"}),
+                                    flexmock(id=22222, name="build_rpm", status="failed", stage="test", artifacts_file=None),
+                                    flexmock(id=33333, name="trigger_tests", status="success", stage="test", artifacts_file=None),
+                                ]).mock()
+                            )
+                        ).mock()
+                    )
+                ),
+            )
+        ).mock()
+    )
+
+    result = await get_failed_pipeline_jobs_from_merge_request(merge_request_url=merge_request_url)
+
+    assert len(result) == 2
+    assert result[0].id == "11111"
+    assert result[0].name == "check-tickets"
+    assert result[0].status == "failed"
+    assert result[0].stage == "build"
+    assert "/-/jobs/11111" in result[0].url
+    assert result[0].artifacts_url == "https://gitlab.com/redhat/centos-stream/rpms/bash/-/jobs/11111/artifacts/browse"
+
+    assert result[1].id == "22222"
+    assert result[1].name == "build_rpm"
+    assert result[1].status == "failed"
+    assert result[1].stage == "test"
+    assert "/-/jobs/22222" in result[1].url
+    assert result[1].artifacts_url == ""
+
+
+@pytest.mark.asyncio
+async def test_get_failed_pipeline_jobs_from_merge_request_no_pipelines():
+    merge_request_url = "https://gitlab.com/redhat/centos-stream/rpms/bash/-/merge_requests/123"
+
+    flexmock(GitlabService).should_receive("get_project_from_url").with_args(
+        url="https://gitlab.com/redhat/centos-stream/rpms/bash"
+    ).and_return(
+        flexmock().should_receive("get_pr").with_args(123).and_return(
+            flexmock(_raw_pr=flexmock(head_pipeline=None))
+        ).mock()
+    )
+
+    result = await get_failed_pipeline_jobs_from_merge_request(merge_request_url=merge_request_url)
+
+    assert len(result) == 0
+
+
+@pytest.mark.asyncio
+async def test_get_failed_pipeline_jobs_from_merge_request_invalid_url():
+    merge_request_url = "https://github.com/user/repo/pull/123"
+
+    with pytest.raises(Exception) as exc_info:
+        await get_failed_pipeline_jobs_from_merge_request(merge_request_url=merge_request_url)
+
+    assert "Could not parse merge request URL" in str(exc_info.value)