@vybestack/llxprt-code-core 0.1.12

package/dist/src/telemetry/types.js

@@ -0,0 +1,217 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { ToolConfirmationOutcome } from '../tools/tools.js';
+import { AuthType } from '../core/contentGenerator.js';
+export var ToolCallDecision;
+(function (ToolCallDecision) {
+    ToolCallDecision["ACCEPT"] = "accept";
+    ToolCallDecision["REJECT"] = "reject";
+    ToolCallDecision["MODIFY"] = "modify";
+})(ToolCallDecision || (ToolCallDecision = {}));
+export function getDecisionFromOutcome(outcome) {
+    switch (outcome) {
+        case ToolConfirmationOutcome.ProceedOnce:
+        case ToolConfirmationOutcome.ProceedAlways:
+        case ToolConfirmationOutcome.ProceedAlwaysServer:
+        case ToolConfirmationOutcome.ProceedAlwaysTool:
+            return ToolCallDecision.ACCEPT;
+        case ToolConfirmationOutcome.ModifyWithEditor:
+            return ToolCallDecision.MODIFY;
+        case ToolConfirmationOutcome.Cancel:
+        default:
+            return ToolCallDecision.REJECT;
+    }
+}
+export class StartSessionEvent {
+    'event.name';
+    'event.timestamp'; // ISO 8601
+    model;
+    embedding_model;
+    sandbox_enabled;
+    core_tools_enabled;
+    approval_mode;
+    api_key_enabled;
+    vertex_ai_enabled;
+    debug_enabled;
+    mcp_servers;
+    telemetry_enabled;
+    telemetry_log_user_prompts_enabled;
+    file_filtering_respect_git_ignore;
+    constructor(config) {
+        const generatorConfig = config.getContentGeneratorConfig();
+        const mcpServers = config.getMcpServers();
+        let useGemini = false;
+        let useVertex = false;
+        if (generatorConfig && generatorConfig.authType) {
+            useGemini = generatorConfig.authType === AuthType.USE_GEMINI;
+            useVertex = generatorConfig.authType === AuthType.USE_VERTEX_AI;
+        }
+        this['event.name'] = 'cli_config';
+        this.model = config.getModel();
+        this.embedding_model = config.getEmbeddingModel();
+        this.sandbox_enabled =
+            typeof config.getSandbox() === 'string' || !!config.getSandbox();
+        this.core_tools_enabled = (config.getCoreTools() ?? []).join(',');
+        this.approval_mode = config.getApprovalMode();
+        this.api_key_enabled = useGemini || useVertex;
+        this.vertex_ai_enabled = useVertex;
+        this.debug_enabled = config.getDebugMode();
+        this.mcp_servers = mcpServers ? Object.keys(mcpServers).join(',') : '';
+        this.telemetry_enabled = config.getTelemetryEnabled();
+        this.telemetry_log_user_prompts_enabled =
+            config.getTelemetryLogPromptsEnabled();
+        this.file_filtering_respect_git_ignore =
+            config.getFileFilteringRespectGitIgnore();
+    }
+}
+export class EndSessionEvent {
+    'event.name';
+    'event.timestamp'; // ISO 8601
+    session_id;
+    constructor(config) {
+        this['event.name'] = 'end_session';
+        this['event.timestamp'] = new Date().toISOString();
+        this.session_id = config?.getSessionId();
+    }
+}
+export class UserPromptEvent {
+    'event.name';
+    'event.timestamp'; // ISO 8601
+    prompt_length;
+    prompt_id;
+    auth_type;
+    prompt;
+    constructor(prompt_length, prompt_Id, auth_type, prompt) {
+        this['event.name'] = 'user_prompt';
+        this['event.timestamp'] = new Date().toISOString();
+        this.prompt_length = prompt_length;
+        this.prompt_id = prompt_Id;
+        this.auth_type = auth_type;
+        this.prompt = prompt;
+    }
+}
+export class ToolCallEvent {
+    'event.name';
+    'event.timestamp'; // ISO 8601
+    function_name;
+    function_args;
+    duration_ms;
+    success;
+    decision;
+    error;
+    error_type;
+    prompt_id;
+    constructor(call) {
+        this['event.name'] = 'tool_call';
+        this['event.timestamp'] = new Date().toISOString();
+        this.function_name = call.request.name;
+        this.function_args = call.request.args;
+        this.duration_ms = call.durationMs ?? 0;
+        this.success = call.status === 'success';
+        this.decision = call.outcome
+            ? getDecisionFromOutcome(call.outcome)
+            : undefined;
+        this.error = call.response.error?.message;
+        this.error_type = call.response.error?.name;
+        this.prompt_id = call.request.prompt_id;
+    }
+}
+export class ApiRequestEvent {
+    'event.name';
+    'event.timestamp'; // ISO 8601
+    model;
+    prompt_id;
+    request_text;
+    constructor(model, prompt_id, request_text) {
+        this['event.name'] = 'api_request';
+        this['event.timestamp'] = new Date().toISOString();
+        this.model = model;
+        this.prompt_id = prompt_id;
+        this.request_text = request_text;
+    }
+}
+export class ApiErrorEvent {
+    'event.name';
+    'event.timestamp'; // ISO 8601
+    model;
+    error;
+    error_type;
+    status_code;
+    duration_ms;
+    prompt_id;
+    auth_type;
+    constructor(model, error, duration_ms, prompt_id, auth_type, error_type, status_code) {
+        this['event.name'] = 'api_error';
+        this['event.timestamp'] = new Date().toISOString();
+        this.model = model;
+        this.error = error;
+        this.error_type = error_type;
+        this.status_code = status_code;
+        this.duration_ms = duration_ms;
+        this.prompt_id = prompt_id;
+        this.auth_type = auth_type;
+    }
+}
+export class ApiResponseEvent {
+    'event.name';
+    'event.timestamp'; // ISO 8601
+    model;
+    status_code;
+    duration_ms;
+    error;
+    input_token_count;
+    output_token_count;
+    cached_content_token_count;
+    thoughts_token_count;
+    tool_token_count;
+    total_token_count;
+    response_text;
+    prompt_id;
+    auth_type;
+    constructor(model, duration_ms, prompt_id, auth_type, usage_data, response_text, error) {
+        this['event.name'] = 'api_response';
+        this['event.timestamp'] = new Date().toISOString();
+        this.model = model;
+        this.duration_ms = duration_ms;
+        this.status_code = 200;
+        this.input_token_count = usage_data?.promptTokenCount ?? 0;
+        this.output_token_count = usage_data?.candidatesTokenCount ?? 0;
+        this.cached_content_token_count = usage_data?.cachedContentTokenCount ?? 0;
+        this.thoughts_token_count = usage_data?.thoughtsTokenCount ?? 0;
+        this.tool_token_count = usage_data?.toolUsePromptTokenCount ?? 0;
+        this.total_token_count = usage_data?.totalTokenCount ?? 0;
+        this.response_text = response_text;
+        this.error = error;
+        this.prompt_id = prompt_id;
+        this.auth_type = auth_type;
+    }
+}
+export class FlashFallbackEvent {
+    'event.name';
+    'event.timestamp'; // ISO 8601
+    auth_type;
+    constructor(auth_type) {
+        this['event.name'] = 'flash_fallback';
+        this['event.timestamp'] = new Date().toISOString();
+        this.auth_type = auth_type;
+    }
+}
+export var LoopType;
+(function (LoopType) {
+    LoopType["CONSECUTIVE_IDENTICAL_TOOL_CALLS"] = "consecutive_identical_tool_calls";
+    LoopType["CHANTING_IDENTICAL_SENTENCES"] = "chanting_identical_sentences";
+})(LoopType || (LoopType = {}));
+export class LoopDetectedEvent {
+    'event.name';
+    'event.timestamp'; // ISO 8601
+    loop_type;
+    constructor(loop_type) {
+        this['event.name'] = 'loop_detected';
+        this['event.timestamp'] = new Date().toISOString();
+        this.loop_type = loop_type;
+    }
+}
+//# sourceMappingURL=types.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/telemetry/types.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAKH,OAAO,EAAE,uBAAuB,EAAE,MAAM,mBAAmB,CAAC;AAC5D,OAAO,EAAE,QAAQ,EAAE,MAAM,6BAA6B,CAAC;AAEvD,MAAM,CAAN,IAAY,gBAIX;AAJD,WAAY,gBAAgB;IAC1B,qCAAiB,CAAA;IACjB,qCAAiB,CAAA;IACjB,qCAAiB,CAAA;AACnB,CAAC,EAJW,gBAAgB,KAAhB,gBAAgB,QAI3B;AAED,MAAM,UAAU,sBAAsB,CACpC,OAAgC;IAEhC,QAAQ,OAAO,EAAE,CAAC;QAChB,KAAK,uBAAuB,CAAC,WAAW,CAAC;QACzC,KAAK,uBAAuB,CAAC,aAAa,CAAC;QAC3C,KAAK,uBAAuB,CAAC,mBAAmB,CAAC;QACjD,KAAK,uBAAuB,CAAC,iBAAiB;YAC5C,OAAO,gBAAgB,CAAC,MAAM,CAAC;QACjC,KAAK,uBAAuB,CAAC,gBAAgB;YAC3C,OAAO,gBAAgB,CAAC,MAAM,CAAC;QACjC,KAAK,uBAAuB,CAAC,MAAM,CAAC;QACpC;YACE,OAAO,gBAAgB,CAAC,MAAM,CAAC;IACnC,CAAC;AACH,CAAC;AAED,MAAM,OAAO,iBAAiB;IAC5B,YAAY,CAAe;IAC3B,iBAAiB,CAAS,CAAC,WAAW;IACtC,KAAK,CAAS;IACd,eAAe,CAAS;IACxB,eAAe,CAAU;IACzB,kBAAkB,CAAS;IAC3B,aAAa,CAAS;IACtB,eAAe,CAAU;IACzB,iBAAiB,CAAU;IAC3B,aAAa,CAAU;IACvB,WAAW,CAAS;IACpB,iBAAiB,CAAU;IAC3B,kCAAkC,CAAU;IAC5C,iCAAiC,CAAU;IAE3C,YAAY,MAAc;QACxB,MAAM,eAAe,GAAG,MAAM,CAAC,yBAAyB,EAAE,CAAC;QAC3D,MAAM,UAAU,GAAG,MAAM,CAAC,aAAa,EAAE,CAAC;QAE1C,IAAI,SAAS,GAAG,KAAK,CAAC;QACtB,IAAI,SAAS,GAAG,KAAK,CAAC;QACtB,IAAI,eAAe,IAAI,eAAe,CAAC,QAAQ,EAAE,CAAC;YAChD,SAAS,GAAG,eAAe,CAAC,QAAQ,KAAK,QAAQ,CAAC,UAAU,CAAC;YAC7D,SAAS,GAAG,eAAe,CAAC,QAAQ,KAAK,QAAQ,CAAC,aAAa,CAAC;QAClE,CAAC;QAED,IAAI,CAAC,YAAY,CAAC,GAAG,YAAY,CAAC;QAClC,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC;QAC/B,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,iBAAiB,EAAE,CAAC;QAClD,IAAI,CAAC,eAAe;YAClB,OAAO,MAAM,CAAC,UAAU,EAAE,KAAK,QAAQ,IAAI,CAAC,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;QACnE,IAAI,CAAC,kBAAkB,GAAG,CAAC,MAAM,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAClE,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC,eAAe,EAAE,CAAC;QAC9C,IAAI,CAAC,eAAe,GAAG,SAAS,IAAI,SAAS,CAAC;QAC9C,IAAI,CAAC,iBAAiB,GAAG,SAAS,CAAC;QACnC,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC,YAAY,EAAE,CAAC;QAC3C,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACvE,IAAI,CAAC,iBAAiB,GAAG,MAAM,CAAC,mBAAmB,EAAE,CAAC;QACtD,IAAI,CAAC,kCAAkC;YACrC,MAAM,CAAC,6BAA6B,EAAE,CAAC;QACzC,IAAI,CAAC,iCAAiC;YACpC,MAAM,CAAC,gCAAgC,EAAE,CAAC;IAC9C,CAAC;CACF;AAED,MAAM,OAAO,eAAe;IAC1B,YAAY,CAAgB;IAC5B,iBAAiB,CAAS,CAAC,WAAW;IACtC,UAAU,CAAU;IAEpB,YAAY,MAAe;QACzB,IAAI,CAAC,YAAY,CAAC,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,iBAAiB,CAAC,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACnD,IAAI,CAAC,UAAU,GAAG,MAAM,EAAE,YAAY,EAAE,CAAC;IAC3C,CAAC;CACF;AAED,MAAM,OAAO,eAAe;IAC1B,YAAY,CAAgB;IAC5B,iBAAiB,CAAS,CAAC,WAAW;IACtC,aAAa,CAAS;IACtB,SAAS,CAAS;IAClB,SAAS,CAAU;IACnB,MAAM,CAAU;IAEhB,YACE,aAAqB,EACrB,SAAiB,EACjB,SAAkB,EAClB,MAAe;QAEf,IAAI,CAAC,YAAY,CAAC,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,iBAAiB,CAAC,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACnD,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;CACF;AAED,MAAM,OAAO,aAAa;IACxB,YAAY,CAAc;IAC1B,iBAAiB,CAAS,CAAC,WAAW;IACtC,aAAa,CAAS;IACtB,aAAa,CAA0B;IACvC,WAAW,CAAS;IACpB,OAAO,CAAU;IACjB,QAAQ,CAAoB;IAC5B,KAAK,CAAU;IACf,UAAU,CAAU;IACpB,SAAS,CAAS;IAElB,YAAY,IAAuB;QACjC,IAAI,CAAC,YAAY,CAAC,GAAG,WAAW,CAAC;QACjC,IAAI,CAAC,iBAAiB,CAAC,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACnD,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;QACvC,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;QACvC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,UAAU,IAAI,CAAC,CAAC;QACxC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,MAAM,KAAK,SAAS,CAAC;QACzC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,OAAO;YAC1B,CAAC,CAAC,sBAAsB,CAAC,IAAI,CAAC,OAAO,CAAC;YACtC,CAAC,CAAC,SAAS,CAAC;QACd,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,OAAO,CAAC;QAC1C,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,IAAI,CAAC;QAC5C,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC;IAC1C,CAAC;CACF;AAED,MAAM,OAAO,eAAe;IAC1B,YAAY,CAAgB;IAC5B,iBAAiB,CAAS,CAAC,WAAW;IACtC,KAAK,CAAS;IACd,SAAS,CAAS;IAClB,YAAY,CAAU;IAEtB,YAAY,KAAa,EAAE,SAAiB,EAAE,YAAqB;QACjE,IAAI,CAAC,YAAY,CAAC,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,iBAAiB,CAAC,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACnD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;IACnC,CAAC;CACF;AAED,MAAM,OAAO,aAAa;IACxB,YAAY,CAAc;IAC1B,iBAAiB,CAAS,CAAC,WAAW;IACtC,KAAK,CAAS;IACd,KAAK,CAAS;IACd,UAAU,CAAU;IACpB,WAAW,CAAmB;IAC9B,WAAW,CAAS;IACpB,SAAS,CAAS;IAClB,SAAS,CAAU;IAEnB,YACE,KAAa,EACb,KAAa,EACb,WAAmB,EACnB,SAAiB,EACjB,SAAkB,EAClB,UAAmB,EACnB,WAA6B;QAE7B,IAAI,CAAC,YAAY,CAAC,GAAG,WAAW,CAAC;QACjC,IAAI,CAAC,iBAAiB,CAAC,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACnD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;CACF;AAED,MAAM,OAAO,gBAAgB;IAC3B,YAAY,CAAiB;IAC7B,iBAAiB,CAAS,CAAC,WAAW;IACtC,KAAK,CAAS;IACd,WAAW,CAAmB;IAC9B,WAAW,CAAS;IACpB,KAAK,CAAU;IACf,iBAAiB,CAAS;IAC1B,kBAAkB,CAAS;IAC3B,0BAA0B,CAAS;IACnC,oBAAoB,CAAS;IAC7B,gBAAgB,CAAS;IACzB,iBAAiB,CAAS;IAC1B,aAAa,CAAU;IACvB,SAAS,CAAS;IAClB,SAAS,CAAU;IAEnB,YACE,KAAa,EACb,WAAmB,EACnB,SAAiB,EACjB,SAAkB,EAClB,UAAiD,EACjD,aAAsB,EACtB,KAAc;QAEd,IAAI,CAAC,YAAY,CAAC,GAAG,cAAc,CAAC;QACpC,IAAI,CAAC,iBAAiB,CAAC,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACnD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,WAAW,GAAG,GAAG,CAAC;QACvB,IAAI,CAAC,iBAAiB,GAAG,UAAU,EAAE,gBAAgB,IAAI,CAAC,CAAC;QAC3D,IAAI,CAAC,kBAAkB,GAAG,UAAU,EAAE,oBAAoB,IAAI,CAAC,CAAC;QAChE,IAAI,CAAC,0BAA0B,GAAG,UAAU,EAAE,uBAAuB,IAAI,CAAC,CAAC;QAC3E,IAAI,CAAC,oBAAoB,GAAG,UAAU,EAAE,kBAAkB,IAAI,CAAC,CAAC;QAChE,IAAI,CAAC,gBAAgB,GAAG,UAAU,EAAE,uBAAuB,IAAI,CAAC,CAAC;QACjE,IAAI,CAAC,iBAAiB,GAAG,UAAU,EAAE,eAAe,IAAI,CAAC,CAAC;QAC1D,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;CACF;AAED,MAAM,OAAO,kBAAkB;IAC7B,YAAY,CAAmB;IAC/B,iBAAiB,CAAS,CAAC,WAAW;IACtC,SAAS,CAAS;IAElB,YAAY,SAAiB;QAC3B,IAAI,CAAC,YAAY,CAAC,GAAG,gBAAgB,CAAC;QACtC,IAAI,CAAC,iBAAiB,CAAC,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACnD,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;CACF;AAED,MAAM,CAAN,IAAY,QAGX;AAHD,WAAY,QAAQ;IAClB,iFAAqE,CAAA;IACrE,yEAA6D,CAAA;AAC/D,CAAC,EAHW,QAAQ,KAAR,QAAQ,QAGnB;AAED,MAAM,OAAO,iBAAiB;IAC5B,YAAY,CAAkB;IAC9B,iBAAiB,CAAS,CAAC,WAAW;IACtC,SAAS,CAAW;IAEpB,YAAY,SAAmB;QAC7B,IAAI,CAAC,YAAY,CAAC,GAAG,eAAe,CAAC;QACrC,IAAI,CAAC,iBAAiB,CAAC,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACnD,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;CACF"}

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

@@ -0,0 +1,67 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { EventEmitter } from 'events';
+import { EVENT_API_ERROR, EVENT_API_RESPONSE, EVENT_TOOL_CALL } from './constants.js';
+import { ApiErrorEvent, ApiResponseEvent, ToolCallEvent, ToolCallDecision } from './types.js';
+export type UiEvent = (ApiResponseEvent & {
+    'event.name': typeof EVENT_API_RESPONSE;
+}) | (ApiErrorEvent & {
+    'event.name': typeof EVENT_API_ERROR;
+}) | (ToolCallEvent & {
+    'event.name': typeof EVENT_TOOL_CALL;
+});
+export interface ToolCallStats {
+    count: number;
+    success: number;
+    fail: number;
+    durationMs: number;
+    decisions: {
+        [ToolCallDecision.ACCEPT]: number;
+        [ToolCallDecision.REJECT]: number;
+        [ToolCallDecision.MODIFY]: number;
+    };
+}
+export interface ModelMetrics {
+    api: {
+        totalRequests: number;
+        totalErrors: number;
+        totalLatencyMs: number;
+    };
+    tokens: {
+        prompt: number;
+        candidates: number;
+        total: number;
+        cached: number;
+        thoughts: number;
+        tool: number;
+    };
+}
+export interface SessionMetrics {
+    models: Record<string, ModelMetrics>;
+    tools: {
+        totalCalls: number;
+        totalSuccess: number;
+        totalFail: number;
+        totalDurationMs: number;
+        totalDecisions: {
+            [ToolCallDecision.ACCEPT]: number;
+            [ToolCallDecision.REJECT]: number;
+            [ToolCallDecision.MODIFY]: number;
+        };
+        byName: Record<string, ToolCallStats>;
+    };
+}
+export declare class UiTelemetryService extends EventEmitter {
+    #private;
+    addEvent(event: UiEvent): void;
+    getMetrics(): SessionMetrics;
+    getLastPromptTokenCount(): number;
+    private getOrCreateModelMetrics;
+    private processApiResponse;
+    private processApiError;
+    private processToolCall;
+}
+export declare const uiTelemetryService: UiTelemetryService;

package/dist/src/telemetry/uiTelemetry.js

@@ -0,0 +1,131 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { EventEmitter } from 'events';
+import { EVENT_API_ERROR, EVENT_API_RESPONSE, EVENT_TOOL_CALL, } from './constants.js';
+import { ToolCallDecision, } from './types.js';
+const createInitialModelMetrics = () => ({
+    api: {
+        totalRequests: 0,
+        totalErrors: 0,
+        totalLatencyMs: 0,
+    },
+    tokens: {
+        prompt: 0,
+        candidates: 0,
+        total: 0,
+        cached: 0,
+        thoughts: 0,
+        tool: 0,
+    },
+});
+const createInitialMetrics = () => ({
+    models: {},
+    tools: {
+        totalCalls: 0,
+        totalSuccess: 0,
+        totalFail: 0,
+        totalDurationMs: 0,
+        totalDecisions: {
+            [ToolCallDecision.ACCEPT]: 0,
+            [ToolCallDecision.REJECT]: 0,
+            [ToolCallDecision.MODIFY]: 0,
+        },
+        byName: {},
+    },
+});
+export class UiTelemetryService extends EventEmitter {
+    #metrics = createInitialMetrics();
+    #lastPromptTokenCount = 0;
+    addEvent(event) {
+        switch (event['event.name']) {
+            case EVENT_API_RESPONSE:
+                this.processApiResponse(event);
+                break;
+            case EVENT_API_ERROR:
+                this.processApiError(event);
+                break;
+            case EVENT_TOOL_CALL:
+                this.processToolCall(event);
+                break;
+            default:
+                // We should not emit update for any other event metric.
+                return;
+        }
+        this.emit('update', {
+            metrics: this.#metrics,
+            lastPromptTokenCount: this.#lastPromptTokenCount,
+        });
+    }
+    getMetrics() {
+        return this.#metrics;
+    }
+    getLastPromptTokenCount() {
+        return this.#lastPromptTokenCount;
+    }
+    getOrCreateModelMetrics(modelName) {
+        if (!this.#metrics.models[modelName]) {
+            this.#metrics.models[modelName] = createInitialModelMetrics();
+        }
+        return this.#metrics.models[modelName];
+    }
+    processApiResponse(event) {
+        const modelMetrics = this.getOrCreateModelMetrics(event.model);
+        modelMetrics.api.totalRequests++;
+        modelMetrics.api.totalLatencyMs += event.duration_ms;
+        modelMetrics.tokens.prompt += event.input_token_count;
+        modelMetrics.tokens.candidates += event.output_token_count;
+        modelMetrics.tokens.total += event.total_token_count;
+        modelMetrics.tokens.cached += event.cached_content_token_count;
+        modelMetrics.tokens.thoughts += event.thoughts_token_count;
+        modelMetrics.tokens.tool += event.tool_token_count;
+        this.#lastPromptTokenCount = event.input_token_count;
+    }
+    processApiError(event) {
+        const modelMetrics = this.getOrCreateModelMetrics(event.model);
+        modelMetrics.api.totalRequests++;
+        modelMetrics.api.totalErrors++;
+        modelMetrics.api.totalLatencyMs += event.duration_ms;
+    }
+    processToolCall(event) {
+        const { tools } = this.#metrics;
+        tools.totalCalls++;
+        tools.totalDurationMs += event.duration_ms;
+        if (event.success) {
+            tools.totalSuccess++;
+        }
+        else {
+            tools.totalFail++;
+        }
+        if (!tools.byName[event.function_name]) {
+            tools.byName[event.function_name] = {
+                count: 0,
+                success: 0,
+                fail: 0,
+                durationMs: 0,
+                decisions: {
+                    [ToolCallDecision.ACCEPT]: 0,
+                    [ToolCallDecision.REJECT]: 0,
+                    [ToolCallDecision.MODIFY]: 0,
+                },
+            };
+        }
+        const toolStats = tools.byName[event.function_name];
+        toolStats.count++;
+        toolStats.durationMs += event.duration_ms;
+        if (event.success) {
+            toolStats.success++;
+        }
+        else {
+            toolStats.fail++;
+        }
+        if (event.decision) {
+            tools.totalDecisions[event.decision]++;
+            toolStats.decisions[event.decision]++;
+        }
+    }
+}
+export const uiTelemetryService = new UiTelemetryService();
+//# sourceMappingURL=uiTelemetry.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"uiTelemetry.js","sourceRoot":"","sources":["../../../src/telemetry/uiTelemetry.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EACL,eAAe,EACf,kBAAkB,EAClB,eAAe,GAChB,MAAM,gBAAgB,CAAC;AAExB,OAAO,EAIL,gBAAgB,GACjB,MAAM,YAAY,CAAC;AAmDpB,MAAM,yBAAyB,GAAG,GAAiB,EAAE,CAAC,CAAC;IACrD,GAAG,EAAE;QACH,aAAa,EAAE,CAAC;QAChB,WAAW,EAAE,CAAC;QACd,cAAc,EAAE,CAAC;KAClB;IACD,MAAM,EAAE;QACN,MAAM,EAAE,CAAC;QACT,UAAU,EAAE,CAAC;QACb,KAAK,EAAE,CAAC;QACR,MAAM,EAAE,CAAC;QACT,QAAQ,EAAE,CAAC;QACX,IAAI,EAAE,CAAC;KACR;CACF,CAAC,CAAC;AAEH,MAAM,oBAAoB,GAAG,GAAmB,EAAE,CAAC,CAAC;IAClD,MAAM,EAAE,EAAE;IACV,KAAK,EAAE;QACL,UAAU,EAAE,CAAC;QACb,YAAY,EAAE,CAAC;QACf,SAAS,EAAE,CAAC;QACZ,eAAe,EAAE,CAAC;QAClB,cAAc,EAAE;YACd,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;YAC5B,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;YAC5B,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;SAC7B;QACD,MAAM,EAAE,EAAE;KACX;CACF,CAAC,CAAC;AAEH,MAAM,OAAO,kBAAmB,SAAQ,YAAY;IAClD,QAAQ,GAAmB,oBAAoB,EAAE,CAAC;IAClD,qBAAqB,GAAG,CAAC,CAAC;IAE1B,QAAQ,CAAC,KAAc;QACrB,QAAQ,KAAK,CAAC,YAAY,CAAC,EAAE,CAAC;YAC5B,KAAK,kBAAkB;gBACrB,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;gBAC/B,MAAM;YACR,KAAK,eAAe;gBAClB,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;gBAC5B,MAAM;YACR,KAAK,eAAe;gBAClB,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;gBAC5B,MAAM;YACR;gBACE,wDAAwD;gBACxD,OAAO;QACX,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE;YAClB,OAAO,EAAE,IAAI,CAAC,QAAQ;YACtB,oBAAoB,EAAE,IAAI,CAAC,qBAAqB;SACjD,CAAC,CAAC;IACL,CAAC;IAED,UAAU;QACR,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED,uBAAuB;QACrB,OAAO,IAAI,CAAC,qBAAqB,CAAC;IACpC,CAAC;IAEO,uBAAuB,CAAC,SAAiB;QAC/C,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;YACrC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,yBAAyB,EAAE,CAAC;QAChE,CAAC;QACD,OAAO,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IACzC,CAAC;IAEO,kBAAkB,CAAC,KAAuB;QAChD,MAAM,YAAY,GAAG,IAAI,CAAC,uBAAuB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAE/D,YAAY,CAAC,GAAG,CAAC,aAAa,EAAE,CAAC;QACjC,YAAY,CAAC,GAAG,CAAC,cAAc,IAAI,KAAK,CAAC,WAAW,CAAC;QAErD,YAAY,CAAC,MAAM,CAAC,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC;QACtD,YAAY,CAAC,MAAM,CAAC,UAAU,IAAI,KAAK,CAAC,kBAAkB,CAAC;QAC3D,YAAY,CAAC,MAAM,CAAC,KAAK,IAAI,KAAK,CAAC,iBAAiB,CAAC;QACrD,YAAY,CAAC,MAAM,CAAC,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC;QAC/D,YAAY,CAAC,MAAM,CAAC,QAAQ,IAAI,KAAK,CAAC,oBAAoB,CAAC;QAC3D,YAAY,CAAC,MAAM,CAAC,IAAI,IAAI,KAAK,CAAC,gBAAgB,CAAC;QAEnD,IAAI,CAAC,qBAAqB,GAAG,KAAK,CAAC,iBAAiB,CAAC;IACvD,CAAC;IAEO,eAAe,CAAC,KAAoB;QAC1C,MAAM,YAAY,GAAG,IAAI,CAAC,uBAAuB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC/D,YAAY,CAAC,GAAG,CAAC,aAAa,EAAE,CAAC;QACjC,YAAY,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC;QAC/B,YAAY,CAAC,GAAG,CAAC,cAAc,IAAI,KAAK,CAAC,WAAW,CAAC;IACvD,CAAC;IAEO,eAAe,CAAC,KAAoB;QAC1C,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC;QAChC,KAAK,CAAC,UAAU,EAAE,CAAC;QACnB,KAAK,CAAC,eAAe,IAAI,KAAK,CAAC,WAAW,CAAC;QAE3C,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAClB,KAAK,CAAC,YAAY,EAAE,CAAC;QACvB,CAAC;aAAM,CAAC;YACN,KAAK,CAAC,SAAS,EAAE,CAAC;QACpB,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,EAAE,CAAC;YACvC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,GAAG;gBAClC,KAAK,EAAE,CAAC;gBACR,OAAO,EAAE,CAAC;gBACV,IAAI,EAAE,CAAC;gBACP,UAAU,EAAE,CAAC;gBACb,SAAS,EAAE;oBACT,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;oBAC5B,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;oBAC5B,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;iBAC7B;aACF,CAAC;QACJ,CAAC;QAED,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;QACpD,SAAS,CAAC,KAAK,EAAE,CAAC;QAClB,SAAS,CAAC,UAAU,IAAI,KAAK,CAAC,WAAW,CAAC;QAC1C,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAClB,SAAS,CAAC,OAAO,EAAE,CAAC;QACtB,CAAC;aAAM,CAAC;YACN,SAAS,CAAC,IAAI,EAAE,CAAC;QACnB,CAAC;QAED,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;YACnB,KAAK,CAAC,cAAc,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC;YACvC,SAAS,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC;QACxC,CAAC;IACH,CAAC;CACF;AAED,MAAM,CAAC,MAAM,kBAAkB,GAAG,IAAI,kBAAkB,EAAE,CAAC"}

package/dist/src/telemetry/uiTelemetry.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};