update default investigation name with auto LLM generated name (#256)

Signed-off-by: Hailong Cui <[email protected]>

Author: Hailong-am

common/constants/notebooks.ts

@@ -30,3 +30,5 @@ export const NOTEBOOK_APP_NAME = 'investigate-notebook';
 export const OBSERVABILITY_VISUALIZATION_TYPE = 'observability-visualization';
 
 export const DASHBOARDS_VISUALIZATION_TYPE = 'visualization';
+
+export const DEFAULT_INVESTIGATION_NAME = 'Discover investigation';

common/types/notebooks.ts

@@ -159,6 +159,7 @@ export interface PERAgentHypothesisItem {
 export interface PERAgentInvestigationResponse {
   findings: PERAgentHypothesisFinding[];
   hypotheses: PERAgentHypothesisItem[];
+  investigationName?: string;
 }
 
 export interface NotebookComponentProps {

public/components/notebooks/components/discover_explorer/__tests__/start_investigation_modal.test.tsx

@@ -13,7 +13,10 @@ import {
   httpServiceMock,
   notificationServiceMock,
 } from '../../../../../../../../src/core/public/mocks';
-import { NOTEBOOKS_API_PREFIX } from '../../../../../../common/constants/notebooks';
+import {
+  DEFAULT_INVESTIGATION_NAME,
+  NOTEBOOKS_API_PREFIX,
+} from '../../../../../../common/constants/notebooks';
 
 describe('<StartInvestigationModal /> spec', () => {
   const httpMock = httpServiceMock.createStartContract();
@@ -134,7 +137,7 @@ describe('<StartInvestigationModal /> spec', () => {
       expect(httpMock.post).toHaveBeenCalledWith(
         `${NOTEBOOKS_API_PREFIX}/note/savedNotebook`,
         expect.objectContaining({
-          body: expect.stringContaining('Discover investigation'),
+          body: expect.stringContaining(DEFAULT_INVESTIGATION_NAME),
         })
       );
     });

public/components/notebooks/components/discover_explorer/start_investigation_modal.tsx

@@ -23,7 +23,10 @@ import {
 import React, { useState, useMemo } from 'react';
 import { useOpenSearchDashboards } from '../../../../../../../src/plugins/opensearch_dashboards_react/public';
 import type { NoteBookServices } from '../../../../types';
-import { NOTEBOOKS_API_PREFIX } from '../../../../../common/constants/notebooks';
+import {
+  DEFAULT_INVESTIGATION_NAME,
+  NOTEBOOKS_API_PREFIX,
+} from '../../../../../common/constants/notebooks';
 
 const suggestedActions = [
   {
@@ -116,7 +119,7 @@ export const StartInvestigationModal = ({ log, closeModal }: StartInvestigationM
     }
     setDisabled(true);
     try {
-      const id = await createNotebook('Discover investigation');
+      const id = await createNotebook(DEFAULT_INVESTIGATION_NAME);
       const path = `#/agentic/${id}`;
       application.navigateToApp('investigation-notebooks', {
         path,

public/hooks/use_investigation.ts

@@ -25,6 +25,7 @@ import { AgenticMemeory, PERAgentInvestigationResponse } from '../../common/type
 import { isValidPERAgentInvestigationResponse } from '../../common/utils/per_agent';
 import { useNotebook } from './use_notebook';
 import { generateContextPromptFromParagraphs } from '../services/helpers/per_agent';
+import { DEFAULT_INVESTIGATION_NAME, NOTEBOOKS_API_PREFIX } from '../../common/constants/notebooks';
 
 const getFindingFromParagraph = (paragraph: ParagraphStateValue<unknown>) => {
   return `
@@ -144,6 +145,34 @@ ${finding.evidence}
     [updateHypotheses, batchCreateParagraphs, batchRunParagraphs]
   );
 
+  /**
+   * Update investigation title based on the suggested title from the agent only if investigation use the default value
+   */
+  const updateInvestigationName = useCallback(
+    async (suggestedName: string) => {
+      const isDefaultName = contextStateValue?.title === DEFAULT_INVESTIGATION_NAME;
+      if (isDefaultName && suggestedName) {
+        try {
+          const autoGeneratedName = suggestedName.substring(0, 50);
+          const { id: openedNoteId } = context.state.value;
+          await http.put(`${NOTEBOOKS_API_PREFIX}/note/savedNotebook/rename`, {
+            body: JSON.stringify({
+              name: autoGeneratedName,
+              noteId: openedNoteId,
+            }),
+          });
+
+          // Update local state to reflect invesitgation header
+          context.state.updateValue({ title: autoGeneratedName, path: autoGeneratedName });
+        } catch (error) {
+          // Don't fail the entire investigation if title update fails
+          console.error('Failed to update investigation title:', error);
+        }
+      }
+    },
+    [contextStateValue?.title, context.state, http]
+  );
+
   /**
    * Poll for investigation completion and process the response
    * @returns Promise that resolves when investigation is complete or rejects on error
@@ -219,6 +248,11 @@ ${finding.evidence}
                 throw error;
               }
 
+              // Update notebook title if suggested_title is provided and name is default investigation name
+              if (responseJson.investigationName) {
+                await updateInvestigationName(responseJson.investigationName);
+              }
+
               context.state.updateValue({
                 historyMemory: runningMemory,
                 investigationError: undefined,
@@ -253,6 +287,7 @@ ${finding.evidence}
       notifications,
       context.state,
       storeInvestigationResponse,
+      updateInvestigationName,
       updateHypotheses,
       batchDeleteParagraphs,
     ]

server/routes/notebooks/agent_router.ts

@@ -51,7 +51,8 @@ Final result must be a stringified JSON object:
 \`\`\`json
 {
     "findings": array[object],
-    "hypotheses": array[object]
+    "hypotheses": array[object],
+    "investigationName": "string object which will be the auto generated name for the whole investigation, max 50 characters"
 }
 \`\`\`
 
@@ -107,7 +108,7 @@ Your final result JSON must include:
 \`\`\`json
 {
   "steps": [],
-  "result": "{\"findings\":[{\"id\":\"F1\",\"description\":\"High error rate detected\",\"importance\":90,\"evidence\":\"500+ errors in last hour\"}],\"hypotheses\":[{\"id\":\"H1\",\"title\":\"Database Connection Issue\",\"description\":\"Application errors caused by database connectivity problems\",\"likelihood\":85,\"supporting_findings\":[\"F1\"]}]}"
+  "result": "{\"investigationName\": \"Invalid payment token Investigation\",\"findings\":[{\"id\":\"F1\",\"description\":\"High error rate detected\",\"importance\":90,\"evidence\":\"500+ errors in last hour\"}],\"hypotheses\":[{\"id\":\"H1\",\"title\":\"Database Connection Issue\",\"description\":\"Application errors caused by database connectivity problems\",\"likelihood\":85,\"supporting_findings\":[\"F1\"]}]}"
 }
 \`\`\`