chrome-devtools-frontend 1.0.1376716 → 1.0.1386602

package/front_end/panels/freestyler/AiHistoryStorage.test.ts

@@ -0,0 +1,159 @@
+// Copyright 2023 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+import {
+  describeWithEnvironment,
+} from '../../testing/EnvironmentHelpers.js';
+
+import * as Freestyler from './freestyler.js';
+
+describeWithEnvironment('Freestyler.AiHistoryStorage', () => {
+  const agent1: Freestyler.SerializedAgent = {
+    id: 'id1',
+    type: Freestyler.AgentType.FREESTYLER,
+    history: [],
+  };
+  const agent2: Freestyler.SerializedAgent = {
+    id: 'id2',
+    type: Freestyler.AgentType.DRJONES_FILE,
+    history: [],
+  };
+  const agent3: Freestyler.SerializedAgent = {
+    id: 'id3',
+    type: Freestyler.AgentType.DRJONES_NETWORK_REQUEST,
+    history: [],
+  };
+
+  afterEach(() => {
+    Freestyler.AiHistoryStorage.instance().clearForTest();
+  });
+
+  it('should create and retrieve history entry', async () => {
+    const storage = Freestyler.AiHistoryStorage.instance();
+    await storage.upsertHistoryEntry(agent1);
+    assert.deepEqual(
+        storage.getHistory(),
+        [{
+          id: 'id1',
+          type: 'freestyler' as Freestyler.AgentType,
+          history: [],
+        }],
+    );
+    await storage.upsertHistoryEntry(agent2);
+    assert.deepEqual(
+        storage.getHistory(),
+        [
+          {
+            id: 'id1',
+            type: 'freestyler' as Freestyler.AgentType,
+            history: [],
+          },
+          {
+            id: 'id2',
+            type: 'drjones-file' as Freestyler.AgentType,
+            history: [],
+          },
+        ],
+    );
+  });
+
+  it('should update history entries correctly', async () => {
+    const storage = Freestyler.AiHistoryStorage.instance();
+    await storage.upsertHistoryEntry(agent1);
+    await storage.upsertHistoryEntry(agent2);
+    await storage.upsertHistoryEntry({
+      ...agent1,
+      history: [
+        {
+          type: Freestyler.ResponseType.USER_QUERY,
+          query: 'text',
+        },
+      ],
+    });
+    assert.deepEqual(
+        storage.getHistory(),
+        [
+          {
+            id: 'id1',
+            type: 'freestyler' as Freestyler.AgentType,
+            history: [
+              {
+                type: Freestyler.ResponseType.USER_QUERY,
+                query: 'text',
+              },
+            ],
+          },
+          {
+            id: 'id2',
+            type: 'drjones-file' as Freestyler.AgentType,
+            history: [],
+          },
+        ],
+    );
+
+    await storage.upsertHistoryEntry(agent3);
+    assert.deepEqual(
+        storage.getHistory(),
+        [
+          {
+            id: 'id1',
+            type: 'freestyler' as Freestyler.AgentType,
+            history: [
+              {
+                type: Freestyler.ResponseType.USER_QUERY,
+                query: 'text',
+              },
+            ],
+          },
+          {
+            id: 'id2',
+            type: 'drjones-file' as Freestyler.AgentType,
+            history: [],
+          },
+          {
+            id: 'id3',
+            type: 'drjones-network-request' as Freestyler.AgentType,
+            history: [],
+          },
+        ],
+    );
+  });
+
+  it('should delete a single entry', async () => {
+    const storage = Freestyler.AiHistoryStorage.instance();
+    await storage.upsertHistoryEntry(agent1);
+    await storage.upsertHistoryEntry(agent2);
+    await storage.upsertHistoryEntry(agent3);
+    await storage.deleteHistoryEntry('id2');
+    assert.deepEqual(
+        storage.getHistory(),
+        [
+          {
+            id: 'id1',
+            type: 'freestyler' as Freestyler.AgentType,
+            history: [
+
+            ],
+          },
+          {
+            id: 'id3',
+            type: 'drjones-network-request' as Freestyler.AgentType,
+            history: [],
+          },
+        ],
+    );
+  });
+
+  it('should delete all entries', async () => {
+    const storage = Freestyler.AiHistoryStorage.instance();
+    await storage.upsertHistoryEntry(agent1);
+    await storage.upsertHistoryEntry(agent2);
+    await storage.upsertHistoryEntry(agent3);
+    await storage.deleteAll();
+    assert.deepEqual(
+        storage.getHistory(),
+        [],
+    );
+  });
+});

package/front_end/panels/freestyler/AiHistoryStorage.ts

@@ -0,0 +1,73 @@
+
+// Copyright 2024 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+import * as Common from '../../core/common/common.js';
+
+import type {SerializedAgent} from './AiAgent.js';
+
+let instance: AiHistoryStorage|null = null;
+export class AiHistoryStorage {
+  #historySetting: Common.Settings.Setting<SerializedAgent[]>;
+  #mutex = new Common.Mutex.Mutex();
+
+  constructor() {
+    this.#historySetting = Common.Settings.Settings.instance().createSetting(
+        'ai-assistance-history-entries',
+        [],
+    );
+  }
+
+  clearForTest(): void {
+    this.#historySetting.set([]);
+  }
+
+  async upsertHistoryEntry(agentEntry: SerializedAgent): Promise<void> {
+    const release = await this.#mutex.acquire();
+    try {
+      const history = await this.#historySetting.forceGet();
+      const historyEntryIndex = history.findIndex(entry => entry.id === agentEntry.id);
+      if (historyEntryIndex !== -1) {
+        history[historyEntryIndex] = agentEntry;
+      } else {
+        history.push(agentEntry);
+      }
+      this.#historySetting.set(history);
+    } finally {
+      release();
+    }
+  }
+
+  async deleteHistoryEntry(id: string): Promise<void> {
+    const release = await this.#mutex.acquire();
+    try {
+      const history = await this.#historySetting.forceGet();
+      this.#historySetting.set(
+          history.filter(entry => entry.id !== id),
+      );
+    } finally {
+      release();
+    }
+  }
+
+  async deleteAll(): Promise<void> {
+    const release = await this.#mutex.acquire();
+    try {
+      this.#historySetting.set([]);
+    } finally {
+      release();
+    }
+  }
+
+  getHistory(): SerializedAgent[] {
+    return this.#historySetting.get();
+  }
+
+  static instance(): AiHistoryStorage {
+    if (!instance) {
+      instance = new AiHistoryStorage();
+    }
+    return instance;
+  }
+}

package/front_end/panels/freestyler/DrJonesFileAgent.test.ts

@@ -16,7 +16,7 @@ import {describeWithMockConnection} from '../../testing/MockConnection.js';
 import {loadBasicSourceMapExample} from '../../testing/SourceMapHelpers.js';
 import {createContentProviderUISourceCodes} from '../../testing/UISourceCodeHelpers.js';
 
-import {DrJonesFileAgent, formatSourceMapDetails, ResponseType} from './freestyler.js';
+import {DrJonesFileAgent, FileContext, formatFile, formatSourceMapDetails, ResponseType} from './freestyler.js';
 
 describeWithMockConnection('DrJonesFileAgent', () => {
   function mockHostConfig(modelId?: string, temperature?: number) {
@@ -84,21 +84,20 @@ describeWithMockConnection('DrJonesFileAgent', () => {
         serverSideLoggingEnabled: true,
       });
       sinon.stub(agent, 'preamble').value('preamble');
-      agent.chatNewHistoryForTesting = new Map([
-        [
-          0,
-          [
-            {
-              type: ResponseType.QUERYING,
-              query: 'question',
-            },
-            {
-              type: ResponseType.ANSWER,
-              text: 'answer',
-            },
-          ],
-        ],
-      ]);
+      agent.chatNewHistoryForTesting = [
+        {
+          type: ResponseType.USER_QUERY,
+          query: 'question',
+        },
+        {
+          type: ResponseType.QUERYING,
+          query: 'question',
+        },
+        {
+          type: ResponseType.ANSWER,
+          text: 'answer',
+        },
+      ];
       assert.deepStrictEqual(
           agent.buildRequest({
             input: 'test input',
@@ -133,6 +132,34 @@ describeWithMockConnection('DrJonesFileAgent', () => {
     });
   });
 
+  async function createUISourceCode(options?: {
+    content?: string,
+    mimeType?: string,
+    url?: Platform.DevToolsPath.UrlString,
+    resourceType?: Common.ResourceType.ResourceType,
+  }): Promise<Workspace.UISourceCode.UISourceCode> {
+    const url = options?.url ?? 'http://example.test/script.js' as Platform.DevToolsPath.UrlString;
+    const {project} = createContentProviderUISourceCodes({
+      items: [
+        {
+          url,
+          mimeType: options?.mimeType ?? 'application/javascript',
+          resourceType: options?.resourceType ?? Common.ResourceType.resourceTypes.Script,
+          content: options?.content ?? undefined,
+        },
+      ],
+      target: createTarget(),
+    });
+
+    const uiSourceCode = project.uiSourceCodeForURL(url);
+
+    if (!uiSourceCode) {
+      throw new Error('Failed to create a test uiSourceCode');
+    }
+    await uiSourceCode.requestContentData();
+    return uiSourceCode;
+  }
+
   describe('run', () => {
     it('generates an answer', async () => {
       async function* generateAnswer() {
@@ -149,20 +176,9 @@ describeWithMockConnection('DrJonesFileAgent', () => {
         aidaClient: mockAidaClient(generateAnswer),
       });
 
-      const url = 'http://example.com/script.js' as Platform.DevToolsPath.UrlString;
-      const {project} = createContentProviderUISourceCodes({
-        items: [
-          {
-            url,
-            mimeType: 'application/javascript',
-            resourceType: Common.ResourceType.resourceTypes.Script,
-          },
-        ],
-        target: createTarget(),
-      });
-
-      const uiSourceCode = project.uiSourceCodeForURL(url);
-      const responses = await Array.fromAsync(agent.run('test', {selected: uiSourceCode}));
+      const uiSourceCode = await createUISourceCode();
+      const responses =
+          await Array.fromAsync(agent.run('test', {selected: uiSourceCode ? new FileContext(uiSourceCode) : null}));
 
       assert.deepStrictEqual(responses, [
         {
@@ -175,17 +191,28 @@ describeWithMockConnection('DrJonesFileAgent', () => {
           details: [
             {
               title: 'Selected file',
-              text: `File Name: script.js
-URL: http://example.com/script.js\n
-File Content:\n\`\`\`
+              text: `File name: script.js
+URL: http://example.test/script.js
+File content:
+\`\`\`
+
 \`\`\``,
             },
           ],
         },
         {
           type: ResponseType.QUERYING,
-          query:
-              '# Selected file\nFile Name: script.js\nURL: http://example.com/script.js\n\nFile Content:\n\`\`\`\n\`\`\`\n\n# User request\n\ntest',
+          query: `# Selected file
+File name: script.js
+URL: http://example.test/script.js
+File content:
+\`\`\`
+
+\`\`\`
+
+# User request
+
+test`,
         },
         {
           type: ResponseType.ANSWER,
@@ -198,9 +225,17 @@ File Content:\n\`\`\`
       assert.deepStrictEqual(agent.chatHistoryForTesting, [
         {
           entity: 1,
-          text: `# Selected file\nFile Name: script.js
-URL: http://example.com/script.js\n
-File Content:\n\`\`\`\n\`\`\`\n\n# User request\n\ntest`,
+          text: `# Selected file
+File name: script.js
+URL: http://example.test/script.js
+File content:
+\`\`\`
+
+\`\`\`
+
+# User request
+
+test`,
         },
         {
           entity: 2,
@@ -225,4 +260,44 @@ File Content:\n\`\`\`\n\`\`\`\n\n# User request\n\ntest`,
       assert.strictEqual(response, 'Source map: file://gen.js.map');
     });
   });
+
+  describe('formatFile', () => {
+    it('formats file content', async () => {
+      const uiSourceCode = await createUISourceCode({
+        content: 'lorem ipsum',
+      });
+      assert.strictEqual(formatFile(uiSourceCode), `File name: script.js
+URL: http://example.test/script.js
+File content:
+\`\`\`
+lorem ipsum
+\`\`\``);
+    });
+
+    it('formats file content of a binary file', async () => {
+      const uiSourceCode = await createUISourceCode({
+        resourceType: Common.ResourceType.resourceTypes.Image,
+        mimeType: 'application/png',
+        url: 'http://example.test/test.png' as Platform.DevToolsPath.UrlString,
+      });
+      assert.strictEqual(formatFile(uiSourceCode), `File name: test.png
+URL: http://example.test/test.png
+File content:
+\`\`\`
+<binary data>
+\`\`\``);
+    });
+
+    it('truncates long file content', async () => {
+      const uiSourceCode = await createUISourceCode({
+        content: 'lorem ipsum'.repeat(10_000),
+      });
+      assert.strictEqual(formatFile(uiSourceCode), `File name: script.js
+URL: http://example.test/script.js
+File content:
+\`\`\`
+lorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsumlorem ipsuml...
+\`\`\``);
+    });
+  });
 });

package/front_end/panels/freestyler/DrJonesFileAgent.ts

@@ -7,14 +7,16 @@ import * as Host from '../../core/host/host.js';
 import * as i18n from '../../core/i18n/i18n.js';
 import * as Bindings from '../../models/bindings/bindings.js';
 import type * as Workspace from '../../models/workspace/workspace.js';
+import * as PanelUtils from '../utils/utils.js';
 
 import {
   AgentType,
   AiAgent,
-  type AidaRequestOptions,
   type ContextDetail,
   type ContextResponse,
+  ConversationContext,
   type ParsedResponse,
+  type RequestOptions,
   ResponseType,
 } from './AiAgent.js';
 
@@ -69,24 +71,47 @@ const lockedString = i18n.i18n.lockedString;
 
 const MAX_FILE_SIZE = 10000;
 
+export class FileContext extends ConversationContext<Workspace.UISourceCode.UISourceCode> {
+  #file: Workspace.UISourceCode.UISourceCode;
+
+  constructor(file: Workspace.UISourceCode.UISourceCode) {
+    super();
+    this.#file = file;
+  }
+
+  override getOrigin(): string {
+    return new URL(this.#file.url()).origin;
+  }
+
+  override getItem(): Workspace.UISourceCode.UISourceCode {
+    return this.#file;
+  }
+
+  override getIcon(): HTMLElement {
+    return PanelUtils.PanelUtils.getIconForSourceFile(this.#file);
+  }
+
+  override getTitle(): string {
+    return this.#file.displayName();
+  }
+}
+
 /**
  * One agent instance handles one conversation. Create a new agent
  * instance for a new conversation.
  */
 export class DrJonesFileAgent extends AiAgent<Workspace.UISourceCode.UISourceCode> {
-  override type = AgentType.DRJONES_FILE;
+  override readonly type = AgentType.DRJONES_FILE;
   readonly preamble = preamble;
   readonly clientFeature = Host.AidaClient.ClientFeature.CHROME_DRJONES_FILE_AGENT;
   get userTier(): string|undefined {
     const config = Common.Settings.Settings.instance().getHostConfig();
-    return config.devToolsAiAssistanceFileAgent?.userTier ?? config.devToolsAiAssistanceFileAgentDogfood?.userTier;
+    return config.devToolsAiAssistanceFileAgent?.userTier;
   }
-  get options(): AidaRequestOptions {
+  get options(): RequestOptions {
     const config = Common.Settings.Settings.instance().getHostConfig();
-    const temperature =
-        config.devToolsAiAssistanceFileAgent?.temperature ?? config.devToolsAiAssistanceFileAgentDogfood?.temperature;
-    const modelId =
-        config.devToolsAiAssistanceFileAgent?.modelId ?? config.devToolsAiAssistanceFileAgentDogfood?.modelId;
+    const temperature = config.devToolsAiAssistanceFileAgent?.temperature;
+    const modelId = config.devToolsAiAssistanceFileAgent?.modelId;
 
     return {
       temperature,
@@ -95,7 +120,7 @@ export class DrJonesFileAgent extends AiAgent<Workspace.UISourceCode.UISourceCod
   }
 
   async *
-      handleContextDetails(selectedFile: Workspace.UISourceCode.UISourceCode|null):
+      handleContextDetails(selectedFile: ConversationContext<Workspace.UISourceCode.UISourceCode>|null):
           AsyncGenerator<ContextResponse, void, void> {
     if (!selectedFile) {
       return;
@@ -108,9 +133,10 @@ export class DrJonesFileAgent extends AiAgent<Workspace.UISourceCode.UISourceCod
     };
   }
 
-  override async enhanceQuery(query: string, selectedFile: Workspace.UISourceCode.UISourceCode|null): Promise<string> {
+  override async enhanceQuery(
+      query: string, selectedFile: ConversationContext<Workspace.UISourceCode.UISourceCode>|null): Promise<string> {
     const fileEnchantmentQuery =
-        selectedFile ? `# Selected file\n${formatFile(selectedFile)}\n\n# User request\n\n` : '';
+        selectedFile ? `# Selected file\n${formatFile(selectedFile.getItem())}\n\n# User request\n\n` : '';
     return `${fileEnchantmentQuery}${query}`;
   }
 
@@ -121,28 +147,35 @@ export class DrJonesFileAgent extends AiAgent<Workspace.UISourceCode.UISourceCod
   }
 }
 
-function createContextDetailsForDrJonesFileAgent(selectedFile: Workspace.UISourceCode.UISourceCode):
-    [ContextDetail, ...ContextDetail[]] {
+function createContextDetailsForDrJonesFileAgent(
+    selectedFile: ConversationContext<Workspace.UISourceCode.UISourceCode>): [ContextDetail, ...ContextDetail[]] {
   return [
     {
       title: 'Selected file',
-      text: formatFile(selectedFile),
+      text: formatFile(selectedFile.getItem()),
     },
   ];
 }
 
-function formatFile(selectedFile: Workspace.UISourceCode.UISourceCode): string {
+export function formatFile(selectedFile: Workspace.UISourceCode.UISourceCode): string {
   const debuggerWorkspaceBinding = Bindings.DebuggerWorkspaceBinding.DebuggerWorkspaceBinding.instance();
-  return `File Name: ${selectedFile.displayName()}
-URL: ${selectedFile.url()}
-${formatSourceMapDetails(selectedFile, debuggerWorkspaceBinding)}
-File Content:
-${formatFileContent(selectedFile.content())}`;
+  const sourceMapDetails = formatSourceMapDetails(selectedFile, debuggerWorkspaceBinding);
+  const lines = [
+    `File name: ${selectedFile.displayName()}`,
+    `URL: ${selectedFile.url()}`,
+    sourceMapDetails,
+    `File content:
+${formatFileContent(selectedFile)}`,
+  ];
+  return lines.filter(line => line.trim() !== '').join('\n');
 }
 
-function formatFileContent(content: string): string {
-  const formattedContent = content.length > MAX_FILE_SIZE ? content.slice(0, MAX_FILE_SIZE) + '...\n' : content + '\n';
-  return '```' + formattedContent + '```';
+function formatFileContent(selectedFile: Workspace.UISourceCode.UISourceCode): string {
+  const content = selectedFile.contentType().isTextType() ? selectedFile.content() : '<binary data>';
+  const truncated = content.length > MAX_FILE_SIZE ? content.slice(0, MAX_FILE_SIZE) + '...' : content;
+  return `\`\`\`
+${truncated}
+\`\`\``;
 }
 
 export function formatSourceMapDetails(
@@ -179,14 +212,3 @@ export function formatSourceMapDetails(
   }
   return sourceMapDetails;
 }
-
-function setDebugFreestylerEnabled(enabled: boolean): void {
-  if (enabled) {
-    localStorage.setItem('debugFreestylerEnabled', 'true');
-  } else {
-    localStorage.removeItem('debugFreestylerEnabled');
-  }
-}
-
-// @ts-ignore
-globalThis.setDebugFreestylerEnabled = setDebugFreestylerEnabled;