gencode-ai 0.3.0 → 0.4.0

package/RELEASE_NOTES_v0.4.0.md

@@ -0,0 +1,140 @@
+# Release Notes - v0.4.0
+
+## 🔄 Breaking Changes
+
+### Provider Rename: Gemini → Google
+
+We've renamed the "Gemini" provider to "Google" for better naming consistency.
+
+**Rationale:**
+- Provider = Company (Google, OpenAI, Anthropic)
+- Model = Product (Gemini, GPT, Claude)
+- Aligns with industry naming patterns
+
+**Changes:**
+- Provider ID: `'gemini'` → `'google'`
+- TypeScript: `GeminiProvider` → `GoogleProvider`, `GeminiConfig` → `GoogleConfig`
+
+**Unchanged:**
+- ✅ Model names: `gemini-2.0-flash`, `gemini-1.5-pro`, etc.
+- ✅ Environment variables: `GOOGLE_API_KEY` and `GEMINI_API_KEY` both supported
+- ✅ All functionality identical
+
+### Migration Required
+
+**Automatic Migration (Recommended):**
+
+```bash
+# macOS
+sed -i '' 's/"gemini":/"google":/g' ~/.gen/providers.json
+sed -i '' 's/"gemini:/"google:/g' ~/.gen/providers.json
+sed -i '' 's/"provider": "gemini"/"provider": "google"/g' ~/.gen/settings.json
+
+# Linux
+sed -i 's/"gemini":/"google":/g' ~/.gen/providers.json
+sed -i 's/"gemini:/"google:/g' ~/.gen/providers.json
+sed -i 's/"provider": "gemini"/"provider": "google"/g' ~/.gen/settings.json
+```
+
+**Manual Migration:**
+
+1. **`~/.gen/settings.json`**: Change `"provider": "gemini"` → `"provider": "google"`
+2. **`~/.gen/providers.json`**:
+   - Rename connection: `"gemini": {...}` → `"google": {...}`
+   - Rename model cache: `"gemini:api_key": {...}` → `"google:api_key": {...}`
+
+**For SDK Users:**
+
+```typescript
+// Before
+import { GeminiProvider, type GeminiConfig } from 'gencode-ai';
+const provider = new GeminiProvider();
+
+// After
+import { GoogleProvider, type GoogleConfig } from 'gencode-ai';
+const provider = new GoogleProvider();
+```
+
+---
+
+## ✨ New Features
+
+### Context Management (Proposal 0007)
+Manual control over conversation context with new CLI commands:
+
+- **`/compact`** - Trigger conversation compaction manually
+  - Summarizes older messages to reduce context
+  - Shows before/after statistics
+
+- **`/context`** - View context usage statistics
+  - Active vs total messages
+  - Compression status
+  - Visual progress indicator
+
+### Checkpoint Persistence (Proposal 0008)
+Fixed critical persistence bug:
+
+- ✅ Checkpoints saved to disk with sessions
+- ✅ Restored when resuming sessions
+- ✅ Preserved when forking sessions
+- ✅ Session compression works across restarts
+
+### Gemini 3+ Thinking Support
+Added support for Gemini 3+ model reasoning:
+
+- Captures `thoughtSignature` from Gemini 3+ models
+- Streams reasoning chunks with `type: 'reasoning'`
+- Preserves thinking process in tool calls
+
+---
+
+## 🐛 Bug Fixes
+
+- Fixed checkpoint data not persisting across session restarts
+- Fixed provider display name for Google in registry
+- Improved error messages for unknown provider IDs
+
+---
+
+## 📚 Documentation
+
+- Added session compression implementation guide with flowcharts
+- Updated provider documentation for Google rename
+- Added migration guide for breaking changes
+
+---
+
+## 🧹 Housekeeping
+
+- Removed temporary test scripts and documentation
+- Cleaned up code comments and logging
+- Improved checkpoint serialization logic
+
+---
+
+## 📦 Installation
+
+```bash
+# New installation
+npm install -g gencode-ai@0.4.0
+
+# Update from v0.3.x
+npm update -g gencode-ai
+```
+
+---
+
+## ⚠️ Important Notes
+
+- **Breaking changes** - Follow migration guide above
+- **Backup recommended** - Back up `~/.gen/` before upgrading
+- **Config update required** - Use migration script or update manually
+- **SDK users** - Update imports from `GeminiProvider` to `GoogleProvider`
+
+---
+
+## 🔗 Links
+
+- [Full Changelog](https://github.com/your-repo/gencode/compare/v0.3.0...v0.4.0)
+- [Documentation](https://github.com/your-repo/gencode/docs)
+- [Report Issues](https://github.com/your-repo/gencode/issues)

package/dist/agent/agent.d.ts

@@ -17,7 +17,6 @@ export declare class Agent {
     private memoryManager;
     private planModeManager;
     private config;
-    private messages;
     private sessionId;
     private loadedMemory;
     private askUserCallback;
@@ -119,6 +118,13 @@ export declare class Agent {
      * Get current provider
      */
     getProvider(): Provider;
+    /**
+     * Get model information for compression
+     */
+    getModelInfo(): {
+        contextWindow: number;
+        outputLimit?: number;
+    };
     /**
      * List available models from the provider API
      */
@@ -157,7 +163,7 @@ export declare class Agent {
     /**
      * Run a single query through the agent
      */
-    run(prompt: string): AsyncGenerator<AgentEvent, void, unknown>;
+    run(prompt: string, signal?: AbortSignal): AsyncGenerator<AgentEvent, void, unknown>;
     /**
      * Clear conversation history
      */
@@ -171,5 +177,14 @@ export declare class Agent {
      * Get conversation history
      */
     getHistory(): Message[];
+    /**
+     * Get compression statistics
+     */
+    getCompressionStats(): {
+        totalMessages: number;
+        activeMessages: number;
+        summaryCount: number;
+        compressionRatio: number;
+    } | null;
 }
 //# sourceMappingURL=agent.d.ts.map

package/dist/agent/agent.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../../src/agent/agent.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAe,OAAO,EAAqB,QAAQ,EAAc,MAAM,uBAAuB,CAAC;AAG3G,OAAO,EACL,iBAAiB,EACjB,KAAK,cAAc,EACnB,KAAK,kBAAkB,EACvB,KAAK,gBAAgB,EACrB,KAAK,kBAAkB,EACxB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAE,aAAa,EAAE,KAAK,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACtE,OAAO,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAE1D,OAAO,KAAK,EAAE,QAAQ,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAClE,OAAO,EAEL,KAAK,eAAe,EACpB,KAAK,QAAQ,EACb,KAAK,aAAa,EACnB,MAAM,sBAAsB,CAAC;AAG9B,MAAM,MAAM,eAAe,GAAG,CAAC,SAAS,EAAE,QAAQ,EAAE,KAAK,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;AAEnF,qBAAa,KAAK;IAChB,OAAO,CAAC,QAAQ,CAAc;IAC9B,OAAO,CAAC,QAAQ,CAAe;IAC/B,OAAO,CAAC,WAAW,CAAoB;IACvC,OAAO,CAAC,cAAc,CAAiB;IACvC,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,eAAe,CAAkB;IACzC,OAAO,CAAC,MAAM,CAAc;IAC5B,OAAO,CAAC,QAAQ,CAAiB;IACjC,OAAO,CAAC,SAAS,CAAuB;IACxC,OAAO,CAAC,YAAY,CAA6B;IACjD,OAAO,CAAC,eAAe,CAAgC;gBAE3C,MAAM,EAAE,WAAW;IAqB/B;;OAEG;IACG,qBAAqB,CAAC,QAAQ,CAAC,EAAE,kBAAkB,GAAG,OAAO,CAAC,IAAI,CAAC;IAIzE;;OAEG;IACH,kBAAkB,CAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,KAAK,OAAO,CAAC,OAAO,CAAC,GAAG,IAAI;IAItF;;OAEG;IACH,0BAA0B,CACxB,QAAQ,EAAE,CACR,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,OAAO,EACd,WAAW,EAAE,kBAAkB,EAAE,KAC9B,OAAO,CAAC,cAAc,CAAC,GAC3B,IAAI;IAIP;;OAEG;IACH,iBAAiB,CAAC,OAAO,EAAE,gBAAgB,EAAE,GAAG,IAAI;IAIpD;;OAEG;IACH,mBAAmB,IAAI,IAAI;IAI3B;;;OAGG;IACH,mBAAmB,CAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI;IAItF;;OAEG;IACH,oBAAoB,IAAI,iBAAiB;IAIzC;;;OAGG;IACH,kBAAkB,CAAC,QAAQ,EAAE,eAAe,GAAG,IAAI;IAInD;;OAEG;IACH,gBAAgB,IAAI,aAAa;IAIjC;;OAEG;IACH,eAAe,IAAI,YAAY,GAAG,IAAI;IAItC;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,YAAY,CAAC;IA2BzC;;OAEG;IACH,kBAAkB,IAAI,eAAe;IAIrC;;OAEG;IACH,gBAAgB,IAAI,OAAO;IAI3B;;OAEG;IACH,cAAc,IAAI,QAAQ;IAI1B;;OAEG;IACG,aAAa,CAAC,eAAe,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAQ9D;;OAEG;IACH,YAAY,CAAC,QAAQ,GAAE,OAAe,GAAG,IAAI;IAW7C;;OAEG;IACG,cAAc,IAAI,OAAO,CAAC,IAAI,CAAC;IAQrC;;OAEG;IACH,sBAAsB,IAAI,aAAa,EAAE;IAQzC;;OAEG;IACH,YAAY,IAAI,MAAM,GAAG,IAAI;IAI7B;;OAEG;IACH,iBAAiB,IAAI,cAAc;IAInC;;;;;OAKG;IACH,QAAQ,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI;IAuBrE;;OAEG;IACH,QAAQ,IAAI,MAAM;IAIlB;;OAEG;IACH,WAAW,IAAI,QAAQ;IAIvB;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;IAI3D;;OAEG;IACG,YAAY,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAcnD;;OAEG;IACG,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAYxD;;OAEG;IACG,YAAY,IAAI,OAAO,CAAC,OAAO,CAAC;IAYtC;;OAEG;IACG,WAAW,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAWzD;;OAEG;IACG,YAAY;IAIlB;;OAEG;IACG,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAIxD;;OAEG;IACG,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC;IAQlC;;OAEG;IACI,GAAG,CAAC,MAAM,EAAE,MAAM,GAAG,cAAc,CAAC,UAAU,EAAE,IAAI,EAAE,OAAO,CAAC;IAgHrE;;OAEG;IACH,YAAY,IAAI,IAAI;IAKpB;;;OAGG;IACH,yBAAyB,IAAI,IAAI;IAuBjC;;OAEG;IACH,UAAU,IAAI,OAAO,EAAE;CAGxB"}
+{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../../src/agent/agent.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAe,OAAO,EAAqB,QAAQ,EAAc,MAAM,uBAAuB,CAAC;AAG3G,OAAO,EACL,iBAAiB,EACjB,KAAK,cAAc,EACnB,KAAK,kBAAkB,EACvB,KAAK,gBAAgB,EACrB,KAAK,kBAAkB,EACxB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAE,aAAa,EAAE,KAAK,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACtE,OAAO,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAE1D,OAAO,KAAK,EAAE,QAAQ,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAClE,OAAO,EAEL,KAAK,eAAe,EACpB,KAAK,QAAQ,EACb,KAAK,aAAa,EACnB,MAAM,sBAAsB,CAAC;AAI9B,MAAM,MAAM,eAAe,GAAG,CAAC,SAAS,EAAE,QAAQ,EAAE,KAAK,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;AAEnF,qBAAa,KAAK;IAChB,OAAO,CAAC,QAAQ,CAAc;IAC9B,OAAO,CAAC,QAAQ,CAAe;IAC/B,OAAO,CAAC,WAAW,CAAoB;IACvC,OAAO,CAAC,cAAc,CAAiB;IACvC,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,eAAe,CAAkB;IACzC,OAAO,CAAC,MAAM,CAAc;IAC5B,OAAO,CAAC,SAAS,CAAuB;IACxC,OAAO,CAAC,YAAY,CAA6B;IACjD,OAAO,CAAC,eAAe,CAAgC;gBAE3C,MAAM,EAAE,WAAW;IA0B/B;;OAEG;IACG,qBAAqB,CAAC,QAAQ,CAAC,EAAE,kBAAkB,GAAG,OAAO,CAAC,IAAI,CAAC;IAIzE;;OAEG;IACH,kBAAkB,CAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,KAAK,OAAO,CAAC,OAAO,CAAC,GAAG,IAAI;IAItF;;OAEG;IACH,0BAA0B,CACxB,QAAQ,EAAE,CACR,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,OAAO,EACd,WAAW,EAAE,kBAAkB,EAAE,KAC9B,OAAO,CAAC,cAAc,CAAC,GAC3B,IAAI;IAIP;;OAEG;IACH,iBAAiB,CAAC,OAAO,EAAE,gBAAgB,EAAE,GAAG,IAAI;IAIpD;;OAEG;IACH,mBAAmB,IAAI,IAAI;IAI3B;;;OAGG;IACH,mBAAmB,CAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI;IAItF;;OAEG;IACH,oBAAoB,IAAI,iBAAiB;IAIzC;;;OAGG;IACH,kBAAkB,CAAC,QAAQ,EAAE,eAAe,GAAG,IAAI;IAInD;;OAEG;IACH,gBAAgB,IAAI,aAAa;IAIjC;;OAEG;IACH,eAAe,IAAI,YAAY,GAAG,IAAI;IAItC;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,YAAY,CAAC;IA2BzC;;OAEG;IACH,kBAAkB,IAAI,eAAe;IAIrC;;OAEG;IACH,gBAAgB,IAAI,OAAO;IAI3B;;OAEG;IACH,cAAc,IAAI,QAAQ;IAI1B;;OAEG;IACG,aAAa,CAAC,eAAe,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAQ9D;;OAEG;IACH,YAAY,CAAC,QAAQ,GAAE,OAAe,GAAG,IAAI;IAW7C;;OAEG;IACG,cAAc,IAAI,OAAO,CAAC,IAAI,CAAC;IAQrC;;OAEG;IACH,sBAAsB,IAAI,aAAa,EAAE;IAQzC;;OAEG;IACH,YAAY,IAAI,MAAM,GAAG,IAAI;IAI7B;;OAEG;IACH,iBAAiB,IAAI,cAAc;IAInC;;;;;OAKG;IACH,QAAQ,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI;IAyBrE;;OAEG;IACH,QAAQ,IAAI,MAAM;IAIlB;;OAEG;IACH,WAAW,IAAI,QAAQ;IAIvB;;OAEG;IACH,YAAY,IAAI;QAAE,aAAa,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,MAAM,CAAA;KAAE;IA6B/D;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;IAI3D;;OAEG;IACG,YAAY,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAgBnD;;OAEG;IACG,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAcxD;;OAEG;IACG,YAAY,IAAI,OAAO,CAAC,OAAO,CAAC;IActC;;OAEG;IACG,WAAW,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAWzD;;OAEG;IACG,YAAY;IAIlB;;OAEG;IACG,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAIxD;;OAEG;IACG,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC;IAOlC;;OAEG;IACI,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,cAAc,CAAC,UAAU,EAAE,IAAI,EAAE,OAAO,CAAC;IAqU3F;;OAEG;IACH,YAAY,IAAI,IAAI;IAIpB;;;OAGG;IACH,yBAAyB,IAAI,IAAI;IAiBjC;;OAEG;IACH,UAAU,IAAI,OAAO,EAAE;IAIvB;;OAEG;IACH,mBAAmB,IAAI;QACrB,aAAa,EAAE,MAAM,CAAC;QACtB,cAAc,EAAE,MAAM,CAAC;QACvB,YAAY,EAAE,MAAM,CAAC;QACrB,gBAAgB,EAAE,MAAM,CAAC;KAC1B,GAAG,IAAI;CAGT"}

package/dist/agent/agent.js

@@ -8,6 +8,7 @@ import { SessionManager } from '../session/index.js';
 import { MemoryManager } from '../memory/index.js';
 import { buildSystemPromptForModel, debugPromptLoading } from '../prompts/index.js';
 import { getPlanModeManager, } from '../planning/index.js';
+import { initCheckpointManager } from '../checkpointing/index.js';
 export class Agent {
     provider;
     registry;
@@ -16,7 +17,6 @@ export class Agent {
     memoryManager;
     planModeManager;
     config;
-    messages = [];
     sessionId = null;
     loadedMemory = null;
     askUserCallback = null;
@@ -35,9 +35,13 @@ export class Agent {
             config: config.permissions,
             projectPath: config.cwd,
         });
-        this.sessionManager = new SessionManager();
+        this.sessionManager = new SessionManager({
+            compression: config.compression,
+        });
         this.memoryManager = new MemoryManager();
         this.planModeManager = getPlanModeManager();
+        // Set compression engine with current model
+        this.sessionManager.setCompressionEngine(this.provider, this.config.model);
     }
     /**
      * Initialize permission system (load persisted rules)
@@ -215,6 +219,8 @@ export class Agent {
                 provider: newProvider,
                 authMethod: newAuthMethod,
             });
+            // Update compression engine with new provider and model
+            this.sessionManager.setCompressionEngine(this.provider, model);
         }
     }
     /**
@@ -229,6 +235,32 @@ export class Agent {
     getProvider() {
         return this.config.provider;
     }
+    /**
+     * Get model information for compression
+     */
+    getModelInfo() {
+        // Try to get from provider if available
+        if (this.provider.getModelInfo) {
+            const info = this.provider.getModelInfo(this.config.model);
+            if (info.contextWindow) {
+                return { contextWindow: info.contextWindow, outputLimit: info.outputLimit };
+            }
+        }
+        // Fallback: rough estimates based on model name
+        // These should eventually be moved to provider implementations
+        const model = this.config.model.toLowerCase();
+        if (model.includes('claude')) {
+            return { contextWindow: 200_000, outputLimit: 8192 };
+        }
+        if (model.includes('gpt-4') || model.includes('gpt-3.5')) {
+            return { contextWindow: 128_000, outputLimit: 4096 };
+        }
+        if (model.includes('gemini')) {
+            return { contextWindow: 1_000_000, outputLimit: 8192 };
+        }
+        // Default fallback
+        return { contextWindow: 128_000, outputLimit: 4096 };
+    }
     /**
      * List available models from the provider API
      */
@@ -246,7 +278,8 @@ export class Agent {
             title,
         });
         this.sessionId = session.metadata.id;
-        this.messages = [];
+        // Initialize checkpoint manager for this session
+        initCheckpointManager(this.sessionId);
         return this.sessionId;
     }
     /**
@@ -258,7 +291,8 @@ export class Agent {
             return false;
         }
         this.sessionId = session.metadata.id;
-        this.messages = session.messages;
+        // CheckpointManager already restored by SessionManager.load()
+        // No need to call initCheckpointManager again
         return true;
     }
     /**
@@ -270,7 +304,8 @@ export class Agent {
             return false;
         }
         this.sessionId = session.metadata.id;
-        this.messages = session.messages;
+        // CheckpointManager already restored by SessionManager.load()
+        // No need to call initCheckpointManager again
         return true;
     }
     /**
@@ -302,26 +337,47 @@ export class Agent {
     async saveSession() {
         const current = this.sessionManager.getCurrent();
         if (current) {
-            current.messages = this.messages;
             await this.sessionManager.save(current);
         }
     }
     /**
      * Run a single query through the agent
      */
-    async *run(prompt) {
+    async *run(prompt, signal) {
+        // Check for abort before starting
+        if (signal?.aborted) {
+            yield { type: 'error', error: new Error('Operation cancelled') };
+            return;
+        }
         // Auto-create session if none exists
-        if (!this.sessionId) {
-            await this.startSession();
+        try {
+            if (!this.sessionId) {
+                await this.startSession();
+            }
+            // Load memory if not already loaded
+            if (!this.loadedMemory) {
+                await this.loadMemory();
+            }
         }
-        // Load memory if not already loaded
-        if (!this.loadedMemory) {
-            await this.loadMemory();
+        catch (error) {
+            yield {
+                type: 'error',
+                error: error instanceof Error ? error : new Error(String(error))
+            };
+            return;
         }
         // Add user message
         const userMessage = { role: 'user', content: prompt };
-        this.messages.push(userMessage);
-        await this.sessionManager.addMessage(userMessage);
+        try {
+            await this.sessionManager.addMessage(userMessage, this.getModelInfo());
+        }
+        catch (error) {
+            yield {
+                type: 'error',
+                error: new Error(`Failed to save user message: ${error instanceof Error ? error.message : String(error)}`)
+            };
+            return;
+        }
         let turns = 0;
         const maxTurns = this.config.maxTurns ?? 10;
         while (turns < maxTurns) {
@@ -330,6 +386,9 @@ export class Agent {
             const toolDefs = this.registry.getFilteredDefinitions(this.config.tools);
             // Call LLM
             let response;
+            const processingStartTime = Date.now();
+            // Determine if streaming is enabled
+            const useStreaming = process.env.GEN_STREAM === '1' || this.config.streaming;
             try {
                 // Debug prompt loading (enabled with GENCODE_DEBUG_PROMPTS=1)
                 debugPromptLoading(this.config.model, this.config.provider);
@@ -339,35 +398,179 @@ export class Agent {
                     buildSystemPromptForModel(this.config.model, this.config.cwd ?? process.cwd(), true, // Assume git repo for now
                     this.loadedMemory?.context, this.config.provider // Fallback provider if model lookup fails
                     );
-                response = await this.provider.complete({
-                    model: this.config.model,
-                    messages: this.messages,
-                    tools: toolDefs,
-                    systemPrompt,
-                    maxTokens: 4096,
-                });
+                if (useStreaming) {
+                    // === STREAMING PATH ===
+                    // Build response incrementally from stream chunks
+                    const responseBuilder = {
+                        content: [],
+                        textBuffer: '',
+                        toolCalls: new Map(),
+                        stopReason: 'end_turn',
+                        usage: undefined,
+                        cost: undefined,
+                    };
+                    // Process stream chunks
+                    for await (const chunk of this.provider.stream({
+                        model: this.config.model,
+                        messages: this.sessionManager.getMessagesForLLM(),
+                        tools: toolDefs,
+                        systemPrompt,
+                        maxTokens: 4096,
+                        signal, // Pass abort signal for cancellation
+                    })) {
+                        // Check for abort
+                        if (signal?.aborted) {
+                            yield { type: 'error', error: new Error('Operation cancelled by user') };
+                            return;
+                        }
+                        switch (chunk.type) {
+                            case 'text':
+                                // Accumulate text and yield immediately for real-time display
+                                responseBuilder.textBuffer += chunk.text;
+                                yield { type: 'text', text: chunk.text };
+                                break;
+                            case 'reasoning':
+                                // Forward reasoning content (o1/o3/Gemini 3+ thinking)
+                                yield { type: 'reasoning_delta', text: chunk.text };
+                                break;
+                            case 'tool_start':
+                                // Initialize tool call tracking
+                                responseBuilder.toolCalls.set(chunk.id, {
+                                    id: chunk.id,
+                                    name: chunk.name,
+                                    inputBuffer: '',
+                                });
+                                break;
+                            case 'tool_input':
+                                // Accumulate incremental JSON input and forward delta
+                                const tool = responseBuilder.toolCalls.get(chunk.id);
+                                if (tool) {
+                                    tool.inputBuffer += chunk.input;
+                                    // Emit incremental tool input for progressive display
+                                    yield { type: 'tool_input_delta', id: chunk.id, delta: chunk.input };
+                                }
+                                break;
+                            case 'done':
+                                // Save final metadata
+                                responseBuilder.stopReason = chunk.response.stopReason;
+                                responseBuilder.usage = chunk.response.usage;
+                                responseBuilder.cost = chunk.response.cost;
+                                break;
+                            case 'error':
+                                yield { type: 'error', error: chunk.error };
+                                return;
+                        }
+                    }
+                    // Build complete response from accumulated data
+                    if (responseBuilder.textBuffer) {
+                        responseBuilder.content.push({
+                            type: 'text',
+                            text: responseBuilder.textBuffer,
+                        });
+                    }
+                    for (const [_id, tool] of responseBuilder.toolCalls) {
+                        try {
+                            responseBuilder.content.push({
+                                type: 'tool_use',
+                                id: tool.id,
+                                name: tool.name,
+                                input: JSON.parse(tool.inputBuffer || '{}'),
+                            });
+                        }
+                        catch (error) {
+                            // If JSON parsing fails, treat as malformed tool call
+                            yield {
+                                type: 'error',
+                                error: new Error(`Failed to parse tool input for ${tool.name}: ${error instanceof Error ? error.message : String(error)}`),
+                            };
+                            return;
+                        }
+                    }
+                    response = {
+                        content: responseBuilder.content,
+                        stopReason: responseBuilder.stopReason,
+                        usage: responseBuilder.usage,
+                        cost: responseBuilder.cost,
+                    };
+                }
+                else {
+                    // === TRADITIONAL PATH (COMPLETE) ===
+                    response = await this.provider.complete({
+                        model: this.config.model,
+                        messages: this.sessionManager.getMessagesForLLM(),
+                        tools: toolDefs,
+                        systemPrompt,
+                        maxTokens: 4096,
+                    });
+                }
             }
             catch (error) {
                 yield { type: 'error', error: error };
                 return;
             }
+            // Validate response completeness
+            if (!response || !response.content) {
+                yield {
+                    type: 'error',
+                    error: new Error('Provider returned null or undefined response')
+                };
+                return;
+            }
+            // Validate content is not empty (excluding max_tokens case)
+            if (response.content.length === 0 && response.stopReason !== 'max_tokens') {
+                yield {
+                    type: 'error',
+                    error: new Error(`Provider returned empty content (stopReason: ${response.stopReason}, ` +
+                        `usage: ${JSON.stringify(response.usage)})`)
+                };
+                return;
+            }
             // Process response content
             const toolCalls = [];
             let textContent = '';
             for (const content of response.content) {
                 if (content.type === 'text') {
                     textContent += content.text;
-                    yield { type: 'text', text: content.text };
+                    // Only yield text if not in streaming mode (streaming already yielded chunks)
+                    if (!useStreaming) {
+                        yield { type: 'text', text: content.text };
+                    }
                 }
                 else if (content.type === 'tool_use') {
                     toolCalls.push({ id: content.id, name: content.name, input: content.input });
                 }
             }
             // Add assistant message and check if done
-            this.messages.push({ role: 'assistant', content: response.content });
-            await this.sessionManager.addMessage({ role: 'assistant', content: response.content });
+            try {
+                await this.sessionManager.addMessage({ role: 'assistant', content: response.content }, this.getModelInfo());
+            }
+            catch (error) {
+                yield {
+                    type: 'error',
+                    error: new Error(`Failed to save assistant message: ${error instanceof Error ? error.message : String(error)}`)
+                };
+                return;
+            }
             if (response.stopReason !== 'tool_use' || toolCalls.length === 0) {
                 yield { type: 'done', text: textContent, usage: response.usage, cost: response.cost };
+                // Save completion metadata for UI restoration
+                if (response.usage || response.cost) {
+                    const current = this.sessionManager.getCurrent();
+                    if (current) {
+                        if (!current.metadata.completions) {
+                            current.metadata.completions = [];
+                        }
+                        current.metadata.completions.push({
+                            afterMessageIndex: current.messages.length - 1,
+                            durationMs: Date.now() - processingStartTime,
+                            usage: response.usage ? {
+                                inputTokens: response.usage.inputTokens,
+                                outputTokens: response.usage.outputTokens,
+                            } : undefined,
+                            cost: response.cost,
+                        });
+                    }
+                }
                 return;
             }
             // Execute tool calls
@@ -380,21 +583,48 @@ export class Agent {
             };
             for (const call of toolCalls) {
                 yield { type: 'tool_start', id: call.id, name: call.name, input: call.input };
-                const allowed = await this.permissions.requestPermission(call.name, call.input);
-                const result = allowed
-                    ? await this.registry.execute(call.name, call.input, toolContext)
-                    : { success: false, output: '', error: 'Permission denied by user' };
-                yield { type: 'tool_result', id: call.id, name: call.name, result };
-                toolResults.push({
-                    type: 'tool_result',
-                    toolUseId: call.id,
-                    content: result.success ? result.output : (result.error ?? 'Unknown error'),
-                    isError: !result.success,
-                });
+                try {
+                    // Protect permission check and tool execution
+                    const allowed = await this.permissions.requestPermission(call.name, call.input);
+                    const result = allowed
+                        ? await this.registry.execute(call.name, call.input, toolContext)
+                        : { success: false, output: '', error: 'Permission denied by user' };
+                    yield { type: 'tool_result', id: call.id, name: call.name, result };
+                    toolResults.push({
+                        type: 'tool_result',
+                        toolUseId: call.id,
+                        content: result.success ? result.output : (result.error ?? 'Unknown error'),
+                        isError: !result.success,
+                    });
+                }
+                catch (error) {
+                    // Catch permission check or tool execution errors
+                    const errorMsg = error instanceof Error ? error.message : String(error);
+                    const errorResult = {
+                        success: false,
+                        output: '',
+                        error: `Tool execution error: ${errorMsg}`
+                    };
+                    yield { type: 'tool_result', id: call.id, name: call.name, result: errorResult };
+                    toolResults.push({
+                        type: 'tool_result',
+                        toolUseId: call.id,
+                        content: errorMsg,
+                        isError: true,
+                    });
+                }
             }
             // Add tool results as user message
-            this.messages.push({ role: 'user', content: toolResults });
-            await this.sessionManager.addMessage({ role: 'user', content: toolResults });
+            try {
+                await this.sessionManager.addMessage({ role: 'user', content: toolResults }, this.getModelInfo());
+            }
+            catch (error) {
+                yield {
+                    type: 'error',
+                    error: new Error(`Failed to save tool results: ${error instanceof Error ? error.message : String(error)}`)
+                };
+                return;
+            }
         }
         yield { type: 'error', error: new Error(`Max turns (${maxTurns}) exceeded`) };
     }
@@ -402,7 +632,6 @@ export class Agent {
      * Clear conversation history
      */
     clearHistory() {
-        this.messages = [];
         this.sessionManager.clearMessages();
     }
     /**
@@ -410,21 +639,16 @@ export class Agent {
      * Removes the last assistant message if it contains tool_use without corresponding tool_result
      */
     cleanupIncompleteMessages() {
-        if (this.messages.length === 0)
+        const messages = this.sessionManager.getMessages();
+        if (messages.length === 0)
             return;
-        const lastMessage = this.messages[this.messages.length - 1];
+        const lastMessage = messages[messages.length - 1];
         // Check if last message is an assistant message with tool_use
         if (lastMessage.role === 'assistant' && Array.isArray(lastMessage.content)) {
             const hasToolUse = lastMessage.content.some((c) => c.type === 'tool_use');
             if (hasToolUse) {
-                // Remove the incomplete assistant message
-                this.messages.pop();
-                // Also remove from session manager
-                // Note: SessionManager should have corresponding cleanup method
-                const messages = this.sessionManager.getMessages();
-                if (messages.length > 0 && messages[messages.length - 1].role === 'assistant') {
-                    this.sessionManager.removeLastMessage();
-                }
+                // Remove the incomplete assistant message from session manager
+                this.sessionManager.removeLastMessage();
             }
         }
     }
@@ -432,7 +656,13 @@ export class Agent {
      * Get conversation history
      */
     getHistory() {
-        return [...this.messages];
+        return this.sessionManager.getMessages();
+    }
+    /**
+     * Get compression statistics
+     */
+    getCompressionStats() {
+        return this.sessionManager.getCompressionStats();
     }
 }
 //# sourceMappingURL=agent.js.map