promptopskit 0.3.5 → 0.3.6

package/dist/index.cjs

@@ -253,7 +253,10 @@ var HistorySchema = import_zod.z.object({
 });
 var ContextInputDefinitionObjectSchema = import_zod.z.object({
   name: import_zod.z.string(),
-  max_size: import_zod.z.number().int().positive().optional()
+  max_size: import_zod.z.number().int().positive().optional(),
+  trim: import_zod.z.union([import_zod.z.boolean(), import_zod.z.enum(["start", "end", "both"])]).optional(),
+  allow_regex: import_zod.z.string().optional(),
+  deny_regex: import_zod.z.string().optional()
 });
 var ContextInputDefinitionSchema = import_zod.z.union([
   import_zod.z.string(),
@@ -596,6 +599,139 @@ function resolveInlinePromptSource(source, options = {}) {
   });
 }
 
+// src/context.ts
+var textEncoder = new TextEncoder();
+function getContextInputs(asset) {
+  return (asset.context?.inputs ?? []).map(normalizeContextInput);
+}
+function getContextInputNames(asset) {
+  return getContextInputs(asset).map((input) => input.name);
+}
+function normalizeContextInput(input) {
+  if (typeof input === "string") {
+    return { name: input };
+  }
+  return {
+    name: input.name,
+    max_size: input.max_size,
+    trim: input.trim,
+    allow_regex: input.allow_regex,
+    deny_regex: input.deny_regex
+  };
+}
+function isTrimEnabled(mode) {
+  return mode === true || mode === "start" || mode === "end" || mode === "both";
+}
+function normalizeTrimMode(mode) {
+  if (mode === "start") {
+    return "start";
+  }
+  return "end";
+}
+function trimToMaxSize(value, maxSize, mode) {
+  const measured = measureContextValueSize(value);
+  if (measured <= maxSize) {
+    return value;
+  }
+  const characters = Array.from(value);
+  const normalizedMode = normalizeTrimMode(mode);
+  if (normalizedMode === "start") {
+    let collected2 = "";
+    let size2 = 0;
+    for (let i = characters.length - 1; i >= 0; i -= 1) {
+      const next = characters[i];
+      const charSize = measureContextValueSize(next);
+      if (size2 + charSize > maxSize) {
+        break;
+      }
+      collected2 = next + collected2;
+      size2 += charSize;
+    }
+    return collected2;
+  }
+  let collected = "";
+  let size = 0;
+  for (const char of characters) {
+    const charSize = measureContextValueSize(char);
+    if (size + charSize > maxSize) {
+      break;
+    }
+    collected += char;
+    size += charSize;
+  }
+  return collected;
+}
+function sanitizeContextVariables(asset, variables = {}, options = {}) {
+  const { onContextOverflow } = options;
+  const sanitized = { ...variables };
+  for (const input of getContextInputs(asset)) {
+    const value = sanitized[input.name];
+    if (value === void 0) {
+      continue;
+    }
+    let candidate = value;
+    if (input.max_size !== void 0) {
+      const actualSize = measureContextValueSize(candidate);
+      if (actualSize > input.max_size && onContextOverflow) {
+        candidate = onContextOverflow({
+          promptId: asset.id,
+          variable: input.name,
+          value: candidate,
+          maxSize: input.max_size,
+          actualSize
+        });
+      }
+    }
+    if (isTrimEnabled(input.trim) && input.max_size !== void 0) {
+      candidate = trimToMaxSize(candidate, input.max_size, input.trim);
+    }
+    sanitized[input.name] = candidate;
+    if (input.allow_regex) {
+      const candidate2 = sanitized[input.name];
+      const matcher = new RegExp(input.allow_regex);
+      if (!matcher.test(candidate2)) {
+        throw new Error(
+          `POK031: Context variable "${input.name}" failed allow_regex validation for prompt "${asset.id}".`
+        );
+      }
+    }
+    if (input.deny_regex) {
+      const candidate2 = sanitized[input.name];
+      const matcher = new RegExp(input.deny_regex);
+      if (matcher.test(candidate2)) {
+        throw new Error(
+          `POK032: Context variable "${input.name}" matched deny_regex for prompt "${asset.id}".`
+        );
+      }
+    }
+  }
+  return sanitized;
+}
+function measureContextValueSize(value) {
+  return textEncoder.encode(value).length;
+}
+function collectContextSizeWarnings(asset, variables = {}) {
+  const warnings = [];
+  for (const input of getContextInputs(asset)) {
+    if (input.max_size === void 0) {
+      continue;
+    }
+    const value = variables[input.name];
+    if (value === void 0) {
+      continue;
+    }
+    const actualSize = measureContextValueSize(value);
+    if (actualSize > input.max_size) {
+      warnings.push({
+        variable: input.name,
+        maxSize: input.max_size,
+        actualSize
+      });
+    }
+  }
+  return warnings;
+}
+
 // src/providers/prompt-input.ts
 function isPromptLookup(input) {
   return "path" in input && typeof input.path === "string";
@@ -619,7 +755,13 @@ function withPromptInputSupport(adapter) {
   };
   const renderPrompt2 = async (input, runtime) => {
     const resolved = await resolveProviderPromptInput(input, runtime);
-    return adapter.render(resolved, runtime);
+    const variables = sanitizeContextVariables(resolved, runtime.variables, {
+      onContextOverflow: runtime.onContextOverflow
+    });
+    return adapter.render(resolved, {
+      ...runtime,
+      variables
+    });
   };
   return {
     ...adapter,
@@ -906,48 +1048,6 @@ function getAdapter(provider) {
   return adapter;
 }
 
-// src/context.ts
-var textEncoder = new TextEncoder();
-function getContextInputs(asset) {
-  return (asset.context?.inputs ?? []).map(normalizeContextInput);
-}
-function getContextInputNames(asset) {
-  return getContextInputs(asset).map((input) => input.name);
-}
-function normalizeContextInput(input) {
-  if (typeof input === "string") {
-    return { name: input };
-  }
-  return {
-    name: input.name,
-    max_size: input.max_size
-  };
-}
-function measureContextValueSize(value) {
-  return textEncoder.encode(value).length;
-}
-function collectContextSizeWarnings(asset, variables = {}) {
-  const warnings = [];
-  for (const input of getContextInputs(asset)) {
-    if (input.max_size === void 0) {
-      continue;
-    }
-    const value = variables[input.name];
-    if (value === void 0) {
-      continue;
-    }
-    const actualSize = measureContextValueSize(value);
-    if (actualSize > input.max_size) {
-      warnings.push({
-        variable: input.name,
-        maxSize: input.max_size,
-        actualSize
-      });
-    }
-  }
-  return warnings;
-}
-
 // src/validation/levenshtein.ts
 function levenshtein(a, b) {
   const m = a.length;
@@ -1059,6 +1159,30 @@ function validateAsset(asset, frontMatterKeys, filePath) {
       });
     }
   }
+  for (const input of getContextInputs(asset)) {
+    if (input.trim !== void 0 && input.trim !== false && input.max_size === void 0) {
+      warnings.push({
+        code: "POK014",
+        message: `Context input "${input.name}" sets trim but has no max_size; trim-to-budget will be skipped.`,
+        filePath
+      });
+    }
+    const checks = [];
+    if (input.allow_regex) checks.push({ pattern: input.allow_regex, kind: "allow_regex" });
+    if (input.deny_regex) checks.push({ pattern: input.deny_regex, kind: "deny_regex" });
+    for (const check of checks) {
+      try {
+        new RegExp(check.pattern);
+      } catch (error) {
+        const reason = error instanceof Error ? error.message : String(error);
+        errors.push({
+          code: "POK013",
+          message: `Invalid context ${check.kind} for "${input.name}": ${reason}`,
+          filePath
+        });
+      }
+    }
+  }
   return {
     valid: errors.length === 0,
     errors,
@@ -1540,7 +1664,10 @@ var PromptOpsKit = class {
 ` + validation.errors.map((e) => `  - ${e}`).join("\n")
       );
     }
-    const contextSizeWarnings = collectContextSizeWarnings(resolved, options.variables).map(
+    const sanitizedVariables = sanitizeContextVariables(resolved, options.variables, {
+      onContextOverflow: options.onContextOverflow
+    });
+    const contextSizeWarnings = collectContextSizeWarnings(resolved, sanitizedVariables).map(
       (warning) => formatContextSizeWarning(resolved, warning)
     );
     const contextWarningPolicy = this.config.warnings?.contextSize;
@@ -1550,7 +1677,7 @@ var PromptOpsKit = class {
       }
     }
     const request = adapter.render(resolved, {
-      variables: options.variables,
+      variables: sanitizedVariables,
       history: options.history,
       toolRegistry: options.toolRegistry,
       strict: options.strict