scai 0.1.138 → 0.1.139

package/dist/agents/MainAgent.js

@@ -230,8 +230,8 @@ export class MainAgent {
         userOutput("All input/output logs can be found at ~/.scai/input_output.log");
         logLine("RUN", "complete", stopRun());
         // ───────────── LOG FINAL CONTEXT ─────────────
-        /* console.log("\n[DEBUG] Final MainAgent context:");
-        console.dir(this.context, { depth: null, colors: true }); */
+        console.log("\n[DEBUG] Final MainAgent context:");
+        console.dir(this.context, { depth: null, colors: true });
         return stepIO;
     }
     /* ───────────── helpers ───────────── */

package/dist/agents/writeFileStep.js

@@ -1,6 +1,7 @@
 // File: src/modules/writeFileModule.ts
 import fs from "fs/promises";
 import path from "path";
+import chalk from "chalk";
 import { normalizePath } from "../utils/contentUtils.js";
 import { logInputOutput } from "../utils/promptLogHelper.js";
 export const writeFileStep = {
@@ -10,54 +11,54 @@ export const writeFileStep = {
     groups: ["finalize"],
     run: async (input) => {
         var _a;
+        console.log(">>> writeFileStep starting");
         const context = input.context;
         const mode = input.data?.mode ?? "overwrite";
         const writtenFiles = [];
         const errors = [];
-        /* ───────────── guard ───────────── */
         if (!context) {
-            return {
-                query: input.query,
-                data: {
-                    writeMode: mode,
-                    writtenFiles,
-                    errors: ["Missing execution context"]
-                }
-            };
+            console.error("writeFileStep: Missing execution context");
+            return { query: input.query, data: { writeMode: mode, writtenFiles, errors: ["Missing execution context"] } };
         }
+        console.log("writeFileStep: context exists");
+        console.log("writeFileStep: mode =", mode);
+        console.log("writeFileStep: existing touchedFiles =", context.plan?.touchedFiles);
         const artifacts = context.execution?.codeTransformArtifacts?.files ?? [];
-        if (!artifacts.length) {
-            return {
-                query: input.query,
-                data: {
-                    writeMode: mode,
-                    writtenFiles,
-                    errors: ["No codeTransform artifacts found to write"]
-                }
-            };
+        console.log("writeFileStep: found codeTransformArtifacts.files =", artifacts.length);
+        if (artifacts.length) {
+            console.log("writeFileStep: filePaths =", artifacts.map(f => f.filePath));
+        }
+        else {
+            console.warn("writeFileStep: No artifacts found to write");
         }
-        /* ───────────── write files ───────────── */
         for (const f of artifacts) {
+            console.log("Processing artifact:", f.filePath);
             const filePath = normalizePath(f.filePath);
             if (!filePath) {
-                errors.push(`Invalid filePath: ${f.filePath}`);
+                const msg = `Invalid filePath: ${f.filePath}`;
+                console.error(msg);
+                errors.push(msg);
                 continue;
             }
             if (typeof f.content !== "string" || !f.content.trim()) {
-                errors.push(`No content to write for ${filePath}`);
+                const msg = `No content to write for ${filePath}`;
+                console.error(msg);
+                errors.push(msg);
                 continue;
             }
             try {
                 await fs.mkdir(path.dirname(filePath), { recursive: true });
                 await fs.writeFile(filePath, f.content, "utf-8");
+                console.log(chalk.green(`✅ Wrote file: ${filePath}`));
                 writtenFiles.push(filePath);
             }
             catch (err) {
                 const msg = err?.message ?? String(err);
+                console.error(chalk.red(`❌ Failed writing ${filePath}: ${msg}`));
                 errors.push(`${filePath}: ${msg}`);
             }
         }
-        /* ───────────── mark touched files ───────────── */
+        // mark files as touched if not already
         context.plan ?? (context.plan = {});
         (_a = context.plan).touchedFiles ?? (_a.touchedFiles = []);
         for (const file of writtenFiles) {
@@ -65,15 +66,13 @@ export const writeFileStep = {
                 context.plan.touchedFiles.push(file);
             }
         }
+        console.log("writeFileStep: touchedFiles after run =", context.plan.touchedFiles);
         const output = {
             query: input.query,
-            data: {
-                writeMode: mode,
-                writtenFiles,
-                errors
-            }
+            data: { writeMode: mode, writtenFiles, errors }
         };
         logInputOutput("writeFile", "output", output.data);
+        console.log(">>> writeFileStep complete");
         return output;
     }
 };

package/dist/pipeline/modules/codeTransformModule.js

@@ -31,7 +31,7 @@ export const codeTransformModule = {
     description: "Transforms a single file specified in the current plan step based on user instruction.",
     groups: ["transform"],
     run: async (input) => {
-        var _a, _b, _c, _d;
+        var _a, _b, _c;
         const query = typeof input.query === "string" ? input.query : String(input.query ?? "");
         const context = input.context;
         if (!context) {
@@ -48,10 +48,8 @@ export const codeTransformModule = {
             return { query, data: { files: [], errors: [`Target file not found or missing code: ${targetFile}`] } };
         }
         const normalizedQuery = context.analysis?.intent?.normalizedQuery ?? query;
-        // 🔹 NEW: extract proposed changes for this file (if any)
-        const fileAnalysis = context.analysis?.fileAnalysis?.[file.path]; // ✅ correct
+        const fileAnalysis = context.analysis?.fileAnalysis?.[file.path];
         const proposed = fileAnalysis?.proposedChanges;
-        // 🔹 NEW: pre-render guidance block for prompts
         const proposedChangesBlock = proposed
             ? `
 Proposed changes for this file:
@@ -64,6 +62,8 @@ ${proposed.rationale ? `- Rationale: ${proposed.rationale}` : ""}
         const outputs = [];
         const perFileErrors = [];
         const tokenCount = countTokens(file.code);
+        context.execution || (context.execution = {});
+        (_a = context.execution).codeTransformArtifacts || (_a.codeTransformArtifacts = { files: [] });
         // ───────────── SMALL FILE ─────────────
         if (tokenCount <= SINGLE_SHOT_TOKEN_LIMIT) {
             logInputOutput("codeTransform", "output", {
@@ -109,29 +109,32 @@ JSON schema:
                     responseSnippet: String(llmResponse.data ?? "").slice(0, 200),
                 });
                 const cleaned = await cleanupModule.run({ query, content: llmResponse.data });
-                // Optional debug: show that cleanup returned raw code because it wasn't JSON
+                let finalContent;
+                let notes = undefined;
                 if (typeof cleaned.data === 'string') {
-                    console.debug(chalk.yellow(` - [cleanupModule] Could not parse JSON. Likely non-JSON output (code), using cleaned content.`));
+                    // Raw code returned
+                    finalContent = cleaned.data;
+                    console.debug(chalk.yellow(` - [cleanupModule] Using raw code output for ${file.path}`));
+                }
+                else if (cleaned.data &&
+                    typeof cleaned.data === 'object' &&
+                    Array.isArray(cleaned.data.files)) {
+                    // Structured JSON returned
+                    const structured = cleaned.data;
+                    const out = structured.files.find(f => f.filePath === file.path);
+                    finalContent = out?.content ?? file.code;
+                    notes = out?.notes;
+                    perFileErrors.push(...(structured.errors ?? []));
                 }
-                const structured = cleaned.data;
-                const out = Array.isArray(structured?.files)
-                    ? structured.files.find((f) => f.filePath === file.path)
-                    : null;
-                if (!out || typeof out.content !== "string") {
-                    return { query, data: { files: [], errors: [`Model did not return valid content for ${file.path}`] } };
+                else {
+                    // Fallback
+                    finalContent = file.code;
                 }
-                const finalContent = out.content.trim() ? out.content : file.code;
-                outputs.push({ filePath: file.path, content: finalContent, notes: out.notes });
-                perFileErrors.push(...(structured.errors ?? []));
-                context.execution || (context.execution = {});
-                (_a = context.execution).codeTransformArtifacts || (_a.codeTransformArtifacts = { files: [] });
+                outputs.push({ filePath: file.path, content: finalContent, notes });
+                // ensure artifacts are updated
                 context.execution.codeTransformArtifacts.files =
                     context.execution.codeTransformArtifacts.files.filter(f => f.filePath !== file.path);
-                context.execution.codeTransformArtifacts.files.push({
-                    filePath: file.path,
-                    content: finalContent,
-                    notes: perFileErrors.length ? perFileErrors.join("; ") : undefined,
-                });
+                context.execution.codeTransformArtifacts.files.push(outputs[0]);
                 context.plan || (context.plan = {});
                 (_b = context.plan).touchedFiles || (_b.touchedFiles = []);
                 if (!context.plan.touchedFiles.includes(file.path))
@@ -141,12 +144,16 @@ JSON schema:
                 return output;
             }
             catch (err) {
-                return { query, data: { files: [], errors: [`LLM call or cleanup failed: ${err.message}`] } };
+                const finalContent = file.code;
+                outputs.push({ filePath: file.path, content: finalContent });
+                perFileErrors.push(`LLM call or cleanup failed: ${err.message}`);
+                context.execution.codeTransformArtifacts.files.push(outputs[0]);
+                const output = { query, data: { files: outputs, errors: perFileErrors } };
+                return output;
             }
         }
         // ───────────── LARGE FILE (chunked) ─────────────
         const chunks = splitCodeIntoChunks(file.code, CHUNK_TOKEN_LIMIT);
-        // ⚠️ USER-FACING WARNING
         console.warn(chalk.yellow([
             "",
             "⚠️  Large file detected — falling back to chunked transformation",
@@ -215,13 +222,13 @@ ${chunk}
                 logInputOutput("codeTransform", "output", { file: file.path, chunkIndex: i + 1, error: err.message, message: "Chunk transformation failed, original preserved" });
             }
         }
-        logInputOutput("codeTransform", "output", { file: file.path, message: "Finished all chunks", totalChunks: chunks.length });
-        outputs.push({ filePath: file.path, content: transformedChunks.join("\n") });
-        context.execution || (context.execution = {});
-        (_c = context.execution).codeTransformArtifacts || (_c.codeTransformArtifacts = { files: [] });
-        context.execution.codeTransformArtifacts.files.push(...outputs);
+        const finalContent = transformedChunks.join("\n");
+        outputs.push({ filePath: file.path, content: finalContent });
+        context.execution.codeTransformArtifacts.files =
+            context.execution.codeTransformArtifacts.files.filter(f => f.filePath !== file.path);
+        context.execution.codeTransformArtifacts.files.push(outputs[0]);
         context.plan || (context.plan = {});
-        (_d = context.plan).touchedFiles || (_d.touchedFiles = []);
+        (_c = context.plan).touchedFiles || (_c.touchedFiles = []);
         if (!context.plan.touchedFiles.includes(file.path))
             context.plan.touchedFiles.push(file.path);
         const output = { query, data: { files: outputs, errors: perFileErrors } };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "scai",
-  "version": "0.1.138",
+  "version": "0.1.139",
   "type": "module",
   "bin": {
     "scai": "./dist/index.js"