morpheus-cli 0.4.0 → 0.4.2

package/dist/devkit/tools/system.js

@@ -0,0 +1,132 @@
+import { tool } from '@langchain/core/tools';
+import { z } from 'zod';
+import { ShellAdapter } from '../adapters/shell.js';
+import { registerToolFactory } from '../registry.js';
+import { platform } from 'os';
+export function createSystemTools(ctx) {
+    const shell = ShellAdapter.create();
+    const isWindows = platform() === 'win32';
+    const isMac = platform() === 'darwin';
+    return [
+        tool(async ({ title, message, urgency }) => {
+            try {
+                if (isWindows) {
+                    // PowerShell toast notification
+                    const ps = `[Windows.UI.Notifications.ToastNotificationManager, Windows.UI.Notifications, ContentType=WindowsRuntime] | Out-Null; $template = [Windows.UI.Notifications.ToastNotificationManager]::GetTemplateContent([Windows.UI.Notifications.ToastTemplateType]::ToastText02); $template.GetElementsByTagName('text')[0].AppendChild($template.CreateTextNode('${title}')); $template.GetElementsByTagName('text')[1].AppendChild($template.CreateTextNode('${message}')); [Windows.UI.Notifications.ToastNotificationManager]::CreateToastNotifier('Morpheus').Show([Windows.UI.Notifications.ToastNotification]::new($template))`;
+                    await shell.run('powershell', ['-Command', ps], { cwd: ctx.working_dir, timeout_ms: 5_000 });
+                }
+                else if (isMac) {
+                    await shell.run('osascript', ['-e', `display notification "${message}" with title "${title}"`], {
+                        cwd: ctx.working_dir, timeout_ms: 5_000,
+                    });
+                }
+                else {
+                    // Linux — notify-send
+                    const args = [title, message];
+                    if (urgency)
+                        args.unshift(`-u`, urgency);
+                    await shell.run('notify-send', args, { cwd: ctx.working_dir, timeout_ms: 5_000 });
+                }
+                return JSON.stringify({ success: true });
+            }
+            catch (err) {
+                return JSON.stringify({ success: false, error: err.message });
+            }
+        }, {
+            name: 'notify',
+            description: 'Send a desktop notification.',
+            schema: z.object({
+                title: z.string(),
+                message: z.string(),
+                urgency: z.enum(['low', 'normal', 'critical']).optional().describe('Linux urgency level'),
+            }),
+        }),
+        tool(async () => {
+            try {
+                let result;
+                if (isWindows) {
+                    result = await shell.run('powershell', ['-Command', 'Get-Clipboard'], {
+                        cwd: ctx.working_dir, timeout_ms: 5_000,
+                    });
+                }
+                else if (isMac) {
+                    result = await shell.run('pbpaste', [], { cwd: ctx.working_dir, timeout_ms: 5_000 });
+                }
+                else {
+                    result = await shell.run('xclip', ['-selection', 'clipboard', '-o'], {
+                        cwd: ctx.working_dir, timeout_ms: 5_000,
+                    });
+                }
+                return JSON.stringify({ success: result.exitCode === 0, content: result.stdout });
+            }
+            catch (err) {
+                return JSON.stringify({ success: false, error: err.message });
+            }
+        }, {
+            name: 'read_clipboard',
+            description: 'Read the current clipboard contents.',
+            schema: z.object({}),
+        }),
+        tool(async ({ content }) => {
+            try {
+                let result;
+                if (isWindows) {
+                    result = await shell.run('powershell', ['-Command', `Set-Clipboard -Value '${content.replace(/'/g, "''")}'`], {
+                        cwd: ctx.working_dir, timeout_ms: 5_000,
+                    });
+                }
+                else if (isMac) {
+                    result = await shell.run('sh', ['-c', `printf '%s' '${content.replace(/'/g, "'\\''")}' | pbcopy`], {
+                        cwd: ctx.working_dir, timeout_ms: 5_000,
+                    });
+                }
+                else {
+                    result = await shell.run('sh', ['-c', `printf '%s' '${content.replace(/'/g, "'\\''")}' | xclip -selection clipboard`], {
+                        cwd: ctx.working_dir, timeout_ms: 5_000,
+                    });
+                }
+                return JSON.stringify({ success: result.exitCode === 0 });
+            }
+            catch (err) {
+                return JSON.stringify({ success: false, error: err.message });
+            }
+        }, {
+            name: 'write_clipboard',
+            description: 'Write content to the clipboard.',
+            schema: z.object({ content: z.string() }),
+        }),
+        tool(async ({ url }) => {
+            try {
+                const open = isWindows ? 'start' : isMac ? 'open' : 'xdg-open';
+                const result = await shell.run(isWindows ? 'cmd' : open, isWindows ? ['/c', 'start', url] : [url], {
+                    cwd: ctx.working_dir, timeout_ms: 5_000,
+                });
+                return JSON.stringify({ success: result.exitCode === 0, url });
+            }
+            catch (err) {
+                return JSON.stringify({ success: false, error: err.message });
+            }
+        }, {
+            name: 'open_url',
+            description: 'Open a URL in the default browser.',
+            schema: z.object({ url: z.string() }),
+        }),
+        tool(async ({ file_path }) => {
+            try {
+                const open = isWindows ? 'start' : isMac ? 'open' : 'xdg-open';
+                const result = await shell.run(isWindows ? 'cmd' : open, isWindows ? ['/c', 'start', '""', file_path] : [file_path], {
+                    cwd: ctx.working_dir, timeout_ms: 5_000,
+                });
+                return JSON.stringify({ success: result.exitCode === 0, file_path });
+            }
+            catch (err) {
+                return JSON.stringify({ success: false, error: err.message });
+            }
+        }, {
+            name: 'open_file',
+            description: 'Open a file with the default application.',
+            schema: z.object({ file_path: z.string() }),
+        }),
+    ];
+}
+registerToolFactory(createSystemTools);

package/dist/devkit/types.js

@@ -0,0 +1 @@
+export const MAX_OUTPUT_BYTES = 50 * 1024; // 50 KB

package/dist/devkit/utils.js

@@ -0,0 +1,45 @@
+import path from 'path';
+import { MAX_OUTPUT_BYTES } from './types.js';
+/**
+ * Truncates a string to MAX_OUTPUT_BYTES (50 KB) if needed.
+ * Returns a UTF-8-safe truncation with a note when truncated.
+ */
+export function truncateOutput(output) {
+    const bytes = Buffer.byteLength(output, 'utf8');
+    if (bytes <= MAX_OUTPUT_BYTES)
+        return output;
+    const truncated = Buffer.from(output).subarray(0, MAX_OUTPUT_BYTES).toString('utf8');
+    return truncated + `\n\n[OUTPUT TRUNCATED: ${bytes} bytes total, showing first ${MAX_OUTPUT_BYTES} bytes]`;
+}
+/**
+ * Returns true if filePath is inside dir (or equal to dir).
+ * Both paths are resolved before comparison.
+ */
+export function isWithinDir(filePath, dir) {
+    const resolved = path.resolve(filePath);
+    const resolvedDir = path.resolve(dir);
+    return resolved === resolvedDir || resolved.startsWith(resolvedDir + path.sep);
+}
+/**
+ * Extracts the binary base name from a command string.
+ * Handles full paths (/usr/bin/node, C:\bin\node.exe) and plain names.
+ */
+export function extractBinaryName(command) {
+    // Take first token (before any space), then get the basename, strip extension
+    const firstToken = command.split(/\s+/)[0] ?? command;
+    const base = path.basename(firstToken);
+    return base.replace(/\.(exe|cmd|bat|sh|ps1)$/i, '').toLowerCase();
+}
+/**
+ * Checks if a command is allowed based on the allowlist.
+ * Empty allowlist means ALL commands are allowed (Merovingian mode).
+ */
+export function isCommandAllowed(command, allowedCommands) {
+    if (allowedCommands.length === 0)
+        return true;
+    const binary = extractBinaryName(command);
+    return allowedCommands.some(allowed => {
+        const allowedBinary = extractBinaryName(allowed);
+        return allowedBinary === binary;
+    });
+}

package/dist/http/api.js

@@ -213,6 +213,82 @@ export function createApiRouter(oracle) {
             res.status(500).json({ error: error.message });
         }
     });
+    // --- Model Pricing ---
+    const ModelPricingSchema = z.object({
+        provider: z.string().min(1),
+        model: z.string().min(1),
+        input_price_per_1m: z.number().nonnegative(),
+        output_price_per_1m: z.number().nonnegative()
+    });
+    router.get('/model-pricing', (req, res) => {
+        try {
+            const h = new SQLiteChatMessageHistory({ sessionId: 'api-reader' });
+            const entries = h.listModelPricing();
+            h.close();
+            res.json(entries);
+        }
+        catch (error) {
+            res.status(500).json({ error: error.message });
+        }
+    });
+    router.post('/model-pricing', (req, res) => {
+        const parsed = ModelPricingSchema.safeParse(req.body);
+        if (!parsed.success) {
+            return res.status(400).json({ error: 'Invalid payload', details: parsed.error.issues });
+        }
+        try {
+            const h = new SQLiteChatMessageHistory({ sessionId: 'api-reader' });
+            h.upsertModelPricing(parsed.data);
+            h.close();
+            res.json({ success: true });
+        }
+        catch (error) {
+            res.status(500).json({ error: error.message });
+        }
+    });
+    router.put('/model-pricing/:provider/:model', (req, res) => {
+        const { provider, model } = req.params;
+        const partial = z.object({
+            input_price_per_1m: z.number().nonnegative().optional(),
+            output_price_per_1m: z.number().nonnegative().optional()
+        }).safeParse(req.body);
+        if (!partial.success) {
+            return res.status(400).json({ error: 'Invalid payload', details: partial.error.issues });
+        }
+        try {
+            const h = new SQLiteChatMessageHistory({ sessionId: 'api-reader' });
+            const existing = h.listModelPricing().find(e => e.provider === provider && e.model === model);
+            if (!existing) {
+                h.close();
+                return res.status(404).json({ error: 'Pricing entry not found' });
+            }
+            h.upsertModelPricing({
+                provider,
+                model,
+                input_price_per_1m: partial.data.input_price_per_1m ?? existing.input_price_per_1m,
+                output_price_per_1m: partial.data.output_price_per_1m ?? existing.output_price_per_1m
+            });
+            h.close();
+            res.json({ success: true });
+        }
+        catch (error) {
+            res.status(500).json({ error: error.message });
+        }
+    });
+    router.delete('/model-pricing/:provider/:model', (req, res) => {
+        const { provider, model } = req.params;
+        try {
+            const h = new SQLiteChatMessageHistory({ sessionId: 'api-reader' });
+            const changes = h.deleteModelPricing(provider, model);
+            h.close();
+            if (changes === 0)
+                return res.status(404).json({ error: 'Pricing entry not found' });
+            res.json({ success: true });
+        }
+        catch (error) {
+            res.status(500).json({ error: error.message });
+        }
+    });
     // Calculate diff between two objects
     const getDiff = (obj1, obj2, prefix = '') => {
         const changes = [];
@@ -310,6 +386,52 @@ export function createApiRouter(oracle) {
             res.status(500).json({ error: error.message });
         }
     });
+    // Apoc config endpoints
+    router.get('/config/apoc', (req, res) => {
+        try {
+            const apocConfig = configManager.getApocConfig();
+            res.json(apocConfig);
+        }
+        catch (error) {
+            res.status(500).json({ error: error.message });
+        }
+    });
+    router.post('/config/apoc', async (req, res) => {
+        try {
+            const config = configManager.get();
+            await configManager.save({ ...config, apoc: req.body });
+            const display = DisplayManager.getInstance();
+            display.log('Apoc configuration updated via UI', {
+                source: 'Zaion',
+                level: 'info'
+            });
+            res.json({ success: true });
+        }
+        catch (error) {
+            if (error.name === 'ZodError') {
+                res.status(400).json({ error: 'Validation failed', details: error.errors });
+            }
+            else {
+                res.status(500).json({ error: error.message });
+            }
+        }
+    });
+    router.delete('/config/apoc', async (req, res) => {
+        try {
+            const config = configManager.get();
+            const { apoc: _apoc, ...restConfig } = config;
+            await configManager.save(restConfig);
+            const display = DisplayManager.getInstance();
+            display.log('Apoc configuration removed via UI (falling back to Oracle config)', {
+                source: 'Zaion',
+                level: 'info'
+            });
+            res.json({ success: true });
+        }
+        catch (error) {
+            res.status(500).json({ error: error.message });
+        }
+    });
     // Sati memories endpoints
     router.get('/sati/memories', async (req, res) => {
         try {

package/dist/runtime/apoc.js

@@ -0,0 +1,110 @@
+import { HumanMessage, SystemMessage } from "@langchain/core/messages";
+import { ConfigManager } from "../config/manager.js";
+import { ProviderFactory } from "./providers/factory.js";
+import { ProviderError } from "./errors.js";
+import { DisplayManager } from "./display.js";
+import { buildDevKit } from "../devkit/index.js";
+/**
+ * Apoc is a subagent of Oracle specialized in devtools operations.
+ * It receives delegated tasks from Oracle and executes them using DevKit tools
+ * (filesystem, shell, git, network, processes, packages, system).
+ *
+ * Oracle calls Apoc via the `apoc_delegate` tool when the user requests
+ * dev-related tasks such as running commands, reading/writing files,
+ * managing git, or inspecting system state.
+ */
+export class Apoc {
+    static instance = null;
+    agent;
+    config;
+    display = DisplayManager.getInstance();
+    constructor(config) {
+        this.config = config || ConfigManager.getInstance().get();
+    }
+    static getInstance(config) {
+        if (!Apoc.instance) {
+            Apoc.instance = new Apoc(config);
+        }
+        return Apoc.instance;
+    }
+    static resetInstance() {
+        Apoc.instance = null;
+    }
+    async initialize() {
+        const apocConfig = this.config.apoc || this.config.llm;
+        console.log(`Apoc configuration: ${JSON.stringify(apocConfig)}`);
+        const working_dir = this.config.apoc?.working_dir || process.cwd();
+        const timeout_ms = this.config.apoc?.timeout_ms || 30_000;
+        // Import all devkit tool factories (side-effect registration)
+        await import("../devkit/index.js");
+        const tools = buildDevKit({
+            working_dir,
+            allowed_commands: [], // no restriction — Oracle is trusted orchestrator
+            timeout_ms,
+        });
+        this.display.log(`Apoc initialized with ${tools.length} DevKit tools (working_dir: ${working_dir})`, { source: "Apoc" });
+        try {
+            this.agent = await ProviderFactory.createBare(apocConfig, tools);
+        }
+        catch (err) {
+            throw new ProviderError(apocConfig.provider, err, "Apoc subagent initialization failed");
+        }
+    }
+    /**
+     * Execute a devtools task delegated by Oracle.
+     * @param task Natural language task description
+     * @param context Optional additional context from the ongoing conversation
+     */
+    async execute(task, context) {
+        if (!this.agent) {
+            await this.initialize();
+        }
+        this.display.log(`Executing delegated task: ${task.slice(0, 80)}...`, {
+            source: "Apoc",
+        });
+        const systemMessage = new SystemMessage(`
+You are Apoc, a specialized devtools subagent within the Morpheus system.
+
+You are called by Oracle when the user needs dev operations performed.
+Your job is to execute the requested task accurately using your available tools.
+
+Available capabilities:
+- Read, write, append, and delete files
+- Execute shell commands
+- Inspect and manage processes
+- Run git operations (status, log, diff, clone, commit, etc.)
+- Perform network operations (curl, DNS, ping)
+- Manage packages (npm, yarn)
+- Inspect system information
+
+OPERATING RULES:
+1. Use tools to accomplish the task. Do not speculate.
+2. Always verify results after execution.
+3. Report clearly what was done and what the result was.
+4. If something fails, report the error and what you tried.
+5. Stay focused on the delegated task only.
+
+${context ? `CONTEXT FROM ORACLE:\n${context}` : ""}
+    `);
+        const userMessage = new HumanMessage(task);
+        const messages = [systemMessage, userMessage];
+        try {
+            const response = await this.agent.invoke({ messages });
+            const lastMessage = response.messages[response.messages.length - 1];
+            const content = typeof lastMessage.content === "string"
+                ? lastMessage.content
+                : JSON.stringify(lastMessage.content);
+            this.display.log("Apoc task completed.", { source: "Apoc" });
+            return content;
+        }
+        catch (err) {
+            throw new ProviderError(this.config.apoc?.provider || this.config.llm.provider, err, "Apoc task execution failed");
+        }
+    }
+    /** Reload with updated config (called when settings change) */
+    async reload() {
+        this.config = ConfigManager.getInstance().get();
+        this.agent = undefined;
+        await this.initialize();
+    }
+}

package/dist/runtime/memory/sati/index.js

@@ -41,7 +41,7 @@ export class SatiMemoryMiddleware {
             return null;
         }
     }
-    async afterAgent(generatedResponse, history) {
+    async afterAgent(generatedResponse, history, userSessionId) {
         try {
             await this.service.evaluateAndPersist([
                 ...history.slice(-5).map(m => ({
@@ -49,7 +49,7 @@ export class SatiMemoryMiddleware {
                     content: m.content.toString()
                 })),
                 { role: 'assistant', content: generatedResponse }
-            ]);
+            ], userSessionId);
         }
         catch (error) {
             display.log(`Error in afterAgent: ${error}`, { source: 'Sati' });

package/dist/runtime/memory/sati/service.js

@@ -50,7 +50,7 @@ export class SatiService {
             }))
         };
     }
-    async evaluateAndPersist(conversation) {
+    async evaluateAndPersist(conversation, userSessionId) {
         try {
             const satiConfig = ConfigManager.getInstance().getSatiConfig();
             if (!satiConfig)
@@ -74,7 +74,8 @@ export class SatiService {
                 new SystemMessage(SATI_EVALUATION_PROMPT),
                 new HumanMessage(JSON.stringify(inputPayload, null, 2))
             ];
-            const history = new SQLiteChatMessageHistory({ sessionId: 'sati-evaluation' });
+            const satiSessionId = userSessionId ? `sati-evaluation-${userSessionId}` : 'sati-evaluation';
+            const history = new SQLiteChatMessageHistory({ sessionId: satiSessionId });
             try {
                 const inputMsg = new ToolMessage({
                     content: JSON.stringify(inputPayload, null, 2),

package/dist/runtime/memory/sqlite.js

@@ -14,6 +14,9 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
     sessionId;
     limit;
     titleSet = false; // cache: skip setSessionTitleIfNeeded after title is set
+    get currentSessionId() {
+        return this.sessionId;
+    }
     constructor(fields) {
         super();
         this.sessionId = fields.sessionId && fields.sessionId !== '' ? fields.sessionId : '';
@@ -108,9 +111,10 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
           total_tokens INTEGER,
           cache_read_tokens INTEGER,
           provider TEXT,
-          model TEXT
+          model TEXT,
+          audio_duration_seconds REAL
         );
-        
+
         CREATE INDEX IF NOT EXISTS idx_messages_session_id
         ON messages(session_id);
 
@@ -130,6 +134,33 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
           embedding_status TEXT CHECK (embedding_status IN ('none', 'pending', 'embedded', 'failed')) NOT NULL DEFAULT 'none'
         );
 
+        CREATE TABLE IF NOT EXISTS model_pricing (
+          provider TEXT NOT NULL,
+          model TEXT NOT NULL,
+          input_price_per_1m REAL NOT NULL DEFAULT 0,
+          output_price_per_1m REAL NOT NULL DEFAULT 0,
+          PRIMARY KEY (provider, model)
+        );
+
+        INSERT OR IGNORE INTO model_pricing (provider, model, input_price_per_1m, output_price_per_1m) VALUES
+          ('anthropic', 'claude-opus-4-6', 15.0, 75.0),
+          ('anthropic', 'claude-sonnet-4-5-20250929', 3.0, 15.0),
+          ('anthropic', 'claude-haiku-4-5-20251001', 0.8, 4.0),
+          ('anthropic', 'claude-3-5-sonnet-20241022', 3.0, 15.0),
+          ('anthropic', 'claude-3-5-haiku-20241022', 0.8, 4.0),
+          ('anthropic', 'claude-3-opus-20240229', 15.0, 75.0),
+          ('openai', 'gpt-4o', 2.5, 10.0),
+          ('openai', 'gpt-4o-mini', 0.15, 0.6),
+          ('openai', 'gpt-4-turbo', 10.0, 30.0),
+          ('openai', 'gpt-3.5-turbo', 0.5, 1.5),
+          ('openai', 'o1', 15.0, 60.0),
+          ('openai', 'o1-mini', 3.0, 12.0),
+          ('google', 'gemini-2.5-flash', 0.15, 0.6),
+          ('google', 'gemini-2.5-flash-lite', 0.075, 0.3),
+          ('google', 'gemini-2.0-flash', 0.1, 0.4),
+          ('google', 'gemini-1.5-pro', 1.25, 5.0),
+          ('google', 'gemini-1.5-flash', 0.075, 0.3);
+
       `);
             this.migrateTable();
         }
@@ -151,13 +182,15 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
                 'total_tokens',
                 'cache_read_tokens',
                 'provider',
-                'model'
+                'model',
+                'audio_duration_seconds'
             ];
             const integerColumns = new Set(['input_tokens', 'output_tokens', 'total_tokens', 'cache_read_tokens']);
+            const realColumns = new Set(['audio_duration_seconds']);
             for (const col of newColumns) {
                 if (!columns.has(col)) {
                     try {
-                        const type = integerColumns.has(col) ? 'INTEGER' : 'TEXT';
+                        const type = integerColumns.has(col) ? 'INTEGER' : realColumns.has(col) ? 'REAL' : 'TEXT';
                         this.db.exec(`ALTER TABLE messages ADD COLUMN ${col} ${type}`);
                     }
                     catch (e) {
@@ -296,6 +329,7 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
             // Extract provider metadata
             const provider = anyMsg.provider_metadata?.provider ?? null;
             const model = anyMsg.provider_metadata?.model ?? null;
+            const audioDurationSeconds = usage?.audio_duration_seconds ?? null;
             // Handle special content serialization for Tools
             let finalContent = "";
             if (type === 'ai' && (message.tool_calls?.length ?? 0) > 0) {
@@ -318,8 +352,8 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
                     ? message.content
                     : JSON.stringify(message.content);
             }
-            const stmt = this.db.prepare("INSERT INTO messages (session_id, type, content, created_at, input_tokens, output_tokens, total_tokens, cache_read_tokens, provider, model) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)");
-            stmt.run(this.sessionId, type, finalContent, Date.now(), inputTokens, outputTokens, totalTokens, cacheReadTokens, provider, model);
+            const stmt = this.db.prepare("INSERT INTO messages (session_id, type, content, created_at, input_tokens, output_tokens, total_tokens, cache_read_tokens, provider, model, audio_duration_seconds) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)");
+            stmt.run(this.sessionId, type, finalContent, Date.now(), inputTokens, outputTokens, totalTokens, cacheReadTokens, provider, model, audioDurationSeconds);
             // Verificar se a sessão tem título e definir automaticamente se necessário
             await this.setSessionTitleIfNeeded();
         }
@@ -346,7 +380,7 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
     async addMessages(messages) {
         if (messages.length === 0)
             return;
-        const stmt = this.db.prepare("INSERT INTO messages (session_id, type, content, created_at, input_tokens, output_tokens, total_tokens, cache_read_tokens, provider, model) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)");
+        const stmt = this.db.prepare("INSERT INTO messages (session_id, type, content, created_at, input_tokens, output_tokens, total_tokens, cache_read_tokens, provider, model, audio_duration_seconds) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)");
         const insertAll = this.db.transaction((msgs) => {
             for (const message of msgs) {
                 let type;
@@ -373,7 +407,7 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
                 else {
                     finalContent = typeof message.content === "string" ? message.content : JSON.stringify(message.content);
                 }
-                stmt.run(this.sessionId, type, finalContent, Date.now(), usage?.input_tokens ?? null, usage?.output_tokens ?? null, usage?.total_tokens ?? null, usage?.input_token_details?.cache_read ?? usage?.cache_read_tokens ?? null, anyMsg.provider_metadata?.provider ?? null, anyMsg.provider_metadata?.model ?? null);
+                stmt.run(this.sessionId, type, finalContent, Date.now(), usage?.input_tokens ?? null, usage?.output_tokens ?? null, usage?.total_tokens ?? null, usage?.input_token_details?.cache_read ?? usage?.cache_read_tokens ?? null, anyMsg.provider_metadata?.provider ?? null, anyMsg.provider_metadata?.model ?? null, usage?.audio_duration_seconds ?? null);
             }
         });
         try {
@@ -440,9 +474,18 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
         try {
             const stmt = this.db.prepare("SELECT SUM(input_tokens) as totalInput, SUM(output_tokens) as totalOutput FROM messages");
             const row = stmt.get();
+            // Calculate total estimated cost by summing per-model costs
+            const costStmt = this.db.prepare(`SELECT
+          SUM((COALESCE(m.input_tokens, 0) / 1000000.0) * p.input_price_per_1m
+            + (COALESCE(m.output_tokens, 0) / 1000000.0) * p.output_price_per_1m) as totalCost
+        FROM messages m
+        INNER JOIN model_pricing p ON p.provider = m.provider AND p.model = COALESCE(m.model, 'unknown')
+        WHERE m.provider IS NOT NULL`);
+            const costRow = costStmt.get();
             return {
                 totalInputTokens: row.totalInput || 0,
-                totalOutputTokens: row.totalOutput || 0
+                totalOutputTokens: row.totalOutput || 0,
+                totalEstimatedCostUsd: costRow.totalCost ?? null
             };
         }
         catch (error) {
@@ -474,31 +517,58 @@ export class SQLiteChatMessageHistory extends BaseListChatMessageHistory {
      */
     async getUsageStatsByProviderAndModel() {
         try {
-            const stmt = this.db.prepare(`SELECT 
-          provider,
-          COALESCE(model, 'unknown') as model,
-          SUM(input_tokens) as totalInputTokens,
-          SUM(output_tokens) as totalOutputTokens,
-          SUM(total_tokens) as totalTokens,
-          COUNT(*) as messageCount
-        FROM messages
-        WHERE provider IS NOT NULL
-        GROUP BY provider, COALESCE(model, 'unknown')
-        ORDER BY provider, model`);
+            const stmt = this.db.prepare(`SELECT
+          m.provider,
+          COALESCE(m.model, 'unknown') as model,
+          SUM(m.input_tokens) as totalInputTokens,
+          SUM(m.output_tokens) as totalOutputTokens,
+          SUM(m.total_tokens) as totalTokens,
+          COUNT(*) as messageCount,
+          COALESCE(SUM(m.audio_duration_seconds), 0) as totalAudioSeconds,
+          p.input_price_per_1m,
+          p.output_price_per_1m
+        FROM messages m
+        LEFT JOIN model_pricing p ON p.provider = m.provider AND p.model = COALESCE(m.model, 'unknown')
+        WHERE m.provider IS NOT NULL
+        GROUP BY m.provider, COALESCE(m.model, 'unknown')
+        ORDER BY m.provider, m.model`);
             const rows = stmt.all();
-            return rows.map((row) => ({
-                provider: row.provider,
-                model: row.model,
-                totalInputTokens: row.totalInputTokens || 0,
-                totalOutputTokens: row.totalOutputTokens || 0,
-                totalTokens: row.totalTokens || 0,
-                messageCount: row.messageCount || 0
-            }));
+            return rows.map((row) => {
+                const inputTokens = row.totalInputTokens || 0;
+                const outputTokens = row.totalOutputTokens || 0;
+                let estimatedCostUsd = null;
+                if (row.input_price_per_1m != null && row.output_price_per_1m != null) {
+                    estimatedCostUsd = (inputTokens / 1_000_000) * row.input_price_per_1m
+                        + (outputTokens / 1_000_000) * row.output_price_per_1m;
+                }
+                return {
+                    provider: row.provider,
+                    model: row.model,
+                    totalInputTokens: inputTokens,
+                    totalOutputTokens: outputTokens,
+                    totalTokens: row.totalTokens || 0,
+                    messageCount: row.messageCount || 0,
+                    totalAudioSeconds: row.totalAudioSeconds || 0,
+                    estimatedCostUsd
+                };
+            });
         }
         catch (error) {
             throw new Error(`Failed to get grouped usage stats: ${error}`);
         }
     }
+    // --- Model Pricing CRUD ---
+    listModelPricing() {
+        const rows = this.db.prepare('SELECT provider, model, input_price_per_1m, output_price_per_1m FROM model_pricing ORDER BY provider, model').all();
+        return rows;
+    }
+    upsertModelPricing(entry) {
+        this.db.prepare('INSERT INTO model_pricing (provider, model, input_price_per_1m, output_price_per_1m) VALUES (?, ?, ?, ?) ON CONFLICT(provider, model) DO UPDATE SET input_price_per_1m = excluded.input_price_per_1m, output_price_per_1m = excluded.output_price_per_1m').run(entry.provider, entry.model, entry.input_price_per_1m, entry.output_price_per_1m);
+    }
+    deleteModelPricing(provider, model) {
+        const result = this.db.prepare('DELETE FROM model_pricing WHERE provider = ? AND model = ?').run(provider, model);
+        return result.changes;
+    }
     /**
      * Clears all messages for the current session from the database.
      */