@deepagents/context 0.38.0 → 0.39.0

package/dist/index.js

@@ -199,6 +199,12 @@ function assistant(message2) {
     }
   };
 }
+function toMessageFragment(item) {
+  if (isFragment(item) && isMessageFragment(item)) {
+    return item;
+  }
+  return message(item);
+}
 function message(content) {
   const message2 = typeof content === "string" ? {
     id: generateId(),
@@ -1396,27 +1402,27 @@ import {
 } from "ai";
 import chalk from "chalk";
 function createRepairToolCall(model, abortSignal) {
-  return async ({ toolCall, tools, inputSchema, error }) => {
+  return async ({ toolCall: toolCall2, tools, inputSchema, error }) => {
     if (NoSuchToolError.isInstance(error)) {
       return null;
     }
     console.log(
-      `Debug: ${chalk.yellow("RepairingToolCall")}: ${toolCall.toolName}`
+      `Debug: ${chalk.yellow("RepairingToolCall")}: ${toolCall2.toolName}`
     );
-    const tool3 = tools[toolCall.toolName];
+    const tool3 = tools[toolCall2.toolName];
     const { output } = await generateText({
       abortSignal,
       model,
       output: Output.object({ schema: tool3.inputSchema }),
       prompt: [
-        `The model tried to call the tool "${toolCall.toolName}" with the following inputs:`,
-        JSON.stringify(toolCall.input),
+        `The model tried to call the tool "${toolCall2.toolName}" with the following inputs:`,
+        JSON.stringify(toolCall2.input),
         `The tool accepts the following schema:`,
-        JSON.stringify(inputSchema(toolCall)),
+        JSON.stringify(inputSchema(toolCall2)),
         "Please fix the inputs."
       ].join("\n")
     });
-    return { ...toolCall, input: JSON.stringify(output) };
+    return { ...toolCall2, input: JSON.stringify(output) };
   };
 }
 
@@ -1464,10 +1470,10 @@ var Agent = class _Agent {
       ),
       toolChoice: this.#options.toolChoice,
       onStepFinish: (step) => {
-        const toolCall = step.toolCalls.at(-1);
-        if (toolCall) {
+        const toolCall2 = step.toolCalls.at(-1);
+        if (toolCall2) {
           console.log(
-            `Debug: ${chalk2.yellow("ToolCalled")}: ${toolCall.toolName}(${JSON.stringify(toolCall.input)})`
+            `Debug: ${chalk2.yellow("ToolCalled")}: ${toolCall2.toolName}(${JSON.stringify(toolCall2.input)})`
           );
         }
       }
@@ -1538,10 +1544,10 @@ var Agent = class _Agent {
       experimental_context: contextVariables,
       toolChoice: this.#options.toolChoice,
       onStepFinish: (step) => {
-        const toolCall = step.toolCalls.at(-1);
-        if (toolCall) {
+        const toolCall2 = step.toolCalls.at(-1);
+        if (toolCall2) {
           console.log(
-            `Debug: (${runId}) ${chalk2.bold.yellow("ToolCalled")}: ${toolCall.toolName}(${JSON.stringify(toolCall.input)})`
+            `Debug: (${runId}) ${chalk2.bold.yellow("ToolCalled")}: ${toolCall2.toolName}(${JSON.stringify(toolCall2.input)})`
           );
         }
       }
@@ -1869,7 +1875,6 @@ import {
   NoSuchToolError as NoSuchToolError2,
   ToolCallRepairError,
   createUIMessageStream as createUIMessageStream2,
-  generateId as generateId4,
   isToolUIPart
 } from "ai";
 
@@ -1877,7 +1882,10 @@ import {
 import { z as z2 } from "zod";
 
 // packages/context/src/lib/engine.ts
-import { validateUIMessages } from "ai";
+import {
+  generateId as generateId4,
+  validateUIMessages
+} from "ai";
 import { mergeWith } from "lodash-es";
 
 // packages/context/src/lib/estimate.ts
@@ -2585,6 +2593,9 @@ function estimateMessageContent(data) {
 function isLanguageModelUsage(value) {
   return typeof value === "object" && value !== null && "totalTokens" in value;
 }
+function isEmptyAssistantPlaceholder(message2) {
+  return message2.role === "assistant" && message2.parts.length === 0;
+}
 var ContextEngine = class _ContextEngine {
   /** Non-message fragments (role, hints, etc.) - not persisted in graph */
   #fragments = [];
@@ -2733,6 +2744,7 @@ var ContextEngine = class _ContextEngine {
     let lastMessageAt;
     let lastMessage;
     let lastAssistantMessage;
+    const lastAssistantMessages = [];
     if (this.#branch?.headMessageId) {
       const chain = await this.#store.getMessageChain(
         this.#branch.headMessageId
@@ -2740,7 +2752,9 @@ var ContextEngine = class _ContextEngine {
       for (const msg of chain) {
         messageCount++;
         if (msg.name === "assistant") {
-          lastAssistantMessage = msg.data;
+          const assistantMsg = msg.data;
+          lastAssistantMessage = assistantMsg;
+          lastAssistantMessages.push(assistantMsg);
         }
         if (msg.name !== "user") {
           continue;
@@ -2759,13 +2773,128 @@ var ContextEngine = class _ContextEngine {
       messageCount,
       lastMessageAt,
       lastMessage,
-      lastAssistantMessage
+      lastAssistantMessage,
+      lastAssistantMessages
     };
   }
   async getTurnCount() {
     const { turn } = await this.#getChainContext();
     return turn;
   }
+  async firstUserMessage() {
+    await this.#ensureInitialized();
+    if (this.#branch?.headMessageId) {
+      const chain = await this.#store.getMessageChain(
+        this.#branch.headMessageId
+      );
+      for (const msg of chain) {
+        if (msg.name === "user") return msg.data;
+      }
+    }
+    for (const fragment2 of this.#pendingMessages) {
+      if (fragment2.name !== "user") continue;
+      if (!fragment2.codec) {
+        throw new Error(`Fragment "${fragment2.name}" is missing codec.`);
+      }
+      return fragment2.codec.encode();
+    }
+    return void 0;
+  }
+  /**
+   * Return the head of the conversation — pending tail or persisted branch head.
+   *
+   * Includes empty assistant placeholders (use this for id-lookup, not for
+   * building model prompts — see `getMessages()` for prompt-ready output).
+   *
+   * @throws if the pending tail is missing an id (programming error).
+   */
+  async headMessage() {
+    await this.#ensureInitialized();
+    if (this.#pendingMessages.length > 0) {
+      const tail = this.#pendingMessages[this.#pendingMessages.length - 1];
+      if (!tail.id) {
+        throw new Error(
+          `headMessage: pending fragment "${tail.name}" is missing id`
+        );
+      }
+      return { id: tail.id, name: tail.name };
+    }
+    if (this.#branch?.headMessageId) {
+      const msg = await this.#store.getMessage(this.#branch.headMessageId);
+      if (msg) return { id: msg.id, name: msg.name };
+    }
+    return void 0;
+  }
+  /**
+   * Return the model-ready conversation: persisted chain plus pending fragments,
+   * with empty assistant placeholders filtered out.
+   *
+   * For id-lookup use `headMessage()` instead — that one keeps placeholders.
+   */
+  async getMessages() {
+    await this.#ensureInitialized();
+    const messages = [];
+    if (this.#branch?.headMessageId) {
+      const chain = await this.#store.getMessageChain(
+        this.#branch.headMessageId
+      );
+      for (const msg of chain) {
+        const data = msg.data;
+        if (isEmptyAssistantPlaceholder(data)) continue;
+        messages.push(data);
+      }
+    }
+    for (const fragment2 of this.#pendingMessages) {
+      if (!fragment2.codec) {
+        throw new Error(`Fragment "${fragment2.name}" is missing codec.`);
+      }
+      const encoded = fragment2.codec.encode();
+      if (isEmptyAssistantPlaceholder(encoded)) continue;
+      messages.push(encoded);
+    }
+    return messages.length === 0 ? [] : validateUIMessages({ messages });
+  }
+  /**
+   * Advance the conversation by one turn. Required setup before `chat()`.
+   *
+   * - User input → persists the message AND appends an empty assistant
+   *   placeholder reserving the id of the next streamed response.
+   * - Assistant input (tool-resume / continuation) → persists in-place
+   *   (`branch: false`), reusing the input's id.
+   *
+   * Always leaves the chain head as an assistant fragment, satisfying chat()'s
+   * precondition.
+   *
+   * @returns the assistant id that will receive the streamed response — useful
+   *   for telemetry, optimistic UI, or correlating logs before the stream starts.
+   * @throws if assistant input is missing an id.
+   *
+   * @example
+   * ```ts
+   * const assistantId = await context.continue(user('hi'));
+   * const stream = await chat(agent); // streams into assistantId
+   * ```
+   */
+  async continue(input) {
+    const fragment2 = toMessageFragment(input);
+    const isAssistantUpdate = fragment2.name === "assistant";
+    let assistantId;
+    if (isAssistantUpdate) {
+      if (!fragment2.id) {
+        throw new Error("continue: assistant input is missing id");
+      }
+      assistantId = fragment2.id;
+      this.set(fragment2);
+    } else {
+      assistantId = generateId4();
+      this.set(
+        fragment2,
+        assistant({ id: assistantId, role: "assistant", parts: [] })
+      );
+    }
+    await this.save({ branch: !isAssistantUpdate });
+    return assistantId;
+  }
   /**
    * Add fragments to the context.
    *
@@ -2812,25 +2941,8 @@ var ContextEngine = class _ContextEngine {
   async resolve(options) {
     await this.#ensureInitialized();
     const systemPrompt = options.renderer.render(this.#renderableFragments);
-    const messages = [];
-    if (this.#branch?.headMessageId) {
-      const chain = await this.#store.getMessageChain(
-        this.#branch.headMessageId
-      );
-      for (const msg of chain) {
-        messages.push(msg.data);
-      }
-    }
-    for (const fragment2 of this.#pendingMessages) {
-      if (!fragment2.codec) {
-        throw new Error(`Fragment "${fragment2.name}" is missing codec.`);
-      }
-      messages.push(fragment2.codec.encode());
-    }
-    return {
-      systemPrompt,
-      messages: messages.length === 0 ? [] : await validateUIMessages({ messages })
-    };
+    const messages = await this.getMessages();
+    return { systemPrompt, messages };
   }
   /**
    * Save pending messages to the graph.
@@ -2881,7 +2993,8 @@ var ContextEngine = class _ContextEngine {
             messageCount,
             lastMessageAt,
             lastMessage,
-            lastAssistantMessage
+            lastAssistantMessage,
+            lastAssistantMessages
           } = await this.#getChainContext();
           const original = lastUserFragment.codec.encode();
           const plainText = extractPlainText(original);
@@ -2899,7 +3012,8 @@ var ContextEngine = class _ContextEngine {
             branch: this.#branchName,
             elapsed,
             messageCount,
-            lastAssistantMessage
+            lastAssistantMessage,
+            lastAssistantMessages
           };
           const configs = conditionalReminders.map(
             (it) => it.metadata.reminder
@@ -3746,55 +3860,75 @@ Examples:
 - "hi" -> New Conversation
 - "debug my python code" -> Python Debugging`;
 var titleSchema = z2.object({ title: z2.string() });
-function extractText(message2) {
-  const cleaned = stripReminders(message2);
-  const textPart = cleaned.parts.find((p) => p.type === "text");
-  return textPart && "text" in textPart ? textPart.text : "";
-}
-function truncateTitle(text) {
-  if (!text) return "New Chat";
-  return text.length > 100 ? text.slice(0, 100) + "..." : text;
-}
-function staticChatTitle(message2) {
-  return truncateTitle(extractText(message2));
-}
-async function generateChatTitle(options) {
-  const text = extractText(options.message);
-  const fallback = truncateTitle(text);
-  if (!text) return fallback;
-  const store2 = new InMemoryContextStore();
-  const context = new ContextEngine({
-    store: store2,
-    chatId: crypto.randomUUID(),
-    userId: "system"
-  });
-  context.set(role(TITLE_PROMPT), user(text));
-  try {
+var TitleGenerator = class {
+  #context;
+  constructor(options) {
+    this.#context = options.context;
+  }
+  async ensure(options) {
+    const msg = await this.#firstUntitledUser();
+    if (!msg) return null;
+    try {
+      const title = await this.#generateTitle(msg, options);
+      return this.#applyTitle(title, "llm");
+    } catch (error) {
+      console.warn(
+        "TitleGenerator: LLM title generation failed, falling back to static.",
+        error
+      );
+      return this.#applyTitle(this.#staticTitle(msg), "static");
+    }
+  }
+  async ensureStatic() {
+    const msg = await this.#firstUntitledUser();
+    if (!msg) return null;
+    return this.#applyTitle(this.#staticTitle(msg), "static");
+  }
+  #staticTitle(message2) {
+    return this.#truncateTitle(this.#extractText(message2));
+  }
+  async #generateTitle(message2, options) {
+    const text = this.#extractText(message2);
+    if (!text) {
+      throw new Error(
+        "Cannot generate chat title: message has no text content."
+      );
+    }
+    const store2 = new InMemoryContextStore();
+    const context = new ContextEngine({
+      store: store2,
+      chatId: crypto.randomUUID(),
+      userId: "system"
+    });
+    context.set(role(TITLE_PROMPT), user(text));
     const { title } = await structuredOutput({
       context,
       model: options.model,
       schema: titleSchema
     }).generate({}, { abortSignal: options.abortSignal });
-    return title || fallback;
-  } catch (error) {
-    console.warn("Title generation failed, using fallback:", error);
-    return fallback;
+    if (!title) {
+      throw new Error("Title generation returned an empty string.");
+    }
+    return title;
   }
-}
-
-// packages/context/src/lib/chat.ts
-function toMessageFragment(item) {
-  if (isFragment(item) && isMessageFragment(item)) {
-    return item;
+  #extractText(message2) {
+    return extractPlainText(stripReminders(message2));
   }
-  return message(item);
-}
-function chatMessageToUIMessage(item) {
-  if (isFragment(item) && isMessageFragment(item)) {
-    return item.codec.encode();
+  #truncateTitle(text) {
+    if (!text) return "New Chat";
+    return text.length > 100 ? text.slice(0, 100) + "..." : text;
   }
-  return item;
-}
+  async #applyTitle(title, source) {
+    await this.#context.updateChat({ title });
+    return { title, source };
+  }
+  async #firstUntitledUser() {
+    if (this.#context.chat?.title) return void 0;
+    return this.#context.firstUserMessage();
+  }
+};
+
+// packages/context/src/lib/chat.ts
 var defaultChatMessageMetadata = ({
   part
 }) => {
@@ -3806,7 +3940,7 @@ var defaultChatMessageMetadata = ({
   }
   return void 0;
 };
-async function chat(agent2, messages, options) {
+async function chat(agent2, options = {}) {
   const context = agent2.context;
   const sandbox = agent2.sandbox;
   if (!context) {
@@ -3814,53 +3948,36 @@ async function chat(agent2, messages, options) {
       "Agent is missing a context. Provide context when creating the agent."
     );
   }
-  if (messages.length === 0) {
-    throw new Error("messages must not be empty");
-  }
-  const lastItem = messages[messages.length - 1];
-  const lastFragment = toMessageFragment(lastItem);
-  const lastUIMessage = chatMessageToUIMessage(lastItem);
-  let assistantMsgId;
-  if (lastUIMessage.role === "assistant") {
-    context.set(lastFragment);
-    await context.save({ branch: false });
-    assistantMsgId = lastUIMessage.id;
-  } else {
-    context.set(lastFragment);
-    await context.save();
-    assistantMsgId = generateId4();
-  }
-  const uiMessages = messages.map(chatMessageToUIMessage);
-  let title = null;
-  if (!context.chat?.title) {
-    const firstUserMsg = uiMessages.find((m) => m.role === "user");
-    if (firstUserMsg) {
-      if (options?.generateTitle && agent2.model) {
-        title = await generateChatTitle({
-          message: firstUserMsg,
-          model: agent2.model,
-          abortSignal: options?.abortSignal
-        });
-      } else {
-        title = staticChatTitle(firstUserMsg);
-      }
-      await context.updateChat({ title });
-    }
+  const head = await context.headMessage();
+  if (head?.name !== "assistant") {
+    throw new Error(
+      "chat: expected an assistant message at head. Call context.continue(input) before chat()."
+    );
   }
-  const streamContextVariables = options?.contextVariables === void 0 ? {} : options.contextVariables;
-  const result = await agent2.stream(streamContextVariables, {
-    transform: options?.transform,
-    abortSignal: options?.abortSignal
-  });
+  const assistantMsgId = head.id;
+  const uiMessages = await context.getMessages();
+  const streamContextVariables = options.contextVariables === void 0 ? {} : options.contextVariables;
+  const [title, result] = await Promise.all([
+    makeTitle({
+      context,
+      model: agent2.model,
+      generateTitle: options.generateTitle,
+      abortSignal: options.abortSignal
+    }),
+    agent2.stream(streamContextVariables, {
+      transform: options.transform,
+      abortSignal: options.abortSignal
+    })
+  ]);
   const uiStream = result.toUIMessageStream({
-    onError: options?.onError ?? formatChatError,
+    onError: options.onError ?? formatChatError,
     sendStart: true,
     sendFinish: true,
     sendReasoning: true,
     sendSources: true,
     originalMessages: uiMessages,
     generateMessageId: () => assistantMsgId,
-    messageMetadata: options?.messageMetadata ?? defaultChatMessageMetadata
+    messageMetadata: options.messageMetadata ?? defaultChatMessageMetadata
   });
   return createUIMessageStream2({
     originalMessages: uiMessages,
@@ -3883,7 +4000,7 @@ async function chat(agent2, messages, options) {
       }
       const drained = sandbox.drainFileEvents?.() ?? [];
       const fileEvents = isAborted ? [] : drained;
-      const finalMetadata = await options?.finalAssistantMetadata?.(normalizedMessage);
+      const finalMetadata = await options.finalAssistantMetadata?.(normalizedMessage);
       const mergedMetadata = {
         ...normalizedMessage.metadata ?? {},
         ...fileEvents.length > 0 ? { fileEvents } : {},
@@ -3920,9 +4037,7 @@ function sanitizeAbortedParts(parts) {
       sanitized.push(part);
       continue;
     }
-    if (part.state === "input-streaming") {
-      continue;
-    }
+    if (part.state === "input-streaming") continue;
     sanitized.push({
       ...part,
       state: "output-error",
@@ -3947,6 +4062,19 @@ function formatChatError(error) {
   }
   return JSON.stringify(error);
 }
+async function makeTitle(options) {
+  const titler = new TitleGenerator({ context: options.context });
+  if (options.generateTitle && !options.model) {
+    console.warn(
+      "chat: generateTitle=true but agent.model is unset; using static title."
+    );
+  }
+  const result = options.generateTitle && options.model ? await titler.ensure({
+    model: options.model,
+    abortSignal: options.abortSignal
+  }) : await titler.ensureStatic();
+  return result?.title ?? null;
+}
 
 // packages/context/src/lib/fragments/user.ts
 function identity(input) {
@@ -4393,6 +4521,32 @@ function or(...predicates) {
 function not(predicate) {
   return async (ctx) => !await predicate(ctx);
 }
+function withinLastN(n, predicate) {
+  return async (ctx) => {
+    if (n <= 0) return false;
+    const candidates = (ctx.lastAssistantMessages ?? []).slice(-n);
+    for (const message2 of candidates) {
+      if (await predicate({ ...ctx, lastAssistantMessage: message2 })) {
+        return true;
+      }
+    }
+    return false;
+  };
+}
+function everyOfLastN(n, predicate) {
+  return async (ctx) => {
+    if (n <= 0) return false;
+    const history = ctx.lastAssistantMessages ?? [];
+    if (history.length < n) return false;
+    const candidates = history.slice(-n);
+    for (const message2 of candidates) {
+      if (!await predicate({ ...ctx, lastAssistantMessage: message2 })) {
+        return false;
+      }
+    }
+    return true;
+  };
+}
 
 // packages/context/src/lib/fragments/reminders/content-predicates.ts
 function contentMatches(topics, options) {
@@ -4418,6 +4572,37 @@ function contentPattern(pattern) {
   };
 }
 
+// packages/context/src/lib/fragments/reminders/message-predicates.ts
+function assertCountSpec(spec) {
+  const hasEq = spec.eq !== void 0;
+  const hasRange = spec.gte !== void 0 || spec.lte !== void 0;
+  if (!hasEq && !hasRange) {
+    throw new Error("CountSpec must include at least one of gte/lte/eq");
+  }
+  if (hasEq && hasRange) {
+    throw new Error("CountSpec.eq cannot be combined with gte/lte");
+  }
+}
+function checkCount(count, spec) {
+  if (spec.eq !== void 0) return count === spec.eq;
+  if (spec.gte !== void 0 && count < spec.gte) return false;
+  if (spec.lte !== void 0 && count > spec.lte) return false;
+  return true;
+}
+function lastAssistantLength(spec) {
+  assertCountSpec(spec);
+  return (ctx) => {
+    const message2 = ctx.lastAssistantMessage;
+    const text = message2 ? extractPlainText(message2) : "";
+    return checkCount(text.length, spec);
+  };
+}
+
+// packages/context/src/lib/fragments/reminders/temporal/elapsed.ts
+function elapsedExceeds(ms) {
+  return (ctx) => (ctx.elapsed ?? 0) >= ms;
+}
+
 // packages/context/src/lib/fragments/reminders/temporal/predicates.ts
 var LOCALE_METADATA_KEY = "locale";
 function getLocaleFromMessage(message2) {
@@ -4689,6 +4874,73 @@ function temporalReminder(options) {
   ];
 }
 
+// packages/context/src/lib/fragments/reminders/tool-predicates.ts
+import { isStaticToolUIPart } from "ai";
+var COMPLETED_STATES = /* @__PURE__ */ new Set([
+  "input-available",
+  "output-available",
+  "output-error"
+]);
+function matchesName(spec, name) {
+  return typeof spec === "function" ? spec(name) : spec === name;
+}
+function toolNameOf(part) {
+  return part.type.slice("tool-".length);
+}
+function toolPartsOf(message2) {
+  if (!message2) return [];
+  return message2.parts.filter(isStaticToolUIPart);
+}
+function toolCall(options) {
+  return (ctx) => {
+    const parts = toolPartsOf(ctx.lastAssistantMessage);
+    return parts.some((part) => {
+      if (options.state) {
+        if (part.state !== options.state) return false;
+      } else if (!COMPLETED_STATES.has(part.state)) {
+        return false;
+      }
+      if (options.name !== void 0 && !matchesName(options.name, toolNameOf(part))) {
+        return false;
+      }
+      if (options.input && !options.input(part.input)) {
+        return false;
+      }
+      if (options.output) {
+        if (part.state !== "output-available") return false;
+        if (!options.output(part.output))
+          return false;
+      }
+      if (options.errorText) {
+        if (part.state !== "output-error") return false;
+        const text = part.errorText ?? "";
+        if (!options.errorText(text)) return false;
+      }
+      return true;
+    });
+  };
+}
+function toolCalled(name) {
+  return toolCall({ name });
+}
+function toolFailed(name) {
+  return toolCall({ name, state: "output-error" });
+}
+function anyToolCalled() {
+  return (ctx) => toolPartsOf(ctx.lastAssistantMessage).some(
+    (part) => COMPLETED_STATES.has(part.state)
+  );
+}
+function toolCallCount(name, spec) {
+  assertCountSpec(spec);
+  return (ctx) => {
+    const count = toolPartsOf(ctx.lastAssistantMessage).filter(
+      (part) => COMPLETED_STATES.has(part.state) && matchesName(name, toolNameOf(part))
+    ).length;
+    return checkCount(count, spec);
+  };
+}
+
 // packages/context/src/lib/fragments/reminders/turn-predicates.ts
 function everyNTurns(n) {
   return ({ turn }) => turn % n === 0;
@@ -4703,6 +4955,11 @@ function afterTurn(n) {
   return ({ turn }) => turn > n;
 }
 
+// packages/context/src/lib/fragments/reminders/usage-predicates.ts
+function usageExceeds(totalTokens) {
+  return (ctx) => (ctx.usage?.totalTokens ?? 0) >= totalTokens;
+}
+
 // packages/context/src/lib/fragments/socratic.ts
 function socraticPrompting() {
   return [
@@ -9211,6 +9468,7 @@ export {
   SqliteStreamStore,
   StreamManager,
   StreamStore,
+  TitleGenerator,
   TomlRenderer,
   ToonRenderer,
   XmlRenderer,
@@ -9223,18 +9481,20 @@ export {
   alias,
   analogy,
   and,
+  anyToolCalled,
   applyInlineReminder,
   applyPartReminder,
   applyReminderToMessage,
   asStaticWordPartText,
   asStaticWordText,
+  assertCountSpec,
   assistant,
   assistantText,
   buildSchemaSubcommand,
   buildSubcommandRepair,
   chainHooks,
   chat,
-  chatMessageToUIMessage,
+  checkCount,
   clarification,
   classifies,
   contentIncludes,
@@ -9257,10 +9517,12 @@ export {
   defaultTokenizer,
   defineSubcommandGroup,
   discoverSkillsInDirectory,
+  elapsedExceeds,
   encodeSerializedValue,
   errorRecoveryGuardrail,
   estimate,
   everyNTurns,
+  everyOfLastN,
   example,
   executorContext,
   explain,
@@ -9272,7 +9534,6 @@ export {
   formatResponse,
   fragment,
   fromFragment,
-  generateChatTitle,
   getFragmentData,
   getLocaleFromMessage,
   getModelsRegistry,
@@ -9291,6 +9552,7 @@ export {
   isInstallable,
   isMessageFragment,
   isRecord,
+  lastAssistantLength,
   loadSkillMetadata,
   localeReminder,
   mapGenerateErrorToCode,
@@ -9335,7 +9597,6 @@ export {
   skillsReminder,
   socraticPrompting,
   soul,
-  staticChatTitle,
   stop,
   stripQuoteArtifacts,
   stripReminders,
@@ -9347,13 +9608,19 @@ export {
   timeReminder,
   toFragment,
   toMessageFragment,
+  toolCall,
+  toolCallCount,
+  toolCalled,
+  toolFailed,
   uploadSkills,
+  usageExceeds,
   useAgentOsSandbox,
   useBashMeta,
   useSandbox,
   user,
   visualizeGraph,
   weekChanged,
+  withinLastN,
   workflow,
   yearChanged,
   yearReminder