@blackbox_ai/blackbox-cli-core 0.0.7 → 0.8.1

package/dist/src/core/encryptedGeminiClient.js

@@ -0,0 +1,177 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { EncryptedTurn } from './encryptedTurn.js';
+import { getCoreSystemPrompt } from './prompts.js';
+import { ApprovalMode } from '../config/config.js';
+import { getEnvironmentContext } from '../utils/environmentContext.js';
+/**
+ * EncryptedGeminiClient provides the same interface as GeminiClient
+ * but routes requests through the encrypted provider and emits the correct events
+ * for tool execution.
+ */
+export class EncryptedGeminiClient {
+    encryptedService;
+    config;
+    tools = [];
+    constructor(encryptedService, config) {
+        this.encryptedService = encryptedService;
+        this.config = config;
+    }
+    /**
+     * Set the available tools for this client
+     */
+    setTools(tools) {
+        this.tools = tools;
+    }
+    /**
+     * Send a message and get streaming events that match the regular GeminiClient interface
+     * This is the key method that bridges the encrypted provider with the event system
+     */
+    async *sendMessageStream(request, signal, prompt_id) {
+        try {
+            // Convert request to text
+            const messageText = this.convertRequestToText(request);
+            // Create tools system prompt if we have tools
+            const toolsSystemPrompt = EncryptedTurn.createToolsSystemPrompt(this.tools, 'deepseek');
+            const messageWithTools = toolsSystemPrompt ? `${toolsSystemPrompt}\n\n${messageText}` : messageText;
+            // Initialize the encrypted service if needed
+            if (!this.encryptedService.isReady()) {
+                // Use the same core system prompt as Gemini, combined with tools
+                const coreSystemPrompt = await this.getCoreSystemPrompt();
+                const fullSystemPrompt = toolsSystemPrompt
+                    ? `${coreSystemPrompt}\n\n${toolsSystemPrompt}`
+                    : coreSystemPrompt;
+                await this.encryptedService.initialize(fullSystemPrompt);
+            }
+            // Send message through encrypted service
+            const encryptedStream = await this.encryptedService.sendMessage(messageWithTools);
+            // Read the entire response
+            const reader = encryptedStream.getReader();
+            const decoder = new TextDecoder();
+            let fullResponse = '';
+            try {
+                while (true) {
+                    if (signal.aborted) {
+                        yield { type: 'user_cancelled' };
+                        return;
+                    }
+                    const { done, value } = await reader.read();
+                    if (done)
+                        break;
+                    const chunk = decoder.decode(value, { stream: true });
+                    fullResponse += chunk;
+                    // Yield content events for streaming text (without tool calls)
+                    if (!this.containsToolCalls(chunk)) {
+                        yield {
+                            type: 'content',
+                            value: chunk
+                        };
+                    }
+                }
+            }
+            finally {
+                reader.releaseLock();
+            }
+            // Process the complete response through EncryptedTurn to emit tool events
+            const encryptedTurn = new EncryptedTurn(this, prompt_id);
+            // Only process for tool calls if we have tools and the response contains tool calls
+            if (this.tools.length > 0 && this.containsToolCalls(fullResponse)) {
+                // Process the response and emit tool call events
+                for await (const event of encryptedTurn.processEncryptedResponse(fullResponse, true)) {
+                    // Skip content events since we already yielded them during streaming
+                    if (event.type !== 'content') {
+                        yield event;
+                    }
+                }
+            }
+            else {
+                // No tool calls, just emit finished event
+                yield {
+                    type: 'finished',
+                    value: 'STOP'
+                };
+            }
+        }
+        catch (error) {
+            console.error('❌ EncryptedGeminiClient.sendMessageStream failed:', error);
+            yield {
+                type: 'error',
+                value: {
+                    error: {
+                        message: error instanceof Error ? error.message : 'Unknown error in encrypted provider',
+                        status: 500
+                    }
+                }
+            };
+        }
+    }
+    /**
+     * Convert PartListUnion request to text string
+     */
+    convertRequestToText(request) {
+        if (typeof request === 'string') {
+            return request;
+        }
+        if (Array.isArray(request)) {
+            return request
+                .map(part => {
+                if (typeof part === 'string') {
+                    return part;
+                }
+                else if (part && typeof part === 'object' && 'text' in part) {
+                    return part.text;
+                }
+                else if (part && typeof part === 'object' && 'functionResponse' in part) {
+                    // Handle function response parts
+                    const funcResp = part.functionResponse;
+                    return `Function ${funcResp?.name} returned: ${JSON.stringify(funcResp?.response)}`;
+                }
+                return '';
+            })
+                .join('\n');
+        }
+        return String(request);
+    }
+    /**
+     * Check if text contains tool call markup
+     */
+    containsToolCalls(text) {
+        return text.includes('<tool_call>') ||
+            text.includes('<｜tool▁calls▁begin｜>') ||
+            text.includes('<｜tool▁call▁begin｜>');
+    }
+    /**
+     * Get available tools
+     */
+    getTools() {
+        return this.tools;
+    }
+    /**
+     * Check if client is ready
+     */
+    isReady() {
+        return this.encryptedService.isReady();
+    }
+    /**
+     * Get the core system prompt like Gemini does
+     */
+    async getCoreSystemPrompt() {
+        if (!this.config) {
+            return 'You are a helpful AI assistant.';
+        }
+        const userMemory = this.config.getUserMemory();
+        const coreSystemPrompt = getCoreSystemPrompt(userMemory, {}, this.config.getModel(), this.config.getApprovalMode() === ApprovalMode.YOLO);
+        // CRITICAL FIX: Include environment context like regular Gemini does
+        const envParts = await getEnvironmentContext(this.config);
+        const environmentContext = envParts.map(part => typeof part === 'string' ? part : part.text || '').join('\n');
+        // Combine core system prompt with environment context (like regular Gemini)
+        const systemPromptWithEnvironment = environmentContext.trim()
+            ? `${coreSystemPrompt}\n\n${environmentContext}`
+            : coreSystemPrompt;
+        return systemPromptWithEnvironment;
+    }
+}
+//# sourceMappingURL=encryptedGeminiClient.js.map

package/dist/src/core/encryptedGeminiClient.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"encryptedGeminiClient.js","sourceRoot":"","sources":["../../../src/core/encryptedGeminiClient.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAWH,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,mBAAmB,EAAE,MAAM,cAAc,CAAC;AAEnD,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,qBAAqB,EAAE,MAAM,gCAAgC,CAAC;AAcvE;;;;GAIG;AACH,MAAM,OAAO,qBAAqB;IAIb;IACA;IAJX,KAAK,GAA0B,EAAE,CAAC;IAE1C,YACmB,gBAAsC,EACtC,MAAe;QADf,qBAAgB,GAAhB,gBAAgB,CAAsB;QACtC,WAAM,GAAN,MAAM,CAAS;IAC/B,CAAC;IAEJ;;OAEG;IACH,QAAQ,CAAC,KAA4B;QACnC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,CAAC,iBAAiB,CACtB,OAAsB,EACtB,MAAmB,EACnB,SAAiB;QAEjB,IAAI,CAAC;YACH,0BAA0B;YAC1B,MAAM,WAAW,GAAG,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;YACvD,8CAA8C;YAC9C,MAAM,iBAAiB,GAAG,aAAa,CAAC,uBAAuB,CAAC,IAAI,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;YACxF,MAAM,gBAAgB,GAAG,iBAAiB,CAAC,CAAC,CAAC,GAAG,iBAAiB,OAAO,WAAW,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC;YACpG,6CAA6C;YAC7C,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,EAAE,CAAC;gBACrC,iEAAiE;gBACjE,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBAC1D,MAAM,gBAAgB,GAAG,iBAAiB;oBACxC,CAAC,CAAC,GAAG,gBAAgB,OAAO,iBAAiB,EAAE;oBAC/C,CAAC,CAAC,gBAAgB,CAAC;gBACrB,MAAM,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC;YAC3D,CAAC;YAED,yCAAyC;YACzC,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,WAAW,CAAC,gBAAgB,CAAC,CAAC;YAElF,2BAA2B;YAC3B,MAAM,MAAM,GAAG,eAAe,CAAC,SAAS,EAAE,CAAC;YAC3C,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;YAClC,IAAI,YAAY,GAAG,EAAE,CAAC;YAEtB,IAAI,CAAC;gBACH,OAAO,IAAI,EAAE,CAAC;oBACZ,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;wBACnB,MAAM,EAAE,IAAI,EAAE,gBAAiD,EAAE,CAAC;wBAClE,OAAO;oBACT,CAAC;oBAED,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;oBAC5C,IAAI,IAAI;wBAAE,MAAM;oBAEhB,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;oBACtD,YAAY,IAAI,KAAK,CAAC;oBAEtB,+DAA+D;oBAC/D,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,EAAE,CAAC;wBACnC,MAAM;4BACJ,IAAI,EAAE,SAAoC;4BAC1C,KAAK,EAAE,KAAK;yBACb,CAAC;oBACJ,CAAC;gBACH,CAAC;YACH,CAAC;oBAAS,CAAC;gBACT,MAAM,CAAC,WAAW,EAAE,CAAC;YACvB,CAAC;YAED,0EAA0E;YAC1E,MAAM,aAAa,GAAG,IAAI,aAAa,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;YAEzD,oFAAoF;YACpF,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,EAAE,CAAC;gBAClE,iDAAiD;gBACjD,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,aAAa,CAAC,wBAAwB,CAAC,YAAY,EAAE,IAAI,CAAC,EAAE,CAAC;oBACrF,qEAAqE;oBACrE,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;wBAC7B,MAAM,KAAK,CAAC;oBACd,CAAC;gBACH,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,0CAA0C;gBAC1C,MAAM;oBACJ,IAAI,EAAE,UAAsC;oBAC5C,KAAK,EAAE,MAAsB;iBAC9B,CAAC;YACJ,CAAC;QAEH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,mDAAmD,EAAE,KAAK,CAAC,CAAC;YAE1E,MAAM;gBACJ,IAAI,EAAE,OAAgC;gBACtC,KAAK,EAAE;oBACL,KAAK,EAAE;wBACL,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,qCAAqC;wBACvF,MAAM,EAAE,GAAG;qBACZ;iBACF;aACF,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,oBAAoB,CAAC,OAAsB;QACjD,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YAChC,OAAO,OAAO,CAAC;QACjB,CAAC;QAED,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAC3B,OAAO,OAAO;iBACX,GAAG,CAAC,IAAI,CAAC,EAAE;gBACV,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;oBAC7B,OAAO,IAAI,CAAC;gBACd,CAAC;qBAAM,IAAI,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,MAAM,IAAI,IAAI,EAAE,CAAC;oBAC9D,OAAO,IAAI,CAAC,IAAI,CAAC;gBACnB,CAAC;qBAAM,IAAI,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,kBAAkB,IAAI,IAAI,EAAE,CAAC;oBAC1E,iCAAiC;oBACjC,MAAM,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC;oBACvC,OAAO,YAAY,QAAQ,EAAE,IAAI,cAAc,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,QAAQ,CAAC,EAAE,CAAC;gBACtF,CAAC;gBACD,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC;iBACD,IAAI,CAAC,IAAI,CAAC,CAAC;QAChB,CAAC;QAED,OAAO,MAAM,CAAC,OAAO,CAAC,CAAC;IACzB,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,IAAY;QACpC,OAAO,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC;YAC5B,IAAI,CAAC,QAAQ,CAAC,sBAAsB,CAAC;YACrC,IAAI,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC;IAC9C,CAAC;IAED;;OAEG;IACH,QAAQ;QACN,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAED;;OAEG;IACH,OAAO;QACL,OAAO,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC;IACzC,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,mBAAmB;QAC/B,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,OAAO,iCAAiC,CAAC;QAC3C,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;QAC/C,MAAM,gBAAgB,GAAG,mBAAmB,CAC1C,UAAU,EACV,EAAE,EACF,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,EACtB,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,KAAK,YAAY,CAAC,IAAI,CACpD,CAAC;QAEF,qEAAqE;QACrE,MAAM,QAAQ,GAAG,MAAM,qBAAqB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1D,MAAM,kBAAkB,GAAG,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAC7C,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAClD,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACb,4EAA4E;QAC5E,MAAM,2BAA2B,GAAG,kBAAkB,CAAC,IAAI,EAAE;YAC3D,CAAC,CAAC,GAAG,gBAAgB,OAAO,kBAAkB,EAAE;YAChD,CAAC,CAAC,gBAAgB,CAAC;QAErB,OAAO,2BAA2B,CAAC;IACrC,CAAC;CACF"}

package/dist/src/core/encryptedGeminiClientBridge.d.ts

@@ -0,0 +1,107 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { PartListUnion, FunctionDeclaration, Content } from '@google/genai';
+import type { ServerGeminiStreamEvent } from './turn.js';
+import type { Config } from '../config/config.js';
+/**
+ * Interface for the encrypted chat service
+ */
+export interface EncryptedChatService {
+    sendMessage(message: string, messageHistory?: Array<{
+        role: 'system' | 'user' | 'assistant';
+        content: string;
+    }>): Promise<ReadableStream<Uint8Array>>;
+    sendMessageWithToolResults(message: string, toolResults: Array<{
+        toolName: string;
+        result: unknown;
+        callId?: string;
+    }>, messageHistory?: Array<{
+        role: 'system' | 'user' | 'assistant';
+        content: string;
+    }>): Promise<ReadableStream<Uint8Array>>;
+    initialize(systemPrompt?: string): Promise<void>;
+    isReady(): boolean;
+}
+/**
+ * EncryptedGeminiClientBridge provides a bridge between the encrypted provider
+ * and the existing GeminiClient interface, enabling tool execution through
+ * the event-based system that the UI layer expects.
+ */
+export declare class EncryptedGeminiClientBridge {
+    private readonly encryptedService;
+    private readonly config?;
+    private tools;
+    private history;
+    private pendingToolResults;
+    private messageTruncator;
+    constructor(encryptedService: EncryptedChatService, config?: Config | undefined);
+    /**
+     * Set the available tools for this client
+     */
+    setTools(tools: FunctionDeclaration[]): void;
+    /**
+     * Add content to history
+     */
+    addHistory(content: Content): void;
+    /**
+     * Get current history
+     */
+    getHistory(): Content[];
+    /**
+     * Send a message and get streaming events that match the regular GeminiClient interface
+     * This is the key method that bridges the encrypted provider with the event system
+     */
+    sendMessageStream(request: PartListUnion, signal: AbortSignal, prompt_id: string): AsyncGenerator<ServerGeminiStreamEvent>;
+    /**
+     * Convert PartListUnion request to text string
+     */
+    private convertRequestToText;
+    /**
+     * Convert Content[] history to message format expected by EncryptedChatService
+     * Now uses TOKEN-BASED truncation like the normal Blackbox provider
+     */
+    private convertHistoryToMessageFormat;
+    /**
+     * Check if text contains tool call markup
+     */
+    private containsToolCalls;
+    /**
+     * Get available tools
+     */
+    getTools(): FunctionDeclaration[];
+    /**
+     * Add tool execution result for processing
+     */
+    addToolResult(callId: string, toolName: string, result: unknown): void;
+    /**
+     * Send tool results back to encrypted service and continue conversation
+     */
+    sendToolResultsAndContinue(signal: AbortSignal, prompt_id: string): AsyncGenerator<ServerGeminiStreamEvent>;
+    /**
+     * Format tool results into a readable message
+     */
+    private formatToolResultsMessage;
+    /**
+     * Check if there are pending tool results
+     */
+    hasPendingToolResults(): boolean;
+    /**
+     * Clear pending tool results
+     */
+    clearPendingToolResults(): void;
+    /**
+     * Reset all chat history and state
+     */
+    resetHistory(): void;
+    /**
+     * Check if client is ready
+     */
+    isReady(): boolean;
+    /**
+     * Format think content with special styling to distinguish from assistant messages
+     */
+    private formatThinkContent;
+}