chrome-ai-bridge 1.0.0

package/build/node_modules/chrome-devtools-frontend/front_end/core/host/AidaClient.js

@@ -0,0 +1,434 @@
+// Copyright 2023 The Chromium Authors
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+import * as Common from '../common/common.js';
+import * as Root from '../root/root.js';
+import * as DispatchHttpRequestClient from './DispatchHttpRequestClient.js';
+import { InspectorFrontendHostInstance } from './InspectorFrontendHost.js';
+import { bindOutputStream } from './ResourceLoader.js';
+export var Role;
+(function (Role) {
+    /** Provide this role when giving a function call response  */
+    Role[Role["ROLE_UNSPECIFIED"] = 0] = "ROLE_UNSPECIFIED";
+    /** Tags the content came from the user */
+    Role[Role["USER"] = 1] = "USER";
+    /** Tags the content came from the LLM */
+    Role[Role["MODEL"] = 2] = "MODEL";
+})(Role || (Role = {}));
+export var FunctionalityType;
+(function (FunctionalityType) {
+    // Unspecified functionality type.
+    FunctionalityType[FunctionalityType["FUNCTIONALITY_TYPE_UNSPECIFIED"] = 0] = "FUNCTIONALITY_TYPE_UNSPECIFIED";
+    // The generic AI chatbot functionality.
+    FunctionalityType[FunctionalityType["CHAT"] = 1] = "CHAT";
+    // The explain error functionality.
+    FunctionalityType[FunctionalityType["EXPLAIN_ERROR"] = 2] = "EXPLAIN_ERROR";
+    FunctionalityType[FunctionalityType["AGENTIC_CHAT"] = 5] = "AGENTIC_CHAT";
+})(FunctionalityType || (FunctionalityType = {}));
+/** See: cs/aida.proto (google3). **/
+export var ClientFeature;
+(function (ClientFeature) {
+    // Unspecified client feature.
+    ClientFeature[ClientFeature["CLIENT_FEATURE_UNSPECIFIED"] = 0] = "CLIENT_FEATURE_UNSPECIFIED";
+    // Chrome console insights feature.
+    ClientFeature[ClientFeature["CHROME_CONSOLE_INSIGHTS"] = 1] = "CHROME_CONSOLE_INSIGHTS";
+    // Chrome AI Assistance Styling Agent.
+    ClientFeature[ClientFeature["CHROME_STYLING_AGENT"] = 2] = "CHROME_STYLING_AGENT";
+    // Chrome AI Assistance Network Agent.
+    ClientFeature[ClientFeature["CHROME_NETWORK_AGENT"] = 7] = "CHROME_NETWORK_AGENT";
+    // Chrome AI Annotations Performance Agent
+    ClientFeature[ClientFeature["CHROME_PERFORMANCE_ANNOTATIONS_AGENT"] = 20] = "CHROME_PERFORMANCE_ANNOTATIONS_AGENT";
+    // Chrome AI Assistance File Agent.
+    ClientFeature[ClientFeature["CHROME_FILE_AGENT"] = 9] = "CHROME_FILE_AGENT";
+    // Chrome AI Patch Agent.
+    ClientFeature[ClientFeature["CHROME_PATCH_AGENT"] = 12] = "CHROME_PATCH_AGENT";
+    // Chrome AI Assistance Performance Agent.
+    ClientFeature[ClientFeature["CHROME_PERFORMANCE_FULL_AGENT"] = 24] = "CHROME_PERFORMANCE_FULL_AGENT";
+    // Removed features (for reference).
+    // Chrome AI Assistance Performance Insights Agent.
+    // CHROME_PERFORMANCE_INSIGHTS_AGENT = 13,
+    // Chrome AI Assistance Performance Agent (call trees).
+    // CHROME_PERFORMANCE_AGENT = 8,
+})(ClientFeature || (ClientFeature = {}));
+export var UserTier;
+(function (UserTier) {
+    // Unspecified user tier.
+    UserTier[UserTier["USER_TIER_UNSPECIFIED"] = 0] = "USER_TIER_UNSPECIFIED";
+    // Users who are internal testers.
+    UserTier[UserTier["TESTERS"] = 1] = "TESTERS";
+    // Users who are early adopters.
+    UserTier[UserTier["BETA"] = 2] = "BETA";
+    // Users in the general public.
+    UserTier[UserTier["PUBLIC"] = 3] = "PUBLIC";
+})(UserTier || (UserTier = {}));
+/* eslint-enable @typescript-eslint/naming-convention */
+export var EditType;
+(function (EditType) {
+    // Unknown edit type
+    EditType[EditType["EDIT_TYPE_UNSPECIFIED"] = 0] = "EDIT_TYPE_UNSPECIFIED";
+    // User typed code/text into file
+    EditType[EditType["ADD"] = 1] = "ADD";
+    // User deleted code/text from file
+    EditType[EditType["DELETE"] = 2] = "DELETE";
+    // User pasted into file (this includes smart paste)
+    EditType[EditType["PASTE"] = 3] = "PASTE";
+    // User performs an undo action
+    EditType[EditType["UNDO"] = 4] = "UNDO";
+    // User performs a redo action
+    EditType[EditType["REDO"] = 5] = "REDO";
+    // User accepted a completion from AIDA
+    EditType[EditType["ACCEPT_COMPLETION"] = 6] = "ACCEPT_COMPLETION";
+})(EditType || (EditType = {}));
+export var Reason;
+(function (Reason) {
+    // Unknown reason.
+    Reason[Reason["UNKNOWN"] = 0] = "UNKNOWN";
+    // The file is currently open.
+    Reason[Reason["CURRENTLY_OPEN"] = 1] = "CURRENTLY_OPEN";
+    // The file is opened recently.
+    Reason[Reason["RECENTLY_OPENED"] = 2] = "RECENTLY_OPENED";
+    // The file is edited recently.
+    Reason[Reason["RECENTLY_EDITED"] = 3] = "RECENTLY_EDITED";
+    // The file is located within the same directory.
+    Reason[Reason["COLOCATED"] = 4] = "COLOCATED";
+    // Included based on relation to code around the cursor (e.g: could be
+    // provided by local IDE analysis)
+    Reason[Reason["RELATED_FILE"] = 5] = "RELATED_FILE";
+})(Reason || (Reason = {}));
+/* eslint-enable @typescript-eslint/naming-convention */
+export var UseCase;
+(function (UseCase) {
+    // Unspecified usecase.
+    UseCase[UseCase["USE_CASE_UNSPECIFIED"] = 0] = "USE_CASE_UNSPECIFIED";
+    // Code generation use case is expected to generate code from scratch
+    UseCase[UseCase["CODE_GENERATION"] = 1] = "CODE_GENERATION";
+})(UseCase || (UseCase = {}));
+/* eslint-enable @typescript-eslint/naming-convention */
+export var RecitationAction;
+(function (RecitationAction) {
+    RecitationAction["ACTION_UNSPECIFIED"] = "ACTION_UNSPECIFIED";
+    RecitationAction["CITE"] = "CITE";
+    RecitationAction["BLOCK"] = "BLOCK";
+    RecitationAction["NO_ACTION"] = "NO_ACTION";
+    RecitationAction["EXEMPT_FOUND_IN_PROMPT"] = "EXEMPT_FOUND_IN_PROMPT";
+})(RecitationAction || (RecitationAction = {}));
+export var CitationSourceType;
+(function (CitationSourceType) {
+    CitationSourceType["CITATION_SOURCE_TYPE_UNSPECIFIED"] = "CITATION_SOURCE_TYPE_UNSPECIFIED";
+    CitationSourceType["TRAINING_DATA"] = "TRAINING_DATA";
+    CitationSourceType["WORLD_FACTS"] = "WORLD_FACTS";
+    CitationSourceType["LOCAL_FACTS"] = "LOCAL_FACTS";
+    CitationSourceType["INDIRECT"] = "INDIRECT";
+})(CitationSourceType || (CitationSourceType = {}));
+const AidaLanguageToMarkdown = {
+    CPP: 'cpp',
+    PYTHON: 'py',
+    KOTLIN: 'kt',
+    JAVA: 'java',
+    JAVASCRIPT: 'js',
+    GO: 'go',
+    TYPESCRIPT: 'ts',
+    HTML: 'html',
+    BASH: 'sh',
+    CSS: 'css',
+    DART: 'dart',
+    JSON: 'json',
+    MARKDOWN: 'md',
+    VUE: 'vue',
+    XML: 'xml',
+};
+export const CLIENT_NAME = 'CHROME_DEVTOOLS';
+export const SERVICE_NAME = 'aidaService';
+const CODE_CHUNK_SEPARATOR = (lang = '') => ('\n`````' + lang + '\n');
+export class AidaAbortError extends Error {
+}
+export class AidaBlockError extends Error {
+}
+export class AidaClient {
+    static buildConsoleInsightsRequest(input) {
+        const disallowLogging = Root.Runtime.hostConfig.aidaAvailability?.disallowLogging ?? true;
+        const chromeVersion = Root.Runtime.getChromeVersion();
+        if (!chromeVersion) {
+            throw new Error('Cannot determine Chrome version');
+        }
+        const request = {
+            current_message: { parts: [{ text: input }], role: Role.USER },
+            client: CLIENT_NAME,
+            functionality_type: FunctionalityType.EXPLAIN_ERROR,
+            client_feature: ClientFeature.CHROME_CONSOLE_INSIGHTS,
+            metadata: {
+                disable_user_content_logging: disallowLogging,
+                client_version: chromeVersion,
+            },
+        };
+        let temperature = -1;
+        let modelId;
+        if (Root.Runtime.hostConfig.devToolsConsoleInsights?.enabled) {
+            temperature = Root.Runtime.hostConfig.devToolsConsoleInsights.temperature ?? -1;
+            modelId = Root.Runtime.hostConfig.devToolsConsoleInsights.modelId;
+        }
+        if (temperature >= 0) {
+            request.options ??= {};
+            request.options.temperature = temperature;
+        }
+        if (modelId) {
+            request.options ??= {};
+            request.options.model_id = modelId;
+        }
+        return request;
+    }
+    static async checkAccessPreconditions() {
+        if (!navigator.onLine) {
+            return "no-internet" /* AidaAccessPreconditions.NO_INTERNET */;
+        }
+        const syncInfo = await new Promise(resolve => InspectorFrontendHostInstance.getSyncInformation(syncInfo => resolve(syncInfo)));
+        if (!syncInfo.accountEmail) {
+            return "no-account-email" /* AidaAccessPreconditions.NO_ACCOUNT_EMAIL */;
+        }
+        if (syncInfo.isSyncPaused) {
+            return "sync-is-paused" /* AidaAccessPreconditions.SYNC_IS_PAUSED */;
+        }
+        return "available" /* AidaAccessPreconditions.AVAILABLE */;
+    }
+    async *doConversation(request, options) {
+        if (!InspectorFrontendHostInstance.doAidaConversation) {
+            throw new Error('doAidaConversation is not available');
+        }
+        const stream = (() => {
+            let { promise, resolve, reject } = Promise.withResolvers();
+            options?.signal?.addEventListener('abort', () => {
+                reject(new AidaAbortError());
+            }, { once: true });
+            return {
+                write: async (data) => {
+                    resolve(data);
+                    ({ promise, resolve, reject } = Promise.withResolvers());
+                },
+                close: async () => {
+                    resolve(null);
+                },
+                read: () => {
+                    return promise;
+                },
+                fail: (e) => reject(e),
+            };
+        })();
+        const streamId = bindOutputStream(stream);
+        InspectorFrontendHostInstance.doAidaConversation(JSON.stringify(request), streamId, result => {
+            if (result.statusCode === 403) {
+                stream.fail(new Error('Server responded: permission denied'));
+            }
+            else if (result.error) {
+                stream.fail(new Error(`Cannot send request: ${result.error} ${result.detail || ''}`));
+            }
+            else if (result.netErrorName === 'net::ERR_TIMED_OUT') {
+                stream.fail(new Error('doAidaConversation timed out'));
+            }
+            else if (result.statusCode !== 200) {
+                stream.fail(new Error(`Request failed: ${JSON.stringify(result)}`));
+            }
+            else {
+                void stream.close();
+            }
+        });
+        let chunk;
+        const text = [];
+        let inCodeChunk = false;
+        const functionCalls = [];
+        let metadata = { rpcGlobalId: 0 };
+        while ((chunk = await stream.read())) {
+            let textUpdated = false;
+            // The AIDA response is a JSON array of objects, split at the object
+            // boundary. Therefore each chunk may start with `[` or `,` and possibly
+            // followed by `]`. Each chunk may include one or more objects, so we
+            // make sure that each chunk becomes a well-formed JSON array when we
+            // parse it by adding `[` and `]` and removing `,` where appropriate.
+            if (!chunk.length) {
+                continue;
+            }
+            if (chunk.startsWith(',')) {
+                chunk = chunk.slice(1);
+            }
+            if (!chunk.startsWith('[')) {
+                chunk = '[' + chunk;
+            }
+            if (!chunk.endsWith(']')) {
+                chunk = chunk + ']';
+            }
+            let results;
+            try {
+                results = JSON.parse(chunk);
+            }
+            catch (error) {
+                throw new Error('Cannot parse chunk: ' + chunk, { cause: error });
+            }
+            for (const result of results) {
+                if ('metadata' in result) {
+                    metadata = result.metadata;
+                    if (metadata?.attributionMetadata?.attributionAction === RecitationAction.BLOCK) {
+                        throw new AidaBlockError();
+                    }
+                }
+                if ('textChunk' in result) {
+                    if (inCodeChunk) {
+                        text.push(CODE_CHUNK_SEPARATOR());
+                        inCodeChunk = false;
+                    }
+                    text.push(result.textChunk.text);
+                    textUpdated = true;
+                }
+                else if ('codeChunk' in result) {
+                    if (!inCodeChunk) {
+                        const language = AidaLanguageToMarkdown[result.codeChunk.inferenceLanguage] ?? '';
+                        text.push(CODE_CHUNK_SEPARATOR(language));
+                        inCodeChunk = true;
+                    }
+                    text.push(result.codeChunk.code);
+                    textUpdated = true;
+                }
+                else if ('functionCallChunk' in result) {
+                    functionCalls.push({
+                        name: result.functionCallChunk.functionCall.name,
+                        args: result.functionCallChunk.functionCall.args,
+                    });
+                }
+                else if ('error' in result) {
+                    throw new Error(`Server responded: ${JSON.stringify(result)}`);
+                }
+                else {
+                    throw new Error('Unknown chunk result');
+                }
+            }
+            if (textUpdated) {
+                yield {
+                    explanation: text.join('') + (inCodeChunk ? CODE_CHUNK_SEPARATOR() : ''),
+                    metadata,
+                    completed: false,
+                };
+            }
+        }
+        yield {
+            explanation: text.join('') + (inCodeChunk ? CODE_CHUNK_SEPARATOR() : ''),
+            metadata,
+            functionCalls: functionCalls.length ? functionCalls :
+                undefined,
+            completed: true,
+        };
+    }
+    registerClientEvent(clientEvent) {
+        const { promise, resolve } = Promise.withResolvers();
+        InspectorFrontendHostInstance.registerAidaClientEvent(JSON.stringify({
+            client: CLIENT_NAME,
+            event_time: new Date().toISOString(),
+            ...clientEvent,
+        }), resolve);
+        return promise;
+    }
+    async completeCode(request) {
+        if (!InspectorFrontendHostInstance.aidaCodeComplete) {
+            throw new Error('aidaCodeComplete is not available');
+        }
+        const { promise, resolve } = Promise.withResolvers();
+        InspectorFrontendHostInstance.aidaCodeComplete(JSON.stringify(request), resolve);
+        const completeCodeResult = await promise;
+        if (completeCodeResult.error) {
+            throw new Error(`Cannot send request: ${completeCodeResult.error} ${completeCodeResult.detail || ''}`);
+        }
+        const response = completeCodeResult.response;
+        if (!response?.length) {
+            throw new Error('Empty response');
+        }
+        let parsedResponse;
+        try {
+            parsedResponse = JSON.parse(response);
+        }
+        catch (error) {
+            throw new Error('Cannot parse response: ' + response, { cause: error });
+        }
+        const generatedSamples = [];
+        let metadata = { rpcGlobalId: 0 };
+        if ('metadata' in parsedResponse) {
+            metadata = parsedResponse.metadata;
+        }
+        if ('generatedSamples' in parsedResponse) {
+            for (const generatedSample of parsedResponse.generatedSamples) {
+                const sample = {
+                    generationString: generatedSample.generationString,
+                    score: generatedSample.score,
+                    sampleId: generatedSample.sampleId,
+                };
+                if ('metadata' in generatedSample && 'attributionMetadata' in generatedSample.metadata) {
+                    sample.attributionMetadata = generatedSample.metadata.attributionMetadata;
+                }
+                generatedSamples.push(sample);
+            }
+        }
+        else {
+            return null;
+        }
+        return { generatedSamples, metadata };
+    }
+    async generateCode(request, options) {
+        const response = await DispatchHttpRequestClient.makeHttpRequest({
+            service: SERVICE_NAME,
+            path: '/v1/aida:generateCode',
+            method: 'POST',
+            body: JSON.stringify(request),
+        }, options);
+        return response;
+    }
+}
+export function convertToUserTierEnum(userTier) {
+    if (userTier) {
+        switch (userTier) {
+            case 'TESTERS':
+                return UserTier.TESTERS;
+            case 'BETA':
+                return UserTier.BETA;
+            case 'PUBLIC':
+                return UserTier.PUBLIC;
+        }
+    }
+    return UserTier.PUBLIC;
+}
+let hostConfigTrackerInstance;
+export class HostConfigTracker extends Common.ObjectWrapper.ObjectWrapper {
+    #pollTimer;
+    #aidaAvailability;
+    constructor() {
+        super();
+    }
+    static instance() {
+        if (!hostConfigTrackerInstance) {
+            hostConfigTrackerInstance = new HostConfigTracker();
+        }
+        return hostConfigTrackerInstance;
+    }
+    addEventListener(eventType, listener) {
+        const isFirst = !this.hasEventListeners(eventType);
+        const eventDescriptor = super.addEventListener(eventType, listener);
+        if (isFirst) {
+            window.clearTimeout(this.#pollTimer);
+            void this.pollAidaAvailability();
+        }
+        return eventDescriptor;
+    }
+    removeEventListener(eventType, listener) {
+        super.removeEventListener(eventType, listener);
+        if (!this.hasEventListeners(eventType)) {
+            window.clearTimeout(this.#pollTimer);
+        }
+    }
+    async pollAidaAvailability() {
+        this.#pollTimer = window.setTimeout(() => this.pollAidaAvailability(), 2000);
+        const currentAidaAvailability = await AidaClient.checkAccessPreconditions();
+        if (currentAidaAvailability !== this.#aidaAvailability) {
+            this.#aidaAvailability = currentAidaAvailability;
+            const config = await new Promise(resolve => InspectorFrontendHostInstance.getHostConfig(resolve));
+            Object.assign(Root.Runtime.hostConfig, config);
+            // TODO(crbug.com/442545623): Send `currentAidaAvailability` to the listeners as part of the event so that
+            // `await AidaClient.checkAccessPreconditions()` does not need to be called again in the event handlers.
+            this.dispatchEventToListeners("aidaAvailabilityChanged" /* Events.AIDA_AVAILABILITY_CHANGED */);
+        }
+    }
+}

package/build/node_modules/chrome-devtools-frontend/front_end/core/host/DispatchHttpRequestClient.js

@@ -0,0 +1,66 @@
+// Copyright 2025 The Chromium Authors
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+import { InspectorFrontendHostInstance } from './InspectorFrontendHost.js';
+export var ErrorType;
+(function (ErrorType) {
+    ErrorType["HTTP_RESPONSE_UNAVAILABLE"] = "HTTP_RESPONSE_UNAVAILABLE";
+    ErrorType["NOT_FOUND"] = "NOT_FOUND";
+    ErrorType["ABORT"] = "ABORT";
+})(ErrorType || (ErrorType = {}));
+export class DispatchHttpRequestError extends Error {
+    type;
+    constructor(type, options) {
+        super(undefined, options);
+        this.type = type;
+    }
+}
+export async function makeHttpRequest(request, options) {
+    const signal = options?.signal;
+    if (signal?.aborted) {
+        throw new DispatchHttpRequestError(ErrorType.ABORT);
+    }
+    const response = await new Promise((resolve, reject) => {
+        const onAbort = () => {
+            reject(new DispatchHttpRequestError(ErrorType.ABORT));
+        };
+        signal?.addEventListener('abort', onAbort, { once: true });
+        InspectorFrontendHostInstance.dispatchHttpRequest(request, result => {
+            signal?.removeEventListener('abort', onAbort);
+            resolve(result);
+        });
+    });
+    debugLog({ request, response });
+    if (response.statusCode === 404) {
+        throw new DispatchHttpRequestError(ErrorType.NOT_FOUND);
+    }
+    if ('response' in response && response.statusCode === 200) {
+        try {
+            return JSON.parse(response.response);
+        }
+        catch (err) {
+            throw new DispatchHttpRequestError(ErrorType.HTTP_RESPONSE_UNAVAILABLE, { cause: err });
+        }
+    }
+    throw new DispatchHttpRequestError(ErrorType.HTTP_RESPONSE_UNAVAILABLE);
+}
+function isDebugMode() {
+    return Boolean(localStorage.getItem('debugDispatchHttpRequestEnabled'));
+}
+function debugLog(...log) {
+    if (!isDebugMode()) {
+        return;
+    }
+    // eslint-disable-next-line no-console
+    console.log('debugLog', ...log);
+}
+function setDebugDispatchHttpRequestEnabled(enabled) {
+    if (enabled) {
+        localStorage.setItem('debugDispatchHttpRequestEnabled', 'true');
+    }
+    else {
+        localStorage.removeItem('debugDispatchHttpRequestEnabled');
+    }
+}
+// @ts-expect-error
+globalThis.setDebugDispatchHttpRequestEnabled = setDebugDispatchHttpRequestEnabled;