@anthropic-field/core 0.1.0

package/dist/src/code_assist/server.js

@@ -0,0 +1,228 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import * as readline from 'node:readline';
+import { UserTierId } from './types.js';
+import { fromCountTokenResponse, fromGenerateContentResponse, toCountTokenRequest, toGenerateContentRequest, } from './converter.js';
+import { formatProtoJsonDuration, recordConversationOffered, } from './telemetry.js';
+import { getClientMetadata } from './experiments/client_metadata.js';
+export const CODE_ASSIST_ENDPOINT = 'https://cloudcode-pa.googleapis.com';
+export const CODE_ASSIST_API_VERSION = 'v1internal';
+export class CodeAssistServer {
+    client;
+    projectId;
+    httpOptions;
+    sessionId;
+    userTier;
+    userTierName;
+    constructor(client, projectId, httpOptions = {}, sessionId, userTier, userTierName) {
+        this.client = client;
+        this.projectId = projectId;
+        this.httpOptions = httpOptions;
+        this.sessionId = sessionId;
+        this.userTier = userTier;
+        this.userTierName = userTierName;
+    }
+    async generateContentStream(req, userPromptId) {
+        const responses = await this.requestStreamingPost('streamGenerateContent', toGenerateContentRequest(req, userPromptId, this.projectId, this.sessionId), req.config?.abortSignal);
+        const streamingLatency = {};
+        const start = Date.now();
+        let isFirst = true;
+        return (async function* (server) {
+            for await (const response of responses) {
+                if (isFirst) {
+                    streamingLatency.firstMessageLatency = formatProtoJsonDuration(Date.now() - start);
+                    isFirst = false;
+                }
+                streamingLatency.totalLatency = formatProtoJsonDuration(Date.now() - start);
+                const translatedResponse = fromGenerateContentResponse(response);
+                await recordConversationOffered(server, response.traceId, translatedResponse, streamingLatency, req.config?.abortSignal);
+                yield translatedResponse;
+            }
+        })(this);
+    }
+    async generateContent(req, userPromptId) {
+        const start = Date.now();
+        const response = await this.requestPost('generateContent', toGenerateContentRequest(req, userPromptId, this.projectId, this.sessionId), req.config?.abortSignal);
+        const duration = formatProtoJsonDuration(Date.now() - start);
+        const streamingLatency = {
+            totalLatency: duration,
+            firstMessageLatency: duration,
+        };
+        const translatedResponse = fromGenerateContentResponse(response);
+        await recordConversationOffered(this, response.traceId, translatedResponse, streamingLatency, req.config?.abortSignal);
+        return translatedResponse;
+    }
+    async onboardUser(req) {
+        return this.requestPost('onboardUser', req);
+    }
+    async getOperation(name) {
+        return this.requestGetOperation(name);
+    }
+    async loadCodeAssist(req) {
+        try {
+            return await this.requestPost('loadCodeAssist', req);
+        }
+        catch (e) {
+            if (isVpcScAffectedUser(e)) {
+                return {
+                    currentTier: { id: UserTierId.STANDARD },
+                };
+            }
+            else {
+                throw e;
+            }
+        }
+    }
+    async fetchAdminControls(req) {
+        return this.requestPost('fetchAdminControls', req);
+    }
+    async getCodeAssistGlobalUserSetting() {
+        return this.requestGet('getCodeAssistGlobalUserSetting');
+    }
+    async setCodeAssistGlobalUserSetting(req) {
+        return this.requestPost('setCodeAssistGlobalUserSetting', req);
+    }
+    async countTokens(req) {
+        const resp = await this.requestPost('countTokens', toCountTokenRequest(req));
+        return fromCountTokenResponse(resp);
+    }
+    async embedContent(_req) {
+        throw Error();
+    }
+    async listExperiments(metadata) {
+        if (!this.projectId) {
+            throw new Error('projectId is not defined for CodeAssistServer.');
+        }
+        const projectId = this.projectId;
+        const req = {
+            project: projectId,
+            metadata: { ...metadata, duetProject: projectId },
+        };
+        return this.requestPost('listExperiments', req);
+    }
+    async retrieveUserQuota(req) {
+        return this.requestPost('retrieveUserQuota', req);
+    }
+    async recordConversationOffered(conversationOffered) {
+        if (!this.projectId) {
+            return;
+        }
+        await this.recordCodeAssistMetrics({
+            project: this.projectId,
+            metadata: await getClientMetadata(),
+            metrics: [{ conversationOffered, timestamp: new Date().toISOString() }],
+        });
+    }
+    async recordConversationInteraction(interaction) {
+        if (!this.projectId) {
+            return;
+        }
+        await this.recordCodeAssistMetrics({
+            project: this.projectId,
+            metadata: await getClientMetadata(),
+            metrics: [
+                {
+                    conversationInteraction: interaction,
+                    timestamp: new Date().toISOString(),
+                },
+            ],
+        });
+    }
+    async recordCodeAssistMetrics(request) {
+        return this.requestPost('recordCodeAssistMetrics', request);
+    }
+    async requestPost(method, req, signal) {
+        const res = await this.client.request({
+            url: this.getMethodUrl(method),
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+                ...this.httpOptions.headers,
+            },
+            responseType: 'json',
+            body: JSON.stringify(req),
+            signal,
+        });
+        return res.data;
+    }
+    async makeGetRequest(url, signal) {
+        const res = await this.client.request({
+            url,
+            method: 'GET',
+            headers: {
+                'Content-Type': 'application/json',
+                ...this.httpOptions.headers,
+            },
+            responseType: 'json',
+            signal,
+        });
+        return res.data;
+    }
+    async requestGet(method, signal) {
+        return this.makeGetRequest(this.getMethodUrl(method), signal);
+    }
+    async requestGetOperation(name, signal) {
+        return this.makeGetRequest(this.getOperationUrl(name), signal);
+    }
+    async requestStreamingPost(method, req, signal) {
+        const res = await this.client.request({
+            url: this.getMethodUrl(method),
+            method: 'POST',
+            params: {
+                alt: 'sse',
+            },
+            headers: {
+                'Content-Type': 'application/json',
+                ...this.httpOptions.headers,
+            },
+            responseType: 'stream',
+            body: JSON.stringify(req),
+            signal,
+        });
+        return (async function* () {
+            const rl = readline.createInterface({
+                input: res.data,
+                crlfDelay: Infinity, // Recognizes '\r\n' and '\n' as line breaks
+            });
+            let bufferedLines = [];
+            for await (const line of rl) {
+                if (line.startsWith('data: ')) {
+                    bufferedLines.push(line.slice(6).trim());
+                }
+                else if (line === '') {
+                    if (bufferedLines.length === 0) {
+                        continue; // no data to yield
+                    }
+                    yield JSON.parse(bufferedLines.join('\n'));
+                    bufferedLines = []; // Reset the buffer after yielding
+                }
+                // Ignore other lines like comments or id fields
+            }
+        })();
+    }
+    getBaseUrl() {
+        const endpoint = process.env['CODE_ASSIST_ENDPOINT'] ?? CODE_ASSIST_ENDPOINT;
+        const version = process.env['CODE_ASSIST_API_VERSION'] || CODE_ASSIST_API_VERSION;
+        return `${endpoint}/${version}`;
+    }
+    getMethodUrl(method) {
+        return `${this.getBaseUrl()}:${method}`;
+    }
+    getOperationUrl(name) {
+        return `${this.getBaseUrl()}/${name}`;
+    }
+}
+function isVpcScAffectedUser(error) {
+    if (error && typeof error === 'object' && 'response' in error) {
+        const gaxiosError = error;
+        const response = gaxiosError.response?.data;
+        if (Array.isArray(response?.error?.details)) {
+            return response.error.details.some((detail) => detail.reason === 'SECURITY_POLICY_VIOLATED');
+        }
+    }
+    return false;
+}
+//# sourceMappingURL=server.js.map

package/dist/src/code_assist/server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../../../src/code_assist/server.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAiCH,OAAO,KAAK,QAAQ,MAAM,eAAe,CAAC;AAE1C,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAKxC,OAAO,EACL,sBAAsB,EACtB,2BAA2B,EAC3B,mBAAmB,EACnB,wBAAwB,GACzB,MAAM,gBAAgB,CAAC;AACxB,OAAO,EACL,uBAAuB,EACvB,yBAAyB,GAC1B,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,iBAAiB,EAAE,MAAM,kCAAkC,CAAC;AAOrE,MAAM,CAAC,MAAM,oBAAoB,GAAG,qCAAqC,CAAC;AAC1E,MAAM,CAAC,MAAM,uBAAuB,GAAG,YAAY,CAAC;AAEpD,MAAM,OAAO,gBAAgB;IAEhB;IACA;IACA;IACA;IACA;IACA;IANX,YACW,MAAkB,EAClB,SAAkB,EAClB,cAA2B,EAAE,EAC7B,SAAkB,EAClB,QAAqB,EACrB,YAAqB;QALrB,WAAM,GAAN,MAAM,CAAY;QAClB,cAAS,GAAT,SAAS,CAAS;QAClB,gBAAW,GAAX,WAAW,CAAkB;QAC7B,cAAS,GAAT,SAAS,CAAS;QAClB,aAAQ,GAAR,QAAQ,CAAa;QACrB,iBAAY,GAAZ,YAAY,CAAS;IAC7B,CAAC;IAEJ,KAAK,CAAC,qBAAqB,CACzB,GAA8B,EAC9B,YAAoB;QAEpB,MAAM,SAAS,GACb,MAAM,IAAI,CAAC,oBAAoB,CAC7B,uBAAuB,EACvB,wBAAwB,CACtB,GAAG,EACH,YAAY,EACZ,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,SAAS,CACf,EACD,GAAG,CAAC,MAAM,EAAE,WAAW,CACxB,CAAC;QAEJ,MAAM,gBAAgB,GAAqB,EAAE,CAAC;QAC9C,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACzB,IAAI,OAAO,GAAG,IAAI,CAAC;QAEnB,OAAO,CAAC,KAAK,SAAS,CAAC,EACrB,MAAwB;YAExB,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;gBACvC,IAAI,OAAO,EAAE,CAAC;oBACZ,gBAAgB,CAAC,mBAAmB,GAAG,uBAAuB,CAC5D,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CACnB,CAAC;oBACF,OAAO,GAAG,KAAK,CAAC;gBAClB,CAAC;gBAED,gBAAgB,CAAC,YAAY,GAAG,uBAAuB,CACrD,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CACnB,CAAC;gBAEF,MAAM,kBAAkB,GAAG,2BAA2B,CAAC,QAAQ,CAAC,CAAC;gBAEjE,MAAM,yBAAyB,CAC7B,MAAM,EACN,QAAQ,CAAC,OAAO,EAChB,kBAAkB,EAClB,gBAAgB,EAChB,GAAG,CAAC,MAAM,EAAE,WAAW,CACxB,CAAC;gBAEF,MAAM,kBAAkB,CAAC;YAC3B,CAAC;QACH,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACX,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,GAA8B,EAC9B,YAAoB;QAEpB,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACzB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CACrC,iBAAiB,EACjB,wBAAwB,CACtB,GAAG,EACH,YAAY,EACZ,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,SAAS,CACf,EACD,GAAG,CAAC,MAAM,EAAE,WAAW,CACxB,CAAC;QACF,MAAM,QAAQ,GAAG,uBAAuB,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,CAAC;QAC7D,MAAM,gBAAgB,GAAqB;YACzC,YAAY,EAAE,QAAQ;YACtB,mBAAmB,EAAE,QAAQ;SAC9B,CAAC;QAEF,MAAM,kBAAkB,GAAG,2BAA2B,CAAC,QAAQ,CAAC,CAAC;QAEjE,MAAM,yBAAyB,CAC7B,IAAI,EACJ,QAAQ,CAAC,OAAO,EAChB,kBAAkB,EAClB,gBAAgB,EAChB,GAAG,CAAC,MAAM,EAAE,WAAW,CACxB,CAAC;QAEF,OAAO,kBAAkB,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,WAAW,CACf,GAAuB;QAEvB,OAAO,IAAI,CAAC,WAAW,CAA+B,aAAa,EAAE,GAAG,CAAC,CAAC;IAC5E,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,IAAY;QAC7B,OAAO,IAAI,CAAC,mBAAmB,CAA+B,IAAI,CAAC,CAAC;IACtE,CAAC;IAED,KAAK,CAAC,cAAc,CAClB,GAA0B;QAE1B,IAAI,CAAC;YACH,OAAO,MAAM,IAAI,CAAC,WAAW,CAC3B,gBAAgB,EAChB,GAAG,CACJ,CAAC;QACJ,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,mBAAmB,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC3B,OAAO;oBACL,WAAW,EAAE,EAAE,EAAE,EAAE,UAAU,CAAC,QAAQ,EAAE;iBACzC,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,CAAC;YACV,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,kBAAkB,CACtB,GAA8B;QAE9B,OAAO,IAAI,CAAC,WAAW,CACrB,oBAAoB,EACpB,GAAG,CACJ,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,8BAA8B;QAClC,OAAO,IAAI,CAAC,UAAU,CACpB,gCAAgC,CACjC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,8BAA8B,CAClC,GAA0C;QAE1C,OAAO,IAAI,CAAC,WAAW,CACrB,gCAAgC,EAChC,GAAG,CACJ,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,GAA0B;QAC1C,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,WAAW,CACjC,aAAa,EACb,mBAAmB,CAAC,GAAG,CAAC,CACzB,CAAC;QACF,OAAO,sBAAsB,CAAC,IAAI,CAAC,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,IAA4B;QAE5B,MAAM,KAAK,EAAE,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,QAAwB;QAExB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;QACpE,CAAC;QACD,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;QACjC,MAAM,GAAG,GAA2B;YAClC,OAAO,EAAE,SAAS;YAClB,QAAQ,EAAE,EAAE,GAAG,QAAQ,EAAE,WAAW,EAAE,SAAS,EAAE;SAClD,CAAC;QACF,OAAO,IAAI,CAAC,WAAW,CAA0B,iBAAiB,EAAE,GAAG,CAAC,CAAC;IAC3E,CAAC;IAED,KAAK,CAAC,iBAAiB,CACrB,GAA6B;QAE7B,OAAO,IAAI,CAAC,WAAW,CACrB,mBAAmB,EACnB,GAAG,CACJ,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,yBAAyB,CAC7B,mBAAwC;QAExC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,OAAO;QACT,CAAC;QAED,MAAM,IAAI,CAAC,uBAAuB,CAAC;YACjC,OAAO,EAAE,IAAI,CAAC,SAAS;YACvB,QAAQ,EAAE,MAAM,iBAAiB,EAAE;YACnC,OAAO,EAAE,CAAC,EAAE,mBAAmB,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,CAAC;SACxE,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,6BAA6B,CACjC,WAAoC;QAEpC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,OAAO;QACT,CAAC;QAED,MAAM,IAAI,CAAC,uBAAuB,CAAC;YACjC,OAAO,EAAE,IAAI,CAAC,SAAS;YACvB,QAAQ,EAAE,MAAM,iBAAiB,EAAE;YACnC,OAAO,EAAE;gBACP;oBACE,uBAAuB,EAAE,WAAW;oBACpC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;iBACpC;aACF;SACF,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,uBAAuB,CAC3B,OAAuC;QAEvC,OAAO,IAAI,CAAC,WAAW,CAAO,yBAAyB,EAAE,OAAO,CAAC,CAAC;IACpE,CAAC;IAED,KAAK,CAAC,WAAW,CACf,MAAc,EACd,GAAW,EACX,MAAoB;QAEpB,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;YACpC,GAAG,EAAE,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC;YAC9B,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO;aAC5B;YACD,YAAY,EAAE,MAAM;YACpB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC;YACzB,MAAM;SACP,CAAC,CAAC;QACH,OAAO,GAAG,CAAC,IAAS,CAAC;IACvB,CAAC;IAEO,KAAK,CAAC,cAAc,CAC1B,GAAW,EACX,MAAoB;QAEpB,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;YACpC,GAAG;YACH,MAAM,EAAE,KAAK;YACb,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO;aAC5B;YACD,YAAY,EAAE,MAAM;YACpB,MAAM;SACP,CAAC,CAAC;QACH,OAAO,GAAG,CAAC,IAAS,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,UAAU,CAAI,MAAc,EAAE,MAAoB;QACtD,OAAO,IAAI,CAAC,cAAc,CAAI,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC,CAAC;IACnE,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAI,IAAY,EAAE,MAAoB;QAC7D,OAAO,IAAI,CAAC,cAAc,CAAI,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC,CAAC;IACpE,CAAC;IAED,KAAK,CAAC,oBAAoB,CACxB,MAAc,EACd,GAAW,EACX,MAAoB;QAEpB,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;YACpC,GAAG,EAAE,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC;YAC9B,MAAM,EAAE,MAAM;YACd,MAAM,EAAE;gBACN,GAAG,EAAE,KAAK;aACX;YACD,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO;aAC5B;YACD,YAAY,EAAE,QAAQ;YACtB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC;YACzB,MAAM;SACP,CAAC,CAAC;QAEH,OAAO,CAAC,KAAK,SAAS,CAAC;YACrB,MAAM,EAAE,GAAG,QAAQ,CAAC,eAAe,CAAC;gBAClC,KAAK,EAAE,GAAG,CAAC,IAA6B;gBACxC,SAAS,EAAE,QAAQ,EAAE,4CAA4C;aAClE,CAAC,CAAC;YAEH,IAAI,aAAa,GAAa,EAAE,CAAC;YACjC,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,EAAE,EAAE,CAAC;gBAC5B,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAC9B,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC3C,CAAC;qBAAM,IAAI,IAAI,KAAK,EAAE,EAAE,CAAC;oBACvB,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;wBAC/B,SAAS,CAAC,mBAAmB;oBAC/B,CAAC;oBACD,MAAM,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAM,CAAC;oBAChD,aAAa,GAAG,EAAE,CAAC,CAAC,kCAAkC;gBACxD,CAAC;gBACD,gDAAgD;YAClD,CAAC;QACH,CAAC,CAAC,EAAE,CAAC;IACP,CAAC;IAEO,UAAU;QAChB,MAAM,QAAQ,GACZ,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,IAAI,oBAAoB,CAAC;QAC9D,MAAM,OAAO,GACX,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,IAAI,uBAAuB,CAAC;QACpE,OAAO,GAAG,QAAQ,IAAI,OAAO,EAAE,CAAC;IAClC,CAAC;IAED,YAAY,CAAC,MAAc;QACzB,OAAO,GAAG,IAAI,CAAC,UAAU,EAAE,IAAI,MAAM,EAAE,CAAC;IAC1C,CAAC;IAED,eAAe,CAAC,IAAY;QAC1B,OAAO,GAAG,IAAI,CAAC,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC;IACxC,CAAC;CACF;AAED,SAAS,mBAAmB,CAAC,KAAc;IACzC,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,UAAU,IAAI,KAAK,EAAE,CAAC;QAC9D,MAAM,WAAW,GAAG,KAInB,CAAC;QACF,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAQ,EAAE,IAE1B,CAAC;QACd,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,KAAK,EAAE,OAAO,CAAC,EAAE,CAAC;YAC5C,OAAO,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAChC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,KAAK,0BAA0B,CACzD,CAAC;QACJ,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC"}

package/dist/src/code_assist/setup.d.ts

@@ -0,0 +1,46 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { UserTierId } from './types.js';
+import type { AuthClient } from 'google-auth-library';
+import type { ValidationHandler } from '../fallback/types.js';
+export declare class ProjectIdRequiredError extends Error {
+    constructor();
+}
+/**
+ * Error thrown when user cancels the validation process.
+ * This is a non-recoverable error that should result in auth failure.
+ */
+export declare class ValidationCancelledError extends Error {
+    constructor();
+}
+export interface UserData {
+    projectId: string;
+    userTier: UserTierId;
+    userTierName?: string;
+}
+/**
+ * Sets up the user by loading their Code Assist configuration and onboarding if needed.
+ *
+ * Tier eligibility:
+ * - FREE tier: Eligibility is determined by the Code Assist server response.
+ * - STANDARD tier: User is always eligible if they have a valid project ID.
+ *
+ * If no valid project ID is available (from env var or server response):
+ * - Surfaces ineligibility reasons for the FREE tier from the server.
+ * - Throws ProjectIdRequiredError if no ineligibility reasons are available.
+ *
+ * Handles VALIDATION_REQUIRED via the optional validation handler, allowing
+ * retry, auth change, or cancellation.
+ *
+ * @param client - The authenticated client to use for API calls
+ * @param validationHandler - Optional handler for account validation flow
+ * @returns The user's project ID, tier ID, and tier name
+ * @throws {ValidationRequiredError} If account validation is required
+ * @throws {ProjectIdRequiredError} If no project ID is available and required
+ * @throws {ValidationCancelledError} If user cancels validation
+ * @throws {ChangeAuthRequestedError} If user requests to change auth method
+ */
+export declare function setupUser(client: AuthClient, validationHandler?: ValidationHandler): Promise<UserData>;

package/dist/src/code_assist/setup.js

@@ -0,0 +1,179 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { UserTierId, IneligibleTierReasonCode } from './types.js';
+import { CodeAssistServer } from './server.js';
+import { ChangeAuthRequestedError } from '../utils/errors.js';
+import { ValidationRequiredError } from '../utils/googleQuotaErrors.js';
+export class ProjectIdRequiredError extends Error {
+    constructor() {
+        super('This account requires setting the GOOGLE_CLOUD_PROJECT or GOOGLE_CLOUD_PROJECT_ID env var. See https://goo.gle/gemini-cli-auth-docs#workspace-gca');
+    }
+}
+/**
+ * Error thrown when user cancels the validation process.
+ * This is a non-recoverable error that should result in auth failure.
+ */
+export class ValidationCancelledError extends Error {
+    constructor() {
+        super('User cancelled account validation');
+    }
+}
+/**
+ * Sets up the user by loading their Code Assist configuration and onboarding if needed.
+ *
+ * Tier eligibility:
+ * - FREE tier: Eligibility is determined by the Code Assist server response.
+ * - STANDARD tier: User is always eligible if they have a valid project ID.
+ *
+ * If no valid project ID is available (from env var or server response):
+ * - Surfaces ineligibility reasons for the FREE tier from the server.
+ * - Throws ProjectIdRequiredError if no ineligibility reasons are available.
+ *
+ * Handles VALIDATION_REQUIRED via the optional validation handler, allowing
+ * retry, auth change, or cancellation.
+ *
+ * @param client - The authenticated client to use for API calls
+ * @param validationHandler - Optional handler for account validation flow
+ * @returns The user's project ID, tier ID, and tier name
+ * @throws {ValidationRequiredError} If account validation is required
+ * @throws {ProjectIdRequiredError} If no project ID is available and required
+ * @throws {ValidationCancelledError} If user cancels validation
+ * @throws {ChangeAuthRequestedError} If user requests to change auth method
+ */
+export async function setupUser(client, validationHandler) {
+    const projectId = process.env['GOOGLE_CLOUD_PROJECT'] ||
+        process.env['GOOGLE_CLOUD_PROJECT_ID'] ||
+        undefined;
+    const caServer = new CodeAssistServer(client, projectId, {}, '', undefined, undefined);
+    const coreClientMetadata = {
+        ideType: 'IDE_UNSPECIFIED',
+        platform: 'PLATFORM_UNSPECIFIED',
+        pluginType: 'GEMINI',
+    };
+    let loadRes;
+    while (true) {
+        loadRes = await caServer.loadCodeAssist({
+            cloudaicompanionProject: projectId,
+            metadata: {
+                ...coreClientMetadata,
+                duetProject: projectId,
+            },
+        });
+        try {
+            validateLoadCodeAssistResponse(loadRes);
+            break;
+        }
+        catch (e) {
+            if (e instanceof ValidationRequiredError && validationHandler) {
+                const intent = await validationHandler(e.validationLink, e.validationDescription);
+                if (intent === 'verify') {
+                    continue;
+                }
+                if (intent === 'change_auth') {
+                    throw new ChangeAuthRequestedError();
+                }
+                throw new ValidationCancelledError();
+            }
+            throw e;
+        }
+    }
+    if (loadRes.currentTier) {
+        if (!loadRes.cloudaicompanionProject) {
+            if (projectId) {
+                return {
+                    projectId,
+                    userTier: loadRes.currentTier.id,
+                    userTierName: loadRes.currentTier.name,
+                };
+            }
+            // If user is not setup for standard tier, inform them about all other tiers they are ineligible for.
+            if (loadRes.ineligibleTiers && loadRes.ineligibleTiers.length > 0) {
+                const reasons = loadRes.ineligibleTiers
+                    .map((t) => t.reasonMessage)
+                    .join(', ');
+                throw new Error(reasons);
+            }
+            throw new ProjectIdRequiredError();
+        }
+        return {
+            projectId: loadRes.cloudaicompanionProject,
+            userTier: loadRes.currentTier.id,
+            userTierName: loadRes.currentTier.name,
+        };
+    }
+    const tier = getOnboardTier(loadRes);
+    let onboardReq;
+    if (tier.id === UserTierId.FREE) {
+        // The free tier uses a managed google cloud project. Setting a project in the `onboardUser` request causes a `Precondition Failed` error.
+        onboardReq = {
+            tierId: tier.id,
+            cloudaicompanionProject: undefined,
+            metadata: coreClientMetadata,
+        };
+    }
+    else {
+        onboardReq = {
+            tierId: tier.id,
+            cloudaicompanionProject: projectId,
+            metadata: {
+                ...coreClientMetadata,
+                duetProject: projectId,
+            },
+        };
+    }
+    let lroRes = await caServer.onboardUser(onboardReq);
+    if (!lroRes.done && lroRes.name) {
+        const operationName = lroRes.name;
+        while (!lroRes.done) {
+            await new Promise((f) => setTimeout(f, 5000));
+            lroRes = await caServer.getOperation(operationName);
+        }
+    }
+    if (!lroRes.response?.cloudaicompanionProject?.id) {
+        if (projectId) {
+            return {
+                projectId,
+                userTier: tier.id,
+                userTierName: tier.name,
+            };
+        }
+        throw new ProjectIdRequiredError();
+    }
+    return {
+        projectId: lroRes.response.cloudaicompanionProject.id,
+        userTier: tier.id,
+        userTierName: tier.name,
+    };
+}
+function getOnboardTier(res) {
+    for (const tier of res.allowedTiers || []) {
+        if (tier.isDefault) {
+            return tier;
+        }
+    }
+    return {
+        name: '',
+        description: '',
+        id: UserTierId.LEGACY,
+        userDefinedCloudaicompanionProject: true,
+    };
+}
+function validateLoadCodeAssistResponse(res) {
+    if (!res) {
+        throw new Error('LoadCodeAssist returned empty response');
+    }
+    if (!res.currentTier &&
+        res.ineligibleTiers &&
+        res.ineligibleTiers.length > 0) {
+        const validationTier = res.ineligibleTiers.find((t) => t.validationUrl &&
+            t.reasonCode === IneligibleTierReasonCode.VALIDATION_REQUIRED);
+        const validationUrl = validationTier?.validationUrl;
+        if (validationTier && validationUrl) {
+            throw new ValidationRequiredError(`Account validation required: ${validationTier.reasonMessage}`, undefined, validationUrl, validationTier.reasonMessage);
+        }
+    }
+}
+//# sourceMappingURL=setup.js.map

package/dist/src/code_assist/setup.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"setup.js","sourceRoot":"","sources":["../../../src/code_assist/setup.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAQH,OAAO,EAAE,UAAU,EAAE,wBAAwB,EAAE,MAAM,YAAY,CAAC;AAClE,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAG/C,OAAO,EAAE,wBAAwB,EAAE,MAAM,oBAAoB,CAAC;AAC9D,OAAO,EAAE,uBAAuB,EAAE,MAAM,+BAA+B,CAAC;AAExE,MAAM,OAAO,sBAAuB,SAAQ,KAAK;IAC/C;QACE,KAAK,CACH,mJAAmJ,CACpJ,CAAC;IACJ,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,OAAO,wBAAyB,SAAQ,KAAK;IACjD;QACE,KAAK,CAAC,mCAAmC,CAAC,CAAC;IAC7C,CAAC;CACF;AAQD;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAM,CAAC,KAAK,UAAU,SAAS,CAC7B,MAAkB,EAClB,iBAAqC;IAErC,MAAM,SAAS,GACb,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC;QACnC,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC;QACtC,SAAS,CAAC;IACZ,MAAM,QAAQ,GAAG,IAAI,gBAAgB,CACnC,MAAM,EACN,SAAS,EACT,EAAE,EACF,EAAE,EACF,SAAS,EACT,SAAS,CACV,CAAC;IACF,MAAM,kBAAkB,GAAmB;QACzC,OAAO,EAAE,iBAAiB;QAC1B,QAAQ,EAAE,sBAAsB;QAChC,UAAU,EAAE,QAAQ;KACrB,CAAC;IAEF,IAAI,OAA+B,CAAC;IACpC,OAAO,IAAI,EAAE,CAAC;QACZ,OAAO,GAAG,MAAM,QAAQ,CAAC,cAAc,CAAC;YACtC,uBAAuB,EAAE,SAAS;YAClC,QAAQ,EAAE;gBACR,GAAG,kBAAkB;gBACrB,WAAW,EAAE,SAAS;aACvB;SACF,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,8BAA8B,CAAC,OAAO,CAAC,CAAC;YACxC,MAAM;QACR,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,CAAC,YAAY,uBAAuB,IAAI,iBAAiB,EAAE,CAAC;gBAC9D,MAAM,MAAM,GAAG,MAAM,iBAAiB,CACpC,CAAC,CAAC,cAAc,EAChB,CAAC,CAAC,qBAAqB,CACxB,CAAC;gBACF,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;oBACxB,SAAS;gBACX,CAAC;gBACD,IAAI,MAAM,KAAK,aAAa,EAAE,CAAC;oBAC7B,MAAM,IAAI,wBAAwB,EAAE,CAAC;gBACvC,CAAC;gBACD,MAAM,IAAI,wBAAwB,EAAE,CAAC;YACvC,CAAC;YACD,MAAM,CAAC,CAAC;QACV,CAAC;IACH,CAAC;IAED,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;QACxB,IAAI,CAAC,OAAO,CAAC,uBAAuB,EAAE,CAAC;YACrC,IAAI,SAAS,EAAE,CAAC;gBACd,OAAO;oBACL,SAAS;oBACT,QAAQ,EAAE,OAAO,CAAC,WAAW,CAAC,EAAE;oBAChC,YAAY,EAAE,OAAO,CAAC,WAAW,CAAC,IAAI;iBACvC,CAAC;YACJ,CAAC;YAED,qGAAqG;YACrG,IAAI,OAAO,CAAC,eAAe,IAAI,OAAO,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAClE,MAAM,OAAO,GAAG,OAAO,CAAC,eAAe;qBACpC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC;qBAC3B,IAAI,CAAC,IAAI,CAAC,CAAC;gBACd,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC;YAC3B,CAAC;YACD,MAAM,IAAI,sBAAsB,EAAE,CAAC;QACrC,CAAC;QACD,OAAO;YACL,SAAS,EAAE,OAAO,CAAC,uBAAuB;YAC1C,QAAQ,EAAE,OAAO,CAAC,WAAW,CAAC,EAAE;YAChC,YAAY,EAAE,OAAO,CAAC,WAAW,CAAC,IAAI;SACvC,CAAC;IACJ,CAAC;IAED,MAAM,IAAI,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;IAErC,IAAI,UAA8B,CAAC;IACnC,IAAI,IAAI,CAAC,EAAE,KAAK,UAAU,CAAC,IAAI,EAAE,CAAC;QAChC,0IAA0I;QAC1I,UAAU,GAAG;YACX,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,uBAAuB,EAAE,SAAS;YAClC,QAAQ,EAAE,kBAAkB;SAC7B,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,UAAU,GAAG;YACX,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,uBAAuB,EAAE,SAAS;YAClC,QAAQ,EAAE;gBACR,GAAG,kBAAkB;gBACrB,WAAW,EAAE,SAAS;aACvB;SACF,CAAC;IACJ,CAAC;IAED,IAAI,MAAM,GAAG,MAAM,QAAQ,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;IACpD,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;QAChC,MAAM,aAAa,GAAG,MAAM,CAAC,IAAI,CAAC;QAClC,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACpB,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;YAC9C,MAAM,GAAG,MAAM,QAAQ,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;IAED,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,uBAAuB,EAAE,EAAE,EAAE,CAAC;QAClD,IAAI,SAAS,EAAE,CAAC;YACd,OAAO;gBACL,SAAS;gBACT,QAAQ,EAAE,IAAI,CAAC,EAAE;gBACjB,YAAY,EAAE,IAAI,CAAC,IAAI;aACxB,CAAC;QACJ,CAAC;QACD,MAAM,IAAI,sBAAsB,EAAE,CAAC;IACrC,CAAC;IAED,OAAO;QACL,SAAS,EAAE,MAAM,CAAC,QAAQ,CAAC,uBAAuB,CAAC,EAAE;QACrD,QAAQ,EAAE,IAAI,CAAC,EAAE;QACjB,YAAY,EAAE,IAAI,CAAC,IAAI;KACxB,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CAAC,GAA2B;IACjD,KAAK,MAAM,IAAI,IAAI,GAAG,CAAC,YAAY,IAAI,EAAE,EAAE,CAAC;QAC1C,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IACD,OAAO;QACL,IAAI,EAAE,EAAE;QACR,WAAW,EAAE,EAAE;QACf,EAAE,EAAE,UAAU,CAAC,MAAM;QACrB,kCAAkC,EAAE,IAAI;KACzC,CAAC;AACJ,CAAC;AAED,SAAS,8BAA8B,CAAC,GAA2B;IACjE,IAAI,CAAC,GAAG,EAAE,CAAC;QACT,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;IAC5D,CAAC;IACD,IACE,CAAC,GAAG,CAAC,WAAW;QAChB,GAAG,CAAC,eAAe;QACnB,GAAG,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,EAC9B,CAAC;QACD,MAAM,cAAc,GAAG,GAAG,CAAC,eAAe,CAAC,IAAI,CAC7C,CAAC,CAAC,EAAE,EAAE,CACJ,CAAC,CAAC,aAAa;YACf,CAAC,CAAC,UAAU,KAAK,wBAAwB,CAAC,mBAAmB,CAChE,CAAC;QACF,MAAM,aAAa,GAAG,cAAc,EAAE,aAAa,CAAC;QACpD,IAAI,cAAc,IAAI,aAAa,EAAE,CAAC;YACpC,MAAM,IAAI,uBAAuB,CAC/B,gCAAgC,cAAc,CAAC,aAAa,EAAE,EAC9D,SAAS,EACT,aAAa,EACb,cAAc,CAAC,aAAa,CAC7B,CAAC;QACJ,CAAC;IACH,CAAC;AACH,CAAC"}

package/dist/src/code_assist/telemetry.d.ts

@@ -0,0 +1,14 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { type GenerateContentResponse } from '@google/genai';
+import { type ConversationOffered, type StreamingLatency } from './types.js';
+import type { CompletedToolCall } from '../core/coreToolScheduler.js';
+import type { Config } from '../config/config.js';
+import type { CodeAssistServer } from './server.js';
+export declare function recordConversationOffered(server: CodeAssistServer, traceId: string | undefined, response: GenerateContentResponse, streamingLatency: StreamingLatency, abortSignal: AbortSignal | undefined): Promise<void>;
+export declare function recordToolCallInteractions(config: Config, toolCalls: CompletedToolCall[]): Promise<void>;
+export declare function createConversationOffered(response: GenerateContentResponse, traceId: string, signal: AbortSignal | undefined, streamingLatency: StreamingLatency): ConversationOffered | undefined;
+export declare function formatProtoJsonDuration(milliseconds: number): string;

package/dist/src/code_assist/telemetry.js

@@ -0,0 +1,157 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { FinishReason } from '@google/genai';
+import { getCitations } from '../utils/generateContentResponseUtilities.js';
+import { ActionStatus, ConversationInteractionInteraction, InitiationMethod, } from './types.js';
+import { debugLogger } from '../utils/debugLogger.js';
+import { getCodeAssistServer } from './codeAssist.js';
+import { EDIT_TOOL_NAMES } from '../tools/tool-names.js';
+import { getErrorMessage } from '../utils/errors.js';
+import { ToolConfirmationOutcome } from '../tools/tools.js';
+export async function recordConversationOffered(server, traceId, response, streamingLatency, abortSignal) {
+    try {
+        if (traceId) {
+            const offered = createConversationOffered(response, traceId, abortSignal, streamingLatency);
+            if (offered) {
+                await server.recordConversationOffered(offered);
+            }
+        }
+    }
+    catch (error) {
+        debugLogger.warn(`Error recording tool call interactions: ${getErrorMessage(error)}`);
+    }
+}
+export async function recordToolCallInteractions(config, toolCalls) {
+    // Only send interaction events for responses that contain function calls.
+    if (toolCalls.length === 0) {
+        return;
+    }
+    try {
+        const server = getCodeAssistServer(config);
+        if (!server) {
+            return;
+        }
+        const interaction = summarizeToolCalls(toolCalls);
+        if (interaction) {
+            await server.recordConversationInteraction(interaction);
+        }
+    }
+    catch (error) {
+        debugLogger.warn(`Error recording tool call interactions: ${getErrorMessage(error)}`);
+    }
+}
+export function createConversationOffered(response, traceId, signal, streamingLatency) {
+    // Only send conversation offered events for responses that contain function
+    // calls. Non-function call events don't represent user actionable
+    // 'suggestions'.
+    if ((response.functionCalls?.length || 0) === 0) {
+        return;
+    }
+    const actionStatus = getStatusFromResponse(response, signal);
+    return {
+        citationCount: String(getCitations(response).length),
+        includedCode: includesCode(response),
+        status: actionStatus,
+        traceId,
+        streamingLatency,
+        isAgentic: true,
+        initiationMethod: InitiationMethod.COMMAND,
+    };
+}
+function summarizeToolCalls(toolCalls) {
+    let acceptedToolCalls = 0;
+    let actionStatus = undefined;
+    let traceId = undefined;
+    // Treat file edits as ACCEPT_FILE and everything else as unknown.
+    let isEdit = false;
+    // Iterate the tool calls and summarize them into a single conversation
+    // interaction so that the ConversationOffered and ConversationInteraction
+    // events are 1:1 in telemetry.
+    for (const toolCall of toolCalls) {
+        traceId ||= toolCall.request.traceId;
+        // If any tool call is canceled, we treat the entire interaction as canceled.
+        if (toolCall.status === 'cancelled') {
+            actionStatus = ActionStatus.ACTION_STATUS_CANCELLED;
+            break;
+        }
+        // If any tool call encounters an error, we treat the entire interaction as
+        // having errored.
+        if (toolCall.status === 'error') {
+            actionStatus = ActionStatus.ACTION_STATUS_ERROR_UNKNOWN;
+            break;
+        }
+        // Record if the tool call was accepted.
+        if (toolCall.outcome !== ToolConfirmationOutcome.Cancel) {
+            acceptedToolCalls++;
+            // Edits are ACCEPT_FILE, everything else is UNKNOWN.
+            if (EDIT_TOOL_NAMES.has(toolCall.request.name)) {
+                isEdit ||= true;
+            }
+        }
+    }
+    // Only file interaction telemetry if 100% of the tool calls were accepted.
+    return traceId && acceptedToolCalls / toolCalls.length >= 1
+        ? createConversationInteraction(traceId, actionStatus || ActionStatus.ACTION_STATUS_NO_ERROR, isEdit
+            ? ConversationInteractionInteraction.ACCEPT_FILE
+            : ConversationInteractionInteraction.UNKNOWN)
+        : undefined;
+}
+function createConversationInteraction(traceId, status, interaction) {
+    return {
+        traceId,
+        status,
+        interaction,
+        isAgentic: true,
+    };
+}
+function includesCode(resp) {
+    if (!resp.candidates) {
+        return false;
+    }
+    for (const candidate of resp.candidates) {
+        if (!candidate.content || !candidate.content.parts) {
+            continue;
+        }
+        for (const part of candidate.content.parts) {
+            if ('text' in part && part?.text?.includes('```')) {
+                return true;
+            }
+        }
+    }
+    return false;
+}
+function getStatusFromResponse(response, signal) {
+    if (signal?.aborted) {
+        return ActionStatus.ACTION_STATUS_CANCELLED;
+    }
+    if (hasError(response)) {
+        return ActionStatus.ACTION_STATUS_ERROR_UNKNOWN;
+    }
+    if ((response.candidates?.length ?? 0) <= 0) {
+        return ActionStatus.ACTION_STATUS_EMPTY;
+    }
+    return ActionStatus.ACTION_STATUS_NO_ERROR;
+}
+export function formatProtoJsonDuration(milliseconds) {
+    return `${milliseconds / 1000}s`;
+}
+function hasError(response) {
+    // Non-OK SDK results should be considered an error.
+    if (response.sdkHttpResponse &&
+        !response.sdkHttpResponse?.responseInternal?.ok) {
+        return true;
+    }
+    for (const candidate of response.candidates || []) {
+        // Treat sanitization, SPII, recitation, and forbidden terms as an error.
+        if (candidate.finishReason &&
+            candidate.finishReason !== FinishReason.STOP &&
+            candidate.finishReason !== FinishReason.MAX_TOKENS) {
+            return true;
+        }
+    }
+    return false;
+}
+//# sourceMappingURL=telemetry.js.map

package/dist/src/code_assist/telemetry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"telemetry.js","sourceRoot":"","sources":["../../../src/code_assist/telemetry.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,YAAY,EAAgC,MAAM,eAAe,CAAC;AAC3E,OAAO,EAAE,YAAY,EAAE,MAAM,8CAA8C,CAAC;AAC5E,OAAO,EACL,YAAY,EACZ,kCAAkC,EAClC,gBAAgB,GAIjB,MAAM,YAAY,CAAC;AAGpB,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AACtD,OAAO,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAErD,OAAO,EAAE,uBAAuB,EAAE,MAAM,mBAAmB,CAAC;AAE5D,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC7C,MAAwB,EACxB,OAA2B,EAC3B,QAAiC,EACjC,gBAAkC,EAClC,WAAoC;IAEpC,IAAI,CAAC;QACH,IAAI,OAAO,EAAE,CAAC;YACZ,MAAM,OAAO,GAAG,yBAAyB,CACvC,QAAQ,EACR,OAAO,EACP,WAAW,EACX,gBAAgB,CACjB,CAAC;YACF,IAAI,OAAO,EAAE,CAAC;gBACZ,MAAM,MAAM,CAAC,yBAAyB,CAAC,OAAO,CAAC,CAAC;YAClD,CAAC;QACH,CAAC;IACH,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACxB,WAAW,CAAC,IAAI,CACd,2CAA2C,eAAe,CAAC,KAAK,CAAC,EAAE,CACpE,CAAC;IACJ,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC9C,MAAc,EACd,SAA8B;IAE9B,0EAA0E;IAC1E,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC3B,OAAO;IACT,CAAC;IAED,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,mBAAmB,CAAC,MAAM,CAAC,CAAC;QAC3C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO;QACT,CAAC;QAED,MAAM,WAAW,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAC;QAClD,IAAI,WAAW,EAAE,CAAC;YAChB,MAAM,MAAM,CAAC,6BAA6B,CAAC,WAAW,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACxB,WAAW,CAAC,IAAI,CACd,2CAA2C,eAAe,CAAC,KAAK,CAAC,EAAE,CACpE,CAAC;IACJ,CAAC;AACH,CAAC;AAED,MAAM,UAAU,yBAAyB,CACvC,QAAiC,EACjC,OAAe,EACf,MAA+B,EAC/B,gBAAkC;IAElC,4EAA4E;IAC5E,kEAAkE;IAClE,iBAAiB;IACjB,IAAI,CAAC,QAAQ,CAAC,aAAa,EAAE,MAAM,IAAI,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;QAChD,OAAO;IACT,CAAC;IAED,MAAM,YAAY,GAAG,qBAAqB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IAE7D,OAAO;QACL,aAAa,EAAE,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC;QACpD,YAAY,EAAE,YAAY,CAAC,QAAQ,CAAC;QACpC,MAAM,EAAE,YAAY;QACpB,OAAO;QACP,gBAAgB;QAChB,SAAS,EAAE,IAAI;QACf,gBAAgB,EAAE,gBAAgB,CAAC,OAAO;KAC3C,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CACzB,SAA8B;IAE9B,IAAI,iBAAiB,GAAG,CAAC,CAAC;IAC1B,IAAI,YAAY,GAAG,SAAS,CAAC;IAC7B,IAAI,OAAO,GAAG,SAAS,CAAC;IAExB,kEAAkE;IAClE,IAAI,MAAM,GAAG,KAAK,CAAC;IAEnB,uEAAuE;IACvE,0EAA0E;IAC1E,+BAA+B;IAC/B,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QACjC,OAAO,KAAK,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC;QAErC,6EAA6E;QAC7E,IAAI,QAAQ,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;YACpC,YAAY,GAAG,YAAY,CAAC,uBAAuB,CAAC;YACpD,MAAM;QACR,CAAC;QAED,2EAA2E;QAC3E,kBAAkB;QAClB,IAAI,QAAQ,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;YAChC,YAAY,GAAG,YAAY,CAAC,2BAA2B,CAAC;YACxD,MAAM;QACR,CAAC;QAED,wCAAwC;QACxC,IAAI,QAAQ,CAAC,OAAO,KAAK,uBAAuB,CAAC,MAAM,EAAE,CAAC;YACxD,iBAAiB,EAAE,CAAC;YAEpB,qDAAqD;YACrD,IAAI,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC/C,MAAM,KAAK,IAAI,CAAC;YAClB,CAAC;QACH,CAAC;IACH,CAAC;IAED,2EAA2E;IAC3E,OAAO,OAAO,IAAI,iBAAiB,GAAG,SAAS,CAAC,MAAM,IAAI,CAAC;QACzD,CAAC,CAAC,6BAA6B,CAC3B,OAAO,EACP,YAAY,IAAI,YAAY,CAAC,sBAAsB,EACnD,MAAM;YACJ,CAAC,CAAC,kCAAkC,CAAC,WAAW;YAChD,CAAC,CAAC,kCAAkC,CAAC,OAAO,CAC/C;QACH,CAAC,CAAC,SAAS,CAAC;AAChB,CAAC;AAED,SAAS,6BAA6B,CACpC,OAAe,EACf,MAAoB,EACpB,WAA+C;IAE/C,OAAO;QACL,OAAO;QACP,MAAM;QACN,WAAW;QACX,SAAS,EAAE,IAAI;KAChB,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,IAA6B;IACjD,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;QACrB,OAAO,KAAK,CAAC;IACf,CAAC;IACD,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;QACxC,IAAI,CAAC,SAAS,CAAC,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACnD,SAAS;QACX,CAAC;QACD,KAAK,MAAM,IAAI,IAAI,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YAC3C,IAAI,MAAM,IAAI,IAAI,IAAI,IAAI,EAAE,IAAI,EAAE,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;gBAClD,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,qBAAqB,CAC5B,QAAiC,EACjC,MAA+B;IAE/B,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;QACpB,OAAO,YAAY,CAAC,uBAAuB,CAAC;IAC9C,CAAC;IAED,IAAI,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QACvB,OAAO,YAAY,CAAC,2BAA2B,CAAC;IAClD,CAAC;IAED,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,MAAM,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;QAC5C,OAAO,YAAY,CAAC,mBAAmB,CAAC;IAC1C,CAAC;IAED,OAAO,YAAY,CAAC,sBAAsB,CAAC;AAC7C,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,YAAoB;IAC1D,OAAO,GAAG,YAAY,GAAG,IAAI,GAAG,CAAC;AACnC,CAAC;AAED,SAAS,QAAQ,CAAC,QAAiC;IACjD,oDAAoD;IACpD,IACE,QAAQ,CAAC,eAAe;QACxB,CAAC,QAAQ,CAAC,eAAe,EAAE,gBAAgB,EAAE,EAAE,EAC/C,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,MAAM,SAAS,IAAI,QAAQ,CAAC,UAAU,IAAI,EAAE,EAAE,CAAC;QAClD,yEAAyE;QACzE,IACE,SAAS,CAAC,YAAY;YACtB,SAAS,CAAC,YAAY,KAAK,YAAY,CAAC,IAAI;YAC5C,SAAS,CAAC,YAAY,KAAK,YAAY,CAAC,UAAU,EAClD,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC"}