[AI] Show task-specific widgets insted of CWV widget when a task is selected as context

When the initial context is a call tree, show TIMELINE_RANGE_SUMMARY
and BOTTOM_UP_TREE widgets instead of CORE_VITALS widget.
This provides more relevant context for analyzing specific tasks.

When the context is trace, we still only show CWV widget.

Bug: 498329558
Change-Id: If9163f3f608c1552a5185f85267e6224f9407ee5
Reviewed-on: https://chromium-review.googlesource.com/c/devtools/devtools-frontend/+/7762247
Commit-Queue: Alina Varkki <alinavarkki@chromium.org>
Reviewed-by: Jack Franklin <jacktfranklin@chromium.org>
Auto-Submit: Alina Varkki <alinavarkki@chromium.org>

Author: AlinaVarkki

front_end/models/ai_assistance/agents/PerformanceAgent.snapshot.txt

@@ -11,7 +11,10 @@ Content:
     ],
     "widgets": [
       {
-        "name": "CORE_VITALS"
+        "name": "TIMELINE_RANGE_SUMMARY"
+      },
+      {
+        "name": "BOTTOM_UP_TREE"
       }
     ]
   },

front_end/models/ai_assistance/agents/PerformanceAgent.test.ts

@@ -195,6 +195,33 @@ describeWithMockConnection('PerformanceAgent', function() {
           },
         ]);
       });
+
+      it('yields TIMELINE_RANGE_SUMMARY and BOTTOM_UP_TREE widgets for call tree focus on initialization',
+         async function() {
+           const parsedTrace = await TraceLoader.traceEngine(this, 'web-dev-outermost-frames.json.gz');
+           const events = allThreadEntriesInTrace(parsedTrace);
+           const layoutEvt = events.find(event => event.ts === 465457096322);
+           assert.exists(layoutEvt);
+
+           const aiCallTree = AICallTree.AICallTree.fromEvent(layoutEvt, parsedTrace);
+           assert.exists(aiCallTree);
+
+           const agent = new PerformanceAgent.PerformanceAgent({
+             aidaClient: mockAidaClient([[{explanation: 'done'}]]),
+           });
+
+           const context = PerformanceAgent.PerformanceTraceContext.fromCallTree(aiCallTree);
+           const responses = await Array.fromAsync(agent.run('test', {selected: context}));
+
+           deleteAllWidgetData(responses);
+
+           const contextResponse = responses.find(r => r.type === AiAgent.ResponseType.CONTEXT);
+           assert.exists(contextResponse);
+           assert.exists(contextResponse.widgets);
+           assert.lengthOf(contextResponse.widgets, 2);
+           assert.strictEqual(contextResponse.widgets[0].name, 'TIMELINE_RANGE_SUMMARY');
+           assert.strictEqual(contextResponse.widgets[1].name, 'BOTTOM_UP_TREE');
+         });
     });
 
     describe('enhanceQuery', () => {

front_end/models/ai_assistance/agents/PerformanceAgent.ts

@@ -432,6 +432,7 @@ export class PerformanceAgent extends AiAgent<AgentFocus> {
    * we only show it once.
    */
   #hasShownWidgetForInsightSet = new WeakSet<Trace.Insights.Types.InsightSet>();
+  #hasShownWidgetForCallTree = new WeakSet<AICallTree>();
 
   get preamble(): string {
     return buildPreamble();
@@ -472,16 +473,44 @@ export class PerformanceAgent extends AiAgent<AgentFocus> {
     contextDisclosure.push(...this.#additionalSelectionsForQuery);
 
     const widgets: AiWidget[] = [];
-    const primaryInsightSet = context.getItem().primaryInsightSet;
-    if (primaryInsightSet && !this.#hasShownWidgetForInsightSet.has(primaryInsightSet)) {
-      widgets.push({
-        name: 'CORE_VITALS',
-        data: {
-          parsedTrace: context.getItem().parsedTrace,
-          insightSetKey: primaryInsightSet.id,
-        },
-      });
-      this.#hasShownWidgetForInsightSet.add(primaryInsightSet);
+    const focus = context.getItem();
+
+    // If the user has selected a specific task (call tree) as context, show the summary and bottom-up tree for it.
+    // Otherwise, show the high-level Core Web Vitals widget for the trace or insight.
+    if (focus.callTree && !this.#hasShownWidgetForCallTree.has(focus.callTree)) {
+      const event = focus.callTree.selectedNode?.event;
+      if (event) {
+        const {startTime, endTime} = Trace.Helpers.Timing.eventTimingsMicroSeconds(event);
+        const bounds = Trace.Helpers.Timing.traceWindowFromMicroSeconds(startTime, endTime);
+        widgets.push({
+          name: 'TIMELINE_RANGE_SUMMARY',
+          data: {
+            bounds,
+            parsedTrace: focus.parsedTrace,
+            track: 'main',
+          },
+        });
+        widgets.push({
+          name: 'BOTTOM_UP_TREE',
+          data: {
+            bounds,
+            parsedTrace: focus.parsedTrace,
+          },
+        });
+        this.#hasShownWidgetForCallTree.add(focus.callTree);
+      }
+    } else {
+      const primaryInsightSet = focus.primaryInsightSet;
+      if (primaryInsightSet && !this.#hasShownWidgetForInsightSet.has(primaryInsightSet)) {
+        widgets.push({
+          name: 'CORE_VITALS',
+          data: {
+            parsedTrace: focus.parsedTrace,
+            insightSetKey: primaryInsightSet.id,
+          },
+        });
+        this.#hasShownWidgetForInsightSet.add(primaryInsightSet);
+      }
     }
 
     yield {