[AI Assistance] Add experimental workspace agent

Not intended to be used outside of internal tester groups
and not enabled by default.

Bug: 385082492
Change-Id: I705dd9ee48fe6ccec3b373a2d9fbc76ea45e6551
Reviewed-on: https://chromium-review.googlesource.com/c/devtools/devtools-frontend/+/6097889
Reviewed-by: Simon Zünd <[email protected]>
Commit-Queue: Alex Rudenko <[email protected]>

Author: OrKoN

config/gni/devtools_grd_files.gni

@@ -1161,6 +1161,7 @@ grd_files_debug_sources = [
   "front_end/panels/ai_assistance/agents/AiAgent.js",
   "front_end/panels/ai_assistance/agents/FileAgent.js",
   "front_end/panels/ai_assistance/agents/NetworkAgent.js",
+  "front_end/panels/ai_assistance/agents/PatchAgent.js",
   "front_end/panels/ai_assistance/agents/PerformanceAgent.js",
   "front_end/panels/ai_assistance/agents/StylingAgent.js",
   "front_end/panels/ai_assistance/aiAssistancePanel.css.js",

front_end/core/host/AidaClient.ts

@@ -123,6 +123,8 @@ export enum ClientFeature {
   CHROME_PERFORMANCE_AGENT = 8,
   // Chrome AI Assistance File Agent.
   CHROME_FILE_AGENT = 9,
+  // Chrome AI Patch Agent.
+  CHROME_PATCH_AGENT = 12,
 }
 
 export enum UserTier {

front_end/panels/ai_assistance/AiAssistancePanel.ts

@@ -6,6 +6,7 @@ import * as Host from '../../core/host/host.js';
 import * as i18n from '../../core/i18n/i18n.js';
 import type * as Platform from '../../core/platform/platform.js';
 import * as SDK from '../../core/sdk/sdk.js';
+import * as Persistence from '../../models/persistence/persistence.js';
 import * as Workspace from '../../models/workspace/workspace.js';
 import * as UI from '../../ui/legacy/legacy.js';
 import * as LitHtml from '../../ui/lit-html/lit-html.js';
@@ -34,6 +35,7 @@ import {
   NetworkAgent,
   RequestContext,
 } from './agents/NetworkAgent.js';
+import {PatchAgent, ProjectContext} from './agents/PatchAgent.js';
 import {CallTreeContext, PerformanceAgent} from './agents/PerformanceAgent.js';
 import {NodeContext, StylingAgent} from './agents/StylingAgent.js';
 import styles from './aiAssistancePanel.css.js';
@@ -158,6 +160,13 @@ function createFileContext(file: Workspace.UISourceCode.UISourceCode|null): File
   return new FileContext(file);
 }
 
+function createProjectContext(project: Persistence.FileSystemWorkspaceBinding.FileSystem|null): ProjectContext|null {
+  if (!project) {
+    return null;
+  }
+  return new ProjectContext(project);
+}
+
 function createRequestContext(request: SDK.NetworkRequest.NetworkRequest|null): RequestContext|null {
   if (!request) {
     return null;
@@ -196,6 +205,7 @@ export class AiAssistancePanel extends UI.Panel.Panel {
   #currentAgent?: AiAgent<unknown>;
 
   #previousSameOriginContext?: ConversationContext<unknown>;
+  #project: ProjectContext|null = null;
   #selectedFile: FileContext|null = null;
   #selectedElement: NodeContext|null = null;
   #selectedCallTree: CallTreeContext|null = null;
@@ -330,6 +340,10 @@ export class AiAssistancePanel extends UI.Panel.Panel {
         agent = new PerformanceAgent(options);
         break;
       }
+      case AgentType.PATCH: {
+        agent = new PatchAgent(options);
+        break;
+      }
     }
 
     if (history) {
@@ -421,6 +435,8 @@ export class AiAssistancePanel extends UI.Panel.Panel {
     this.#selectedCallTree =
         createCallTreeContext(UI.Context.Context.instance().flavor(TimelineUtils.AICallTree.AICallTree)),
     this.#selectedFile = createFileContext(UI.Context.Context.instance().flavor(Workspace.UISourceCode.UISourceCode)),
+    this.#project =
+        createProjectContext(UI.Context.Context.instance().flavor(Persistence.FileSystemWorkspaceBinding.FileSystem)),
     this.#viewProps = {
       ...this.#viewProps,
       agentType: this.#currentAgent?.type,
@@ -441,6 +457,8 @@ export class AiAssistancePanel extends UI.Panel.Panel {
         TimelineUtils.AICallTree.AICallTree, this.#handleTraceEntryNodeFlavorChange);
     UI.Context.Context.instance().addFlavorChangeListener(
         Workspace.UISourceCode.UISourceCode, this.#handleUISourceCodeFlavorChange);
+    UI.Context.Context.instance().addFlavorChangeListener(
+        Persistence.FileSystemWorkspaceBinding.FileSystem, this.#handlePersistenceFileSystemChange);
     UI.Context.Context.instance().addFlavorChangeListener(
         ElementsPanel.ElementsPanel.ElementsPanel, this.#selectDefaultAgentIfNeeded, this);
     UI.Context.Context.instance().addFlavorChangeListener(
@@ -566,6 +584,19 @@ export class AiAssistancePanel extends UI.Panel.Panel {
         this.#updateAgentState(this.#currentAgent);
       };
 
+  #handlePersistenceFileSystemChange =
+      (ev: Common.EventTarget.EventTargetEvent<Persistence.FileSystemWorkspaceBinding.FileSystem>): void => {
+        const newProject = ev.data;
+        if (!newProject) {
+          return;
+        }
+        if (this.#project?.getItem() === newProject) {
+          return;
+        }
+        this.#project = new ProjectContext(ev.data as Workspace.Workspace.Project);
+        this.#updateAgentState(this.#currentAgent);
+      };
+
   #handleAiAssistanceEnabledSettingChanged = (): void => {
     const nextChatUiState = this.#getChatUiState();
     if (this.#viewProps.state === nextChatUiState) {
@@ -627,6 +658,11 @@ export class AiAssistancePanel extends UI.Panel.Panel {
 
     let targetAgentType: AgentType|undefined;
     switch (actionId) {
+      case 'ai-assistance.filesystem': {
+        // TODO: metrics if needed.
+        targetAgentType = AgentType.PATCH;
+        break;
+      }
       case 'freestyler.elements-floating-button': {
         Host.userMetrics.actionTaken(Host.UserMetrics.Action.AiAssistanceOpenedFromElementsPanelFloatingButton);
         targetAgentType = AgentType.STYLING;
@@ -806,6 +842,9 @@ export class AiAssistancePanel extends UI.Panel.Panel {
       case AgentType.PERFORMANCE:
         context = this.#selectedCallTree;
         break;
+      case AgentType.PATCH:
+        context = this.#project;
+        break;
     }
     return context;
   }
@@ -966,7 +1005,8 @@ export class ActionDelegate implements UI.ActionRegistration.ActionDelegate {
       case 'drjones.network-panel-context':
       case 'drjones.performance-panel-context':
       case 'drjones.sources-floating-button':
-      case 'drjones.sources-panel-context': {
+      case 'drjones.sources-panel-context':
+      case 'ai-assistance.filesystem': {
         void (async () => {
           const view = UI.ViewManager.ViewManager.instance().view(
               AiAssistancePanel.panelName,

front_end/panels/ai_assistance/BUILD.gn

@@ -26,6 +26,7 @@ devtools_module("ai_assistance") {
     "agents/AiAgent.ts",
     "agents/FileAgent.ts",
     "agents/NetworkAgent.ts",
+    "agents/PatchAgent.ts",
     "agents/PerformanceAgent.ts",
     "agents/StylingAgent.ts",
     "components/ChatView.ts",
@@ -40,6 +41,7 @@ devtools_module("ai_assistance") {
     "../../core/root:bundle",
     "../../models/bindings:bundle",
     "../../models/logs:bundle",
+    "../../models/text_utils:bundle",
     "../../models/trace:bundle",
     "../../models/workspace:bundle",
     "../../panels/elements:bundle",

front_end/panels/ai_assistance/agents/AiAgent.ts

@@ -118,6 +118,7 @@ export const enum AgentType {
   FILE = 'drjones-file',
   NETWORK = 'drjones-network-request',
   PERFORMANCE = 'drjones-performance',
+  PATCH = 'patch',
 }
 
 export interface SerializedAgent {