[AI Assistance] keep track of processed files

Bug: 402034781
Change-Id: I86cd779ec19d39a8c86c1f436aa6f8087715f907
Reviewed-on: https://chromium-review.googlesource.com/c/devtools/devtools-frontend/+/6348899
Reviewed-by: Ergün Erdoğmuş <[email protected]>
Auto-Submit: Alex Rudenko <[email protected]>
Commit-Queue: Ergün Erdoğmuş <[email protected]>

Author: OrKoN

front_end/panels/ai_assistance/AgentProject.test.ts

@@ -59,6 +59,13 @@ describeWithEnvironment('AgentProject', () => {
     assert.deepEqual(project.readFile('index.html'), 'content');
   });
 
+  it('can report processed files', async () => {
+    const {project} = await mockProject();
+    assert.deepEqual(project.getProcessedFiles(), []);
+    project.readFile('index.html');
+    assert.deepEqual(project.getProcessedFiles(), ['index.html']);
+  });
+
   it('can write files files', async () => {
     const {project} = await mockProject();
     project.writeFile('index.html', 'updated');

front_end/panels/ai_assistance/AgentProject.ts

@@ -22,6 +22,7 @@ export class AgentProject {
 
   readonly #maxFilesChanged: number;
   readonly #maxLinesChanged: number;
+  readonly #processedFiles = new Set<string>();
 
   constructor(project: Workspace.Workspace.Project, options: {
     maxFilesChanged: number,
@@ -35,6 +36,14 @@ export class AgentProject {
     this.#maxLinesChanged = options.maxLinesChanged;
   }
 
+  /**
+   * Returns a list of files from the project that has been used for
+   * processing.
+   */
+  getProcessedFiles(): string[] {
+    return Array.from(this.#processedFiles);
+  }
+
   /**
    * Provides file names in the project to the agent.
    */
@@ -52,6 +61,7 @@ export class AgentProject {
     if (!uiSourceCode) {
       return;
     }
+    this.#processedFiles.add(filepath);
     // TODO: needs additional handling for binary files.
     return uiSourceCode.workingCopyContentData().text;
   }

front_end/panels/ai_assistance/PatchWidget.ts

@@ -465,8 +465,7 @@ export class PatchWidget extends UI.Widget.Widget {
       serverSideLoggingEnabled: false,
       project: this.#project,
     });
-    const responses =
-        await Array.fromAsync(agent.applyChanges(changeSummary, {signal: this.#applyPatchAbortController.signal}));
+    const {responses} = await agent.applyChanges(changeSummary, {signal: this.#applyPatchAbortController.signal});
     return responses.at(-1);
   }
 }

front_end/panels/ai_assistance/agents/PatchAgent.test.ts

@@ -37,7 +37,9 @@ describeWithEnvironment('PatchAgent', () => {
       })
     });
 
-    return await Array.fromAsync(agent.applyChanges('summary'));
+    const {responses} = await agent.applyChanges('summary');
+
+    return responses;
   }
 
   it('calls listFiles', async () => {

front_end/panels/ai_assistance/agents/PatchAgent.ts

@@ -183,9 +183,10 @@ ${content}
     });
   }
 
-  async *
-      applyChanges(changeSummary: string, {signal}: {signal?: AbortSignal} = {}):
-          AsyncGenerator<ResponseData, void, void> {
+  async applyChanges(changeSummary: string, {signal}: {signal?: AbortSignal} = {}): Promise<{
+    responses: ResponseData[],
+    processedFiles: string[],
+  }> {
     this.#changeSummary = changeSummary;
     const prompt =
         `I have applied the following CSS changes to my page in Chrome DevTools, what are the files in my source code that I need to change to apply the same change?
@@ -203,7 +204,11 @@ CRITICAL: before searching always call listFiles first.
 CRITICAL: never call updateFiles with files that do not need updates.
 `;
 
-    yield* this.run(prompt, {selected: null, signal});
+    const responses = await Array.fromAsync(this.run(prompt, {selected: null, signal}));
+    return {
+      responses,
+      processedFiles: this.#project.getProcessedFiles(),
+    };
   }
 }