@blackbox_ai/blackbox-cli-core 0.0.9 → 0.8.1

package/dist/src/mcp/oauth-token-storage.js.map

@@ -1 +1 @@
-{"version":3,"file":"oauth-token-storage.js","sourceRoot":"","sources":["../../../src/mcp/oauth-token-storage.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,IAAI,EAAE,EAAE,MAAM,SAAS,CAAC;AACzC,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAC;AAC/C,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAGrD;;GAEG;AACH,MAAM,OAAO,oBAAoB;IAC/B;;;;OAIG;IACK,MAAM,CAAC,gBAAgB;QAC7B,OAAO,OAAO,CAAC,qBAAqB,EAAE,CAAC;IACzC,CAAC;IAED;;OAEG;IACK,MAAM,CAAC,KAAK,CAAC,eAAe;QAClC,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACxD,MAAM,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACjD,CAAC;IAED;;;;OAIG;IACH,MAAM,CAAC,KAAK,CAAC,UAAU;QACrB,MAAM,QAAQ,GAAG,IAAI,GAAG,EAA4B,CAAC;QAErD,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1C,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YACnD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAuB,CAAC;YAEtD,KAAK,MAAM,UAAU,IAAI,MAAM,EAAE,CAAC;gBAChC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;YAClD,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,qDAAqD;YACrD,IAAK,KAA+B,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACvD,OAAO,CAAC,KAAK,CACX,oCAAoC,eAAe,CAAC,KAAK,CAAC,EAAE,CAC7D,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;;;;;;;OAQG;IACH,MAAM,CAAC,KAAK,CAAC,SAAS,CACpB,UAAkB,EAClB,KAAiB,EACjB,QAAiB,EACjB,QAAiB,EACjB,YAAqB;QAErB,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAE7B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAEvC,MAAM,UAAU,GAAqB;YACnC,UAAU;YACV,KAAK;YACL,QAAQ;YACR,QAAQ;YACR,YAAY;YACZ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC;QAEF,MAAM,CAAC,GAAG,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;QAEnC,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;QAC/C,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAE1C,IAAI,CAAC;YACH,MAAM,EAAE,CAAC,SAAS,CAChB,SAAS,EACT,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,EACnC,EAAE,IAAI,EAAE,KAAK,EAAE,CAChB,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CACX,mCAAmC,eAAe,CAAC,KAAK,CAAC,EAAE,CAC5D,CAAC;YACF,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACH,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,UAAkB;QACtC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACvC,OAAO,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC;IACxC,CAAC;IAED;;;;OAIG;IACH,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,UAAkB;QACzC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAEvC,IAAI,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC;YAC9B,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;YAC/C,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAE1C,IAAI,CAAC;gBACH,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBAC5B,gCAAgC;oBAChC,MAAM,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;gBAC7B,CAAC;qBAAM,CAAC;oBACN,MAAM,EAAE,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE;wBACjE,IAAI,EAAE,KAAK;qBACZ,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CACX,qCAAqC,eAAe,CAAC,KAAK,CAAC,EAAE,CAC9D,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACH,MAAM,CAAC,cAAc,CAAC,KAAiB;QACrC,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC;YACrB,OAAO,KAAK,CAAC,CAAC,0BAA0B;QAC1C,CAAC;QAED,kDAAkD;QAClD,MAAM,QAAQ,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;QAC/B,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,IAAI,KAAK,CAAC,SAAS,CAAC;IAClD,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,cAAc;QACzB,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1C,MAAM,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAC7B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAK,KAA+B,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACvD,OAAO,CAAC,KAAK,CACX,qCAAqC,eAAe,CAAC,KAAK,CAAC,EAAE,CAC9D,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;CACF"}
+{"version":3,"file":"oauth-token-storage.js","sourceRoot":"","sources":["../../../src/mcp/oauth-token-storage.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,IAAI,EAAE,EAAE,MAAM,SAAS,CAAC;AACzC,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAC;AAC/C,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAGrD;;GAEG;AACH,MAAM,OAAO,oBAAoB;IAC/B,+DAA+D;IACvD,MAAM,CAAC,UAAU,GAAyC,IAAI,CAAC;IAC/D,MAAM,CAAC,gBAAgB,GAAG,KAAK,CAAC;IAExC;;;;OAIG;IACK,MAAM,CAAC,gBAAgB;QAC7B,OAAO,OAAO,CAAC,qBAAqB,EAAE,CAAC;IACzC,CAAC;IAED;;OAEG;IACK,MAAM,CAAC,KAAK,CAAC,eAAe;QAClC,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACxD,MAAM,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACjD,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,eAAe;QACpB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACvB,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;IAC/B,CAAC;IAED;;;;;OAKG;IACH,MAAM,CAAC,KAAK,CAAC,UAAU;QACrB,wDAAwD;QACxD,IAAI,IAAI,CAAC,UAAU,KAAK,IAAI,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACvD,OAAO,IAAI,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClC,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,GAAG,EAA4B,CAAC;QAErD,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1C,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YACnD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAuB,CAAC;YAEtD,KAAK,MAAM,UAAU,IAAI,MAAM,EAAE,CAAC;gBAChC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;YAClD,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,qDAAqD;YACrD,IAAK,KAA+B,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACvD,OAAO,CAAC,KAAK,CACX,oCAAoC,eAAe,CAAC,KAAK,CAAC,EAAE,CAC7D,CAAC;YACJ,CAAC;QACH,CAAC;QAED,0BAA0B;QAC1B,IAAI,CAAC,UAAU,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,CAAC;QACpC,IAAI,CAAC,gBAAgB,GAAG,KAAK,CAAC;QAE9B,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;;;;;;;OAQG;IACH,MAAM,CAAC,KAAK,CAAC,SAAS,CACpB,UAAkB,EAClB,KAAiB,EACjB,QAAiB,EACjB,QAAiB,EACjB,YAAqB;QAErB,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAE7B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAEvC,MAAM,UAAU,GAAqB;YACnC,UAAU;YACV,KAAK;YACL,QAAQ;YACR,QAAQ;YACR,YAAY;YACZ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC;QAEF,MAAM,CAAC,GAAG,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;QAEnC,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;QAC/C,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAE1C,IAAI,CAAC;YACH,MAAM,EAAE,CAAC,SAAS,CAChB,SAAS,EACT,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,EACnC,EAAE,IAAI,EAAE,KAAK,EAAE,CAChB,CAAC;YAEF,0CAA0C;YAC1C,IAAI,CAAC,eAAe,EAAE,CAAC;QACzB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CACX,mCAAmC,eAAe,CAAC,KAAK,CAAC,EAAE,CAC5D,CAAC;YACF,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACH,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,UAAkB;QACtC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACvC,OAAO,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC;IACxC,CAAC;IAED;;;;OAIG;IACH,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,UAAkB;QACzC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAEvC,IAAI,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC;YAC9B,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;YAC/C,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAE1C,IAAI,CAAC;gBACH,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBAC5B,gCAAgC;oBAChC,MAAM,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;gBAC7B,CAAC;qBAAM,CAAC;oBACN,MAAM,EAAE,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE;wBACjE,IAAI,EAAE,KAAK;qBACZ,CAAC,CAAC;gBACL,CAAC;gBAED,4CAA4C;gBAC5C,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CACX,qCAAqC,eAAe,CAAC,KAAK,CAAC,EAAE,CAC9D,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACH,MAAM,CAAC,cAAc,CAAC,KAAiB;QACrC,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC;YACrB,OAAO,KAAK,CAAC,CAAC,0BAA0B;QAC1C,CAAC;QAED,kDAAkD;QAClD,MAAM,QAAQ,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;QAC/B,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,IAAI,KAAK,CAAC,SAAS,CAAC;IAClD,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,cAAc;QACzB,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1C,MAAM,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAE3B,kCAAkC;YAClC,IAAI,CAAC,eAAe,EAAE,CAAC;QACzB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAK,KAA+B,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACvD,OAAO,CAAC,KAAK,CACX,qCAAqC,eAAe,CAAC,KAAK,CAAC,EAAE,CAC9D,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC"}

package/dist/src/services/EncryptedChatService.d.ts

@@ -0,0 +1,80 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { FileData, SecureSession } from '../encrypt/types.js';
+/**
+ * EncryptedChatService handles encrypted chat sessions and message streaming
+ * This service integrates the encryption system with the CLI chat interface
+ */
+export declare class EncryptedChatService {
+    private chatId;
+    private session?;
+    private ready;
+    constructor(chatId?: string);
+    /**
+     * Generate a unique chat ID
+     */
+    private generateChatId;
+    /**
+     * Get the current chat ID
+     */
+    getChatId(): string;
+    /**
+     * Check if the service is ready for encrypted communication
+     */
+    isReady(): boolean;
+    /**
+     * Initialize the encrypted session
+     */
+    initialize(systemPrompt?: string): Promise<void>;
+    /**
+     * Send an encrypted message and get streaming response
+     */
+    sendMessage(message: string, messageHistory?: Array<{
+        role: 'system' | 'user' | 'assistant';
+        content: string;
+    }>, fileData?: FileData): Promise<ReadableStream<Uint8Array>>;
+    /**
+     * Send an encrypted message with tool results and get streaming response
+     * This method handles tool execution results in the encrypted channel
+     */
+    sendMessageWithToolResults(message: string, toolResults: Array<{
+        toolName: string;
+        result: unknown;
+        callId?: string;
+    }>, messageHistory?: Array<{
+        role: 'system' | 'user' | 'assistant';
+        content: string;
+    }>, fileData?: FileData): Promise<ReadableStream<Uint8Array>>;
+    /**
+     * Format tool results for inclusion in encrypted messages
+     */
+    private formatToolResults;
+    /**
+     * Get the current session
+     */
+    getSession(): SecureSession | undefined;
+    /**
+     * Static method to check if a model requires encryption
+     */
+    static isEncryptedModel(modelName: string): boolean;
+    /**
+     * Create a fallback stream when encryption fails
+     */
+    static createFallbackStream(message: string, error: Error): ReadableStream<Uint8Array>;
+    /**
+     * Reset the service (clear session)
+     */
+    reset(): void;
+    /**
+     * Get session statistics
+     */
+    getSessionStats(): {
+        chatId: string;
+        initialized: boolean;
+        messageCount: number;
+        sessionAge: number;
+    };
+}

package/dist/src/services/EncryptedChatService.js

@@ -0,0 +1,202 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { isEncryptedModel } from '../encrypt/config.js';
+import { initializeSecureSession, getHydratedSessionByChatId } from '../encrypt/sessions.js';
+import { sendMessageStream } from '../encrypt/client.js';
+/**
+ * EncryptedChatService handles encrypted chat sessions and message streaming
+ * This service integrates the encryption system with the CLI chat interface
+ */
+export class EncryptedChatService {
+    chatId;
+    session;
+    ready = false;
+    constructor(chatId) {
+        this.chatId = chatId || this.generateChatId();
+    }
+    /**
+     * Generate a unique chat ID
+     */
+    generateChatId() {
+        return `chat_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
+    }
+    /**
+     * Get the current chat ID
+     */
+    getChatId() {
+        return this.chatId;
+    }
+    /**
+     * Check if the service is ready for encrypted communication
+     */
+    isReady() {
+        return this.ready && !!this.session?.initialized;
+    }
+    /**
+     * Initialize the encrypted session
+     */
+    async initialize(systemPrompt) {
+        try {
+            // Try to get existing session first
+            this.session = await getHydratedSessionByChatId(this.chatId);
+            if (!this.session || !this.session.initialized) {
+                // Create new session if none exists or not initialized
+                this.session = await initializeSecureSession(this.chatId, systemPrompt);
+            }
+            this.ready = true;
+        }
+        catch (error) {
+            console.error('❌ Failed to initialize encrypted session:', error);
+            this.ready = false;
+            throw error;
+        }
+    }
+    /**
+     * Send an encrypted message and get streaming response
+     */
+    async sendMessage(message, messageHistory, fileData) {
+        if (!this.isReady()) {
+            throw new Error('EncryptedChatService not initialized. Call initialize() first.');
+        }
+        // Validate message is not empty
+        if (!message || message.trim().length === 0) {
+            throw new Error('Message cannot be empty');
+        }
+        // Ensure we have a valid message history array
+        const validMessageHistory = messageHistory || [];
+        try {
+            return await sendMessageStream(this.chatId, message, validMessageHistory, fileData);
+        }
+        catch (error) {
+            console.error('❌ Failed to send encrypted message:', error);
+            // Throw the actual error instead of creating a fallback stream
+            // This allows the model to learn from real errors
+            if (error instanceof Error) {
+                const enhancedError = new Error(`Failed: ${error.message}`);
+                enhancedError.stack = error.stack;
+                enhancedError.cause = error;
+                throw enhancedError;
+            }
+            else {
+                throw new Error(`Failed with unknown error: ${String(error)}`);
+            }
+        }
+    }
+    /**
+     * Send an encrypted message with tool results and get streaming response
+     * This method handles tool execution results in the encrypted channel
+     */
+    async sendMessageWithToolResults(message, toolResults, messageHistory, fileData) {
+        if (!this.isReady()) {
+            throw new Error('EncryptedChatService not initialized. Call initialize() first.');
+        }
+        // Format tool results for the encrypted channel
+        const toolResultsText = this.formatToolResults(toolResults);
+        // Combine the original message with tool results
+        const enhancedMessage = `${message}\n\n**Tool Results:**\n${toolResultsText}`;
+        // Ensure we have a valid message history array
+        const validMessageHistory = messageHistory || [];
+        try {
+            return await sendMessageStream(this.chatId, enhancedMessage, validMessageHistory, fileData);
+        }
+        catch (error) {
+            console.error('❌ Failed to send encrypted message with tool results:', error);
+            // Throw the actual error instead of creating a fallback stream
+            // This allows the model to learn from real errors
+            if (error instanceof Error) {
+                const enhancedError = new Error(`Tool results failed: ${error.message}`);
+                enhancedError.stack = error.stack;
+                enhancedError.cause = error;
+                throw enhancedError;
+            }
+            else {
+                throw new Error(`Tool results failed with unknown error: ${String(error)}`);
+            }
+        }
+    }
+    /**
+     * Format tool results for inclusion in encrypted messages
+     */
+    formatToolResults(toolResults) {
+        return toolResults.map(({ toolName, result, callId }) => {
+            const resultText = typeof result === 'string' ? result : JSON.stringify(result, null, 2);
+            const callIdText = callId ? ` (Call ID: ${callId})` : '';
+            return `**${toolName}${callIdText}:**
+\`\`\`
+${resultText}
+\`\`\``;
+        }).join('\n\n');
+    }
+    /**
+     * Get the current session
+     */
+    getSession() {
+        return this.session;
+    }
+    /**
+     * Static method to check if a model requires encryption
+     */
+    static isEncryptedModel(modelName) {
+        return isEncryptedModel(modelName);
+    }
+    /**
+     * Create a fallback stream when encryption fails
+     */
+    static createFallbackStream(message, error) {
+        const encoder = new TextEncoder();
+        return new ReadableStream({
+            async start(controller) {
+                const fallbackResponse = `🔒 **ENCRYPTED RESPONSE (Fallback Mode)**
+
+Your message: "${message}"
+
+The encryption service encountered an error, but your message has been processed securely using fallback encryption protocols.
+
+**Error Details:**
+${error.message}
+
+🛡️ **Fallback Security Features:**
+- ✅ Local encryption processing
+- ✅ Secure session management  
+- ✅ Error recovery protocols
+
+**Note:** This is a fallback response. The full encryption service will be restored automatically.
+
+To troubleshoot:
+1. Check your network connection
+2. Verify the encryption service is available
+3. Try again in a few moments
+
+If the issue persists, please contact support.`;
+                // Stream the response character by character for a realistic effect
+                for (let i = 0; i < fallbackResponse.length; i++) {
+                    await new Promise((resolve) => setTimeout(resolve, 10));
+                    controller.enqueue(encoder.encode(fallbackResponse[i]));
+                }
+                controller.close();
+            },
+        });
+    }
+    /**
+     * Reset the service (clear session)
+     */
+    reset() {
+        this.session = undefined;
+        this.ready = false;
+    }
+    /**
+     * Get session statistics
+     */
+    getSessionStats() {
+        return {
+            chatId: this.chatId,
+            initialized: this.isReady(),
+            messageCount: this.session?.history?.length || 0,
+            sessionAge: this.session?.createdAt ? Date.now() - this.session.createdAt : 0
+        };
+    }
+}
+//# sourceMappingURL=EncryptedChatService.js.map

package/dist/src/services/EncryptedChatService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"EncryptedChatService.js","sourceRoot":"","sources":["../../../src/services/EncryptedChatService.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,EAAE,uBAAuB,EAAE,0BAA0B,EAAE,MAAM,wBAAwB,CAAC;AAC7F,OAAO,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAGzD;;;GAGG;AACH,MAAM,OAAO,oBAAoB;IACvB,MAAM,CAAS;IACf,OAAO,CAAiB;IACxB,KAAK,GAAY,KAAK,CAAC;IAE/B,YAAY,MAAe;QACzB,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;IAChD,CAAC;IAED;;OAEG;IACK,cAAc;QACpB,OAAO,QAAQ,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;IACzE,CAAC;IAED;;OAEG;IACH,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED;;OAEG;IACH,OAAO;QACL,OAAO,IAAI,CAAC,KAAK,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC;IACnD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,YAAqB;QACpC,IAAI,CAAC;YACH,oCAAoC;YACpC,IAAI,CAAC,OAAO,GAAG,MAAM,0BAA0B,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAE7D,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;gBAC/C,uDAAuD;gBACvD,IAAI,CAAC,OAAO,GAAG,MAAM,uBAAuB,CAAC,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAC1E,CAAC;YAED,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QACpB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,2CAA2C,EAAE,KAAK,CAAC,CAAC;YAClE,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;YACnB,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW,CACf,OAAe,EACf,cAAkF,EAClF,QAAmB;QAEnB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,gEAAgE,CAAC,CAAC;QACpF,CAAC;QAED,gCAAgC;QAChC,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5C,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QAC7C,CAAC;QAED,+CAA+C;QAC/C,MAAM,mBAAmB,GAAG,cAAc,IAAI,EAAE,CAAC;QACjD,IAAI,CAAC;YACH,OAAO,MAAM,iBAAiB,CAC5B,IAAI,CAAC,MAAM,EACX,OAAO,EACP,mBAAmB,EACnB,QAAQ,CACT,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,qCAAqC,EAAE,KAAK,CAAC,CAAC;YAE5D,+DAA+D;YAC/D,kDAAkD;YAClD,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC3B,MAAM,aAAa,GAAG,IAAI,KAAK,CAC7B,WAAW,KAAK,CAAC,OAAO,EAAE,CAC3B,CAAC;gBACF,aAAa,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;gBAClC,aAAa,CAAC,KAAK,GAAG,KAAK,CAAC;gBAC5B,MAAM,aAAa,CAAC;YACtB,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,KAAK,CACb,8BAA8B,MAAM,CAAC,KAAK,CAAC,EAAE,CAC9C,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,0BAA0B,CAC9B,OAAe,EACf,WAA0E,EAC1E,cAAkF,EAClF,QAAmB;QAEnB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,gEAAgE,CAAC,CAAC;QACpF,CAAC;QAED,gDAAgD;QAChD,MAAM,eAAe,GAAG,IAAI,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;QAE5D,iDAAiD;QACjD,MAAM,eAAe,GAAG,GAAG,OAAO,0BAA0B,eAAe,EAAE,CAAC;QAE9E,+CAA+C;QAC/C,MAAM,mBAAmB,GAAG,cAAc,IAAI,EAAE,CAAC;QAEjD,IAAI,CAAC;YACH,OAAO,MAAM,iBAAiB,CAC5B,IAAI,CAAC,MAAM,EACX,eAAe,EACf,mBAAmB,EACnB,QAAQ,CACT,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,uDAAuD,EAAE,KAAK,CAAC,CAAC;YAE9E,+DAA+D;YAC/D,kDAAkD;YAClD,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC3B,MAAM,aAAa,GAAG,IAAI,KAAK,CAC7B,wBAAwB,KAAK,CAAC,OAAO,EAAE,CACxC,CAAC;gBACF,aAAa,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;gBAClC,aAAa,CAAC,KAAK,GAAG,KAAK,CAAC;gBAC5B,MAAM,aAAa,CAAC;YACtB,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,KAAK,CACb,2CAA2C,MAAM,CAAC,KAAK,CAAC,EAAE,CAC3D,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,WAA0E;QAClG,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE,EAAE;YACtD,MAAM,UAAU,GAAG,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YACzF,MAAM,UAAU,GAAG,MAAM,CAAC,CAAC,CAAC,cAAc,MAAM,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;YAEzD,OAAO,KAAK,QAAQ,GAAG,UAAU;;EAErC,UAAU;OACL,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAClB,CAAC;IAED;;OAEG;IACH,UAAU;QACR,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,gBAAgB,CAAC,SAAiB;QACvC,OAAO,gBAAgB,CAAC,SAAS,CAAC,CAAC;IACrC,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,oBAAoB,CAAC,OAAe,EAAE,KAAY;QACvD,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;QAElC,OAAO,IAAI,cAAc,CAAa;YACpC,KAAK,CAAC,KAAK,CAAC,UAAU;gBACpB,MAAM,gBAAgB,GAAG;;iBAEhB,OAAO;;;;;EAKtB,KAAK,CAAC,OAAO;;;;;;;;;;;;;;+CAcgC,CAAC;gBAExC,oEAAoE;gBACpE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,gBAAgB,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;oBACjD,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;oBACxD,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC1D,CAAC;gBAED,UAAU,CAAC,KAAK,EAAE,CAAC;YACrB,CAAC;SACF,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,OAAO,GAAG,SAAS,CAAC;QACzB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED;;OAEG;IACH,eAAe;QAMb,OAAO;YACL,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,WAAW,EAAE,IAAI,CAAC,OAAO,EAAE;YAC3B,YAAY,EAAE,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC;YAChD,UAAU,EAAE,IAAI,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;SAC9E,CAAC;IACJ,CAAC;CACF"}

package/dist/src/services/StatsHistoryService.d.ts

@@ -0,0 +1,131 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { SessionMetrics } from '../telemetry/uiTelemetry.js';
+export interface SessionHistoryEntry {
+    sessionId: string;
+    startTime: string;
+    endTime: string;
+    duration: number;
+    metrics: SessionMetrics;
+    promptCount: number;
+    cost?: number;
+}
+export interface AggregatedStats {
+    totalSessions: number;
+    totalDuration: number;
+    totalCost: number;
+    allTimeMetrics: SessionMetrics;
+}
+export interface StatsHistoryData {
+    sessions: SessionHistoryEntry[];
+    aggregated: AggregatedStats;
+    version: string;
+}
+export interface TrendPeriod {
+    period: 'daily' | 'weekly' | 'monthly';
+    startDate: string;
+    endDate: string;
+    sessionCount: number;
+    avgDuration: number;
+    avgMetrics: SessionMetrics;
+}
+export interface SessionFilter {
+    startDate?: Date;
+    endDate?: Date;
+    limit?: number;
+    offset?: number;
+}
+/**
+ * Service for managing persistent storage of session statistics
+ */
+export declare class StatsHistoryService {
+    private static instance;
+    private readonly statsFilePath;
+    private data;
+    private savePromise;
+    private constructor();
+    /**
+     * Get singleton instance
+     */
+    static getInstance(): StatsHistoryService;
+    /**
+     * Initialize the service by loading existing data
+     */
+    initialize(): Promise<void>;
+    /**
+     * Load data from disk
+     */
+    private loadData;
+    /**
+     * Create empty data structure
+     */
+    private createEmptyData;
+    /**
+     * Create empty metrics structure
+     */
+    private createEmptyMetrics;
+    /**
+     * Save current session to history
+     */
+    saveSession(sessionData: {
+        sessionId: string;
+        startTime: Date;
+        endTime: Date;
+        metrics: SessionMetrics;
+        promptCount: number;
+        cost?: number;
+    }): Promise<void>;
+    /**
+     * Update aggregated statistics
+     */
+    private updateAggregatedStats;
+    /**
+     * Save data to disk with debouncing
+     */
+    private saveData;
+    /**
+     * Perform the actual save operation
+     */
+    private performSave;
+    /**
+     * Get sessions with optional filtering
+     */
+    getSessions(filter?: SessionFilter): Promise<SessionHistoryEntry[]>;
+    /**
+     * Get aggregated all-time statistics
+     */
+    getAggregatedStats(): Promise<AggregatedStats>;
+    /**
+     * Calculate trends over a specified period
+     */
+    getTrends(period: 'daily' | 'weekly' | 'monthly'): Promise<TrendPeriod[]>;
+    /**
+     * Calculate average metrics across sessions
+     */
+    private calculateAverageMetrics;
+    /**
+     * Get comparison between current session and historical average
+     */
+    getCurrentVsHistorical(currentMetrics: SessionMetrics): Promise<{
+        current: SessionMetrics;
+        historical: SessionMetrics;
+        percentDiff: {
+            apiCalls: number;
+            tokens: number;
+            toolCalls: number;
+            linesChanged: number;
+        };
+    }>;
+    /**
+     * Export all data as JSON
+     */
+    exportData(): Promise<StatsHistoryData>;
+    /**
+     * Clear all historical data (for testing or user request)
+     */
+    clearHistory(): Promise<void>;
+}
+export declare const statsHistoryService: StatsHistoryService;