codecane 1.0.177 → 1.0.179

package/dist/chat-storage.d.ts

@@ -1,27 +1,2 @@
 import { Message } from './common/types/message';
-interface Chat {
-    id: string;
-    messages: Message[];
-    fileVersions: FileVersion[];
-    createdAt: string;
-    updatedAt: string;
-    timestamp: number;
-}
-interface FileVersion {
-    files: Record<string, string>;
-}
-export declare class ChatStorage {
-    private currentChat;
-    private currentVersionIndex;
-    constructor();
-    getCurrentChat(): Chat;
-    addMessage(chat: Chat, message: Message): void;
-    private saveMessagesToFile;
-    getCurrentVersion(): FileVersion | null;
-    navigateVersion(direction: 'undo' | 'redo'): boolean;
-    saveFilesChanged(filesChanged: string[]): string[];
-    saveCurrentFileState(files: Record<string, string>): void;
-    addNewFileState(files: Record<string, string>): void;
-    private createChat;
-}
-export {};
+export declare function setMessages(messages: Message[]): void;

package/dist/chat-storage.js

@@ -23,147 +23,71 @@ var __importStar = (this && this.__importStar) || function (mod) {
     return result;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ChatStorage = void 0;
+exports.setMessages = setMessages;
 const path = __importStar(require("path"));
 const fs = __importStar(require("fs"));
 const project_files_1 = require("./project-files");
 const string_1 = require("./common/util/string");
 const ts_pattern_1 = require("ts-pattern");
-class ChatStorage {
-    currentChat;
-    currentVersionIndex;
-    constructor() {
-        this.currentChat = this.createChat();
-        this.currentVersionIndex = -1;
-    }
-    getCurrentChat() {
-        return this.currentChat;
-    }
-    addMessage(chat, message) {
-        // Before adding new message, clean up any screenshots and logs in previous messages
-        // Skip the last message as it may not have been processed by the backend yet
-        const lastIndex = chat.messages.length - 1;
-        chat.messages = chat.messages.map((msg, index) => {
-            if (index === lastIndex) {
-                return msg; // Preserve the most recent message in its entirety
-            }
-            // Helper function to clean up message content
-            const cleanContent = (content) => {
-                // Keep only tool logs
-                content = (0, string_1.transformJsonInString)(content, 'logs', (logs) => logs.filter((log) => log.source === 'tool'), '(LOGS_REMOVED)');
-                // Remove metrics
-                content = (0, string_1.transformJsonInString)(content, 'metrics', () => '(METRICS_REMOVED)', '(METRICS_REMOVED)');
-                return content;
-            };
-            // Clean up message content
-            if (!msg.content)
-                return msg;
-            return (0, ts_pattern_1.match)(msg)
-                .with({ content: ts_pattern_1.P.array() }, (message) => ({
-                ...message,
-                content: message.content.reduce((acc, contentObj) => [
-                    ...acc,
-                    ...(0, ts_pattern_1.match)(contentObj)
-                        .with({ type: 'tool_result', content: ts_pattern_1.P.string }, (obj) => [
-                        {
-                            ...obj,
-                            content: cleanContent(obj.content),
-                        },
-                    ])
-                        .with({ type: 'text', text: ts_pattern_1.P.string }, (obj) => [
-                        {
-                            ...obj,
-                            text: cleanContent(obj.text),
-                        },
-                    ])
-                        .otherwise((obj) => [obj]),
-                ], []),
-            }))
-                .with({ content: ts_pattern_1.P.string }, (message) => ({
-                ...message,
-                content: cleanContent(message.content),
-            }))
-                .otherwise((message) => message);
-        });
-        // Add the new message
-        chat.messages.push(message);
-        chat.updatedAt = new Date().toISOString();
-        // Save messages to chat directory
-        this.saveMessagesToFile(chat);
-    }
-    saveMessagesToFile(chat) {
-        try {
-            const chatDir = (0, project_files_1.getCurrentChatDir)();
-            const messagesPath = path.join(chatDir, 'messages.json');
-            const messagesData = {
-                id: chat.id,
-                messages: chat.messages,
-                updatedAt: chat.updatedAt,
-            };
-            fs.writeFileSync(messagesPath, JSON.stringify(messagesData, null, 2));
-        }
-        catch (error) {
-            console.error('Failed to save messages to file:', error);
-        }
-    }
-    getCurrentVersion() {
-        if (this.currentVersionIndex >= 0 &&
-            this.currentVersionIndex < this.currentChat.fileVersions.length) {
-            return this.currentChat.fileVersions[this.currentVersionIndex];
-        }
-        return null;
-    }
-    navigateVersion(direction) {
-        if (direction === 'undo' && this.currentVersionIndex >= 0) {
-            this.currentVersionIndex--;
-            return true;
-        }
-        else if (direction === 'redo' &&
-            this.currentVersionIndex < this.currentChat.fileVersions.length - 1) {
-            this.currentVersionIndex++;
-            return true;
+function setMessages(messages) {
+    // Clean up any screenshots and logs in previous messages
+    // Skip the last message as it may not have been processed by the backend yet
+    const lastIndex = messages.length - 1;
+    const cleanedMessages = messages.map((msg, index) => {
+        if (index === lastIndex) {
+            return msg; // Preserve the most recent message in its entirety
         }
-        return false;
-    }
-    saveFilesChanged(filesChanged) {
-        let currentVersion = this.getCurrentVersion();
-        if (!currentVersion) {
-            this.addNewFileState({});
-            currentVersion = this.getCurrentVersion();
-        }
-        const newFilesChanged = filesChanged.filter((f) => !currentVersion.files[f]);
-        const updatedFiles = (0, project_files_1.getExistingFiles)(newFilesChanged);
-        currentVersion.files = { ...currentVersion.files, ...updatedFiles };
-        return Object.keys(currentVersion.files);
-    }
-    saveCurrentFileState(files) {
-        const currentVersion = this.getCurrentVersion();
-        if (currentVersion) {
-            currentVersion.files = files;
-        }
-        else {
-            this.addNewFileState(files);
-        }
-    }
-    addNewFileState(files) {
-        const newVersion = {
-            files,
+        // Helper function to clean up message content
+        const cleanContent = (content) => {
+            // Keep only tool logs
+            content = (0, string_1.transformJsonInString)(content, 'logs', (logs) => logs.filter((log) => log.source === 'tool'), '(LOGS_REMOVED)');
+            // Remove metrics
+            content = (0, string_1.transformJsonInString)(content, 'metrics', () => '(METRICS_REMOVED)', '(METRICS_REMOVED)');
+            return content;
         };
-        this.currentChat.fileVersions.push(newVersion);
-        this.currentVersionIndex = this.currentChat.fileVersions.length - 1;
-    }
-    createChat(messages = []) {
-        const now = new Date();
-        const chat = {
+        // Clean up message content
+        if (!msg.content)
+            return msg;
+        return (0, ts_pattern_1.match)(msg)
+            .with({ content: ts_pattern_1.P.array() }, (message) => ({
+            ...message,
+            content: message.content.reduce((acc, contentObj) => [
+                ...acc,
+                ...(0, ts_pattern_1.match)(contentObj)
+                    .with({ type: 'tool_result', content: ts_pattern_1.P.string }, (obj) => [
+                    {
+                        ...obj,
+                        content: cleanContent(obj.content),
+                    },
+                ])
+                    .with({ type: 'text', text: ts_pattern_1.P.string }, (obj) => [
+                    {
+                        ...obj,
+                        text: cleanContent(obj.text),
+                    },
+                ])
+                    .otherwise((obj) => [obj]),
+            ], []),
+        }))
+            .with({ content: ts_pattern_1.P.string }, (message) => ({
+            ...message,
+            content: cleanContent(message.content),
+        }))
+            .otherwise((message) => message);
+    });
+    // Save messages to chat directory
+    try {
+        const chatDir = (0, project_files_1.getCurrentChatDir)();
+        const messagesPath = path.join(chatDir, 'messages.json');
+        const messagesData = {
             id: project_files_1.currentChatId,
-            messages,
-            fileVersions: [],
-            createdAt: now.toISOString(),
-            updatedAt: now.toISOString(),
-            timestamp: now.getTime(),
+            messages: cleanedMessages,
+            updatedAt: new Date().toISOString(),
         };
-        return chat;
+        fs.writeFileSync(messagesPath, JSON.stringify(messagesData, null, 2));
+    }
+    catch (error) {
+        console.error('Failed to save messages to file:', error);
     }
 }
-exports.ChatStorage = ChatStorage;
 //# sourceMappingURL=chat-storage.js.map

package/dist/chat-storage.js.map

@@ -1 +1 @@
-{"version":3,"file":"chat-storage.js","sourceRoot":"","sources":["../src/chat-storage.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,2CAA4B;AAC5B,uCAAwB;AAExB,mDAIwB;AACxB,+CAA0D;AAE1D,2CAAqC;AAerC,MAAa,WAAW;IACd,WAAW,CAAM;IACjB,mBAAmB,CAAQ;IAEnC;QACE,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,UAAU,EAAE,CAAA;QACpC,IAAI,CAAC,mBAAmB,GAAG,CAAC,CAAC,CAAA;IAC/B,CAAC;IAED,cAAc;QACZ,OAAO,IAAI,CAAC,WAAW,CAAA;IACzB,CAAC;IAED,UAAU,CAAC,IAAU,EAAE,OAAgB;QACrC,oFAAoF;QACpF,6EAA6E;QAC7E,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAA;QAC1C,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE;YAC/C,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACxB,OAAO,GAAG,CAAA,CAAC,mDAAmD;YAChE,CAAC;YAED,8CAA8C;YAC9C,MAAM,YAAY,GAAG,CAAC,OAAe,EAAE,EAAE;gBACvC,sBAAsB;gBACtB,OAAO,GAAG,IAAA,8BAAqB,EAC7B,OAAO,EACP,MAAM,EACN,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,KAAK,MAAM,CAAC,EACrD,gBAAgB,CACjB,CAAA;gBAED,iBAAiB;gBACjB,OAAO,GAAG,IAAA,8BAAqB,EAC7B,OAAO,EACP,SAAS,EACT,GAAG,EAAE,CAAC,mBAAmB,EACzB,mBAAmB,CACpB,CAAA;gBAED,OAAO,OAAO,CAAA;YAChB,CAAC,CAAA;YAED,2BAA2B;YAC3B,IAAI,CAAC,GAAG,CAAC,OAAO;gBAAE,OAAO,GAAG,CAAA;YAE5B,OAAO,IAAA,kBAAK,EAAC,GAAG,CAAC;iBACd,IAAI,CAAC,EAAE,OAAO,EAAE,cAAC,CAAC,KAAK,EAAE,EAAE,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;gBAC1C,GAAG,OAAO;gBACV,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,MAAM,CAC7B,CAAC,GAAG,EAAE,UAAU,EAAE,EAAE,CAAC;oBACnB,GAAG,GAAG;oBACN,GAAG,IAAA,kBAAK,EAAC,UAAU,CAAC;yBACjB,IAAI,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,cAAC,CAAC,MAAM,EAAE,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC;wBACzD;4BACE,GAAG,GAAG;4BACN,OAAO,EAAE,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC;yBACnC;qBACF,CAAC;yBACD,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,cAAC,CAAC,MAAM,EAAE,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC;wBAC/C;4BACE,GAAG,GAAG;4BACN,IAAI,EAAE,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC;yBAC7B;qBACF,CAAC;yBACD,SAAS,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC;iBAC7B,EACD,EAAE,CACH;aACF,CAAC,CAAC;iBACF,IAAI,CAAC,EAAE,OAAO,EAAE,cAAC,CAAC,MAAM,EAAE,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;gBACzC,GAAG,OAAO;gBACV,OAAO,EAAE,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC;aACvC,CAAC,CAAC;iBACF,SAAS,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,CAAA;QACpC,CAAC,CAAC,CAAA;QAEF,sBAAsB;QACtB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QAC3B,IAAI,CAAC,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAA;QAEzC,kCAAkC;QAClC,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAA;IAC/B,CAAC;IAEO,kBAAkB,CAAC,IAAU;QACnC,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAA,iCAAiB,GAAE,CAAA;YACnC,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,eAAe,CAAC,CAAA;YAExD,MAAM,YAAY,GAAG;gBACnB,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,SAAS,EAAE,IAAI,CAAC,SAAS;aAC1B,CAAA;YAED,EAAE,CAAC,aAAa,CAAC,YAAY,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;QACvE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,KAAK,CAAC,CAAA;QAC1D,CAAC;IACH,CAAC;IAED,iBAAiB;QACf,IACE,IAAI,CAAC,mBAAmB,IAAI,CAAC;YAC7B,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,MAAM,EAC/D,CAAC;YACD,OAAO,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAA;QAChE,CAAC;QACD,OAAO,IAAI,CAAA;IACb,CAAC;IAED,eAAe,CAAC,SAA0B;QACxC,IAAI,SAAS,KAAK,MAAM,IAAI,IAAI,CAAC,mBAAmB,IAAI,CAAC,EAAE,CAAC;YAC1D,IAAI,CAAC,mBAAmB,EAAE,CAAA;YAC1B,OAAO,IAAI,CAAA;QACb,CAAC;aAAM,IACL,SAAS,KAAK,MAAM;YACpB,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,EACnE,CAAC;YACD,IAAI,CAAC,mBAAmB,EAAE,CAAA;YAC1B,OAAO,IAAI,CAAA;QACb,CAAC;QACD,OAAO,KAAK,CAAA;IACd,CAAC;IAED,gBAAgB,CAAC,YAAsB;QACrC,IAAI,cAAc,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAA;QAC7C,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC,CAAA;YACxB,cAAc,GAAG,IAAI,CAAC,iBAAiB,EAAiB,CAAA;QAC1D,CAAC;QACD,MAAM,eAAe,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAA;QAC5E,MAAM,YAAY,GAAG,IAAA,gCAAgB,EAAC,eAAe,CAAC,CAAA;QACtD,cAAc,CAAC,KAAK,GAAG,EAAE,GAAG,cAAc,CAAC,KAAK,EAAE,GAAG,YAAY,EAAE,CAAA;QACnE,OAAO,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAA;IAC1C,CAAC;IAED,oBAAoB,CAAC,KAA6B;QAChD,MAAM,cAAc,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAA;QAC/C,IAAI,cAAc,EAAE,CAAC;YACnB,cAAc,CAAC,KAAK,GAAG,KAAK,CAAA;QAC9B,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAA;QAC7B,CAAC;IACH,CAAC;IAED,eAAe,CAAC,KAA6B;QAC3C,MAAM,UAAU,GAAgB;YAC9B,KAAK;SACN,CAAA;QACD,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QAC9C,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAA;IACrE,CAAC;IAEO,UAAU,CAAC,WAAsB,EAAE;QACzC,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAA;QACtB,MAAM,IAAI,GAAS;YACjB,EAAE,EAAE,6BAAa;YACjB,QAAQ;YACR,YAAY,EAAE,EAAE;YAChB,SAAS,EAAE,GAAG,CAAC,WAAW,EAAE;YAC5B,SAAS,EAAE,GAAG,CAAC,WAAW,EAAE;YAC5B,SAAS,EAAE,GAAG,CAAC,OAAO,EAAE;SACzB,CAAA;QACD,OAAO,IAAI,CAAA;IACb,CAAC;CACF;AAvKD,kCAuKC"}
+{"version":3,"file":"chat-storage.js","sourceRoot":"","sources":["../src/chat-storage.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;AAQA,kCA+EC;AAvFD,2CAA4B;AAC5B,uCAAwB;AAExB,mDAAkE;AAClE,+CAA0D;AAE1D,2CAAqC;AAErC,SAAgB,WAAW,CAAC,QAAmB;IAC7C,yDAAyD;IACzD,6EAA6E;IAC7E,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAA;IACrC,MAAM,eAAe,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE;QAClD,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,OAAO,GAAG,CAAA,CAAC,mDAAmD;QAChE,CAAC;QAED,8CAA8C;QAC9C,MAAM,YAAY,GAAG,CAAC,OAAe,EAAE,EAAE;YACvC,sBAAsB;YACtB,OAAO,GAAG,IAAA,8BAAqB,EAC7B,OAAO,EACP,MAAM,EACN,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,KAAK,MAAM,CAAC,EACrD,gBAAgB,CACjB,CAAA;YAED,iBAAiB;YACjB,OAAO,GAAG,IAAA,8BAAqB,EAC7B,OAAO,EACP,SAAS,EACT,GAAG,EAAE,CAAC,mBAAmB,EACzB,mBAAmB,CACpB,CAAA;YAED,OAAO,OAAO,CAAA;QAChB,CAAC,CAAA;QAED,2BAA2B;QAC3B,IAAI,CAAC,GAAG,CAAC,OAAO;YAAE,OAAO,GAAG,CAAA;QAE5B,OAAO,IAAA,kBAAK,EAAC,GAAG,CAAC;aACd,IAAI,CAAC,EAAE,OAAO,EAAE,cAAC,CAAC,KAAK,EAAE,EAAE,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YAC1C,GAAG,OAAO;YACV,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,MAAM,CAC7B,CAAC,GAAG,EAAE,UAAU,EAAE,EAAE,CAAC;gBACnB,GAAG,GAAG;gBACN,GAAG,IAAA,kBAAK,EAAC,UAAU,CAAC;qBACjB,IAAI,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,cAAC,CAAC,MAAM,EAAE,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC;oBACzD;wBACE,GAAG,GAAG;wBACN,OAAO,EAAE,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC;qBACnC;iBACF,CAAC;qBACD,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,cAAC,CAAC,MAAM,EAAE,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC;oBAC/C;wBACE,GAAG,GAAG;wBACN,IAAI,EAAE,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC;qBAC7B;iBACF,CAAC;qBACD,SAAS,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC;aAC7B,EACD,EAAE,CACH;SACF,CAAC,CAAC;aACF,IAAI,CAAC,EAAE,OAAO,EAAE,cAAC,CAAC,MAAM,EAAE,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YACzC,GAAG,OAAO;YACV,OAAO,EAAE,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC;SACvC,CAAC,CAAC;aACF,SAAS,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,CAAA;IACpC,CAAC,CAAC,CAAA;IAEF,kCAAkC;IAClC,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,IAAA,iCAAiB,GAAE,CAAA;QACnC,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,eAAe,CAAC,CAAA;QAExD,MAAM,YAAY,GAAG;YACnB,EAAE,EAAE,6BAAa;YACjB,QAAQ,EAAE,eAAe;YACzB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAA;QAED,EAAE,CAAC,aAAa,CAAC,YAAY,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;IACvE,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,KAAK,CAAC,CAAA;IAC1D,CAAC;AACH,CAAC"}

package/dist/checkpoint-file-manager.d.ts

@@ -0,0 +1,8 @@
+export declare function initializeCheckpointFileManager(): Promise<void>;
+/**
+ * Stores the current state of all files in the project as a git commit
+ * @param message The commit message to use for this file state
+ * @returns A promise that resolves to the id hash that can be used to restore this file state
+ */
+export declare function storeFileState(message: string): Promise<string>;
+export declare function checkoutFileState(fileStateId: string): Promise<void>;

package/dist/checkpoint-file-manager.js

@@ -0,0 +1,89 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.initializeCheckpointFileManager = initializeCheckpointFileManager;
+exports.storeFileState = storeFileState;
+exports.checkoutFileState = checkoutFileState;
+const crypto = __importStar(require("crypto"));
+const fs = __importStar(require("fs"));
+const git = __importStar(require("isomorphic-git"));
+const path = __importStar(require("path"));
+const project_files_1 = require("./project-files");
+let projectDir;
+let bareRepoPath;
+async function initializeCheckpointFileManager() {
+    projectDir = (0, project_files_1.getProjectRoot)();
+    const bareRepoName = crypto
+        .createHash('sha256')
+        .update(projectDir)
+        .digest('hex');
+    bareRepoPath = path.join((0, project_files_1.getProjectDataDir)(), bareRepoName);
+    // Create the bare repo directory if it doesn't exist
+    fs.mkdirSync(bareRepoPath, { recursive: true });
+    try {
+        // Check if it's already a valid Git repo
+        await git.resolveRef({ fs, dir: bareRepoPath, ref: 'HEAD' });
+        return; // Exit if the repository exists
+    }
+    catch (error) {
+        // Bare repo doesn't exist yet
+    }
+    // Initialize a bare repository
+    await git.init({ fs, dir: bareRepoPath, bare: true });
+    // Commit the files in the bare repo
+    await storeFileState('Initial Commit');
+}
+/**
+ * Stores the current state of all files in the project as a git commit
+ * @param message The commit message to use for this file state
+ * @returns A promise that resolves to the id hash that can be used to restore this file state
+ */
+async function storeFileState(message) {
+    await git.add({
+        fs,
+        dir: projectDir,
+        gitdir: bareRepoPath,
+        filepath: '.',
+    });
+    const commitHash = await git.commit({
+        fs,
+        dir: projectDir,
+        gitdir: bareRepoPath,
+        author: { name: 'codebuff' },
+        message,
+    });
+    return commitHash;
+}
+async function checkoutFileState(fileStateId) {
+    // Checkout the given hash
+    await git.checkout({
+        fs,
+        dir: projectDir,
+        gitdir: bareRepoPath,
+        ref: fileStateId,
+        force: true,
+    });
+}
+//# sourceMappingURL=checkpoint-file-manager.js.map

package/dist/checkpoint-file-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"checkpoint-file-manager.js","sourceRoot":"","sources":["../src/checkpoint-file-manager.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;AAUA,0EAyBC;AAOD,wCAiBC;AAED,8CASC;AAtED,+CAAgC;AAChC,uCAAwB;AACxB,oDAAqC;AACrC,2CAA4B;AAE5B,mDAAmE;AAEnE,IAAI,UAAkB,CAAA;AACtB,IAAI,YAAoB,CAAA;AAEjB,KAAK,UAAU,+BAA+B;IACnD,UAAU,GAAG,IAAA,8BAAc,GAAE,CAAA;IAE7B,MAAM,YAAY,GAAG,MAAM;SACxB,UAAU,CAAC,QAAQ,CAAC;SACpB,MAAM,CAAC,UAAU,CAAC;SAClB,MAAM,CAAC,KAAK,CAAC,CAAA;IAChB,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,IAAA,iCAAiB,GAAE,EAAE,YAAY,CAAC,CAAA;IAE3D,qDAAqD;IACrD,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAA;IAE/C,IAAI,CAAC;QACH,yCAAyC;QACzC,MAAM,GAAG,CAAC,UAAU,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,YAAY,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAA;QAC5D,OAAM,CAAC,gCAAgC;IACzC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,8BAA8B;IAChC,CAAC;IAED,+BAA+B;IAC/B,MAAM,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,YAAY,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;IAErD,oCAAoC;IACpC,MAAM,cAAc,CAAC,gBAAgB,CAAC,CAAA;AACxC,CAAC;AAED;;;;GAIG;AACI,KAAK,UAAU,cAAc,CAAC,OAAe;IAClD,MAAM,GAAG,CAAC,GAAG,CAAC;QACZ,EAAE;QACF,GAAG,EAAE,UAAU;QACf,MAAM,EAAE,YAAY;QACpB,QAAQ,EAAE,GAAG;KACd,CAAC,CAAA;IAEF,MAAM,UAAU,GAAG,MAAM,GAAG,CAAC,MAAM,CAAC;QAClC,EAAE;QACF,GAAG,EAAE,UAAU;QACf,MAAM,EAAE,YAAY;QACpB,MAAM,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE;QAC5B,OAAO;KACR,CAAC,CAAA;IAEF,OAAO,UAAU,CAAA;AACnB,CAAC;AAEM,KAAK,UAAU,iBAAiB,CAAC,WAAmB;IACzD,0BAA0B;IAC1B,MAAM,GAAG,CAAC,QAAQ,CAAC;QACjB,EAAE;QACF,GAAG,EAAE,UAAU;QACf,MAAM,EAAE,YAAY;QACpB,GAAG,EAAE,WAAW;QAChB,KAAK,EAAE,IAAI;KACZ,CAAC,CAAA;AACJ,CAAC"}

package/dist/checkpoints.d.ts

@@ -4,7 +4,7 @@ import { AgentState } from './common/types/agent-state';
  */
 export interface Checkpoint {
     agentStateString: string;
-    fileVersions: Record<string, string>;
+    fileStateIdPromise: Promise<string>;
     historyLength: number;
     id: number;
     timestamp: number;
@@ -15,18 +15,14 @@ export interface Checkpoint {
  */
 export declare class CheckpointManager {
     private checkpoints;
-    private maxCheckpoints;
     private nextId;
-    private oldestId;
-    constructor(maxCheckpoints?: number);
     /**
      * Add a new checkpoint
-     * TODO update this jsdoc and add comments to this function
      * @param agentState - The agent state to checkpoint
      * @param userInput - The user input associated with this checkpoint
      * @returns The ID of the created checkpoint
      */
-    addCheckpoint(agentState: AgentState, userInput: string, parentId?: number | null): number;
+    addCheckpoint(agentState: AgentState, userInput: string): Promise<number>;
     /**
      * Get a checkpoint by ID
      * @param id The checkpoint ID
@@ -43,6 +39,7 @@ export declare class CheckpointManager {
      * @returns The most recent checkpoint or null if none exist
      */
     getLatestCheckpoint(): Checkpoint | null;
+    restoreFileState(id: number): Promise<boolean>;
     /**
      * Clear all checkpoints
      */

package/dist/checkpoints.js

@@ -1,38 +1,50 @@
 "use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.checkpointManager = exports.CheckpointManager = void 0;
 const picocolors_1 = require("picocolors");
+const checkpointFileManager = __importStar(require("./checkpoint-file-manager"));
 /**
  * Simple in-memory checkpoint manager for agent states
  */
 class CheckpointManager {
     checkpoints = new Map();
-    maxCheckpoints;
     nextId = 1;
-    oldestId = 1;
-    constructor(maxCheckpoints = 100) {
-        this.maxCheckpoints = maxCheckpoints;
-    }
     /**
      * Add a new checkpoint
-     * TODO update this jsdoc and add comments to this function
      * @param agentState - The agent state to checkpoint
      * @param userInput - The user input associated with this checkpoint
      * @returns The ID of the created checkpoint
      */
-    addCheckpoint(agentState, userInput, parentId = null) {
+    async addCheckpoint(agentState, userInput) {
         // Use incremental ID starting at 1
         const id = this.nextId++;
-        const parentFileVersions = this.getCheckpoint(parentId)?.fileVersions || {};
-        const currentFileVersions = agentState.fileContext.fileVersions
-            .flat()
-            .reduce((acc, { path, content }) => {
-            acc[path] = content;
-            return acc;
-        }, {});
+        const fileStateIdPromise = checkpointFileManager.storeFileState(`Checkpoint ${id}`);
         const checkpoint = {
             agentStateString: JSON.stringify(agentState), // Deep clone to prevent reference issues
-            fileVersions: { ...parentFileVersions, ...currentFileVersions },
+            fileStateIdPromise,
             historyLength: agentState.messageHistory.length,
             id,
             timestamp: Date.now(),
@@ -40,11 +52,6 @@ class CheckpointManager {
         };
         // Add to map
         this.checkpoints.set(id, checkpoint);
-        // If we exceed the maximum number of checkpoints, remove the oldest one
-        if (this.checkpoints.size > this.maxCheckpoints) {
-            this.checkpoints.delete(this.oldestId);
-            this.oldestId++;
-        }
         return id;
     }
     /**
@@ -73,13 +80,20 @@ class CheckpointManager {
     getLatestCheckpoint() {
         return this.checkpoints.get(this.nextId - 1) || null;
     }
+    async restoreFileState(id) {
+        const checkpoint = this.getCheckpoint(id);
+        if (!checkpoint) {
+            return false;
+        }
+        checkpointFileManager.checkoutFileState(await checkpoint.fileStateIdPromise);
+        return true;
+    }
     /**
      * Clear all checkpoints
      */
     clearCheckpoints() {
         this.checkpoints.clear();
         this.nextId = 1; // Reset the ID counter when clearing
-        this.oldestId = 1;
     }
     /**
      * Get a formatted string representation of all checkpoints

package/dist/checkpoints.js.map

@@ -1 +1 @@
-{"version":3,"file":"checkpoints.js","sourceRoot":"","sources":["../src/checkpoints.ts"],"names":[],"mappings":";;;AAAA,2CAAsE;AAgBtE;;GAEG;AACH,MAAa,iBAAiB;IACpB,WAAW,GAA4B,IAAI,GAAG,EAAE,CAAA;IAChD,cAAc,CAAQ;IACtB,MAAM,GAAW,CAAC,CAAA;IAClB,QAAQ,GAAW,CAAC,CAAA;IAE5B,YAAY,iBAAyB,GAAG;QACtC,IAAI,CAAC,cAAc,GAAG,cAAc,CAAA;IACtC,CAAC;IAED;;;;;;OAMG;IACH,aAAa,CACX,UAAsB,EACtB,SAAiB,EACjB,WAA0B,IAAI;QAE9B,mCAAmC;QACnC,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAA;QACxB,MAAM,kBAAkB,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,EAAE,YAAY,IAAI,EAAE,CAAA;QAE3E,MAAM,mBAAmB,GAAG,UAAU,CAAC,WAAW,CAAC,YAAY;aAC5D,IAAI,EAAE;aACN,MAAM,CACL,CAAC,GAAG,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE;YACzB,GAAG,CAAC,IAAI,CAAC,GAAG,OAAO,CAAA;YACnB,OAAO,GAAG,CAAA;QACZ,CAAC,EACD,EAA4B,CAC7B,CAAA;QAEH,MAAM,UAAU,GAAe;YAC7B,gBAAgB,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,EAAE,yCAAyC;YACvF,YAAY,EAAE,EAAE,GAAG,kBAAkB,EAAE,GAAG,mBAAmB,EAAE;YAC/D,aAAa,EAAE,UAAU,CAAC,cAAc,CAAC,MAAM;YAC/C,EAAE;YACF,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,SAAS;SACV,CAAA;QAED,aAAa;QACb,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,EAAE,UAAU,CAAC,CAAA;QAEpC,wEAAwE;QACxE,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;YAChD,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;YACtC,IAAI,CAAC,QAAQ,EAAE,CAAA;QACjB,CAAC;QAED,OAAO,EAAE,CAAA;IACX,CAAC;IAED;;;;OAIG;IACH,aAAa,CAAC,EAAiB;QAC7B,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC;YAChB,OAAO,IAAI,CAAA;QACb,CAAC;QACD,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;QAC3C,OAAO,UAAU,IAAI,IAAI,CAAA;IAC3B,CAAC;IAED;;;OAGG;IACH,iBAAiB;QACf,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAA;IAC9C,CAAC;IAED;;;OAGG;IACH,mBAAmB;QACjB,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,IAAI,CAAA;IACtD,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAA;QACxB,IAAI,CAAC,MAAM,GAAG,CAAC,CAAA,CAAC,qCAAqC;QACrD,IAAI,CAAC,QAAQ,GAAG,CAAC,CAAA;IACnB,CAAC;IAED;;;;OAIG;IACH,sBAAsB,CAAC,WAAoB,KAAK;QAC9C,MAAM,WAAW,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC,CAAA;QAExE,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,OAAO,IAAA,mBAAM,EAAC,2BAA2B,CAAC,CAAA;QAC5C,CAAC;QAED,MAAM,KAAK,GAAa,CAAC,IAAA,iBAAI,EAAC,IAAA,sBAAS,EAAC,0BAA0B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;QAEzE,WAAW,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE;YACjC,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAA;YAC3C,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,EAAE,CAAA;YAE3C,MAAM,gBAAgB,GAAG,UAAU,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC,CAAA;YACnE,MAAM,SAAS,GACb,gBAAgB,CAAC,MAAM,GAAG,EAAE;gBAC1B,CAAC,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK;gBAC3C,CAAC,CAAC,gBAAgB,CAAA;YAEtB,KAAK,CAAC,IAAI,CACR,GAAG,IAAA,iBAAI,EAAC,IAAA,iBAAI,EAAC,IAAI,UAAU,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,IAAA,iBAAI,EAAC,IAAI,aAAa,GAAG,CAAC,GAAG,CACpE,CAAA;YAED,KAAK,CAAC,IAAI,CAAC,KAAK,IAAA,iBAAI,EAAC,OAAO,CAAC,KAAK,SAAS,EAAE,CAAC,CAAA;YAE9C,IAAI,QAAQ,EAAE,CAAC;gBACb,mDAAmD;gBACnD,MAAM,YAAY,GAAG,UAAU,CAAC,aAAa,CAAA;gBAC7C,KAAK,CAAC,IAAI,CAAC,KAAK,IAAA,iBAAI,EAAC,UAAU,CAAC,KAAK,YAAY,EAAE,CAAC,CAAA;gBAEpD,uDAAuD;gBACvD,8CAA8C;YAChD,CAAC;YAED,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA,CAAC,iCAAiC;QAClD,CAAC,CAAC,CAAA;QAEF,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACzB,CAAC;IAED;;;;OAIG;IACH,oBAAoB,CAAC,EAAU;QAC7B,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,CAAA;QACzC,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO,IAAA,iBAAI,EAAC,iBAAiB,EAAE,aAAa,CAAC,CAAA;QAC/C,CAAC;QAED,MAAM,KAAK,GAAa;YACtB,IAAA,iBAAI,EAAC,wCAAwC,EAAE,GAAG,CAAC;SACpD,CAAA;QAED,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAA;QAC3C,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,EAAE,CAAA;QAC3C,KAAK,CAAC,IAAI,CAAC,GAAG,IAAA,iBAAI,EAAC,YAAY,CAAC,KAAK,aAAa,EAAE,CAAC,CAAA;QAErD,IAAI,UAAU,CAAC,SAAS,EAAE,CAAC;YACzB,KAAK,CAAC,IAAI,CAAC,GAAG,IAAA,iBAAI,EAAC,YAAY,CAAC,KAAK,UAAU,CAAC,SAAS,EAAE,CAAC,CAAA;QAC9D,CAAC;QAED,0DAA0D;QAC1D,MAAM,YAAY,GAAG,UAAU,CAAC,aAAa,CAAA;QAC7C,KAAK,CAAC,IAAI,CAAC,GAAG,IAAA,iBAAI,EAAC,iBAAiB,CAAC,KAAK,YAAY,WAAW,CAAC,CAAA;QAElE,yDAAyD;QAEzD,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACzB,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAA;IACpB,CAAC;CACF;AA/KD,8CA+KC;AAED,iEAAiE;AACpD,QAAA,iBAAiB,GAAG,IAAI,iBAAiB,EAAE,CAAA"}
+{"version":3,"file":"checkpoints.js","sourceRoot":"","sources":["../src/checkpoints.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,2CASmB;AAGnB,iFAAkE;AAclE;;GAEG;AACH,MAAa,iBAAiB;IACpB,WAAW,GAA4B,IAAI,GAAG,EAAE,CAAA;IAChD,MAAM,GAAW,CAAC,CAAA;IAE1B;;;;;OAKG;IACH,KAAK,CAAC,aAAa,CACjB,UAAsB,EACtB,SAAiB;QAEjB,mCAAmC;QACnC,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAA;QAExB,MAAM,kBAAkB,GAAG,qBAAqB,CAAC,cAAc,CAC7D,cAAc,EAAE,EAAE,CACnB,CAAA;QAED,MAAM,UAAU,GAAe;YAC7B,gBAAgB,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,EAAE,yCAAyC;YACvF,kBAAkB;YAClB,aAAa,EAAE,UAAU,CAAC,cAAc,CAAC,MAAM;YAC/C,EAAE;YACF,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,SAAS;SACV,CAAA;QAED,aAAa;QACb,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,EAAE,UAAU,CAAC,CAAA;QAEpC,OAAO,EAAE,CAAA;IACX,CAAC;IAED;;;;OAIG;IACH,aAAa,CAAC,EAAiB;QAC7B,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC;YAChB,OAAO,IAAI,CAAA;QACb,CAAC;QACD,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;QAC3C,OAAO,UAAU,IAAI,IAAI,CAAA;IAC3B,CAAC;IAED;;;OAGG;IACH,iBAAiB;QACf,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAA;IAC9C,CAAC;IAED;;;OAGG;IACH,mBAAmB;QACjB,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,IAAI,CAAA;IACtD,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,EAAU;QAC/B,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,CAAA;QACzC,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO,KAAK,CAAA;QACd,CAAC;QAED,qBAAqB,CAAC,iBAAiB,CAAC,MAAM,UAAU,CAAC,kBAAkB,CAAC,CAAA;QAC5E,OAAO,IAAI,CAAA;IACb,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAA;QACxB,IAAI,CAAC,MAAM,GAAG,CAAC,CAAA,CAAC,qCAAqC;IACvD,CAAC;IAED;;;;OAIG;IACH,sBAAsB,CAAC,WAAoB,KAAK;QAC9C,MAAM,WAAW,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC,CAAA;QAExE,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,OAAO,IAAA,mBAAM,EAAC,2BAA2B,CAAC,CAAA;QAC5C,CAAC;QAED,MAAM,KAAK,GAAa,CAAC,IAAA,iBAAI,EAAC,IAAA,sBAAS,EAAC,0BAA0B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;QAEzE,WAAW,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE;YACjC,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAA;YAC3C,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,EAAE,CAAA;YAE3C,MAAM,gBAAgB,GAAG,UAAU,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC,CAAA;YACnE,MAAM,SAAS,GACb,gBAAgB,CAAC,MAAM,GAAG,EAAE;gBAC1B,CAAC,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK;gBAC3C,CAAC,CAAC,gBAAgB,CAAA;YAEtB,KAAK,CAAC,IAAI,CACR,GAAG,IAAA,iBAAI,EAAC,IAAA,iBAAI,EAAC,IAAI,UAAU,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,IAAA,iBAAI,EAAC,IAAI,aAAa,GAAG,CAAC,GAAG,CACpE,CAAA;YAED,KAAK,CAAC,IAAI,CAAC,KAAK,IAAA,iBAAI,EAAC,OAAO,CAAC,KAAK,SAAS,EAAE,CAAC,CAAA;YAE9C,IAAI,QAAQ,EAAE,CAAC;gBACb,mDAAmD;gBACnD,MAAM,YAAY,GAAG,UAAU,CAAC,aAAa,CAAA;gBAC7C,KAAK,CAAC,IAAI,CAAC,KAAK,IAAA,iBAAI,EAAC,UAAU,CAAC,KAAK,YAAY,EAAE,CAAC,CAAA;gBAEpD,uDAAuD;gBACvD,8CAA8C;YAChD,CAAC;YAED,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA,CAAC,iCAAiC;QAClD,CAAC,CAAC,CAAA;QAEF,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACzB,CAAC;IAED;;;;OAIG;IACH,oBAAoB,CAAC,EAAU;QAC7B,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,CAAA;QACzC,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO,IAAA,iBAAI,EAAC,iBAAiB,EAAE,aAAa,CAAC,CAAA;QAC/C,CAAC;QAED,MAAM,KAAK,GAAa;YACtB,IAAA,iBAAI,EAAC,wCAAwC,EAAE,GAAG,CAAC;SACpD,CAAA;QAED,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAA;QAC3C,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,EAAE,CAAA;QAC3C,KAAK,CAAC,IAAI,CAAC,GAAG,IAAA,iBAAI,EAAC,YAAY,CAAC,KAAK,aAAa,EAAE,CAAC,CAAA;QAErD,IAAI,UAAU,CAAC,SAAS,EAAE,CAAC;YACzB,KAAK,CAAC,IAAI,CAAC,GAAG,IAAA,iBAAI,EAAC,YAAY,CAAC,KAAK,UAAU,CAAC,SAAS,EAAE,CAAC,CAAA;QAC9D,CAAC;QAED,0DAA0D;QAC1D,MAAM,YAAY,GAAG,UAAU,CAAC,aAAa,CAAA;QAC7C,KAAK,CAAC,IAAI,CAAC,GAAG,IAAA,iBAAI,EAAC,iBAAiB,CAAC,KAAK,YAAY,WAAW,CAAC,CAAA;QAElE,yDAAyD;QAEzD,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACzB,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAA;IACpB,CAAC;CACF;AAnKD,8CAmKC;AAED,iEAAiE;AACpD,QAAA,iBAAiB,GAAG,IAAI,iBAAiB,EAAE,CAAA"}

package/dist/cli.d.ts

@@ -2,7 +2,6 @@ import { ProjectFileContext } from './common/util/file';
 import { CliOptions } from './types';
 export declare class CLI {
     private client;
-    private chatStorage;
     private readyPromise;
     private git;
     private costMode;
@@ -14,7 +13,7 @@ export declare class CLI {
     private consecutiveFastInputs;
     private pastedContent;
     private isPasting;
-    constructor(readyPromise: Promise<[void, ProjectFileContext]>, { git, costMode }: CliOptions);
+    constructor(readyPromise: Promise<[void, ProjectFileContext, void]>, { git, costMode }: CliOptions);
     private setupSignalHandlers;
     private initReadlineInterface;
     private completer;
@@ -26,7 +25,6 @@ export declare class CLI {
     private handleUserInput;
     private beforeProcessCommand;
     private processCommand;
-    private addAssistantMessageToChat;
     private forwardUserInput;
     private returnControlToUser;
     private onWebSocketError;