@deepagents/context 0.33.0 → 0.35.0

package/dist/index.js

@@ -151,8 +151,10 @@ var Agent = class _Agent {
       providerOptions: this.#options.providerOptions,
       model: this.#options.model,
       system: systemPrompt,
-      messages: await convertToModelMessages(messages),
-      stopWhen: stepCountIs(25),
+      messages: await convertToModelMessages(messages, {
+        ignoreIncompleteToolCalls: true
+      }),
+      stopWhen: stepCountIs(200),
       tools: this.#options.tools,
       experimental_context: contextVariables,
       experimental_repairToolCall: createRepairToolCall(this.#options.model),
@@ -218,9 +220,11 @@ var Agent = class _Agent {
       providerOptions: this.#options.providerOptions,
       model,
       system: systemPrompt,
-      messages: await convertToModelMessages(messages),
+      messages: await convertToModelMessages(messages, {
+        ignoreIncompleteToolCalls: true
+      }),
       experimental_repairToolCall: createRepairToolCall(model),
-      stopWhen: stepCountIs(50),
+      stopWhen: stepCountIs(200),
       experimental_transform: config?.transform ?? smoothStream(),
       tools: this.#options.tools,
       experimental_context: contextVariables,
@@ -360,8 +364,10 @@ function structuredOutput(options) {
         providerOptions: options.providerOptions,
         model: options.model,
         system: systemPrompt,
-        messages: await convertToModelMessages(messages),
-        stopWhen: stepCountIs(25),
+        messages: await convertToModelMessages(messages, {
+          ignoreIncompleteToolCalls: true
+        }),
+        stopWhen: stepCountIs(200),
         experimental_repairToolCall: createRepairToolCall(options.model),
         experimental_context: contextVariables,
         output: Output.object({ schema: options.schema }),
@@ -385,8 +391,10 @@ function structuredOutput(options) {
         model: options.model,
         system: systemPrompt,
         experimental_repairToolCall: createRepairToolCall(options.model),
-        messages: await convertToModelMessages(messages),
-        stopWhen: stepCountIs(50),
+        messages: await convertToModelMessages(messages, {
+          ignoreIncompleteToolCalls: true
+        }),
+        stopWhen: stepCountIs(200),
         experimental_transform: config?.transform ?? smoothStream(),
         experimental_context: contextVariables,
         output: Output.object({ schema: options.schema }),
@@ -591,432 +599,95 @@ async function estimate(modelId, renderer, ...fragments) {
 }
 
 // packages/context/src/lib/fragments/message/user.ts
-import { generateId as generateId3, isTextUIPart } from "ai";
-function everyNTurns(n) {
-  return ({ turn }) => turn % n === 0;
-}
-function once() {
-  return ({ turn }) => turn === 1;
-}
-function firstN(n) {
-  return ({ turn }) => turn <= n;
-}
-function afterTurn(n) {
-  return ({ turn }) => turn > n;
-}
-function and(...predicates) {
-  return (ctx) => predicates.every((p) => p(ctx));
-}
-function or(...predicates) {
-  return (ctx) => predicates.some((p) => p(ctx));
-}
-function not(predicate) {
-  return (ctx) => !predicate(ctx);
-}
-function contentIncludes(keywords) {
-  const lower = keywords.map((k) => k.toLowerCase());
-  return (ctx) => {
-    const text = ctx.content.toLowerCase();
-    return lower.some((kw) => text.includes(kw));
-  };
-}
-function contentPattern(pattern) {
-  return (ctx) => {
-    pattern.lastIndex = 0;
-    return pattern.test(ctx.content);
-  };
-}
-function toDateParts(date, tz) {
-  const parts = new Intl.DateTimeFormat("en-CA", {
-    timeZone: tz,
-    year: "numeric",
-    month: "2-digit",
-    day: "2-digit",
-    hour: "2-digit",
-    hourCycle: "h23"
-  }).formatToParts(date);
-  const get = (type) => parts.find((p) => p.type === type).value;
-  return {
-    year: get("year"),
-    month: get("month"),
-    day: get("day"),
-    hour: get("hour")
-  };
-}
-function temporalChanged(ctx, tz, getKey) {
-  if (ctx.lastMessageAt === void 0) return true;
-  const nowParts = toDateParts(/* @__PURE__ */ new Date(), tz);
-  const prevParts = toDateParts(new Date(ctx.lastMessageAt), tz);
-  return getKey(nowParts) !== getKey(prevParts);
-}
-function getSeason(month) {
-  if (month >= 2 && month <= 4) return "Spring";
-  if (month >= 5 && month <= 7) return "Summer";
-  if (month >= 8 && month <= 10) return "Fall";
-  return "Winter";
-}
-function isoWeekKey(parts) {
-  const d = new Date(
-    Date.UTC(
-      parseInt(parts.year),
-      parseInt(parts.month) - 1,
-      parseInt(parts.day)
-    )
-  );
-  d.setUTCDate(d.getUTCDate() + 4 - (d.getUTCDay() || 7));
-  const yearStart = new Date(Date.UTC(d.getUTCFullYear(), 0, 1));
-  const weekNo = Math.ceil(
-    ((d.getTime() - yearStart.getTime()) / 864e5 + 1) / 7
-  );
-  return `${d.getUTCFullYear()}-W${String(weekNo).padStart(2, "0")}`;
-}
-function dayChanged(tz = "UTC") {
-  return (ctx) => temporalChanged(ctx, tz, (p) => `${p.year}-${p.month}-${p.day}`);
-}
-function hourChanged(tz = "UTC") {
-  return (ctx) => temporalChanged(ctx, tz, (p) => `${p.year}-${p.month}-${p.day}-${p.hour}`);
-}
-function monthChanged(tz = "UTC") {
-  return (ctx) => temporalChanged(ctx, tz, (p) => `${p.year}-${p.month}`);
-}
-function yearChanged(tz = "UTC") {
-  return (ctx) => temporalChanged(ctx, tz, (p) => p.year);
-}
-function seasonChanged(tz = "UTC") {
-  return (ctx) => temporalChanged(ctx, tz, (p) => getSeason(parseInt(p.month) - 1));
-}
-function weekChanged(tz = "UTC") {
-  return (ctx) => temporalChanged(ctx, tz, isoWeekKey);
-}
-function isConditionalReminder(fragment2) {
-  return fragment2.name === "reminder" && !!fragment2.metadata?.reminder;
-}
-function getConditionalReminder(fragment2) {
-  return fragment2.metadata.reminder;
-}
-var SYSTEM_REMINDER_OPEN_TAG = "<system-reminder>";
-var SYSTEM_REMINDER_CLOSE_TAG = "</system-reminder>";
-function getReminderRanges(metadata) {
-  return metadata?.reminders ?? [];
-}
-function stripTextByRanges(text, ranges) {
-  if (ranges.length === 0) {
-    return text;
+import { generateId as generateId3 } from "ai";
+
+// packages/context/src/lib/renderers/abstract.renderer.ts
+import pluralize from "pluralize";
+import { titlecase } from "stringcase";
+var ContextRenderer = class {
+  options;
+  constructor(options = {}) {
+    this.options = options;
   }
-  const normalized = ranges.map((range) => ({
-    start: Math.max(0, Math.min(text.length, range.start)),
-    end: Math.max(0, Math.min(text.length, range.end))
-  })).filter((range) => range.end > range.start).sort((a, b) => a.start - b.start);
-  if (normalized.length === 0) {
-    return text;
+  /**
+   * Check if data is a primitive (string, number, boolean).
+   */
+  isPrimitive(data) {
+    return typeof data === "string" || typeof data === "number" || typeof data === "boolean";
   }
-  let cursor = 0;
-  let output = "";
-  for (const range of normalized) {
-    if (range.start < cursor) {
-      if (range.end > cursor) {
-        cursor = range.end;
+  /**
+   * Group fragments by name for groupFragments option.
+   */
+  groupByName(fragments) {
+    const groups = /* @__PURE__ */ new Map();
+    for (const fragment2 of fragments) {
+      const existing = groups.get(fragment2.name) ?? [];
+      existing.push(fragment2);
+      groups.set(fragment2.name, existing);
+    }
+    return groups;
+  }
+  /**
+   * Remove null/undefined from fragments and fragment data recursively.
+   * This protects renderers from nullish values and ensures they are ignored
+   * consistently across all output formats.
+   */
+  sanitizeFragments(fragments) {
+    const sanitized = [];
+    for (const fragment2 of fragments) {
+      const cleaned = this.sanitizeFragment(fragment2, /* @__PURE__ */ new WeakSet());
+      if (cleaned) {
+        sanitized.push(cleaned);
       }
-      continue;
     }
-    output += text.slice(cursor, range.start);
-    cursor = range.end;
+    return sanitized;
   }
-  output += text.slice(cursor);
-  return output.trimEnd();
-}
-function stripReminders(message2) {
-  const reminderRanges = getReminderRanges(
-    isRecord(message2.metadata) ? message2.metadata : void 0
-  );
-  const rangesByPartIndex = /* @__PURE__ */ new Map();
-  for (const range of reminderRanges) {
-    const partRanges = rangesByPartIndex.get(range.partIndex) ?? [];
-    partRanges.push({ start: range.start, end: range.end });
-    rangesByPartIndex.set(range.partIndex, partRanges);
+  sanitizeFragment(fragment2, seen) {
+    const data = this.sanitizeData(getFragmentData(fragment2), seen);
+    if (data == null) {
+      return null;
+    }
+    return {
+      ...fragment2,
+      data
+    };
   }
-  const strippedParts = message2.parts.flatMap((part, partIndex) => {
-    const clonedPart = { ...part };
-    const ranges = rangesByPartIndex.get(partIndex);
-    if (clonedPart.type !== "text" || ranges === void 0) {
-      return [clonedPart];
+  sanitizeData(data, seen) {
+    if (data == null) {
+      return void 0;
     }
-    const strippedText = stripTextByRanges(clonedPart.text, ranges);
-    if (strippedText.length === 0) {
-      return [];
+    if (isFragment(data)) {
+      return this.sanitizeFragment(data, seen) ?? void 0;
     }
-    return [{ ...clonedPart, text: strippedText }];
-  });
-  const nextMessage = {
-    ...message2,
-    parts: strippedParts
-  };
-  if (isRecord(message2.metadata)) {
-    const metadata = { ...message2.metadata };
-    delete metadata.reminders;
-    if (Object.keys(metadata).length > 0) {
-      nextMessage.metadata = metadata;
-    } else {
-      delete nextMessage.metadata;
+    if (Array.isArray(data)) {
+      if (seen.has(data)) {
+        return void 0;
+      }
+      seen.add(data);
+      const cleaned = [];
+      for (const item of data) {
+        const sanitizedItem = this.sanitizeData(item, seen);
+        if (sanitizedItem != null) {
+          cleaned.push(sanitizedItem);
+        }
+      }
+      return cleaned;
     }
-  }
-  return nextMessage;
-}
-function extractPlainText(message2) {
-  return message2.parts.filter(isTextUIPart).map((part) => part.text).join(" ");
-}
-function isRecord(value) {
-  return typeof value === "object" && value !== null && !Array.isArray(value);
-}
-function assertReminderText(text) {
-  if (text.trim().length === 0) {
-    throw new Error("Reminder text must not be empty");
-  }
-}
-function formatTaggedReminder(text) {
-  return `${SYSTEM_REMINDER_OPEN_TAG}${text}${SYSTEM_REMINDER_CLOSE_TAG}`;
-}
-function findLastTextPartIndex(message2) {
-  for (let i = message2.parts.length - 1; i >= 0; i--) {
-    if (message2.parts[i].type === "text") {
-      return i;
+    if (isFragmentObject(data)) {
+      if (seen.has(data)) {
+        return void 0;
+      }
+      seen.add(data);
+      const cleaned = {};
+      for (const [key, value] of Object.entries(data)) {
+        const sanitizedValue = this.sanitizeData(value, seen);
+        if (sanitizedValue != null) {
+          cleaned[key] = sanitizedValue;
+        }
+      }
+      return cleaned;
     }
-  }
-  return void 0;
-}
-function ensureTextPart(message2) {
-  const existingIndex = findLastTextPartIndex(message2);
-  if (existingIndex !== void 0) {
-    return existingIndex;
-  }
-  const reminderPart = {
-    type: "text",
-    text: ""
-  };
-  message2.parts.push(reminderPart);
-  return message2.parts.length - 1;
-}
-function applyInlineReminder(message2, value) {
-  const partIndex = ensureTextPart(message2);
-  const textPart = message2.parts[partIndex];
-  if (textPart.type !== "text") {
-    throw new Error("Failed to resolve text part for inline reminder");
-  }
-  const reminderText = formatTaggedReminder(value);
-  const start = textPart.text.length;
-  const updatedText = `${textPart.text}${reminderText}`;
-  message2.parts[partIndex] = { ...textPart, text: updatedText };
-  return {
-    id: generateId3(),
-    text: value,
-    partIndex,
-    start,
-    end: start + reminderText.length,
-    mode: "inline"
-  };
-}
-function applyPartReminder(message2, value) {
-  const part = { type: "text", text: value };
-  message2.parts.push(part);
-  const partIndex = message2.parts.length - 1;
-  return {
-    id: generateId3(),
-    text: value,
-    partIndex,
-    start: 0,
-    end: value.length,
-    mode: "part"
-  };
-}
-function resolveReminderText(item, ctx) {
-  return resolveReminder(item, ctx)?.text ?? "";
-}
-function normalizeReminderResolution(value) {
-  if (typeof value === "string") {
-    return value.trim().length === 0 ? null : { text: value };
-  }
-  if (value.text.trim().length === 0) {
-    return null;
-  }
-  return value;
-}
-function resolveReminder(item, ctx) {
-  const resolvedText = typeof item.text === "function" ? item.text(ctx) : item.text;
-  const resolved = normalizeReminderResolution(resolvedText);
-  if (!resolved) {
-    return null;
-  }
-  const metadata = item.metadata || resolved.metadata ? {
-    ...item.metadata ?? {},
-    ...resolved.metadata ?? {}
-  } : void 0;
-  return metadata ? { ...resolved, metadata } : resolved;
-}
-function mergeMessageMetadata(message2, addedMetadata) {
-  if (Object.keys(addedMetadata).length === 0) {
-    return;
-  }
-  const metadata = isRecord(message2.metadata) ? { ...message2.metadata } : {};
-  message2.metadata = { ...metadata, ...addedMetadata };
-}
-function applyReminderToMessage(message2, item, ctx) {
-  const resolved = resolveReminder(item, ctx);
-  if (!resolved) {
-    return null;
-  }
-  if (resolved.metadata) {
-    mergeMessageMetadata(message2, resolved.metadata);
-  }
-  return item.asPart ? applyPartReminder(message2, resolved.text) : applyInlineReminder(message2, resolved.text);
-}
-function mergeReminderMetadata(message2, addedReminders) {
-  if (addedReminders.length === 0) return;
-  const metadata = isRecord(message2.metadata) ? { ...message2.metadata } : {};
-  const existing = Array.isArray(metadata.reminders) ? metadata.reminders : [];
-  metadata.reminders = [...existing, ...addedReminders];
-  message2.metadata = metadata;
-}
-function reminder(text, options) {
-  if (typeof text === "string") {
-    assertReminderText(text);
-  }
-  if (options && "when" in options && options.when) {
-    return {
-      name: "reminder",
-      metadata: {
-        reminder: {
-          text,
-          when: options.when,
-          asPart: options.asPart ?? false
-        }
-      }
-    };
-  }
-  return {
-    text,
-    asPart: options?.asPart ?? false
-  };
-}
-function user(content, ...reminders) {
-  const message2 = typeof content === "string" ? {
-    id: generateId3(),
-    role: "user",
-    parts: [{ type: "text", text: content }]
-  } : { ...content, role: "user", parts: [...content.parts] };
-  if (reminders.length > 0) {
-    const plainText = extractPlainText(message2);
-    const added = [];
-    for (const item of reminders) {
-      const meta = applyReminderToMessage(message2, item, {
-        content: plainText
-      });
-      if (meta) added.push(meta);
-    }
-    mergeReminderMetadata(message2, added);
-  }
-  return {
-    id: message2.id,
-    name: "user",
-    type: "message",
-    persist: true,
-    codec: {
-      decode() {
-        return message2;
-      },
-      encode() {
-        return message2;
-      }
-    }
-  };
-}
-
-// packages/context/src/lib/renderers/abstract.renderer.ts
-import pluralize from "pluralize";
-import { titlecase } from "stringcase";
-var ContextRenderer = class {
-  options;
-  constructor(options = {}) {
-    this.options = options;
-  }
-  /**
-   * Check if data is a primitive (string, number, boolean).
-   */
-  isPrimitive(data) {
-    return typeof data === "string" || typeof data === "number" || typeof data === "boolean";
-  }
-  /**
-   * Group fragments by name for groupFragments option.
-   */
-  groupByName(fragments) {
-    const groups = /* @__PURE__ */ new Map();
-    for (const fragment2 of fragments) {
-      const existing = groups.get(fragment2.name) ?? [];
-      existing.push(fragment2);
-      groups.set(fragment2.name, existing);
-    }
-    return groups;
-  }
-  /**
-   * Remove null/undefined from fragments and fragment data recursively.
-   * This protects renderers from nullish values and ensures they are ignored
-   * consistently across all output formats.
-   */
-  sanitizeFragments(fragments) {
-    const sanitized = [];
-    for (const fragment2 of fragments) {
-      const cleaned = this.sanitizeFragment(fragment2, /* @__PURE__ */ new WeakSet());
-      if (cleaned) {
-        sanitized.push(cleaned);
-      }
-    }
-    return sanitized;
-  }
-  sanitizeFragment(fragment2, seen) {
-    const data = this.sanitizeData(getFragmentData(fragment2), seen);
-    if (data == null) {
-      return null;
-    }
-    return {
-      ...fragment2,
-      data
-    };
-  }
-  sanitizeData(data, seen) {
-    if (data == null) {
-      return void 0;
-    }
-    if (isFragment(data)) {
-      return this.sanitizeFragment(data, seen) ?? void 0;
-    }
-    if (Array.isArray(data)) {
-      if (seen.has(data)) {
-        return void 0;
-      }
-      seen.add(data);
-      const cleaned = [];
-      for (const item of data) {
-        const sanitizedItem = this.sanitizeData(item, seen);
-        if (sanitizedItem != null) {
-          cleaned.push(sanitizedItem);
-        }
-      }
-      return cleaned;
-    }
-    if (isFragmentObject(data)) {
-      if (seen.has(data)) {
-        return void 0;
-      }
-      seen.add(data);
-      const cleaned = {};
-      for (const [key, value] of Object.entries(data)) {
-        const sanitizedValue = this.sanitizeData(value, seen);
-        if (sanitizedValue != null) {
-          cleaned[key] = sanitizedValue;
-        }
-      }
-      return cleaned;
-    }
-    return data;
+    return data;
   }
   /**
    * Template method - dispatches value to appropriate handler.
@@ -1806,24 +1477,391 @@ ${entries}`;
   }
 };
 
-// packages/context/src/lib/store/store.ts
-var ContextStore = class {
-};
+// packages/context/src/lib/text.ts
+import { isTextUIPart } from "ai";
+function getTextParts(message2) {
+  return message2.parts.filter(isTextUIPart).map((part) => part.text);
+}
+function extractPlainText(message2) {
+  return getTextParts(message2).join(" ");
+}
 
-// packages/context/src/lib/engine.ts
-function estimateMessageContent(data) {
-  return typeof data === "string" ? data : JSON.stringify(data);
+// packages/context/src/lib/fragments/message/user.ts
+function everyNTurns(n) {
+  return ({ turn }) => turn % n === 0;
 }
-var ContextEngine = class {
-  /** Non-message fragments (role, hints, etc.) - not persisted in graph */
-  #fragments = [];
-  /** Pending message fragments to be added to graph */
-  #pendingMessages = [];
-  #store;
-  #chatId;
-  #userId;
-  #branchName;
-  #branch = null;
+function once() {
+  return ({ turn }) => turn === 1;
+}
+function firstN(n) {
+  return ({ turn }) => turn <= n;
+}
+function afterTurn(n) {
+  return ({ turn }) => turn > n;
+}
+function and(...predicates) {
+  return async (ctx) => {
+    for (const it of predicates) {
+      if (!await it(ctx)) return false;
+    }
+    return true;
+  };
+}
+function or(...predicates) {
+  return async (ctx) => {
+    for (const it of predicates) {
+      if (await it(ctx)) return true;
+    }
+    return false;
+  };
+}
+function not(predicate) {
+  return async (ctx) => !await predicate(ctx);
+}
+function contentIncludes(keywords) {
+  const lower = keywords.map((k) => k.toLowerCase());
+  return (ctx) => {
+    const text = ctx.content.toLowerCase();
+    return lower.some((kw) => text.includes(kw));
+  };
+}
+function contentPattern(pattern) {
+  return (ctx) => {
+    pattern.lastIndex = 0;
+    return pattern.test(ctx.content);
+  };
+}
+function toDateParts(date, tz) {
+  const parts = new Intl.DateTimeFormat("en-CA", {
+    timeZone: tz,
+    year: "numeric",
+    month: "2-digit",
+    day: "2-digit",
+    hour: "2-digit",
+    hourCycle: "h23"
+  }).formatToParts(date);
+  const get = (type) => parts.find((p) => p.type === type).value;
+  return {
+    year: get("year"),
+    month: get("month"),
+    day: get("day"),
+    hour: get("hour")
+  };
+}
+function temporalChanged(ctx, tz, getKey) {
+  if (ctx.lastMessageAt === void 0) return true;
+  const nowParts = toDateParts(/* @__PURE__ */ new Date(), tz);
+  const prevParts = toDateParts(new Date(ctx.lastMessageAt), tz);
+  return getKey(nowParts) !== getKey(prevParts);
+}
+function getSeason(month) {
+  if (month >= 2 && month <= 4) return "Spring";
+  if (month >= 5 && month <= 7) return "Summer";
+  if (month >= 8 && month <= 10) return "Fall";
+  return "Winter";
+}
+function isoWeekKey(parts) {
+  const d = new Date(
+    Date.UTC(
+      parseInt(parts.year),
+      parseInt(parts.month) - 1,
+      parseInt(parts.day)
+    )
+  );
+  d.setUTCDate(d.getUTCDate() + 4 - (d.getUTCDay() || 7));
+  const yearStart = new Date(Date.UTC(d.getUTCFullYear(), 0, 1));
+  const weekNo = Math.ceil(
+    ((d.getTime() - yearStart.getTime()) / 864e5 + 1) / 7
+  );
+  return `${d.getUTCFullYear()}-W${String(weekNo).padStart(2, "0")}`;
+}
+function dayChanged(tz = "UTC") {
+  return (ctx) => temporalChanged(ctx, tz, (p) => `${p.year}-${p.month}-${p.day}`);
+}
+function hourChanged(tz = "UTC") {
+  return (ctx) => temporalChanged(ctx, tz, (p) => `${p.year}-${p.month}-${p.day}-${p.hour}`);
+}
+function monthChanged(tz = "UTC") {
+  return (ctx) => temporalChanged(ctx, tz, (p) => `${p.year}-${p.month}`);
+}
+function yearChanged(tz = "UTC") {
+  return (ctx) => temporalChanged(ctx, tz, (p) => p.year);
+}
+function seasonChanged(tz = "UTC") {
+  return (ctx) => temporalChanged(ctx, tz, (p) => getSeason(parseInt(p.month) - 1));
+}
+function weekChanged(tz = "UTC") {
+  return (ctx) => temporalChanged(ctx, tz, isoWeekKey);
+}
+function isConditionalReminder(fragment2) {
+  return fragment2.name === "reminder" && !!fragment2.metadata?.reminder;
+}
+function getConditionalReminder(fragment2) {
+  return fragment2.metadata.reminder;
+}
+var SYSTEM_REMINDER_OPEN_TAG = "<system-reminder>";
+var SYSTEM_REMINDER_CLOSE_TAG = "</system-reminder>";
+function getReminderRanges(metadata) {
+  return metadata?.reminders ?? [];
+}
+function stripTextByRanges(text, ranges) {
+  if (ranges.length === 0) {
+    return text;
+  }
+  const normalized = ranges.map((range) => ({
+    start: Math.max(0, Math.min(text.length, range.start)),
+    end: Math.max(0, Math.min(text.length, range.end))
+  })).filter((range) => range.end > range.start).sort((a, b) => a.start - b.start);
+  if (normalized.length === 0) {
+    return text;
+  }
+  let cursor = 0;
+  let output = "";
+  for (const range of normalized) {
+    if (range.start < cursor) {
+      if (range.end > cursor) {
+        cursor = range.end;
+      }
+      continue;
+    }
+    output += text.slice(cursor, range.start);
+    cursor = range.end;
+  }
+  output += text.slice(cursor);
+  return output.trimEnd();
+}
+function stripReminders(message2) {
+  const reminderRanges = getReminderRanges(
+    isRecord(message2.metadata) ? message2.metadata : void 0
+  );
+  const rangesByPartIndex = /* @__PURE__ */ new Map();
+  for (const range of reminderRanges) {
+    const partRanges = rangesByPartIndex.get(range.partIndex) ?? [];
+    partRanges.push({ start: range.start, end: range.end });
+    rangesByPartIndex.set(range.partIndex, partRanges);
+  }
+  const strippedParts = message2.parts.flatMap((part, partIndex) => {
+    const clonedPart = { ...part };
+    const ranges = rangesByPartIndex.get(partIndex);
+    if (clonedPart.type !== "text" || ranges === void 0) {
+      return [clonedPart];
+    }
+    const strippedText = stripTextByRanges(clonedPart.text, ranges);
+    if (strippedText.length === 0) {
+      return [];
+    }
+    return [{ ...clonedPart, text: strippedText }];
+  });
+  const nextMessage = {
+    ...message2,
+    parts: strippedParts
+  };
+  if (isRecord(message2.metadata)) {
+    const metadata = { ...message2.metadata };
+    delete metadata.reminders;
+    if (Object.keys(metadata).length > 0) {
+      nextMessage.metadata = metadata;
+    } else {
+      delete nextMessage.metadata;
+    }
+  }
+  return nextMessage;
+}
+function isRecord(value) {
+  return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+function assertReminderText(text) {
+  if (text.trim().length === 0) {
+    throw new Error("Reminder text must not be empty");
+  }
+}
+function formatTaggedReminder(text) {
+  return `${SYSTEM_REMINDER_OPEN_TAG}${text}${SYSTEM_REMINDER_CLOSE_TAG}`;
+}
+function findLastTextPartIndex(message2) {
+  for (let i = message2.parts.length - 1; i >= 0; i--) {
+    if (message2.parts[i].type === "text") {
+      return i;
+    }
+  }
+  return void 0;
+}
+function ensureTextPart(message2) {
+  const existingIndex = findLastTextPartIndex(message2);
+  if (existingIndex !== void 0) {
+    return existingIndex;
+  }
+  const reminderPart = {
+    type: "text",
+    text: ""
+  };
+  message2.parts.push(reminderPart);
+  return message2.parts.length - 1;
+}
+function applyInlineReminder(message2, value) {
+  const partIndex = ensureTextPart(message2);
+  const textPart = message2.parts[partIndex];
+  if (textPart.type !== "text") {
+    throw new Error("Failed to resolve text part for inline reminder");
+  }
+  const reminderText = formatTaggedReminder(value);
+  const start = textPart.text.length;
+  const updatedText = `${textPart.text}${reminderText}`;
+  message2.parts[partIndex] = { ...textPart, text: updatedText };
+  return {
+    id: generateId3(),
+    text: value,
+    partIndex,
+    start,
+    end: start + reminderText.length,
+    mode: "inline"
+  };
+}
+function applyPartReminder(message2, value) {
+  const part = { type: "text", text: value };
+  message2.parts.push(part);
+  const partIndex = message2.parts.length - 1;
+  return {
+    id: generateId3(),
+    text: value,
+    partIndex,
+    start: 0,
+    end: value.length,
+    mode: "part"
+  };
+}
+function resolveReminderText(item, ctx) {
+  return resolveReminder(item, ctx)?.text ?? "";
+}
+function normalizeReminderResolution(value) {
+  if (typeof value === "string") {
+    return value.trim().length === 0 ? null : { text: value };
+  }
+  if (value.text.trim().length === 0) {
+    return null;
+  }
+  return value;
+}
+function resolveReminder(item, ctx) {
+  const resolvedText = typeof item.text === "function" ? item.text(ctx) : item.text;
+  const resolved = normalizeReminderResolution(resolvedText);
+  if (!resolved) {
+    return null;
+  }
+  const metadata = item.metadata || resolved.metadata ? {
+    ...item.metadata ?? {},
+    ...resolved.metadata ?? {}
+  } : void 0;
+  return metadata ? { ...resolved, metadata } : resolved;
+}
+async function resolveReminderAsync(item, ctx) {
+  const text = await (typeof item.text === "function" ? item.text(ctx) : item.text);
+  const resolved = normalizeReminderResolution(text);
+  if (!resolved) return null;
+  const metadata = item.metadata || resolved.metadata ? { ...item.metadata ?? {}, ...resolved.metadata ?? {} } : void 0;
+  return metadata ? { ...resolved, metadata } : resolved;
+}
+function mergeMessageMetadata(message2, addedMetadata) {
+  if (Object.keys(addedMetadata).length === 0) {
+    return;
+  }
+  const metadata = isRecord(message2.metadata) ? { ...message2.metadata } : {};
+  message2.metadata = { ...metadata, ...addedMetadata };
+}
+function applyReminderToMessage(message2, item, ctx) {
+  const resolved = resolveReminder(item, ctx);
+  if (!resolved) {
+    return null;
+  }
+  if (resolved.metadata) {
+    mergeMessageMetadata(message2, resolved.metadata);
+  }
+  return item.asPart ? applyPartReminder(message2, resolved.text) : applyInlineReminder(message2, resolved.text);
+}
+function mergeReminderMetadata(message2, addedReminders) {
+  if (addedReminders.length === 0) return;
+  const metadata = isRecord(message2.metadata) ? { ...message2.metadata } : {};
+  const existing = Array.isArray(metadata.reminders) ? metadata.reminders : [];
+  metadata.reminders = [...existing, ...addedReminders];
+  message2.metadata = metadata;
+}
+function reminder(textOrFragment, options) {
+  const text = isFragment(textOrFragment) ? new XmlRenderer().render([textOrFragment]) : textOrFragment;
+  if (typeof text === "string") {
+    assertReminderText(text);
+  }
+  if (options && "when" in options && options.when) {
+    return {
+      name: "reminder",
+      data: null,
+      metadata: {
+        reminder: {
+          text,
+          when: options.when,
+          asPart: options.asPart ?? false
+        }
+      }
+    };
+  }
+  return {
+    text,
+    asPart: options?.asPart ?? false
+  };
+}
+function user(content, ...reminders) {
+  const message2 = typeof content === "string" ? {
+    id: generateId3(),
+    role: "user",
+    parts: [{ type: "text", text: content }]
+  } : { ...content, role: "user", parts: [...content.parts] };
+  if (reminders.length > 0) {
+    const plainText = extractPlainText(message2);
+    const added = [];
+    for (const item of reminders) {
+      const meta = applyReminderToMessage(message2, item, {
+        content: plainText
+      });
+      if (meta) added.push(meta);
+    }
+    mergeReminderMetadata(message2, added);
+  }
+  return {
+    id: message2.id,
+    name: "user",
+    type: "message",
+    persist: true,
+    codec: {
+      decode() {
+        return message2;
+      },
+      encode() {
+        return message2;
+      }
+    }
+  };
+}
+
+// packages/context/src/lib/store/store.ts
+var ContextStore = class {
+};
+
+// packages/context/src/lib/engine.ts
+function estimateMessageContent(data) {
+  return typeof data === "string" ? data : JSON.stringify(data);
+}
+function isLanguageModelUsage(value) {
+  return typeof value === "object" && value !== null && "totalTokens" in value;
+}
+var ContextEngine = class {
+  /** Non-message fragments (role, hints, etc.) - not persisted in graph */
+  #fragments = [];
+  /** Pending message fragments to be added to graph */
+  #pendingMessages = [];
+  #store;
+  #chatId;
+  #userId;
+  #branchName;
+  #branch = null;
   #chatData = null;
   #initialized = false;
   /** Initial metadata to merge on first initialization */
@@ -1949,17 +1987,7 @@ var ContextEngine = class {
    * Returns null if the chat hasn't been initialized yet.
    */
   get chat() {
-    if (!this.#chatData) {
-      return null;
-    }
-    return {
-      id: this.#chatData.id,
-      userId: this.#chatData.userId,
-      createdAt: this.#chatData.createdAt,
-      updatedAt: this.#chatData.updatedAt,
-      title: this.#chatData.title,
-      metadata: this.#chatData.metadata
-    };
+    return this.#chatData;
   }
   /**
    * Count user turns in the conversation and return the previous saved user message context.
@@ -1968,13 +1996,19 @@ var ContextEngine = class {
   async #getChainContext() {
     await this.#ensureInitialized();
     let turn = 0;
+    let messageCount = 0;
     let lastMessageAt;
     let lastMessage;
+    let lastAssistantMessage2;
     if (this.#branch?.headMessageId) {
       const chain = await this.#store.getMessageChain(
         this.#branch.headMessageId
       );
       for (const msg of chain) {
+        messageCount++;
+        if (msg.name === "assistant") {
+          lastAssistantMessage2 = msg.data;
+        }
         if (msg.name !== "user") {
           continue;
         }
@@ -1984,9 +2018,16 @@ var ContextEngine = class {
       }
     }
     for (const fragment2 of this.#pendingMessages) {
+      messageCount++;
       if (fragment2.name === "user") turn++;
     }
-    return { turn, lastMessageAt, lastMessage };
+    return {
+      turn,
+      messageCount,
+      lastMessageAt,
+      lastMessage,
+      lastAssistantMessage: lastAssistantMessage2
+    };
   }
   async getTurnCount() {
     const { turn } = await this.#getChainContext();
@@ -2116,32 +2157,45 @@ var ContextEngine = class {
       if (lastUserIndex >= 0) {
         const lastUserFragment = this.#pendingMessages[lastUserIndex];
         if (lastUserFragment.codec) {
-          const { turn, lastMessageAt, lastMessage } = await this.#getChainContext();
+          const {
+            turn,
+            messageCount,
+            lastMessageAt,
+            lastMessage,
+            lastAssistantMessage: lastAssistantMessage2
+          } = await this.#getChainContext();
           const original = lastUserFragment.codec.encode();
           const plainText = extractPlainText(original);
-          const firedReminders = conditionalReminders.map(getConditionalReminder).filter(
-            (config) => config.when({
-              turn,
-              content: plainText,
-              lastMessageAt,
-              lastMessage
-            })
+          const rawUsage = this.#chatData?.metadata?.usage;
+          const usage = isLanguageModelUsage(rawUsage) ? rawUsage : void 0;
+          const elapsed = lastMessageAt !== void 0 ? Date.now() - lastMessageAt : void 0;
+          const whenCtx = {
+            turn,
+            content: plainText,
+            lastMessageAt,
+            lastMessage,
+            chat: this.#chatData,
+            usage,
+            branch: this.#branchName,
+            elapsed,
+            messageCount,
+            lastAssistantMessage: lastAssistantMessage2
+          };
+          const configs = conditionalReminders.map(getConditionalReminder);
+          const whenResults = await Promise.all(
+            configs.map((it) => it.when(whenCtx))
           );
+          const firedReminders = configs.filter((_, i) => whenResults[i]);
           if (firedReminders.length > 0) {
-            const reminders = firedReminders.flatMap((rem) => {
-              const resolved = resolveReminder(rem, {
-                content: plainText,
-                turn,
-                lastMessageAt,
-                lastMessage
-              });
-              if (!resolved) {
-                return [];
-              }
+            const resolvedOrNull = await Promise.all(
+              firedReminders.map((it) => resolveReminderAsync(it, whenCtx))
+            );
+            const reminders = resolvedOrNull.flatMap((resolved, i) => {
+              if (!resolved) return [];
               return [
                 {
                   text: resolved.text,
-                  asPart: rem.asPart,
+                  asPart: firedReminders[i].asPart,
                   metadata: resolved.metadata
                 }
               ];
@@ -7959,7 +8013,6 @@ export {
   everyNTurns,
   example,
   explain,
-  extractPlainText,
   fail,
   firstN,
   fragment,
@@ -8010,6 +8063,7 @@ export {
   render,
   resetAdaptivePolling,
   resolveReminder,
+  resolveReminderAsync,
   resolveReminderText,
   role,
   runGuardrailChain,