From b06ffae57c8c2a6613421ce5ea81ce659244a3cb Mon Sep 17 00:00:00 2001
From: Catherine Lee <csl@fb.com>
Date: Fri, 11 Jul 2025 10:57:39 -0700
Subject: [PATCH 1/2] tc

---
 .../commit_jobs_query/params.json             |  6 ++-
 .../commit_jobs_query/query.sql               |  8 +++
 torchci/components/commit/CommitStatus.tsx    |  6 +++
 torchci/components/commit/WorkflowBox.tsx     | 47 +++++++++++++++--
 torchci/lib/fetchCommit.ts                    | 50 +++++++++++++++++--
 .../[repoOwner]/[repoName]/commit/[sha].ts    |  5 +-
 6 files changed, 111 insertions(+), 11 deletions(-)

diff --git a/torchci/clickhouse_queries/commit_jobs_query/params.json b/torchci/clickhouse_queries/commit_jobs_query/params.json
index 270d08eaf7..36cf80d8ee 100644
--- a/torchci/clickhouse_queries/commit_jobs_query/params.json
+++ b/torchci/clickhouse_queries/commit_jobs_query/params.json
@@ -1,12 +1,14 @@
 {
   "params": {
     "repo": "String",
-    "sha": "String"
+    "sha": "String",
+    "workflowId": "Int64"
   },
   "tests": [
     {
       "repo": "pytorch/pytorch",
-      "sha": "85df746892d9b0e87e7a5dfa78ef81a84aec6de0"
+      "sha": "85df746892d9b0e87e7a5dfa78ef81a84aec6de0",
+      "workflow_id": 0
     }
   ]
 }
diff --git a/torchci/clickhouse_queries/commit_jobs_query/query.sql b/torchci/clickhouse_queries/commit_jobs_query/query.sql
index 8c84e8f106..d0440b0d06 100644
--- a/torchci/clickhouse_queries/commit_jobs_query/query.sql
+++ b/torchci/clickhouse_queries/commit_jobs_query/query.sql
@@ -46,6 +46,10 @@ WITH job AS (
         AND workflow.event != 'workflow_run' -- Filter out workflow_run-triggered jobs, which have nothing to do with the SHA
         AND workflow.event != 'repository_dispatch' -- Filter out repository_dispatch-triggered jobs, which have nothing to do with the SHA
         AND workflow.id in (select id from materialized_views.workflow_run_by_head_sha where head_sha = {sha: String})
+        AND (
+            {workflowId: Int64} = 0
+            OR workflow.id = {workflowId: Int64} -- If a specific workflow ID is provided, filter by it
+        )
         AND job.id in (select id from materialized_views.workflow_job_by_head_sha where head_sha = {sha: String})
         AND workflow.repository. 'full_name' = {repo: String } --         UNION
         AND workflow.name != 'Upload test stats while running' -- Continuously running cron job that cancels itself to avoid running concurrently
@@ -84,6 +88,10 @@ WITH job AS (
         workflow.event != 'workflow_run' -- Filter out workflow_run-triggered jobs, which have nothing to do with the SHA
         AND workflow.event != 'repository_dispatch' -- Filter out repository_dispatch-triggered jobs, which have nothing to do with the SHA
         AND workflow.id in (select id from materialized_views.workflow_run_by_head_sha where head_sha = {sha: String})
+        AND (
+            {workflowId: Int64} = 0
+            OR workflow.id = {workflowId: Int64} -- If a specific workflow ID is provided, filter by it
+        )
         AND workflow.repository.full_name = {repo: String }
         AND workflow.name != 'Upload test stats while running' -- Continuously running cron job that cancels itself to avoid running concurrently
 )
diff --git a/torchci/components/commit/CommitStatus.tsx b/torchci/components/commit/CommitStatus.tsx
index 1fdcb35a31..7487afe4c3 100644
--- a/torchci/components/commit/CommitStatus.tsx
+++ b/torchci/components/commit/CommitStatus.tsx
@@ -54,10 +54,12 @@ function getBoxOrdering(jobs: JobData[], wideBoxes: Set<string>) {
 function WorkflowsContainer({
   jobs,
   unstableIssues,
+  workflowIdsByName,
   repoFullName,
 }: {
   jobs: JobData[];
   unstableIssues: IssueData[];
+  workflowIdsByName: Record<string, number[]>;
   repoFullName: string;
 }) {
   useScrollTo();
@@ -81,6 +83,7 @@ function WorkflowsContainer({
               repoFullName={repoFullName}
               key={workflowName}
               workflowName={workflowName}
+              allWorkflowIds={workflowIdsByName[workflowName] || []}
               jobs={jobs}
               unstableIssues={unstableIssues}
               wide={wideBoxes.has(workflowName)}
@@ -106,6 +109,7 @@ export default function CommitStatus({
   repoName,
   commit,
   jobs,
+  workflowIdsByName,
   isCommitPage,
   unstableIssues,
 }: {
@@ -113,6 +117,7 @@ export default function CommitStatus({
   repoName: string;
   commit: CommitData;
   jobs: JobData[];
+  workflowIdsByName: Record<string, number[]>;
   isCommitPage: boolean;
   unstableIssues: IssueData[];
 }) {
@@ -174,6 +179,7 @@ export default function CommitStatus({
       />
       <WorkflowsContainer
         jobs={jobs}
+        workflowIdsByName={workflowIdsByName}
         unstableIssues={unstableIssues}
         repoFullName={`${repoOwner}/${repoName}`}
       />
diff --git a/torchci/components/commit/WorkflowBox.tsx b/torchci/components/commit/WorkflowBox.tsx
index e49d67f61e..a83eff54c5 100644
--- a/torchci/components/commit/WorkflowBox.tsx
+++ b/torchci/components/commit/WorkflowBox.tsx
@@ -1,4 +1,4 @@
-import { Button, Stack, styled, Typography } from "@mui/material";
+import { Button, Stack, styled, Tooltip, Typography } from "@mui/material";
 import { TestInfo } from "components/additionalTestInfo/TestInfo";
 import styles from "components/commit/commit.module.css";
 import LogViewer, { SearchLogViewer } from "components/common/log/LogViewer";
@@ -14,7 +14,10 @@ import {
   ListUtilizationMetadataInfoAPIResponse,
   UtilizationMetadataInfo,
 } from "lib/utilization/types";
+import { CommitApiResponse } from "pages/api/[repoOwner]/[repoName]/commit/[sha]";
 import React, { useEffect, useState } from "react";
+import { FaInfoCircle } from "react-icons/fa";
+import useSWR from "swr";
 import useSWRImmutable from "swr/immutable";
 
 function sortJobsByConclusion(jobA: JobData, jobB: JobData): number {
@@ -140,16 +143,32 @@ export default function WorkflowBox({
   unstableIssues,
   wide,
   setWide,
+  allWorkflowIds,
   repoFullName,
 }: {
   workflowName: string;
   jobs: JobData[];
   unstableIssues: IssueData[];
   wide: boolean;
+  allWorkflowIds: number[];
   setWide: any;
   repoFullName: string;
 }) {
-  const workflowId = jobs[0].workflowId;
+  const [selectedWorkflowId, setSelectedWorkflowId] = useState<
+    string | undefined
+  >(undefined);
+  const workflowId = selectedWorkflowId || jobs[0].workflowId;
+
+  const { data: jobsFromSelectedWorkflowId } = useSWR<CommitApiResponse>(
+    selectedWorkflowId &&
+      `/api/${repoFullName}/commit/${jobs[0].sha}?workflowId=${selectedWorkflowId}`,
+    fetcher
+  );
+
+  if (selectedWorkflowId) {
+    jobs = jobsFromSelectedWorkflowId?.jobs || [];
+  }
+
   const isFailed = jobs.some(isFailedJob) !== false;
   const workflowClass = isFailed
     ? styles.workflowBoxFail
@@ -163,6 +182,7 @@ export default function WorkflowBox({
   const { artifacts, error } = useArtifacts(jobs.map((job) => job.workflowId));
   const [artifactsToShow, setArtifactsToShow] = useState(new Set<string>());
   const groupedArtifacts = groupArtifacts(jobs, artifacts);
+
   const [searchString, setSearchString] = useState("");
   const [searchRes, setSearchRes] = useState<{
     results: Map<string, LogSearchResult>;
@@ -196,7 +216,28 @@ export default function WorkflowBox({
             Job Status
           </Typography>
         </Stack>
-        <Stack direction="column" spacing={1} paddingTop={6}>
+        <Stack direction="column" spacing={1}>
+          <Stack direction="row" spacing={1}>
+            <select
+              value={selectedWorkflowId}
+              onChange={(e) => {
+                setSelectedWorkflowId(e.target.value);
+              }}
+              style={{ width: "100%" }}
+            >
+              <option value={""}>Select Workflow ID</option>
+              {allWorkflowIds.sort().map((id) => (
+                <option key={id} value={id}>
+                  {id}
+                </option>
+              ))}
+            </select>
+            <Tooltip title="By default the box will show what it believes to be the latest jobs. Use this to select a specific workflow ID if it's wrong.">
+              <Typography>
+                <FaInfoCircle />
+              </Typography>
+            </Tooltip>
+          </Stack>
           <div>
             {repoFullName == "pytorch/pytorch" && (
               <button
diff --git a/torchci/lib/fetchCommit.ts b/torchci/lib/fetchCommit.ts
index 70f10c2cb9..3434218468 100644
--- a/torchci/lib/fetchCommit.ts
+++ b/torchci/lib/fetchCommit.ts
@@ -1,14 +1,21 @@
 import _ from "lodash";
 import { Octokit } from "octokit";
+import { CommitApiResponse } from "pages/api/[repoOwner]/[repoName]/commit/[sha]";
 import { queryClickhouseSaved } from "./clickhouse";
 import { commitDataFromResponse, getOctokit } from "./github";
 import { removeCancelledJobAfterRetry } from "./jobUtils";
-import { CommitData, JobData } from "./types";
+import { JobData } from "./types";
 
-async function fetchDatabaseInfo(owner: string, repo: string, sha: string) {
+async function fetchDatabaseInfo(
+  owner: string,
+  repo: string,
+  sha: string,
+  workflowId: number
+) {
   const response = await queryClickhouseSaved("commit_jobs_query", {
     repo: `${owner}/${repo}`,
     sha: sha,
+    workflowId,
   });
 
   for (const row of response) {
@@ -20,21 +27,53 @@ async function fetchDatabaseInfo(owner: string, repo: string, sha: string) {
   return response;
 }
 
+/**
+ * Get a mapping of workflow names to all workflow IDs from a list of job data.
+ * @param jobs
+ * @returns
+ */
+function getWorkflowIdsByName(jobs: JobData[]): Record<string, number[]> {
+  return _(jobs)
+    .groupBy((job) => job.workflowName)
+    .map((jobs, key) => {
+      const workflowIds = _(jobs)
+        .map((job) => job.workflowId)
+        .filter((id) => id !== null && id !== undefined)
+        .uniq()
+        .value();
+      return [key, workflowIds];
+    })
+    .fromPairs()
+    .value();
+}
+
+/**
+ *
+ * @param owner
+ * @param repo
+ * @param sha
+ * @param workflowId Optional workflow ID to filter jobs by. If not provided,
+ * all jobs for the commit will be returned.
+ * @returns
+ */
 export default async function fetchCommit(
   owner: string,
   repo: string,
-  sha: string
-): Promise<{ commit: CommitData; jobs: JobData[] }> {
+  sha: string,
+  workflowId: number = 0
+): Promise<CommitApiResponse> {
   // Retrieve commit data from GitHub
   const octokit = await getOctokit(owner, repo);
 
   const [githubResponse, response] = await Promise.all([
     octokit.rest.repos.getCommit({ owner, repo, ref: sha }),
-    await fetchDatabaseInfo(owner, repo, sha),
+    await fetchDatabaseInfo(owner, repo, sha, workflowId),
   ]);
 
   let jobs = response as any[];
 
+  const workflowIdsByName = getWorkflowIdsByName(jobs);
+
   // Subtle: we need to unique jobs by name, taking the most recent job. This is
   // because there might be many periodic jobs with the same name, and we want
   // to avoid noising up the display with many duplicate jobs.
@@ -65,6 +104,7 @@ export default async function fetchCommit(
   return {
     commit: commitDataFromResponse(githubResponse.data),
     jobs: _.concat(filteredJobs, badWorkflows),
+    workflowIdsByName,
   };
 }
 
diff --git a/torchci/pages/api/[repoOwner]/[repoName]/commit/[sha].ts b/torchci/pages/api/[repoOwner]/[repoName]/commit/[sha].ts
index 4ebe65ad12..6d3acaa014 100644
--- a/torchci/pages/api/[repoOwner]/[repoName]/commit/[sha].ts
+++ b/torchci/pages/api/[repoOwner]/[repoName]/commit/[sha].ts
@@ -5,19 +5,22 @@ import type { NextApiRequest, NextApiResponse } from "next";
 export type CommitApiResponse = {
   commit: CommitData;
   jobs: JobData[];
+  workflowIdsByName: Record<string, number[]>;
 };
 
 export default async function handler(
   req: NextApiRequest,
   res: NextApiResponse<CommitApiResponse>
 ) {
+  const workflowId = parseInt(req.query.workflowId as string, 10) || 0;
   res
     .status(200)
     .json(
       await fetchCommit(
         req.query.repoOwner as string,
         req.query.repoName as string,
-        req.query.sha as string
+        req.query.sha as string,
+        workflowId
       )
     );
 }

From d700b50ebebe79f053c3df5052d4768fc041075b Mon Sep 17 00:00:00 2001
From: Catherine Lee <csl@fb.com>
Date: Thu, 17 Jul 2025 13:56:56 -0700
Subject: [PATCH 2/2] tc

---
 torchci/components/commit/CommitInfo.tsx | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/torchci/components/commit/CommitInfo.tsx b/torchci/components/commit/CommitInfo.tsx
index 6a4ffae5dd..1870b0532a 100644
--- a/torchci/components/commit/CommitInfo.tsx
+++ b/torchci/components/commit/CommitInfo.tsx
@@ -43,7 +43,7 @@ export function CommitInfo({
     return <div>Loading...</div>;
   }
 
-  const { commit, jobs } = commitData;
+  const { commit, jobs, workflowIdsByName } = commitData;
 
   return (
     <div>
@@ -53,6 +53,7 @@ export function CommitInfo({
         repoName={repoName}
         commit={commit}
         jobs={jobs}
+        workflowIdsByName={workflowIdsByName}
         isCommitPage={isCommitPage}
         unstableIssues={unstableIssuesData ?? []}
       />