@arnilo/prism-coding-agent 0.0.5 → 0.0.6

package/dist/truncate.js

@@ -9,8 +9,8 @@
  *
  * Never returns partial lines (except the documented tail single-line edge case).
  */
-export const DEFAULT_MAX_LINES = 2000;
-export const DEFAULT_MAX_BYTES = 50 * 1024; // 50KB
+import { DEFAULT_MAX_BYTES, DEFAULT_MAX_LINES, HARD_MAX_BYTES, HARD_MAX_LINES, validateCodingLimit, } from "./limits.js";
+export { DEFAULT_MAX_BYTES, DEFAULT_MAX_LINES } from "./limits.js";
 /**
  * Default char cap for {@link truncateLine}. pi names this GREP_MAX_LINE_LENGTH
  * because grep is its only caller; grep is out of scope for this package, so the
@@ -41,8 +41,8 @@ export function formatSize(bytes) {
  * returns empty content with firstLineExceedsLimit=true.
  */
 export function truncateHead(content, options = {}) {
-    const maxLines = options.maxLines ?? DEFAULT_MAX_LINES;
-    const maxBytes = options.maxBytes ?? DEFAULT_MAX_BYTES;
+    const maxLines = validateCodingLimit("maxLines", options.maxLines ?? DEFAULT_MAX_LINES, HARD_MAX_LINES);
+    const maxBytes = validateCodingLimit("maxBytes", options.maxBytes ?? DEFAULT_MAX_BYTES, HARD_MAX_BYTES);
     const totalBytes = Buffer.byteLength(content, "utf-8");
     const lines = splitLinesForCounting(content);
     const totalLines = lines.length;
@@ -118,8 +118,8 @@ export function truncateHead(content, options = {}) {
  * the byte limit.
  */
 export function truncateTail(content, options = {}) {
-    const maxLines = options.maxLines ?? DEFAULT_MAX_LINES;
-    const maxBytes = options.maxBytes ?? DEFAULT_MAX_BYTES;
+    const maxLines = validateCodingLimit("maxLines", options.maxLines ?? DEFAULT_MAX_LINES, HARD_MAX_LINES);
+    const maxBytes = validateCodingLimit("maxBytes", options.maxBytes ?? DEFAULT_MAX_BYTES, HARD_MAX_BYTES);
     const totalBytes = Buffer.byteLength(content, "utf-8");
     const lines = splitLinesForCounting(content);
     const totalLines = lines.length;

package/dist/write.d.ts

@@ -5,14 +5,21 @@ import type { ExecutionPolicy, ToolDefinition } from "@arnilo/prism";
  */
 export interface WriteOperations {
     /** Write content to a file (creating or overwriting). */
-    writeFile: (absolutePath: string, content: string) => Promise<void>;
+    writeFile: (absolutePath: string, content: string, options?: {
+        maxBytes: number;
+        signal?: AbortSignal;
+    }) => Promise<void>;
     /** Create a directory recursively. */
-    mkdir: (dir: string) => Promise<void>;
+    mkdir: (dir: string, options?: {
+        signal?: AbortSignal;
+    }) => Promise<void>;
 }
 export interface WriteToolOptions {
     /** Structured pre-execution policy checked before filesystem writes. */
     executionPolicy?: ExecutionPolicy;
     /** Custom operations backend (default: local filesystem). */
     operations?: WriteOperations;
+    /** Maximum UTF-8 input bytes accepted before any filesystem mutation (default 8 MiB). */
+    maxInputBytes?: number;
 }
 export declare function createWriteTool(cwd: string, options?: WriteToolOptions): ToolDefinition;

package/dist/write.js

@@ -20,8 +20,9 @@ import { dirname } from "node:path";
 import { enforceExecutionPolicy } from "./execution-policy.js";
 import { resolveToCwd } from "./path-utils.js";
 import { withFileMutationQueue } from "./file-mutation-queue.js";
+import { DEFAULT_MAX_WRITE_BYTES, HARD_MAX_WRITE_BYTES, validateCodingLimit, } from "./limits.js";
 const defaultWriteOperations = {
-    writeFile: (path, content) => fsWriteFile(path, content, "utf-8"),
+    writeFile: (path, content, options) => fsWriteFile(path, content, { encoding: "utf-8", signal: options?.signal }),
     mkdir: (dir) => fsMkdir(dir, { recursive: true }).then(() => { }),
 };
 function errorResult(toolCallId, message) {
@@ -39,6 +40,7 @@ function countLines(content) {
 }
 export function createWriteTool(cwd, options) {
     const ops = options?.operations ?? defaultWriteOperations;
+    const maxInputBytes = validateCodingLimit("maxInputBytes", options?.maxInputBytes ?? DEFAULT_MAX_WRITE_BYTES, HARD_MAX_WRITE_BYTES);
     return {
         name: "write",
         description: "Write content to a file. Creates the file if it doesn't exist, overwrites if it does. Automatically creates parent directories.",
@@ -61,6 +63,10 @@ export function createWriteTool(cwd, options) {
             if (content === undefined) {
                 return errorResult(toolCallId, "content is required and must be a string.");
             }
+            const bytes = Buffer.byteLength(content, "utf-8");
+            if (bytes > maxInputBytes) {
+                return errorResult(toolCallId, `Write input is ${bytes} bytes, exceeds ${maxInputBytes} byte limit.`);
+            }
             try {
                 const absolutePath = resolveToCwd(path, cwd);
                 const policyCheck = await enforceExecutionPolicy(options?.executionPolicy, {
@@ -68,7 +74,7 @@ export function createWriteTool(cwd, options) {
                     operation: "write",
                     paths: [absolutePath],
                     risk: "medium",
-                    metadata: { bytes: Buffer.byteLength(content, "utf-8"), sessionId: context.sessionId, runId: context.runId, signal: context.signal },
+                    metadata: { bytes, sessionId: context.sessionId, runId: context.runId, signal: context.signal },
                 }, toolCallId, "write");
                 if (!policyCheck.allowed)
                     return policyCheck.result;
@@ -79,11 +85,10 @@ export function createWriteTool(cwd, options) {
                     // do NOT throw from an abort listener: that could release the mutation queue mid-operation.
                     if (context.signal?.aborted)
                         return errorResult(toolCallId, "Operation aborted");
-                    await ops.mkdir(dir);
+                    await ops.mkdir(dir, { signal: context.signal });
                     if (context.signal?.aborted)
                         return errorResult(toolCallId, "Operation aborted");
-                    await ops.writeFile(allowedPath, content);
-                    const bytes = Buffer.byteLength(content, "utf-8");
+                    await ops.writeFile(allowedPath, content, { maxBytes: maxInputBytes, signal: context.signal });
                     const lines = countLines(content);
                     return {
                         toolCallId,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@arnilo/prism-coding-agent",
-  "version": "0.0.5",
+  "version": "0.0.6",
   "description": "Optional coding-agent tools (shell, read, write, edit) package for Prism.",
   "type": "module",
   "main": "./dist/index.js",
@@ -28,7 +28,7 @@
     "diff": "^8.0.4"
   },
   "peerDependencies": {
-    "@arnilo/prism": "0.0.5"
+    "@arnilo/prism": "0.0.6"
   },
   "devDependencies": {
     "@arnilo/prism": "file:../.."