@deepagents/context 0.20.0 → 0.22.0

package/dist/index.d.ts

@@ -1,4 +1,5 @@
 export * from './lib/agent.ts';
+export * from './lib/chat.ts';
 export * from './lib/codec.ts';
 export * from './lib/engine.ts';
 export * from './lib/estimate.ts';

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,gBAAgB,CAAC;AAC/B,cAAc,gBAAgB,CAAC;AAC/B,cAAc,iBAAiB,CAAC;AAChC,cAAc,mBAAmB,CAAC;AAClC,cAAc,oBAAoB,CAAC;AACnC,cAAc,2BAA2B,CAAC;AAC1C,cAAc,iCAAiC,CAAC;AAChD,cAAc,yBAAyB,CAAC;AACxC,cAAc,oBAAoB,CAAC;AACnC,cAAc,8CAA8C,CAAC;AAC7D,cAAc,2BAA2B,CAAC;AAC1C,cAAc,iBAAiB,CAAC;AAChC,cAAc,sCAAsC,CAAC;AACrD,cAAc,wBAAwB,CAAC;AACvC,cAAc,uBAAuB,CAAC;AACtC,cAAc,yBAAyB,CAAC;AACxC,cAAc,6BAA6B,CAAC;AAC5C,cAAc,+BAA+B,CAAC;AAC9C,cAAc,6BAA6B,CAAC;AAC5C,cAAc,gCAAgC,CAAC;AAC/C,cAAc,sBAAsB,CAAC;AACrC,cAAc,wBAAwB,CAAC;AACvC,cAAc,gCAAgC,CAAC;AAC/C,cAAc,qCAAqC,CAAC;AACpD,cAAc,gCAAgC,CAAC;AAC/C,cAAc,8BAA8B,CAAC;AAC7C,cAAc,oBAAoB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,gBAAgB,CAAC;AAC/B,cAAc,eAAe,CAAC;AAC9B,cAAc,gBAAgB,CAAC;AAC/B,cAAc,iBAAiB,CAAC;AAChC,cAAc,mBAAmB,CAAC;AAClC,cAAc,oBAAoB,CAAC;AACnC,cAAc,2BAA2B,CAAC;AAC1C,cAAc,iCAAiC,CAAC;AAChD,cAAc,yBAAyB,CAAC;AACxC,cAAc,oBAAoB,CAAC;AACnC,cAAc,8CAA8C,CAAC;AAC7D,cAAc,2BAA2B,CAAC;AAC1C,cAAc,iBAAiB,CAAC;AAChC,cAAc,sCAAsC,CAAC;AACrD,cAAc,wBAAwB,CAAC;AACvC,cAAc,uBAAuB,CAAC;AACtC,cAAc,yBAAyB,CAAC;AACxC,cAAc,6BAA6B,CAAC;AAC5C,cAAc,+BAA+B,CAAC;AAC9C,cAAc,6BAA6B,CAAC;AAC5C,cAAc,gCAAgC,CAAC;AAC/C,cAAc,sBAAsB,CAAC;AACrC,cAAc,wBAAwB,CAAC;AACvC,cAAc,gCAAgC,CAAC;AAC/C,cAAc,qCAAqC,CAAC;AACpD,cAAc,gCAAgC,CAAC;AAC/C,cAAc,8BAA8B,CAAC;AAC7C,cAAc,oBAAoB,CAAC"}

package/dist/index.js

@@ -120,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) {
@@ -190,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,
@@ -227,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({
@@ -390,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";
 
@@ -2256,7 +2364,7 @@ function policy(input) {
 }
 
 // packages/context/src/lib/fragments/message/user.ts
-import { generateId as generateId3 } from "ai";
+import { generateId as generateId4 } from "ai";
 var SYSTEM_REMINDER_OPEN_TAG = "<system-reminder>";
 var SYSTEM_REMINDER_CLOSE_TAG = "</system-reminder>";
 function getReminderRanges(metadata) {
@@ -2288,6 +2396,43 @@ function stripTextByRanges(text, ranges) {
   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);
 }
@@ -2330,7 +2475,7 @@ function applyInlineReminder(message2, value) {
   const updatedText = `${textPart.text}${reminderText}`;
   message2.parts[partIndex] = { ...textPart, text: updatedText };
   return {
-    id: generateId3(),
+    id: generateId4(),
     text: value,
     partIndex,
     start,
@@ -2343,7 +2488,7 @@ function applyPartReminder(message2, value) {
   message2.parts.push(part);
   const partIndex = message2.parts.length - 1;
   return {
-    id: generateId3(),
+    id: generateId4(),
     text: value,
     partIndex,
     start: 0,
@@ -2360,7 +2505,7 @@ function reminder(text, options) {
 }
 function user(content, ...reminders) {
   const message2 = typeof content === "string" ? {
-    id: generateId3(),
+    id: generateId4(),
     role: "user",
     parts: [{ type: "text", text: content }]
   } : { ...content, role: "user", parts: [...content.parts] };
@@ -3519,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,
@@ -3545,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)
@@ -3553,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,
@@ -3757,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);
@@ -4182,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,
@@ -4211,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)
@@ -4219,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];
@@ -4230,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,
@@ -4436,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);
@@ -4941,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)
@@ -4961,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)
@@ -4977,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(`
@@ -5004,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')
@@ -5225,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);
@@ -5941,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";
@@ -6022,7 +6167,7 @@ var StreamManager = class {
       }
     })();
     let pw;
-    const sink = createUIMessageStream2({
+    const sink = createUIMessageStream3({
       execute: async ({ writer }) => {
         pw = await persistedWriter({
           writer,
@@ -6339,6 +6484,7 @@ export {
   analogy,
   assistant,
   assistantText,
+  chat,
   clarification,
   correction,
   createAdaptivePollingState,
@@ -6387,6 +6533,7 @@ export {
   skills,
   soul,
   stop,
+  stripReminders,
   stripTextByRanges,
   structuredOutput,
   styleGuide,