From c7a809996c741026f93896412d419a122ad6ce76 Mon Sep 17 00:00:00 2001
From: Ashton Galloway <ashton.galloway@hpe.com>
Date: Fri, 26 Jul 2024 16:16:39 -0400
Subject: [PATCH] fix: load trial data for single run searches in search view

This loads trial data for single run searches in the search view. this
prevents issues with the continue trial where the trial information was
missing in the experiment config.
---
 webui/react/src/pages/SearchDetails.tsx | 35 ++++++++++++++++++++++---
 1 file changed, 32 insertions(+), 3 deletions(-)

diff --git a/webui/react/src/pages/SearchDetails.tsx b/webui/react/src/pages/SearchDetails.tsx
index 7ab988ca423..766c4b506c2 100644
--- a/webui/react/src/pages/SearchDetails.tsx
+++ b/webui/react/src/pages/SearchDetails.tsx
@@ -1,6 +1,6 @@
 import Pivot, { PivotProps } from 'hew/Pivot';
 import Notes from 'hew/RichTextEditor';
-import { Loadable } from 'hew/utils/loadable';
+import { Loadable, NotLoaded } from 'hew/utils/loadable';
 import { string } from 'io-ts';
 import _ from 'lodash';
 import { useObservable } from 'micro-observables';
@@ -9,14 +9,16 @@ import { unstable_useBlocker, useLocation, useNavigate, useParams } from 'react-
 
 import Page, { BreadCrumbRoute } from 'components/Page';
 import { terminalRunStates } from 'constants/states';
+import { useAsync } from 'hooks/useAsync';
 import usePermissions from 'hooks/usePermissions';
 import usePolling from 'hooks/usePolling';
 import { SettingsConfig, useSettings } from 'hooks/useSettings';
 import { paths } from 'routes/utils';
-import { getExperimentDetails, patchExperiment } from 'services/api';
+import { getExperimentDetails, getExpTrials, patchExperiment } from 'services/api';
 import workspaceStore from 'stores/workspaces';
-import { ExperimentBase, Note, ValueOf, Workspace } from 'types';
+import { ExperimentBase, Note, TrialItem, ValueOf, Workspace } from 'types';
 import handleError, { ErrorLevel, ErrorType } from 'utils/error';
+import { isSingleTrialExperiment } from 'utils/experiment';
 import { isAborted, isNotFound } from 'utils/service';
 
 import ExperimentCodeViewer from './ExperimentDetails/ExperimentCodeViewer';
@@ -120,6 +122,32 @@ const SearchDetails: React.FC = () => {
     }
   }, [pageError, searchId]);
 
+  const singleTrialData = useAsync<TrialItem | undefined>(
+    async (canceler) => {
+      if (!experiment || !isSingleTrialExperiment(experiment)) {
+        return NotLoaded;
+      }
+      try {
+        const trialsResponse = await getExpTrials(
+          { id: experiment.id, limit: 2 },
+          { signal: canceler.signal },
+        );
+        return trialsResponse.trials[0];
+      } catch (e) {
+        if (!canceler.signal.aborted) {
+          handleError(e, {
+            level: ErrorLevel.Error,
+            publicMessage: 'Failed to fetch run information for search',
+            silent: false,
+            type: ErrorType.Server,
+          });
+        }
+      }
+      return NotLoaded;
+    },
+    [experiment],
+  );
+
   const handleNotesUpdate = useCallback(
     async (notes: Note) => {
       const editedNotes = notes.contents;
@@ -237,6 +265,7 @@ const SearchDetails: React.FC = () => {
           <ExperimentDetailsHeader
             experiment={experiment}
             fetchExperimentDetails={fetchExperimentDetails}
+            trial={singleTrialData.getOrElse(undefined)}
           />
         )
       }