snow-ai 0.3.6 → 0.3.8

package/dist/utils/contextCompressor.js

@@ -1,58 +1,29 @@
-import { getOpenAiConfig, getCustomHeaders, getCustomSystemPrompt } from './apiConfig.js';
-import { SYSTEM_PROMPT } from '../api/systemPrompt.js';
+import { getOpenAiConfig, getCustomSystemPrompt } from './apiConfig.js';
+import { getSystemPrompt } from '../api/systemPrompt.js';
+import { createStreamingChatCompletion } from '../api/chat.js';
+import { createStreamingResponse } from '../api/responses.js';
+import { createStreamingGeminiCompletion } from '../api/gemini.js';
+import { createStreamingAnthropicCompletion } from '../api/anthropic.js';
 /**
  * Compression request prompt - asks AI to summarize conversation with focus on task continuity
  */
 const COMPRESSION_PROMPT = 'Please provide a concise summary of our conversation so far. Focus on: 1) The current task or goal we are working on, 2) Key decisions and approaches we have agreed upon, 3) Important context needed to continue, 4) Any pending or unfinished work. Keep it brief but ensure I can seamlessly continue assisting with the task.';
 /**
- * Parse Server-Sent Events (SSE) stream
+ * Prepare messages for compression by adding system prompt and compression request
  */
-async function* parseSSEStream(reader) {
-    const decoder = new TextDecoder();
-    let buffer = '';
-    while (true) {
-        const { done, value } = await reader.read();
-        if (done)
-            break;
-        buffer += decoder.decode(value, { stream: true });
-        const lines = buffer.split('\n');
-        buffer = lines.pop() || '';
-        for (const line of lines) {
-            const trimmed = line.trim();
-            if (!trimmed || trimmed.startsWith(':'))
-                continue;
-            if (trimmed === 'data: [DONE]') {
-                return;
-            }
-            if (trimmed.startsWith('data: ')) {
-                const data = trimmed.slice(6);
-                try {
-                    yield JSON.parse(data);
-                }
-                catch (e) {
-                    console.error('Failed to parse SSE data:', data);
-                }
-            }
-        }
-    }
-}
-/**
- * Compress context using OpenAI Chat Completions API
- */
-async function compressWithChatCompletions(baseUrl, apiKey, modelName, conversationMessages, systemPrompt) {
-    const customHeaders = getCustomHeaders();
-    // Build messages with system prompt support
+function prepareMessagesForCompression(conversationMessages, customSystemPrompt) {
     const messages = [];
-    if (systemPrompt) {
+    // Add system prompt (handled by API modules)
+    if (customSystemPrompt) {
         // If custom system prompt exists: custom as system, default as first user message
-        messages.push({ role: 'system', content: systemPrompt });
-        messages.push({ role: 'user', content: SYSTEM_PROMPT });
+        messages.push({ role: 'system', content: customSystemPrompt });
+        messages.push({ role: 'user', content: getSystemPrompt() });
     }
     else {
         // No custom system prompt: default as system
-        messages.push({ role: 'system', content: SYSTEM_PROMPT });
+        messages.push({ role: 'system', content: getSystemPrompt() });
     }
-    // Add all conversation history (exclude system messages)
+    // Add all conversation history (exclude system and tool messages)
     for (const msg of conversationMessages) {
         if (msg.role !== 'system' && msg.role !== 'tool') {
             messages.push({
@@ -66,42 +37,31 @@ async function compressWithChatCompletions(baseUrl, apiKey, modelName, conversat
         role: 'user',
         content: COMPRESSION_PROMPT,
     });
-    // Build request payload (no tools for compression)
-    const requestPayload = {
-        model: modelName,
-        messages,
-        stream: true,
-        stream_options: { include_usage: true },
-    };
-    const response = await fetch(`${baseUrl}/chat/completions`, {
-        method: 'POST',
-        headers: {
-            'Content-Type': 'application/json',
-            'Authorization': `Bearer ${apiKey}`,
-            ...customHeaders
-        },
-        body: JSON.stringify(requestPayload)
-    });
-    if (!response.ok) {
-        const errorText = await response.text();
-        throw new Error(`OpenAI API error: ${response.status} ${response.statusText} - ${errorText}`);
-    }
-    if (!response.body) {
-        throw new Error('No response body from OpenAI API');
-    }
+    return messages;
+}
+/**
+ * Compress context using OpenAI Chat Completions API (reuses chat.ts)
+ */
+async function compressWithChatCompletions(modelName, conversationMessages, customSystemPrompt) {
+    const messages = prepareMessagesForCompression(conversationMessages, customSystemPrompt);
     let summary = '';
     let usage = {
         prompt_tokens: 0,
         completion_tokens: 0,
         total_tokens: 0,
     };
-    for await (const chunk of parseSSEStream(response.body.getReader())) {
-        const delta = chunk.choices[0]?.delta;
-        if (delta?.content) {
-            summary += delta.content;
+    // Use the existing streaming API from chat.ts (includes proxy support)
+    for await (const chunk of createStreamingChatCompletion({
+        model: modelName,
+        messages,
+        stream: true,
+    })) {
+        // Collect content
+        if (chunk.type === 'content' && chunk.content) {
+            summary += chunk.content;
         }
-        // Collect usage info (usually in the last chunk)
-        if (chunk.usage) {
+        // Collect usage info
+        if (chunk.type === 'usage' && chunk.usage) {
             usage = {
                 prompt_tokens: chunk.usage.prompt_tokens || 0,
                 completion_tokens: chunk.usage.completion_tokens || 0,
@@ -112,334 +72,111 @@ async function compressWithChatCompletions(baseUrl, apiKey, modelName, conversat
     if (!summary) {
         throw new Error('Failed to generate summary from compact model');
     }
-    return {
-        summary,
-        usage,
-    };
+    return { summary, usage };
 }
 /**
- * Compress context using OpenAI Responses API
+ * Compress context using OpenAI Responses API (reuses responses.ts)
  */
-async function compressWithResponses(baseUrl, apiKey, modelName, conversationMessages, systemPrompt) {
-    const customHeaders = getCustomHeaders();
-    // Build instructions
-    const instructions = systemPrompt || SYSTEM_PROMPT;
-    // Build input array with conversation history
-    const input = [];
-    // If custom system prompt exists, add default as first user message
-    if (systemPrompt) {
-        input.push({
-            type: 'message',
-            role: 'user',
-            content: [{ type: 'input_text', text: SYSTEM_PROMPT }],
-        });
-    }
-    // Add all conversation history (exclude system messages)
-    for (const msg of conversationMessages) {
-        if (msg.role !== 'system' && msg.role !== 'tool') {
-            input.push({
-                type: 'message',
-                role: msg.role,
-                content: [{
-                        type: msg.role === 'user' ? 'input_text' : 'output_text',
-                        text: msg.content,
-                    }],
-            });
-        }
-    }
-    // Add compression request as final user message
-    input.push({
-        type: 'message',
-        role: 'user',
-        content: [{
-                type: 'input_text',
-                text: COMPRESSION_PROMPT,
-            }],
-    });
-    // Build request payload (no tools for compression)
-    const requestPayload = {
-        model: modelName,
-        instructions,
-        input,
-        stream: true,
-    };
-    const response = await fetch(`${baseUrl}/responses`, {
-        method: 'POST',
-        headers: {
-            'Content-Type': 'application/json',
-            'Authorization': `Bearer ${apiKey}`,
-            ...customHeaders
-        },
-        body: JSON.stringify(requestPayload)
-    });
-    if (!response.ok) {
-        const errorText = await response.text();
-        throw new Error(`OpenAI Responses API error: ${response.status} ${response.statusText} - ${errorText}`);
-    }
-    if (!response.body) {
-        throw new Error('No response body from OpenAI Responses API');
-    }
+async function compressWithResponses(modelName, conversationMessages, customSystemPrompt) {
+    const messages = prepareMessagesForCompression(conversationMessages, customSystemPrompt);
     let summary = '';
     let usage = {
         prompt_tokens: 0,
         completion_tokens: 0,
         total_tokens: 0,
     };
-    for await (const chunk of parseSSEStream(response.body.getReader())) {
-        const eventType = chunk.type;
-        // Handle text content delta
-        if (eventType === 'response.output_text.delta') {
-            const delta = chunk.delta;
-            if (delta) {
-                summary += delta;
-            }
+    // Use the existing streaming API from responses.ts (includes proxy support)
+    for await (const chunk of createStreamingResponse({
+        model: modelName,
+        messages,
+        stream: true,
+    })) {
+        // Collect content
+        if (chunk.type === 'content' && chunk.content) {
+            summary += chunk.content;
         }
-        // Handle usage info
-        if (eventType === 'response.completed') {
-            const response = chunk.response;
-            if (response?.usage) {
-                usage = {
-                    prompt_tokens: response.usage.input_tokens || 0,
-                    completion_tokens: response.usage.output_tokens || 0,
-                    total_tokens: (response.usage.input_tokens || 0) + (response.usage.output_tokens || 0),
-                };
-            }
+        // Collect usage info
+        if (chunk.type === 'usage' && chunk.usage) {
+            usage = {
+                prompt_tokens: chunk.usage.prompt_tokens || 0,
+                completion_tokens: chunk.usage.completion_tokens || 0,
+                total_tokens: chunk.usage.total_tokens || 0,
+            };
         }
     }
     if (!summary) {
         throw new Error('Failed to generate summary from compact model (Responses API)');
     }
-    return {
-        summary,
-        usage,
-    };
+    return { summary, usage };
 }
 /**
- * Compress context using Gemini API
+ * Compress context using Gemini API (reuses gemini.ts)
  */
-async function compressWithGemini(baseUrl, apiKey, modelName, conversationMessages, systemPrompt) {
-    const customHeaders = getCustomHeaders();
-    // Build system instruction
-    const systemInstruction = systemPrompt || SYSTEM_PROMPT;
-    // Build contents array with conversation history
-    const contents = [];
-    // If custom system prompt exists, add default as first user message
-    if (systemPrompt) {
-        contents.push({
-            role: 'user',
-            parts: [{ text: SYSTEM_PROMPT }],
-        });
-    }
-    // Add all conversation history (exclude system messages)
-    for (const msg of conversationMessages) {
-        if (msg.role !== 'system' && msg.role !== 'tool') {
-            contents.push({
-                role: msg.role === 'assistant' ? 'model' : 'user',
-                parts: [{ text: msg.content }],
-            });
-        }
-    }
-    // Add compression request as final user message
-    contents.push({
-        role: 'user',
-        parts: [{
-                text: COMPRESSION_PROMPT,
-            }],
-    });
-    const requestBody = {
-        contents,
-        systemInstruction: systemInstruction ? { parts: [{ text: systemInstruction }] } : undefined,
-    };
-    // Extract model name
-    const effectiveBaseUrl = baseUrl && baseUrl !== 'https://api.openai.com/v1'
-        ? baseUrl
-        : 'https://generativelanguage.googleapis.com/v1beta';
-    const model = modelName.startsWith('models/') ? modelName : `models/${modelName}`;
-    const url = `${effectiveBaseUrl}/${model}:streamGenerateContent?key=${apiKey}&alt=sse`;
-    const response = await fetch(url, {
-        method: 'POST',
-        headers: {
-            'Content-Type': 'application/json',
-            'Authorization': `Bearer ${apiKey}`,
-            ...customHeaders
-        },
-        body: JSON.stringify(requestBody)
-    });
-    if (!response.ok) {
-        const errorText = await response.text();
-        throw new Error(`Gemini API error: ${response.status} ${response.statusText} - ${errorText}`);
-    }
-    if (!response.body) {
-        throw new Error('No response body from Gemini API');
-    }
+async function compressWithGemini(modelName, conversationMessages, customSystemPrompt) {
+    const messages = prepareMessagesForCompression(conversationMessages, customSystemPrompt);
     let summary = '';
     let usage = {
         prompt_tokens: 0,
         completion_tokens: 0,
         total_tokens: 0,
     };
-    // Parse SSE stream
-    const reader = response.body.getReader();
-    const decoder = new TextDecoder();
-    let buffer = '';
-    while (true) {
-        const { done, value } = await reader.read();
-        if (done)
-            break;
-        buffer += decoder.decode(value, { stream: true });
-        const lines = buffer.split('\n');
-        buffer = lines.pop() || '';
-        for (const line of lines) {
-            const trimmed = line.trim();
-            if (!trimmed || trimmed.startsWith(':'))
-                continue;
-            if (trimmed.startsWith('data: ')) {
-                const data = trimmed.slice(6);
-                try {
-                    const chunk = JSON.parse(data);
-                    // Process candidates
-                    if (chunk.candidates && chunk.candidates.length > 0) {
-                        const candidate = chunk.candidates[0];
-                        if (candidate.content && candidate.content.parts) {
-                            for (const part of candidate.content.parts) {
-                                if (part.text) {
-                                    summary += part.text;
-                                }
-                            }
-                        }
-                    }
-                    // Collect usage info
-                    if (chunk.usageMetadata) {
-                        usage = {
-                            prompt_tokens: chunk.usageMetadata.promptTokenCount || 0,
-                            completion_tokens: chunk.usageMetadata.candidatesTokenCount || 0,
-                            total_tokens: chunk.usageMetadata.totalTokenCount || 0,
-                        };
-                    }
-                }
-                catch (e) {
-                    console.error('Failed to parse Gemini SSE data:', data);
-                }
-            }
+    // Use the existing streaming API from gemini.ts (includes proxy support)
+    for await (const chunk of createStreamingGeminiCompletion({
+        model: modelName,
+        messages,
+    })) {
+        // Collect content
+        if (chunk.type === 'content' && chunk.content) {
+            summary += chunk.content;
+        }
+        // Collect usage info
+        if (chunk.type === 'usage' && chunk.usage) {
+            usage = {
+                prompt_tokens: chunk.usage.prompt_tokens || 0,
+                completion_tokens: chunk.usage.completion_tokens || 0,
+                total_tokens: chunk.usage.total_tokens || 0,
+            };
         }
     }
     if (!summary) {
         throw new Error('Failed to generate summary from Gemini model');
     }
-    return {
-        summary,
-        usage,
-    };
+    return { summary, usage };
 }
 /**
- * Compress context using Anthropic API
+ * Compress context using Anthropic API (reuses anthropic.ts)
  */
-async function compressWithAnthropic(baseUrl, apiKey, modelName, conversationMessages, systemPrompt) {
-    const customHeaders = getCustomHeaders();
-    // Build messages array with conversation history
-    const messages = [];
-    // If custom system prompt exists, add default as first user message
-    if (systemPrompt) {
-        messages.push({ role: 'user', content: SYSTEM_PROMPT });
-    }
-    // Add all conversation history (exclude system messages)
-    for (const msg of conversationMessages) {
-        if (msg.role !== 'system' && msg.role !== 'tool') {
-            messages.push({
-                role: msg.role,
-                content: msg.content,
-            });
-        }
-    }
-    // Add compression request as final user message
-    messages.push({
-        role: 'user',
-        content: COMPRESSION_PROMPT,
-    });
-    // Anthropic uses system parameter separately
-    const systemParam = systemPrompt || SYSTEM_PROMPT;
-    // Build request payload (no tools for compression)
-    const requestPayload = {
-        model: modelName,
-        max_tokens: 4096,
-        system: systemParam,
-        messages,
-        stream: true
-    };
-    const effectiveBaseUrl = baseUrl && baseUrl !== 'https://api.openai.com/v1'
-        ? baseUrl
-        : 'https://api.anthropic.com/v1';
-    const response = await fetch(`${effectiveBaseUrl}/messages`, {
-        method: 'POST',
-        headers: {
-            'Content-Type': 'application/json',
-            'x-api-key': apiKey,
-            'authorization': `Bearer ${apiKey}`,
-            ...customHeaders
-        },
-        body: JSON.stringify(requestPayload)
-    });
-    if (!response.ok) {
-        const errorText = await response.text();
-        throw new Error(`Anthropic API error: ${response.status} ${response.statusText} - ${errorText}`);
-    }
-    if (!response.body) {
-        throw new Error('No response body from Anthropic API');
-    }
+async function compressWithAnthropic(modelName, conversationMessages, customSystemPrompt) {
+    const messages = prepareMessagesForCompression(conversationMessages, customSystemPrompt);
     let summary = '';
     let usage = {
         prompt_tokens: 0,
         completion_tokens: 0,
         total_tokens: 0,
     };
-    // Parse Anthropic SSE stream
-    const reader = response.body.getReader();
-    const decoder = new TextDecoder();
-    let buffer = '';
-    while (true) {
-        const { done, value } = await reader.read();
-        if (done)
-            break;
-        buffer += decoder.decode(value, { stream: true });
-        const lines = buffer.split('\n');
-        buffer = lines.pop() || '';
-        for (const line of lines) {
-            const trimmed = line.trim();
-            if (!trimmed || trimmed.startsWith(':'))
-                continue;
-            if (trimmed.startsWith('event: ')) {
-                continue;
-            }
-            if (trimmed.startsWith('data: ')) {
-                const data = trimmed.slice(6);
-                try {
-                    const event = JSON.parse(data);
-                    if (event.type === 'content_block_delta' && event.delta?.type === 'text_delta') {
-                        summary += event.delta.text;
-                    }
-                    // Collect usage info from message_start event
-                    if (event.type === 'message_start' && event.message?.usage) {
-                        usage.prompt_tokens = event.message.usage.input_tokens || 0;
-                    }
-                    // Collect usage info from message_delta event
-                    if (event.type === 'message_delta' && event.usage) {
-                        usage.completion_tokens = event.usage.output_tokens || 0;
-                        usage.total_tokens = usage.prompt_tokens + usage.completion_tokens;
-                    }
-                }
-                catch (e) {
-                    console.error('Failed to parse Anthropic SSE data:', data);
-                }
-            }
+    // Use the existing streaming API from anthropic.ts (includes proxy support)
+    for await (const chunk of createStreamingAnthropicCompletion({
+        model: modelName,
+        messages,
+        max_tokens: 4096,
+    })) {
+        // Collect content
+        if (chunk.type === 'content' && chunk.content) {
+            summary += chunk.content;
+        }
+        // Collect usage info
+        if (chunk.type === 'usage' && chunk.usage) {
+            usage = {
+                prompt_tokens: chunk.usage.prompt_tokens || 0,
+                completion_tokens: chunk.usage.completion_tokens || 0,
+                total_tokens: chunk.usage.total_tokens || 0,
+            };
         }
     }
     if (!summary) {
         throw new Error('Failed to generate summary from Anthropic model');
     }
-    return {
-        summary,
-        usage,
-    };
+    return { summary, usage };
 }
 /**
  * Compress conversation history using the compact model
@@ -452,30 +189,25 @@ export async function compressContext(messages) {
     if (!config.compactModel || !config.compactModel.modelName) {
         throw new Error('Compact model not configured. Please configure it in API & Model Settings.');
     }
-    // Use shared API credentials
-    const baseUrl = config.baseUrl;
-    const apiKey = config.apiKey;
     const modelName = config.compactModel.modelName;
     const requestMethod = config.requestMethod;
-    if (!baseUrl || !apiKey) {
-        throw new Error('API configuration incomplete. Please configure Base URL and API Key.');
-    }
     // Get custom system prompt if configured
     const customSystemPrompt = getCustomSystemPrompt();
     try {
         // Choose compression method based on request method
+        // All methods now reuse existing API modules which include proxy support
         switch (requestMethod) {
             case 'gemini':
-                return await compressWithGemini(baseUrl, apiKey, modelName, messages, customSystemPrompt || null);
+                return await compressWithGemini(modelName, messages, customSystemPrompt || null);
             case 'anthropic':
-                return await compressWithAnthropic(baseUrl, apiKey, modelName, messages, customSystemPrompt || null);
+                return await compressWithAnthropic(modelName, messages, customSystemPrompt || null);
             case 'responses':
                 // OpenAI Responses API
-                return await compressWithResponses(baseUrl, apiKey, modelName, messages, customSystemPrompt || null);
+                return await compressWithResponses(modelName, messages, customSystemPrompt || null);
             case 'chat':
             default:
                 // OpenAI Chat Completions API
-                return await compressWithChatCompletions(baseUrl, apiKey, modelName, messages, customSystemPrompt || null);
+                return await compressWithChatCompletions(modelName, messages, customSystemPrompt || null);
         }
     }
     catch (error) {

package/dist/utils/fileDialog.d.ts

@@ -0,0 +1,9 @@
+/**
+ * Cross-platform file save dialog
+ * Opens a native file save dialog and returns the selected path
+ */
+export declare function showSaveDialog(defaultFilename?: string, title?: string): Promise<string | null>;
+/**
+ * Check if native file dialogs are available on this platform
+ */
+export declare function isFileDialogSupported(): boolean;

package/dist/utils/fileDialog.js

@@ -0,0 +1,74 @@
+import { exec } from 'child_process';
+import { promisify } from 'util';
+import * as path from 'path';
+import * as os from 'os';
+const execAsync = promisify(exec);
+/**
+ * Cross-platform file save dialog
+ * Opens a native file save dialog and returns the selected path
+ */
+export async function showSaveDialog(defaultFilename = 'export.txt', title = 'Save File') {
+    const platform = os.platform();
+    try {
+        if (platform === 'darwin') {
+            // macOS - use osascript (AppleScript)
+            const defaultPath = path.join(os.homedir(), 'Downloads', defaultFilename);
+            const script = `
+				set defaultPath to POSIX file "${defaultPath}"
+				set saveFile to choose file name with prompt "${title}" default location (POSIX file "${os.homedir()}/Downloads") default name "${defaultFilename}"
+				return POSIX path of saveFile
+			`;
+            const { stdout } = await execAsync(`osascript -e '${script.replace(/'/g, "'\\''")}'`);
+            return stdout.trim();
+        }
+        else if (platform === 'win32') {
+            // Windows - use PowerShell
+            const script = `
+				Add-Type -AssemblyName System.Windows.Forms
+				$dialog = New-Object System.Windows.Forms.SaveFileDialog
+				$dialog.Title = "${title}"
+				$dialog.Filter = "Text files (*.txt)|*.txt|Markdown files (*.md)|*.md|All files (*.*)|*.*"
+				$dialog.FileName = "${defaultFilename}"
+				$dialog.InitialDirectory = "${path.join(os.homedir(), 'Downloads').replace(/\\/g, '\\\\')}"
+				$result = $dialog.ShowDialog()
+				if ($result -eq 'OK') {
+					Write-Output $dialog.FileName
+				}
+			`;
+            const { stdout } = await execAsync(`powershell -NoProfile -Command "${script.replace(/"/g, '\\"')}"`);
+            const result = stdout.trim();
+            return result || null;
+        }
+        else {
+            // Linux - use zenity (most common) or kdialog as fallback
+            try {
+                const defaultPath = path.join(os.homedir(), 'Downloads', defaultFilename);
+                const { stdout } = await execAsync(`zenity --file-selection --save --title="${title}" --filename="${defaultPath}" --confirm-overwrite`);
+                return stdout.trim();
+            }
+            catch (error) {
+                // Try kdialog as fallback for KDE systems
+                try {
+                    const defaultPath = path.join(os.homedir(), 'Downloads', defaultFilename);
+                    const { stdout } = await execAsync(`kdialog --getsavefilename "${defaultPath}" "*.*|All Files" --title "${title}"`);
+                    return stdout.trim();
+                }
+                catch {
+                    // If both fail, return null
+                    return null;
+                }
+            }
+        }
+    }
+    catch (error) {
+        // User cancelled or error occurred
+        return null;
+    }
+}
+/**
+ * Check if native file dialogs are available on this platform
+ */
+export function isFileDialogSupported() {
+    const platform = os.platform();
+    return platform === 'darwin' || platform === 'win32' || platform === 'linux';
+}

package/dist/utils/incrementalSnapshot.d.ts

@@ -75,6 +75,13 @@ declare class IncrementalSnapshotManager {
      * @returns Number of files rolled back
      */
     rollbackToMessageIndex(sessionId: string, targetMessageIndex: number): Promise<number>;
+    /**
+     * Delete all snapshots >= targetMessageIndex
+     * This is used when user rolls back conversation to clean up snapshot files
+     * @param sessionId Session ID
+     * @param targetMessageIndex The message index to delete from (inclusive)
+     */
+    deleteSnapshotsFromIndex(sessionId: string, targetMessageIndex: number): Promise<number>;
     /**
      * Clear all snapshots for a session
      */