@deepagents/context 0.19.0 → 0.22.0

package/dist/index.js

@@ -29,113 +29,6 @@ function fragment(name, ...children) {
     data: children
   };
 }
-var SYSTEM_REMINDER_OPEN_TAG = "<system-reminder>";
-var SYSTEM_REMINDER_CLOSE_TAG = "</system-reminder>";
-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: generateId(),
-    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: generateId(),
-    text: value,
-    partIndex,
-    start: 0,
-    end: value.length,
-    mode: "part"
-  };
-}
-function reminder(text, options) {
-  assertReminderText(text);
-  return {
-    text,
-    asPart: options?.asPart ?? false
-  };
-}
-function user(content, ...reminders) {
-  const message2 = typeof content === "string" ? {
-    id: generateId(),
-    role: "user",
-    parts: [{ type: "text", text: content }]
-  } : { ...content, role: "user", parts: [...content.parts] };
-  if (reminders.length > 0) {
-    const addedReminders = [];
-    for (const item of reminders) {
-      assertReminderText(item.text);
-      addedReminders.push(
-        item.asPart ? applyPartReminder(message2, item.text) : applyInlineReminder(message2, item.text)
-      );
-    }
-    const metadata = isRecord(message2.metadata) ? { ...message2.metadata } : {};
-    const existingReminders = Array.isArray(metadata.reminders) ? metadata.reminders : [];
-    metadata.reminders = [...existingReminders, ...addedReminders];
-    message2.metadata = metadata;
-  }
-  return {
-    id: message2.id,
-    name: "user",
-    data: "content",
-    type: "message",
-    persist: true,
-    codec: {
-      decode() {
-        return message2;
-      },
-      encode() {
-        return message2;
-      }
-    }
-  };
-}
 function assistant(message2) {
   return {
     id: message2.id,
@@ -227,9 +120,11 @@ var Agent = class _Agent {
   #options;
   #guardrails = [];
   tools;
+  context;
   constructor(options) {
     this.#options = options;
     this.tools = options.tools || {};
+    this.context = options.context;
     this.#guardrails = options.guardrails || [];
   }
   async generate(contextVariables, config) {
@@ -297,17 +192,25 @@ var Agent = class _Agent {
    * Create a raw stream without guardrail processing.
    */
   async #createRawStream(contextVariables, config) {
-    const { messages, systemPrompt } = await this.#options.context.resolve({
+    const context = this.#options.context;
+    if (!context) {
+      throw new Error(`Agent ${this.#options.name} is missing a context.`);
+    }
+    const model = this.#options.model;
+    if (!model) {
+      throw new Error(`Agent ${this.#options.name} is missing a model.`);
+    }
+    const { messages, systemPrompt } = await context.resolve({
       renderer: new XmlRenderer()
     });
     const runId = generateId2();
     return streamText({
       abortSignal: config?.abortSignal,
       providerOptions: this.#options.providerOptions,
-      model: this.#options.model,
+      model,
       system: systemPrompt,
       messages: await convertToModelMessages(messages),
-      experimental_repairToolCall: createRepairToolCall(this.#options.model),
+      experimental_repairToolCall: createRepairToolCall(model),
       stopWhen: stepCountIs(50),
       experimental_transform: config?.transform ?? smoothStream(),
       tools: this.#options.tools,
@@ -334,6 +237,9 @@ var Agent = class _Agent {
   #wrapWithGuardrails(result, contextVariables, config) {
     const maxRetries = config?.maxRetries ?? this.#options.maxGuardrailRetries ?? 3;
     const context = this.#options.context;
+    if (!context) {
+      throw new Error(`Agent ${this.#options.name} is missing a context.`);
+    }
     const originalToUIMessageStream = result.toUIMessageStream.bind(result);
     result.toUIMessageStream = (options) => {
       return createUIMessageStream({
@@ -497,6 +403,101 @@ function writeText(writer, text) {
   });
 }
 
+// packages/context/src/lib/chat.ts
+import {
+  APICallError,
+  InvalidToolInputError,
+  NoSuchToolError,
+  ToolCallRepairError,
+  createUIMessageStream as createUIMessageStream2,
+  generateId as generateId3
+} from "ai";
+async function chat(agent2, messages, options) {
+  const context = agent2.context;
+  if (!context) {
+    throw new Error(
+      "Agent is missing a context. Provide context when creating the agent."
+    );
+  }
+  if (messages.length === 0) {
+    throw new Error("messages must not be empty");
+  }
+  const lastMessage = messages[messages.length - 1];
+  let assistantMsgId;
+  if (lastMessage.role === "assistant") {
+    context.set(message(lastMessage));
+    await context.save({ branch: false });
+    assistantMsgId = lastMessage.id;
+  } else {
+    context.set(message(lastMessage));
+    await context.save();
+    assistantMsgId = options?.generateMessageId?.() ?? generateId3();
+  }
+  const streamContextVariables = options?.contextVariables === void 0 ? {} : options.contextVariables;
+  const result = await agent2.stream(streamContextVariables, {
+    transform: options?.transform
+  });
+  const uiStream = result.toUIMessageStream({
+    onError: options?.onError ?? formatChatError,
+    sendStart: true,
+    sendFinish: true,
+    sendReasoning: true,
+    sendSources: true,
+    originalMessages: messages,
+    generateMessageId: () => assistantMsgId,
+    messageMetadata: options?.messageMetadata
+  });
+  return createUIMessageStream2({
+    originalMessages: messages,
+    generateId: () => assistantMsgId,
+    onStepFinish: async ({ responseMessage }) => {
+      const normalizedMessage = {
+        ...responseMessage,
+        id: assistantMsgId
+      };
+      context.set(assistant(normalizedMessage));
+      await context.save({ branch: false });
+    },
+    onFinish: async ({ responseMessage }) => {
+      const normalizedMessage = {
+        ...responseMessage,
+        id: assistantMsgId
+      };
+      const finalMetadata = await options?.finalAssistantMetadata?.(normalizedMessage);
+      const finalMessage = finalMetadata === void 0 ? normalizedMessage : {
+        ...normalizedMessage,
+        metadata: {
+          ...normalizedMessage.metadata ?? {},
+          ...finalMetadata
+        }
+      };
+      context.set(assistant(finalMessage));
+      await context.save({ branch: false });
+      const totalUsage = await result.totalUsage;
+      await context.trackUsage(totalUsage);
+    },
+    execute: async ({ writer }) => {
+      writer.merge(uiStream);
+    }
+  });
+}
+function formatChatError(error) {
+  if (NoSuchToolError.isInstance(error)) {
+    return "The model tried to call an unknown tool.";
+  }
+  if (InvalidToolInputError.isInstance(error)) {
+    return "The model called a tool with invalid arguments.";
+  }
+  if (ToolCallRepairError.isInstance(error)) {
+    return "The model tried to call a tool with invalid arguments, but it was repaired.";
+  }
+  if (APICallError.isInstance(error)) {
+    console.error("Upstream API call failed:", error);
+    return `Upstream API call failed with status ${error.statusCode}: ${error.message}`;
+  }
+  return JSON.stringify(error);
+}
+
 // packages/context/src/lib/engine.ts
 import { mergeWith } from "lodash-es";
 
@@ -2362,6 +2363,182 @@ function policy(input) {
   };
 }
 
+// packages/context/src/lib/fragments/message/user.ts
+import { generateId as generateId4 } from "ai";
+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: generateId4(),
+    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: generateId4(),
+    text: value,
+    partIndex,
+    start: 0,
+    end: value.length,
+    mode: "part"
+  };
+}
+function reminder(text, options) {
+  assertReminderText(text);
+  return {
+    text,
+    asPart: options?.asPart ?? false
+  };
+}
+function user(content, ...reminders) {
+  const message2 = typeof content === "string" ? {
+    id: generateId4(),
+    role: "user",
+    parts: [{ type: "text", text: content }]
+  } : { ...content, role: "user", parts: [...content.parts] };
+  if (reminders.length > 0) {
+    const addedReminders = [];
+    for (const item of reminders) {
+      assertReminderText(item.text);
+      addedReminders.push(
+        item.asPart ? applyPartReminder(message2, item.text) : applyInlineReminder(message2, item.text)
+      );
+    }
+    const metadata = isRecord(message2.metadata) ? { ...message2.metadata } : {};
+    const existingReminders = Array.isArray(metadata.reminders) ? metadata.reminders : [];
+    metadata.reminders = [...existingReminders, ...addedReminders];
+    message2.metadata = metadata;
+  }
+  return {
+    id: message2.id,
+    name: "user",
+    data: "content",
+    type: "message",
+    persist: true,
+    codec: {
+      decode() {
+        return message2;
+      },
+      encode() {
+        return message2;
+      }
+    }
+  };
+}
+
 // packages/context/src/lib/fragments/user.ts
 function identity(input) {
   return {
@@ -3487,22 +3664,22 @@ var SqliteContextStore = class extends ContextStore {
   // ==========================================================================
   // Chat Operations
   // ==========================================================================
-  async createChat(chat) {
+  async createChat(chat2) {
     return this.#useTransaction(() => {
       const row = this.#db.prepare(
         `INSERT INTO chats (id, userId, title, metadata)
            VALUES (?, ?, ?, ?)
            RETURNING *`
       ).get(
-        chat.id,
-        chat.userId,
-        chat.title ?? null,
-        chat.metadata ? JSON.stringify(chat.metadata) : null
+        chat2.id,
+        chat2.userId,
+        chat2.title ?? null,
+        chat2.metadata ? JSON.stringify(chat2.metadata) : null
       );
       this.#db.prepare(
         `INSERT INTO branches (id, chatId, name, headMessageId, isActive, createdAt)
            VALUES (?, ?, 'main', NULL, 1, ?)`
-      ).run(crypto.randomUUID(), chat.id, Date.now());
+      ).run(crypto.randomUUID(), chat2.id, Date.now());
       return {
         id: row.id,
         userId: row.userId,
@@ -3513,7 +3690,7 @@ var SqliteContextStore = class extends ContextStore {
       };
     });
   }
-  async upsertChat(chat) {
+  async upsertChat(chat2) {
     return this.#useTransaction(() => {
       const row = this.#db.prepare(
         `INSERT INTO chats (id, userId, title, metadata)
@@ -3521,15 +3698,15 @@ var SqliteContextStore = class extends ContextStore {
            ON CONFLICT(id) DO UPDATE SET id = excluded.id
            RETURNING *`
       ).get(
-        chat.id,
-        chat.userId,
-        chat.title ?? null,
-        chat.metadata ? JSON.stringify(chat.metadata) : null
+        chat2.id,
+        chat2.userId,
+        chat2.title ?? null,
+        chat2.metadata ? JSON.stringify(chat2.metadata) : null
       );
       this.#db.prepare(
         `INSERT OR IGNORE INTO branches (id, chatId, name, headMessageId, isActive, createdAt)
            VALUES (?, ?, 'main', NULL, 1, ?)`
-      ).run(crypto.randomUUID(), chat.id, Date.now());
+      ).run(crypto.randomUUID(), chat2.id, Date.now());
       return {
         id: row.id,
         userId: row.userId,
@@ -3725,8 +3902,8 @@ var SqliteContextStore = class extends ContextStore {
     return row.hasChildren === 1;
   }
   async getMessages(chatId) {
-    const chat = await this.getChat(chatId);
-    if (!chat) {
+    const chat2 = await this.getChat(chatId);
+    if (!chat2) {
       throw new Error(`Chat "${chatId}" not found`);
     }
     const activeBranch = await this.getActiveBranch(chatId);
@@ -4150,24 +4327,24 @@ var PostgresContextStore = class _PostgresContextStore extends ContextStore {
   // ==========================================================================
   // Chat Operations
   // ==========================================================================
-  async createChat(chat) {
+  async createChat(chat2) {
     return this.#useTransaction(async (client) => {
       const result = await client.query(
         `INSERT INTO ${this.#t("chats")} (id, userId, title, metadata)
          VALUES ($1, $2, $3, $4)
          RETURNING *`,
         [
-          chat.id,
-          chat.userId,
-          chat.title ?? null,
-          chat.metadata ? JSON.stringify(chat.metadata) : null
+          chat2.id,
+          chat2.userId,
+          chat2.title ?? null,
+          chat2.metadata ? JSON.stringify(chat2.metadata) : null
         ]
       );
       const row = result.rows[0];
       await client.query(
         `INSERT INTO ${this.#t("branches")} (id, chatId, name, headMessageId, isActive, createdAt)
          VALUES ($1, $2, 'main', NULL, TRUE, $3)`,
-        [crypto.randomUUID(), chat.id, Date.now()]
+        [crypto.randomUUID(), chat2.id, Date.now()]
       );
       return {
         id: row.id,
@@ -4179,7 +4356,7 @@ var PostgresContextStore = class _PostgresContextStore extends ContextStore {
       };
     });
   }
-  async upsertChat(chat) {
+  async upsertChat(chat2) {
     return this.#useTransaction(async (client) => {
       const result = await client.query(
         `INSERT INTO ${this.#t("chats")} (id, userId, title, metadata)
@@ -4187,10 +4364,10 @@ var PostgresContextStore = class _PostgresContextStore extends ContextStore {
          ON CONFLICT(id) DO UPDATE SET id = EXCLUDED.id
          RETURNING *`,
         [
-          chat.id,
-          chat.userId,
-          chat.title ?? null,
-          chat.metadata ? JSON.stringify(chat.metadata) : null
+          chat2.id,
+          chat2.userId,
+          chat2.title ?? null,
+          chat2.metadata ? JSON.stringify(chat2.metadata) : null
         ]
       );
       const row = result.rows[0];
@@ -4198,7 +4375,7 @@ var PostgresContextStore = class _PostgresContextStore extends ContextStore {
         `INSERT INTO ${this.#t("branches")} (id, chatId, name, headMessageId, isActive, createdAt)
          VALUES ($1, $2, 'main', NULL, TRUE, $3)
          ON CONFLICT(chatId, name) DO NOTHING`,
-        [crypto.randomUUID(), chat.id, Date.now()]
+        [crypto.randomUUID(), chat2.id, Date.now()]
       );
       return {
         id: row.id,
@@ -4404,8 +4581,8 @@ var PostgresContextStore = class _PostgresContextStore extends ContextStore {
     return rows[0].exists;
   }
   async getMessages(chatId) {
-    const chat = await this.getChat(chatId);
-    if (!chat) {
+    const chat2 = await this.getChat(chatId);
+    if (!chat2) {
       throw new Error(`Chat "${chatId}" not found`);
     }
     const activeBranch = await this.getActiveBranch(chatId);
@@ -4909,17 +5086,17 @@ var SqlServerContextStore = class _SqlServerContextStore extends ContextStore {
   // ==========================================================================
   // Chat Operations
   // ==========================================================================
-  async createChat(chat) {
+  async createChat(chat2) {
     return this.#useTransaction(async (transaction) => {
       const mssql = _SqlServerContextStore.#requireMssql();
       const request = transaction.request();
-      request.input("p0", mssql.NVarChar, chat.id);
-      request.input("p1", mssql.NVarChar, chat.userId);
-      request.input("p2", mssql.NVarChar, chat.title ?? null);
+      request.input("p0", mssql.NVarChar, chat2.id);
+      request.input("p1", mssql.NVarChar, chat2.userId);
+      request.input("p2", mssql.NVarChar, chat2.title ?? null);
       request.input(
         "p3",
         mssql.NVarChar,
-        chat.metadata ? JSON.stringify(chat.metadata) : null
+        chat2.metadata ? JSON.stringify(chat2.metadata) : null
       );
       const result = await request.query(`
         INSERT INTO ${this.#t("chats")} (id, userId, title, metadata)
@@ -4929,7 +5106,7 @@ var SqlServerContextStore = class _SqlServerContextStore extends ContextStore {
       const row = result.recordset[0];
       const branchRequest = transaction.request();
       branchRequest.input("p0", mssql.NVarChar, crypto.randomUUID());
-      branchRequest.input("p1", mssql.NVarChar, chat.id);
+      branchRequest.input("p1", mssql.NVarChar, chat2.id);
       branchRequest.input("p2", mssql.BigInt, Date.now());
       await branchRequest.query(`
         INSERT INTO ${this.#t("branches")} (id, chatId, name, headMessageId, isActive, createdAt)
@@ -4945,17 +5122,17 @@ var SqlServerContextStore = class _SqlServerContextStore extends ContextStore {
       };
     });
   }
-  async upsertChat(chat) {
+  async upsertChat(chat2) {
     return this.#useTransaction(async (transaction) => {
       const mssql = _SqlServerContextStore.#requireMssql();
       const request = transaction.request();
-      request.input("p0", mssql.NVarChar, chat.id);
-      request.input("p1", mssql.NVarChar, chat.userId);
-      request.input("p2", mssql.NVarChar, chat.title ?? null);
+      request.input("p0", mssql.NVarChar, chat2.id);
+      request.input("p1", mssql.NVarChar, chat2.userId);
+      request.input("p2", mssql.NVarChar, chat2.title ?? null);
       request.input(
         "p3",
         mssql.NVarChar,
-        chat.metadata ? JSON.stringify(chat.metadata) : null
+        chat2.metadata ? JSON.stringify(chat2.metadata) : null
       );
       request.input("p4", mssql.BigInt, BigInt(Date.now()));
       const result = await request.query(`
@@ -4972,7 +5149,7 @@ var SqlServerContextStore = class _SqlServerContextStore extends ContextStore {
       const row = result.recordset[0];
       const branchRequest = transaction.request();
       branchRequest.input("p0", mssql.NVarChar, crypto.randomUUID());
-      branchRequest.input("p1", mssql.NVarChar, chat.id);
+      branchRequest.input("p1", mssql.NVarChar, chat2.id);
       branchRequest.input("p2", mssql.BigInt, Date.now());
       await branchRequest.query(`
         IF NOT EXISTS (SELECT 1 FROM ${this.#t("branches")} WHERE chatId = @p1 AND name = 'main')
@@ -5193,8 +5370,8 @@ var SqlServerContextStore = class _SqlServerContextStore extends ContextStore {
     return rows[0].hasChildren === 1;
   }
   async getMessages(chatId) {
-    const chat = await this.getChat(chatId);
-    if (!chat) {
+    const chat2 = await this.getChat(chatId);
+    if (!chat2) {
       throw new Error(`Chat "${chatId}" not found`);
     }
     const activeBranch = await this.getActiveBranch(chatId);
@@ -5909,7 +6086,7 @@ var SqliteStreamStore = class extends StreamStore {
 };
 
 // packages/context/src/lib/stream/stream-manager.ts
-import { createUIMessageStream as createUIMessageStream2 } from "ai";
+import { createUIMessageStream as createUIMessageStream3 } from "ai";
 import { setTimeout } from "node:timers/promises";
 function isTerminal(status) {
   return status !== "queued" && status !== "running";
@@ -5990,7 +6167,7 @@ var StreamManager = class {
       }
     })();
     let pw;
-    const sink = createUIMessageStream2({
+    const sink = createUIMessageStream3({
       execute: async ({ writer }) => {
         pw = await persistedWriter({
           writer,
@@ -6307,6 +6484,7 @@ export {
   analogy,
   assistant,
   assistantText,
+  chat,
   clarification,
   correction,
   createAdaptivePollingState,
@@ -6322,6 +6500,7 @@ export {
   fail,
   fragment,
   getModelsRegistry,
+  getReminderRanges,
   glossary,
   guardrail,
   hint,
@@ -6354,6 +6533,8 @@ export {
   skills,
   soul,
   stop,
+  stripReminders,
+  stripTextByRanges,
   structuredOutput,
   styleGuide,
   term,