draftify-cli 1.0.91 → 1.0.94

package/dist/repl.js

@@ -905,18 +905,32 @@ async function startRepl(initialUsername) {
                 let retryCount = 0;
                 let success = false;
                 let partialText = "";
-                while (retryCount < 3 && !success) {
+                // Detect if a successful response was actually truncated mid-content
+                function detectTruncation(text) {
+                    // Check for unclosed FILE_CREATE
+                    const createOpenMatch = text.match(/<FILE_CREATE\s+path="([^"]+)"[^>]*>/);
+                    if (createOpenMatch && !/<\/FILE_CREATE>/i.test(text)) {
+                        return { truncated: true, filePath: createOpenMatch[1], tagType: 'FILE_CREATE' };
+                    }
+                    // Check for unclosed FILE_MODIFY
+                    const modifyOpenMatch = text.match(/<FILE_MODIFY\s+path="([^"]+)"[^>]*>/);
+                    if (modifyOpenMatch && !/<\/FILE_MODIFY>/i.test(text)) {
+                        return { truncated: true, filePath: modifyOpenMatch[1], tagType: 'FILE_MODIFY' };
+                    }
+                    return { truncated: false };
+                }
+                while (retryCount < 4 && !success) {
                     try {
                         let retryHistory = [...conversationHistory];
                         let retryPrompt = finalPrompt;
                         if (partialText) {
                             retryHistory.push({ role: "user", content: finalPrompt });
                             retryHistory.push({ role: "assistant", content: partialText });
-                            retryPrompt = "Continue exactly from where you left off. Do not repeat what you already said, just continue the exact code or sentence you were writing.";
+                            retryPrompt = "Continue exactly from where you left off. Do not repeat anything you already wrote. Continue the file content from the exact character where it was cut off.";
                         }
                         spinner = (0, ui_1.createSpinner)("Working...").start();
                         if (retryCount > 0) {
-                            spinner.setPrefix(`Retrying... (${retryCount}/3)`);
+                            spinner.setPrefix(`Continuing... (${retryCount}/3)`);
                         }
                         let thisAttemptText = "";
                         const attemptResult = await (0, api_1.refactorCodeApi)(requestPayloadFileName, requestPayloadCode, retryPrompt, retryHistory, currentModel, activeSkillsData, (0, config_1.getThinkingLevel)(), (chunk) => {
@@ -948,7 +962,22 @@ async function startRepl(initialUsername) {
                                 spinner.setPrefix("Analyzing...");
                             }
                         }, exports.globalAbortController.signal);
-                        rawResult = partialText + attemptResult;
+                        const combined = partialText + attemptResult;
+                        // --- TRUNCATION DETECTION ---
+                        // If the stream ended cleanly but the content is still incomplete (token-limited),
+                        // treat it like a retry scenario and continue generation.
+                        const truncationInfo = detectTruncation(combined);
+                        if (truncationInfo.truncated && retryCount < 3) {
+                            if (spinner)
+                                spinner.stop();
+                            ui_1.ui.info(`Response truncated mid-file (${truncationInfo.filePath}). Continuing...`);
+                            partialText = combined;
+                            retryCount++;
+                            currentStreamedText = combined;
+                            await new Promise(r => setTimeout(r, 500));
+                            continue; // restart the while loop with updated partialText
+                        }
+                        rawResult = combined;
                         success = true;
                     }
                     catch (e) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "draftify-cli",
-  "version": "1.0.91",
+  "version": "1.0.94",
   "description": "Draftify AI CLI tool",
   "main": "dist/index.js",
   "bin": {