attocode 0.2.3 → 0.2.5

package/dist/src/integrations/streaming/pty-shell.js

@@ -0,0 +1,367 @@
+/**
+ * Persistent PTY Shell
+ *
+ * Provides a persistent shell session that maintains state between commands.
+ * Uses Node.js child_process with PTY-like behavior.
+ *
+ * Key features:
+ * - Persistent shell process (bash/zsh)
+ * - Command history tracking
+ * - Environment inheritance
+ * - Working directory state
+ * - Output buffering with timeout
+ */
+import { spawn } from 'node:child_process';
+import { platform } from 'node:os';
+// =============================================================================
+// PTY SHELL MANAGER
+// =============================================================================
+/**
+ * Manages persistent shell sessions.
+ */
+export class PTYShellManager {
+    config;
+    process = null;
+    outputBuffer = '';
+    commandHistory = [];
+    currentCwd;
+    currentEnv;
+    commandResolve = null;
+    commandStartTime = 0;
+    commandTimeoutId = null;
+    eventListeners = new Set();
+    endMarker;
+    constructor(config = {}) {
+        this.config = {
+            shell: config.shell || this.detectShell(),
+            cwd: config.cwd || process.cwd(),
+            env: config.env || {},
+            timeout: config.timeout ?? 30000,
+            maxOutputSize: config.maxOutputSize ?? 1024 * 1024, // 1MB
+            promptPattern: config.promptPattern || '__CMD_DONE__',
+        };
+        this.currentCwd = this.config.cwd;
+        this.currentEnv = { ...process.env, ...this.config.env };
+        this.endMarker = `\necho "${this.config.promptPattern} $?"`;
+    }
+    /**
+     * Auto-detect the user's preferred shell.
+     */
+    detectShell() {
+        const userShell = process.env.SHELL;
+        if (userShell)
+            return userShell;
+        if (platform() === 'win32') {
+            return 'cmd.exe';
+        }
+        return '/bin/bash';
+    }
+    /**
+     * Start the persistent shell.
+     */
+    async start() {
+        if (this.process && !this.process.killed) {
+            return; // Already running
+        }
+        return new Promise((resolve, reject) => {
+            const env = {
+                ...process.env,
+                ...this.config.env,
+                // Disable shell prompt customization that might interfere
+                PS1: '$ ',
+                PROMPT_COMMAND: '',
+            };
+            this.process = spawn(this.config.shell, [], {
+                cwd: this.config.cwd,
+                env,
+                stdio: ['pipe', 'pipe', 'pipe'],
+                shell: false, // We're spawning a shell directly
+            });
+            // Set up output handlers
+            this.process.stdout?.on('data', (data) => {
+                const output = data.toString();
+                this.handleOutput(output);
+            });
+            this.process.stderr?.on('data', (data) => {
+                const output = data.toString();
+                this.handleOutput(output);
+            });
+            this.process.on('error', (error) => {
+                this.emitEvent({ type: 'error', error: error.message });
+                reject(error);
+            });
+            this.process.on('exit', (code) => {
+                this.emitEvent({ type: 'shell.stopped', exitCode: code });
+                this.process = null;
+            });
+            // Wait a bit for shell to initialize
+            setTimeout(() => {
+                if (this.process && this.process.pid) {
+                    this.emitEvent({ type: 'shell.started', pid: this.process.pid, shell: this.config.shell });
+                    resolve();
+                }
+                else {
+                    reject(new Error('Failed to start shell'));
+                }
+            }, 100);
+        });
+    }
+    /**
+     * Handle output from the shell.
+     */
+    handleOutput(data) {
+        this.outputBuffer += data;
+        this.emitEvent({ type: 'command.output', data });
+        // Trim buffer if too large
+        if (this.outputBuffer.length > this.config.maxOutputSize) {
+            this.outputBuffer = this.outputBuffer.slice(-this.config.maxOutputSize);
+        }
+        // Check for command completion
+        if (this.commandResolve && this.outputBuffer.includes(this.config.promptPattern)) {
+            this.handleCommandComplete();
+        }
+    }
+    /**
+     * Handle command completion.
+     */
+    handleCommandComplete() {
+        if (!this.commandResolve)
+            return;
+        // Clear timeout
+        if (this.commandTimeoutId) {
+            clearTimeout(this.commandTimeoutId);
+            this.commandTimeoutId = null;
+        }
+        // Extract output and exit code
+        const markerIndex = this.outputBuffer.lastIndexOf(this.config.promptPattern);
+        let output = this.outputBuffer.slice(0, markerIndex).trim();
+        const exitCodeMatch = this.outputBuffer.slice(markerIndex).match(/(\d+)/);
+        const exitCode = exitCodeMatch ? parseInt(exitCodeMatch[1], 10) : null;
+        // Remove the echo command from output
+        const echoIndex = output.lastIndexOf('echo "');
+        if (echoIndex !== -1) {
+            output = output.slice(0, echoIndex).trim();
+        }
+        // Also remove any command echo (bash echoes input commands)
+        const lines = output.split('\n');
+        if (lines.length > 1 && this.commandHistory.length > 0) {
+            const lastCmd = this.commandHistory[this.commandHistory.length - 1];
+            if (lines[0].endsWith(lastCmd) || lines[0].includes(lastCmd)) {
+                lines.shift();
+                output = lines.join('\n');
+            }
+        }
+        const result = {
+            output: output.trim(),
+            exitCode,
+            duration: Date.now() - this.commandStartTime,
+            timedOut: false,
+        };
+        this.emitEvent({ type: 'command.complete', result });
+        const resolve = this.commandResolve;
+        this.commandResolve = null;
+        this.outputBuffer = '';
+        resolve(result);
+    }
+    /**
+     * Execute a command in the persistent shell.
+     */
+    async execute(command) {
+        if (!this.process || this.process.killed) {
+            await this.start();
+        }
+        if (!this.process?.stdin) {
+            throw new Error('Shell not running');
+        }
+        // Track command
+        this.commandHistory.push(command);
+        this.commandStartTime = Date.now();
+        this.outputBuffer = '';
+        this.emitEvent({ type: 'command.start', command });
+        return new Promise((resolve) => {
+            this.commandResolve = resolve;
+            // Set timeout
+            this.commandTimeoutId = setTimeout(() => {
+                if (this.commandResolve) {
+                    this.emitEvent({ type: 'command.timeout', command });
+                    const result = {
+                        output: this.outputBuffer.trim(),
+                        exitCode: null,
+                        duration: Date.now() - this.commandStartTime,
+                        timedOut: true,
+                    };
+                    this.commandResolve = null;
+                    this.outputBuffer = '';
+                    resolve(result);
+                }
+            }, this.config.timeout);
+            // Write command with completion marker
+            const fullCommand = `${command}${this.endMarker}\n`;
+            this.process.stdin.write(fullCommand);
+        });
+    }
+    /**
+     * Change the working directory.
+     */
+    async cd(directory) {
+        const result = await this.execute(`cd "${directory}" && pwd`);
+        if (result.exitCode === 0) {
+            this.currentCwd = result.output.trim();
+        }
+        return result;
+    }
+    /**
+     * Set an environment variable.
+     */
+    async setEnv(key, value) {
+        await this.execute(`export ${key}="${value}"`);
+        this.currentEnv[key] = value;
+    }
+    /**
+     * Get the current state.
+     */
+    getState() {
+        return {
+            cwd: this.currentCwd,
+            env: this.currentEnv,
+            history: [...this.commandHistory],
+            isRunning: this.process !== null && !this.process.killed,
+            pid: this.process?.pid,
+        };
+    }
+    /**
+     * Get command history.
+     */
+    getHistory() {
+        return [...this.commandHistory];
+    }
+    /**
+     * Clear command history.
+     */
+    clearHistory() {
+        this.commandHistory = [];
+    }
+    /**
+     * Subscribe to events.
+     */
+    subscribe(listener) {
+        this.eventListeners.add(listener);
+        return () => this.eventListeners.delete(listener);
+    }
+    /**
+     * Emit an event.
+     */
+    emitEvent(event) {
+        for (const listener of this.eventListeners) {
+            try {
+                listener(event);
+            }
+            catch {
+                // Ignore listener errors
+            }
+        }
+    }
+    /**
+     * Stop the shell.
+     */
+    async stop() {
+        if (!this.process)
+            return;
+        return new Promise((resolve) => {
+            if (!this.process) {
+                resolve();
+                return;
+            }
+            const timeout = setTimeout(() => {
+                this.process?.kill('SIGKILL');
+                resolve();
+            }, 1000);
+            this.process.once('exit', () => {
+                clearTimeout(timeout);
+                resolve();
+            });
+            // Try graceful exit first
+            this.process.stdin?.write('exit\n');
+        });
+    }
+    /**
+     * Cleanup resources.
+     */
+    async cleanup() {
+        if (this.commandTimeoutId) {
+            clearTimeout(this.commandTimeoutId);
+        }
+        await this.stop();
+        this.eventListeners.clear();
+    }
+}
+// =============================================================================
+// FACTORY FUNCTIONS
+// =============================================================================
+/**
+ * Create a PTY shell manager.
+ */
+export function createPTYShell(config) {
+    return new PTYShellManager(config);
+}
+/**
+ * Create and start a PTY shell.
+ */
+export async function createAndStartPTYShell(config) {
+    const shell = new PTYShellManager(config);
+    await shell.start();
+    return shell;
+}
+// =============================================================================
+// TOOL FACTORY
+// =============================================================================
+/**
+ * Create a shell execution tool that uses the persistent PTY.
+ */
+export function createPTYShellTool(shell) {
+    return {
+        name: 'shell_exec',
+        description: 'Execute a command in the persistent shell session. Maintains state between calls (working directory, environment variables, etc.)',
+        parameters: {
+            type: 'object',
+            properties: {
+                command: {
+                    type: 'string',
+                    description: 'The command to execute',
+                },
+            },
+            required: ['command'],
+        },
+        dangerLevel: 'dangerous',
+        async execute(args) {
+            const command = args.command;
+            if (!command) {
+                return 'Error: command is required';
+            }
+            const result = await shell.execute(command);
+            if (result.timedOut) {
+                return `Command timed out after ${result.duration}ms. Partial output:\n${result.output}`;
+            }
+            const exitInfo = result.exitCode !== 0 ? ` (exit code: ${result.exitCode})` : '';
+            return `${result.output}${exitInfo}`;
+        },
+    };
+}
+/**
+ * Format shell state for display.
+ */
+export function formatShellState(state) {
+    const lines = [
+        `Shell: ${state.isRunning ? 'Running' : 'Stopped'}${state.pid ? ` (PID: ${state.pid})` : ''}`,
+        `CWD: ${state.cwd}`,
+        `History: ${state.history.length} commands`,
+    ];
+    if (state.history.length > 0) {
+        lines.push('Recent commands:');
+        for (const cmd of state.history.slice(-5)) {
+            lines.push(`  $ ${cmd}`);
+        }
+    }
+    return lines.join('\n');
+}
+//# sourceMappingURL=pty-shell.js.map

package/dist/src/integrations/streaming/pty-shell.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pty-shell.js","sourceRoot":"","sources":["../../../../src/integrations/streaming/pty-shell.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAEH,OAAO,EAAE,KAAK,EAAgB,MAAM,oBAAoB,CAAC;AACzD,OAAO,EAAE,QAAQ,EAAE,MAAM,SAAS,CAAC;AA+CnC,gFAAgF;AAChF,oBAAoB;AACpB,gFAAgF;AAEhF;;GAEG;AACH,MAAM,OAAO,eAAe;IAClB,MAAM,CAA2B;IACjC,OAAO,GAAwB,IAAI,CAAC;IACpC,YAAY,GAAW,EAAE,CAAC;IAC1B,cAAc,GAAa,EAAE,CAAC;IAC9B,UAAU,CAAS;IACnB,UAAU,CAAyB;IACnC,cAAc,GAA6C,IAAI,CAAC;IAChE,gBAAgB,GAAW,CAAC,CAAC;IAC7B,gBAAgB,GAA0B,IAAI,CAAC;IAC/C,cAAc,GAA0B,IAAI,GAAG,EAAE,CAAC;IACzC,SAAS,CAAS;IAEnC,YAAY,SAAyB,EAAE;QACrC,IAAI,CAAC,MAAM,GAAG;YACZ,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,IAAI,CAAC,WAAW,EAAE;YACzC,GAAG,EAAE,MAAM,CAAC,GAAG,IAAI,OAAO,CAAC,GAAG,EAAE;YAChC,GAAG,EAAE,MAAM,CAAC,GAAG,IAAI,EAAE;YACrB,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,KAAK;YAChC,aAAa,EAAE,MAAM,CAAC,aAAa,IAAI,IAAI,GAAG,IAAI,EAAE,MAAM;YAC1D,aAAa,EAAE,MAAM,CAAC,aAAa,IAAI,cAAc;SACtD,CAAC;QACF,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC;QAClC,IAAI,CAAC,UAAU,GAAG,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,EAA4B,CAAC;QACnF,IAAI,CAAC,SAAS,GAAG,WAAW,IAAI,CAAC,MAAM,CAAC,aAAa,MAAM,CAAC;IAC9D,CAAC;IAED;;OAEG;IACK,WAAW;QACjB,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC;QACpC,IAAI,SAAS;YAAE,OAAO,SAAS,CAAC;QAEhC,IAAI,QAAQ,EAAE,KAAK,OAAO,EAAE,CAAC;YAC3B,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,OAAO,WAAW,CAAC;IACrB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YACzC,OAAO,CAAC,kBAAkB;QAC5B,CAAC;QAED,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,GAAG,GAAG;gBACV,GAAG,OAAO,CAAC,GAAG;gBACd,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG;gBAClB,0DAA0D;gBAC1D,GAAG,EAAE,IAAI;gBACT,cAAc,EAAE,EAAE;aACnB,CAAC;YAEF,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,EAAE;gBAC1C,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG;gBACpB,GAAG;gBACH,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;gBAC/B,KAAK,EAAE,KAAK,EAAE,kCAAkC;aACjD,CAAC,CAAC;YAEH,yBAAyB;YACzB,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE;gBAC/C,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAC/B,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;YAC5B,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE;gBAC/C,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAC/B,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;YAC5B,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;gBACjC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;gBACxD,MAAM,CAAC,KAAK,CAAC,CAAC;YAChB,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;gBAC/B,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC1D,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;YACtB,CAAC,CAAC,CAAC;YAEH,qCAAqC;YACrC,UAAU,CAAC,GAAG,EAAE;gBACd,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;oBACrC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;oBAC3F,OAAO,EAAE,CAAC;gBACZ,CAAC;qBAAM,CAAC;oBACN,MAAM,CAAC,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC,CAAC;gBAC7C,CAAC;YACH,CAAC,EAAE,GAAG,CAAC,CAAC;QACV,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,YAAY,CAAC,IAAY;QAC/B,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC;QAC1B,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,CAAC,CAAC;QAEjD,2BAA2B;QAC3B,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;YACzD,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QAC1E,CAAC;QAED,+BAA+B;QAC/B,IAAI,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;YACjF,IAAI,CAAC,qBAAqB,EAAE,CAAC;QAC/B,CAAC;IACH,CAAC;IAED;;OAEG;IACK,qBAAqB;QAC3B,IAAI,CAAC,IAAI,CAAC,cAAc;YAAE,OAAO;QAEjC,gBAAgB;QAChB,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1B,YAAY,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YACpC,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC/B,CAAC;QAED,+BAA+B;QAC/B,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QAC7E,IAAI,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC,IAAI,EAAE,CAAC;QAC5D,MAAM,aAAa,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC1E,MAAM,QAAQ,GAAG,aAAa,CAAC,CAAC,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QAEvE,sCAAsC;QACtC,MAAM,SAAS,GAAG,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAC/C,IAAI,SAAS,KAAK,CAAC,CAAC,EAAE,CAAC;YACrB,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,IAAI,EAAE,CAAC;QAC7C,CAAC;QAED,4DAA4D;QAC5D,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACjC,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvD,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YACpE,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC7D,KAAK,CAAC,KAAK,EAAE,CAAC;gBACd,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;QAED,MAAM,MAAM,GAAkB;YAC5B,MAAM,EAAE,MAAM,CAAC,IAAI,EAAE;YACrB,QAAQ;YACR,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,gBAAgB;YAC5C,QAAQ,EAAE,KAAK;SAChB,CAAC;QAEF,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,MAAM,EAAE,CAAC,CAAC;QAErD,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC;QACpC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC3B,IAAI,CAAC,YAAY,GAAG,EAAE,CAAC;QAEvB,OAAO,CAAC,MAAM,CAAC,CAAC;IAClB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CAAC,OAAe;QAC3B,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YACzC,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;QACrB,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC;YACzB,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;QACvC,CAAC;QAED,gBAAgB;QAChB,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAClC,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACnC,IAAI,CAAC,YAAY,GAAG,EAAE,CAAC;QAEvB,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,CAAC,CAAC;QAEnD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC7B,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC;YAE9B,cAAc;YACd,IAAI,CAAC,gBAAgB,GAAG,UAAU,CAAC,GAAG,EAAE;gBACtC,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;oBACxB,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,OAAO,EAAE,CAAC,CAAC;oBAErD,MAAM,MAAM,GAAkB;wBAC5B,MAAM,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE;wBAChC,QAAQ,EAAE,IAAI;wBACd,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,gBAAgB;wBAC5C,QAAQ,EAAE,IAAI;qBACf,CAAC;oBAEF,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;oBAC3B,IAAI,CAAC,YAAY,GAAG,EAAE,CAAC;oBAEvB,OAAO,CAAC,MAAM,CAAC,CAAC;gBAClB,CAAC;YACH,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAExB,uCAAuC;YACvC,MAAM,WAAW,GAAG,GAAG,OAAO,GAAG,IAAI,CAAC,SAAS,IAAI,CAAC;YACpD,IAAI,CAAC,OAAQ,CAAC,KAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,EAAE,CAAC,SAAiB;QACxB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,SAAS,UAAU,CAAC,CAAC;QAC9D,IAAI,MAAM,CAAC,QAAQ,KAAK,CAAC,EAAE,CAAC;YAC1B,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QACzC,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM,CAAC,GAAW,EAAE,KAAa;QACrC,MAAM,IAAI,CAAC,OAAO,CAAC,UAAU,GAAG,KAAK,KAAK,GAAG,CAAC,CAAC;QAC/C,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,QAAQ;QACN,OAAO;YACL,GAAG,EAAE,IAAI,CAAC,UAAU;YACpB,GAAG,EAAE,IAAI,CAAC,UAAU;YACpB,OAAO,EAAE,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC;YACjC,SAAS,EAAE,IAAI,CAAC,OAAO,KAAK,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM;YACxD,GAAG,EAAE,IAAI,CAAC,OAAO,EAAE,GAAG;SACvB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,UAAU;QACR,OAAO,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC,CAAC;IAClC,CAAC;IAED;;OAEG;IACH,YAAY;QACV,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,QAA0B;QAClC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAClC,OAAO,GAAG,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IACpD,CAAC;IAED;;OAEG;IACK,SAAS,CAAC,KAAe;QAC/B,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YAC3C,IAAI,CAAC;gBACH,QAAQ,CAAC,KAAK,CAAC,CAAC;YAClB,CAAC;YAAC,MAAM,CAAC;gBACP,yBAAyB;YAC3B,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,IAAI;QACR,IAAI,CAAC,IAAI,CAAC,OAAO;YAAE,OAAO;QAE1B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC7B,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;gBAClB,OAAO,EAAE,CAAC;gBACV,OAAO;YACT,CAAC;YAED,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC9B,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;gBAC9B,OAAO,EAAE,CAAC;YACZ,CAAC,EAAE,IAAI,CAAC,CAAC;YAET,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,EAAE;gBAC7B,YAAY,CAAC,OAAO,CAAC,CAAC;gBACtB,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;YAEH,0BAA0B;YAC1B,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO;QACX,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1B,YAAY,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACtC,CAAC;QACD,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;IAC9B,CAAC;CACF;AAED,gFAAgF;AAChF,oBAAoB;AACpB,gFAAgF;AAEhF;;GAEG;AACH,MAAM,UAAU,cAAc,CAAC,MAAuB;IACpD,OAAO,IAAI,eAAe,CAAC,MAAM,CAAC,CAAC;AACrC,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAAC,MAAuB;IAClE,MAAM,KAAK,GAAG,IAAI,eAAe,CAAC,MAAM,CAAC,CAAC;IAC1C,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC;IACpB,OAAO,KAAK,CAAC;AACf,CAAC;AAED,gFAAgF;AAChF,eAAe;AACf,gFAAgF;AAEhF;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,KAAsB;IACvD,OAAO;QACL,IAAI,EAAE,YAAY;QAClB,WAAW,EAAE,mIAAmI;QAChJ,UAAU,EAAE;YACV,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,OAAO,EAAE;oBACP,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,wBAAwB;iBACtC;aACF;YACD,QAAQ,EAAE,CAAC,SAAS,CAAC;SACtB;QACD,WAAW,EAAE,WAAoB;QACjC,KAAK,CAAC,OAAO,CAAC,IAA6B;YACzC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAiB,CAAC;YACvC,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,4BAA4B,CAAC;YACtC,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAE5C,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;gBACpB,OAAO,2BAA2B,MAAM,CAAC,QAAQ,wBAAwB,MAAM,CAAC,MAAM,EAAE,CAAC;YAC3F,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,KAAK,CAAC,CAAC,CAAC,CAAC,gBAAgB,MAAM,CAAC,QAAQ,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;YACjF,OAAO,GAAG,MAAM,CAAC,MAAM,GAAG,QAAQ,EAAE,CAAC;QACvC,CAAC;KACF,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAAC,KAAiB;IAChD,MAAM,KAAK,GAAG;QACZ,UAAU,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QAC7F,QAAQ,KAAK,CAAC,GAAG,EAAE;QACnB,YAAY,KAAK,CAAC,OAAO,CAAC,MAAM,WAAW;KAC5C,CAAC;IAEF,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7B,KAAK,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAC/B,KAAK,MAAM,GAAG,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAC1C,KAAK,CAAC,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC"}

package/dist/src/integrations/streaming/streaming.d.ts

@@ -0,0 +1,102 @@
+/**
+ * Lesson 25: Streaming Response Handler
+ *
+ * Handles streaming LLM responses for real-time output.
+ * Supports text chunks, tool calls, and usage tracking.
+ */
+import type { ToolCall, ChatResponse } from '../../types.js';
+/**
+ * A chunk from a streaming response.
+ */
+export interface StreamChunk {
+    type: 'text' | 'tool_call_start' | 'tool_call_delta' | 'tool_call_end' | 'usage' | 'done' | 'error';
+    content?: string;
+    toolCall?: Partial<ToolCall>;
+    toolCallId?: string;
+    usage?: {
+        inputTokens: number;
+        outputTokens: number;
+    };
+    error?: string;
+}
+/**
+ * Callback for stream chunks.
+ */
+export type StreamCallback = (chunk: StreamChunk) => void;
+/**
+ * Streaming configuration.
+ */
+export interface StreamConfig {
+    /** Enable streaming (default: true) */
+    enabled?: boolean;
+    /** Buffer size before flushing (default: 50 chars for batched updates) */
+    bufferSize?: number;
+    /** Show typing indicator (default: true) */
+    showTypingIndicator?: boolean;
+}
+/**
+ * Stream events.
+ */
+export type StreamEvent = {
+    type: 'stream.start';
+} | {
+    type: 'stream.text';
+    content: string;
+} | {
+    type: 'stream.tool_call';
+    toolCall: ToolCall;
+} | {
+    type: 'stream.complete';
+    response: ChatResponse;
+} | {
+    type: 'stream.error';
+    error: string;
+};
+export type StreamEventListener = (event: StreamEvent) => void;
+/**
+ * Handles streaming responses from LLM providers.
+ */
+export declare class StreamHandler {
+    private config;
+    private listeners;
+    private buffer;
+    constructor(config?: StreamConfig);
+    /**
+     * Process a streaming response.
+     * Accumulates chunks and returns the final ChatResponse.
+     */
+    processStream(stream: AsyncIterable<StreamChunk>, onChunk?: StreamCallback): Promise<ChatResponse>;
+    /**
+     * Process a single chunk.
+     */
+    private processChunk;
+    /**
+     * Subscribe to events.
+     */
+    on(listener: StreamEventListener): () => void;
+    /**
+     * Emit an event.
+     */
+    private emit;
+    /**
+     * Check if streaming is enabled.
+     */
+    isEnabled(): boolean;
+}
+/**
+ * Create a stream handler.
+ */
+export declare function createStreamHandler(config?: StreamConfig): StreamHandler;
+/**
+ * Format a stream chunk for terminal output.
+ */
+export declare function formatChunkForTerminal(chunk: StreamChunk): string;
+/**
+ * Convert OpenRouter SSE stream to StreamChunk iterator.
+ */
+export declare function adaptOpenRouterStream(response: Response): AsyncIterable<StreamChunk>;
+/**
+ * Convert Anthropic stream to StreamChunk iterator.
+ */
+export declare function adaptAnthropicStream(response: Response): AsyncIterable<StreamChunk>;
+//# sourceMappingURL=streaming.d.ts.map

package/dist/src/integrations/streaming/streaming.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"streaming.d.ts","sourceRoot":"","sources":["../../../../src/integrations/streaming/streaming.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAM7D;;GAEG;AACH,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,GAAG,iBAAiB,GAAG,iBAAiB,GAAG,eAAe,GAAG,OAAO,GAAG,MAAM,GAAG,OAAO,CAAC;IACpG,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;IAC7B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,KAAK,CAAC,EAAE;QACN,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;IACF,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG,CAAC,KAAK,EAAE,WAAW,KAAK,IAAI,CAAC;AAE1D;;GAEG;AACH,MAAM,WAAW,YAAY;IAC3B,uCAAuC;IACvC,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,0EAA0E;IAC1E,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,4CAA4C;IAC5C,mBAAmB,CAAC,EAAE,OAAO,CAAC;CAC/B;AAYD;;GAEG;AACH,MAAM,MAAM,WAAW,GACnB;IAAE,IAAI,EAAE,cAAc,CAAA;CAAE,GACxB;IAAE,IAAI,EAAE,aAAa,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GACxC;IAAE,IAAI,EAAE,kBAAkB,CAAC;IAAC,QAAQ,EAAE,QAAQ,CAAA;CAAE,GAChD;IAAE,IAAI,EAAE,iBAAiB,CAAC;IAAC,QAAQ,EAAE,YAAY,CAAA;CAAE,GACnD;IAAE,IAAI,EAAE,cAAc,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,CAAC;AAE5C,MAAM,MAAM,mBAAmB,GAAG,CAAC,KAAK,EAAE,WAAW,KAAK,IAAI,CAAC;AAM/D;;GAEG;AACH,qBAAa,aAAa;IACxB,OAAO,CAAC,MAAM,CAAyB;IACvC,OAAO,CAAC,SAAS,CAA6B;IAC9C,OAAO,CAAC,MAAM,CAAc;gBAEhB,MAAM,GAAE,YAAiB;IAQrC;;;OAGG;IACG,aAAa,CACjB,MAAM,EAAE,aAAa,CAAC,WAAW,CAAC,EAClC,OAAO,CAAC,EAAE,cAAc,GACvB,OAAO,CAAC,YAAY,CAAC;IAoCxB;;OAEG;IACH,OAAO,CAAC,YAAY;IAiFpB;;OAEG;IACH,EAAE,CAAC,QAAQ,EAAE,mBAAmB,GAAG,MAAM,IAAI;IAQ7C;;OAEG;IACH,OAAO,CAAC,IAAI;IAUZ;;OAEG;IACH,SAAS,IAAI,OAAO;CAGrB;AAMD;;GAEG;AACH,wBAAgB,mBAAmB,CAAC,MAAM,CAAC,EAAE,YAAY,GAAG,aAAa,CAExE;AAMD;;GAEG;AACH,wBAAgB,sBAAsB,CAAC,KAAK,EAAE,WAAW,GAAG,MAAM,CAoBjE;AAMD;;GAEG;AACH,wBAAuB,qBAAqB,CAC1C,QAAQ,EAAE,QAAQ,GACjB,aAAa,CAAC,WAAW,CAAC,CAqF5B;AAED;;GAEG;AACH,wBAAuB,oBAAoB,CACzC,QAAQ,EAAE,QAAQ,GACjB,aAAa,CAAC,WAAW,CAAC,CAyF5B"}