codecane 1.0.156 → 1.0.171

package/dist/chat-storage.js

@@ -27,6 +27,8 @@ exports.ChatStorage = void 0;
 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;
@@ -45,46 +47,48 @@ class ChatStorage {
             if (index === lastIndex) {
                 return msg; // Preserve the most recent message in its entirety
             }
-            // Helper function to clean up content string
+            // Helper function to clean up message content
             const cleanContent = (content) => {
-                let result = content;
-                if (content.includes('"screenshot"')) {
-                    result = result.replace(/"screenshot"\s*:\s*"[^"]+"/g, '"screenshot":"[SCREENSHOT_REMOVED]"');
-                }
-                if (content.includes('"logs"')) {
-                    result = result.replace(/"logs"\s*:\s*\[[^\]]*\]/g, '"logs":[LOGS_REMOVED]');
-                }
-                return result;
+                // 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;
-            if (Array.isArray(msg.content)) {
-                return {
-                    ...msg,
-                    content: msg.content.map((contentObj) => {
-                        if (contentObj.type === 'tool_result' && contentObj.content) {
-                            return {
-                                ...contentObj,
-                                content: cleanContent(contentObj.content),
-                            };
-                        }
-                        return contentObj;
-                    }),
-                };
-            }
-            else if (typeof msg.content === 'string') {
-                return {
-                    ...msg,
-                    content: cleanContent(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 .codebuff/messages/messages.json
+        // Save messages to chat directory
         this.saveMessagesToFile(chat);
     }
     saveMessagesToFile(chat) {

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;AAexB,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,6CAA6C;YAC7C,MAAM,YAAY,GAAG,CAAC,OAAe,EAAE,EAAE;gBACvC,IAAI,MAAM,GAAG,OAAO,CAAA;gBACpB,IAAI,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;oBACrC,MAAM,GAAG,MAAM,CAAC,OAAO,CACrB,6BAA6B,EAC7B,qCAAqC,CACtC,CAAA;gBACH,CAAC;gBACD,IAAI,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAC/B,MAAM,GAAG,MAAM,CAAC,OAAO,CACrB,0BAA0B,EAC1B,uBAAuB,CACxB,CAAA;gBACH,CAAC;gBACD,OAAO,MAAM,CAAA;YACf,CAAC,CAAA;YAED,2BAA2B;YAC3B,IAAI,CAAC,GAAG,CAAC,OAAO;gBAAE,OAAO,GAAG,CAAA;YAE5B,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC/B,OAAO;oBACL,GAAG,GAAG;oBACN,OAAO,EAAE,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE;wBACtC,IAAI,UAAU,CAAC,IAAI,KAAK,aAAa,IAAI,UAAU,CAAC,OAAO,EAAE,CAAC;4BAC5D,OAAO;gCACL,GAAG,UAAU;gCACb,OAAO,EAAE,YAAY,CAAC,UAAU,CAAC,OAAO,CAAC;6BAC1C,CAAA;wBACH,CAAC;wBACD,OAAO,UAAU,CAAA;oBACnB,CAAC,CAAC;iBACH,CAAA;YACH,CAAC;iBAAM,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;gBAC3C,OAAO;oBACL,GAAG,GAAG;oBACN,OAAO,EAAE,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC;iBACnC,CAAA;YACH,CAAC;YACD,OAAO,GAAG,CAAA;QACZ,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,oDAAoD;QACpD,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;AA3JD,kCA2JC"}
+{"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"}

package/dist/checkpoints.d.ts

@@ -0,0 +1,64 @@
+import { AgentState } from './common/types/agent-state';
+/**
+ * Interface representing a checkpoint of agent state
+ */
+export interface Checkpoint {
+    agentStateString: string;
+    fileVersions: Record<string, string>;
+    historyLength: number;
+    id: number;
+    timestamp: number;
+    userInput: string;
+}
+/**
+ * Simple in-memory checkpoint manager for agent states
+ */
+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;
+    /**
+     * Get a checkpoint by ID
+     * @param id The checkpoint ID
+     * @returns The checkpoint or null if not found
+     */
+    getCheckpoint(id: number | null): Checkpoint | null;
+    /**
+     * Get all checkpoints
+     * @returns Array of all checkpoints
+     */
+    getAllCheckpoints(): Checkpoint[];
+    /**
+     * Get the most recent checkpoint
+     * @returns The most recent checkpoint or null if none exist
+     */
+    getLatestCheckpoint(): Checkpoint | null;
+    /**
+     * Clear all checkpoints
+     */
+    clearCheckpoints(): void;
+    /**
+     * Get a formatted string representation of all checkpoints
+     * @param detailed Whether to include detailed information about each checkpoint
+     * @returns A formatted string representation of all checkpoints
+     */
+    getCheckpointsAsString(detailed?: boolean): string;
+    /**
+     * Get detailed information about a specific checkpoint
+     * @param id The checkpoint ID
+     * @returns A formatted string with detailed information about the checkpoint, or an error message if not found
+     */
+    getCheckpointDetails(id: number): string;
+    getNextId(): number;
+}
+export declare const checkpointManager: CheckpointManager;

package/dist/checkpoints.js

@@ -0,0 +1,147 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.checkpointManager = exports.CheckpointManager = void 0;
+const picocolors_1 = require("picocolors");
+/**
+ * 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) {
+        // 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 checkpoint = {
+            agentStateString: JSON.stringify(agentState), // Deep clone to prevent reference issues
+            fileVersions: { ...parentFileVersions, ...currentFileVersions },
+            historyLength: agentState.messageHistory.length,
+            id,
+            timestamp: Date.now(),
+            userInput,
+        };
+        // 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;
+    }
+    /**
+     * Get a checkpoint by ID
+     * @param id The checkpoint ID
+     * @returns The checkpoint or null if not found
+     */
+    getCheckpoint(id) {
+        if (id === null) {
+            return null;
+        }
+        const checkpoint = this.checkpoints.get(id);
+        return checkpoint || null;
+    }
+    /**
+     * Get all checkpoints
+     * @returns Array of all checkpoints
+     */
+    getAllCheckpoints() {
+        return Array.from(this.checkpoints.values());
+    }
+    /**
+     * Get the most recent checkpoint
+     * @returns The most recent checkpoint or null if none exist
+     */
+    getLatestCheckpoint() {
+        return this.checkpoints.get(this.nextId - 1) || null;
+    }
+    /**
+     * 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
+     * @param detailed Whether to include detailed information about each checkpoint
+     * @returns A formatted string representation of all checkpoints
+     */
+    getCheckpointsAsString(detailed = false) {
+        const checkpoints = this.getAllCheckpoints().sort((a, b) => a.id - b.id);
+        if (checkpoints.length === 0) {
+            return (0, picocolors_1.yellow)('No checkpoints available.');
+        }
+        const lines = [(0, picocolors_1.bold)((0, picocolors_1.underline)('Agent State Checkpoints:')), ''];
+        checkpoints.forEach((checkpoint) => {
+            const date = new Date(checkpoint.timestamp);
+            const formattedDate = date.toLocaleString();
+            const userInputOneLine = checkpoint.userInput.replaceAll('\n', ' ');
+            const userInput = userInputOneLine.length > 50
+                ? userInputOneLine.substring(0, 47) + '...'
+                : userInputOneLine;
+            lines.push(`${(0, picocolors_1.cyan)((0, picocolors_1.bold)(`#${checkpoint.id}`))} ${(0, picocolors_1.gray)(`[${formattedDate}]`)}:`);
+            lines.push(`  ${(0, picocolors_1.blue)('Input')}: ${userInput}`);
+            if (detailed) {
+                // Add more details about the agent state if needed
+                const messageCount = checkpoint.historyLength;
+                lines.push(`  ${(0, picocolors_1.blue)('Messages')}: ${messageCount}`);
+                // You can add more detailed information here as needed
+                // For example, file context information, etc.
+            }
+            lines.push(''); // Empty line between checkpoints
+        });
+        return lines.join('\n');
+    }
+    /**
+     * Get detailed information about a specific checkpoint
+     * @param id The checkpoint ID
+     * @returns A formatted string with detailed information about the checkpoint, or an error message if not found
+     */
+    getCheckpointDetails(id) {
+        const checkpoint = this.getCheckpoint(id);
+        if (!checkpoint) {
+            return (0, picocolors_1.cyan)(`\nCheckpoint #${id} not found.`);
+        }
+        const lines = [
+            (0, picocolors_1.cyan)(`Detailed information for checkpoint #${id}:`),
+        ];
+        const date = new Date(checkpoint.timestamp);
+        const formattedDate = date.toLocaleString();
+        lines.push(`${(0, picocolors_1.blue)('Created at')}: ${formattedDate}`);
+        if (checkpoint.userInput) {
+            lines.push(`${(0, picocolors_1.blue)('User input')}: ${checkpoint.userInput}`);
+        }
+        // Display more detailed information about the agent state
+        const messageCount = checkpoint.historyLength;
+        lines.push(`${(0, picocolors_1.blue)('Message history')}: ${messageCount} messages`);
+        // You could add more detailed information here as needed
+        return lines.join('\n');
+    }
+    getNextId() {
+        return this.nextId;
+    }
+}
+exports.CheckpointManager = CheckpointManager;
+// Export a singleton instance for use throughout the application
+exports.checkpointManager = new CheckpointManager();
+//# sourceMappingURL=checkpoints.js.map

package/dist/checkpoints.js.map

@@ -0,0 +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"}

package/dist/cli.d.ts

@@ -9,33 +9,39 @@ export declare class CLI {
     private rl;
     private isReceivingResponse;
     private stopResponse;
-    private loadingInterval;
-    private lastChanges;
     private lastSigintTime;
     private lastInputTime;
     private consecutiveFastInputs;
     private pastedContent;
     private isPasting;
     constructor(readyPromise: Promise<[void, ProjectFileContext]>, { git, costMode }: CliOptions);
+    private setupSignalHandlers;
+    private initReadlineInterface;
+    private completer;
+    private setPrompt;
+    private promptWithCheckpointNumber;
+    printInitialPrompt(initialInput?: string): Promise<void>;
+    printDiff(): Promise<void>;
+    private handleLine;
+    private handleUserInput;
+    private beforeProcessCommand;
+    private processCommand;
+    private forwardUserInput;
     private returnControlToUser;
     private onWebSocketError;
     private onWebSocketReconnect;
-    private detectPasting;
-    private handleInput;
-    private setPrompt;
-    printInitialPrompt(initialInput?: string): Promise<void>;
     private handleUndo;
-    private handleRedo;
-    private navigateFileVersion;
+    private handleKeyPress;
+    private handleSigint;
+    private handleEscKey;
     private handleStopResponse;
-    private restoreCursor;
+    private showEasterEgg;
     private handleExit;
-    private handleEscKey;
-    private applyAndDisplayCurrentFileVersion;
-    private startLoadingAnimation;
-    private stopLoadingAnimation;
-    private autoCommitChanges;
+    private detectPasting;
     private handleDiff;
-    private handleUserInput;
-    private sendUserInputAndAwaitResponse;
+    private handleCheckpoints;
+    private handleCheckpointDetail;
+    private handleRestoreCheckpoint;
+    private restoreAgentStateAndFiles;
+    private handleClearCheckpoints;
 }