@deepagents/context 0.33.0 → 0.34.0

package/dist/index.js

@@ -591,7 +591,18 @@ async function estimate(modelId, renderer, ...fragments) {
 }
 
 // packages/context/src/lib/fragments/message/user.ts
-import { generateId as generateId3, isTextUIPart } from "ai";
+import { generateId as generateId3 } from "ai";
+
+// 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/fragments/message/user.ts
 function everyNTurns(n) {
   return ({ turn }) => turn % n === 0;
 }
@@ -762,9 +773,6 @@ function stripReminders(message2) {
   }
   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);
 }
@@ -1814,6 +1822,9 @@ var ContextStore = class {
 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 = [];
@@ -1949,17 +1960,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 +1969,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 +1991,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,25 +2130,34 @@ 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 firedReminders = conditionalReminders.map(getConditionalReminder).filter((config) => config.when(whenCtx));
           if (firedReminders.length > 0) {
             const reminders = firedReminders.flatMap((rem) => {
-              const resolved = resolveReminder(rem, {
-                content: plainText,
-                turn,
-                lastMessageAt,
-                lastMessage
-              });
+              const resolved = resolveReminder(rem, whenCtx);
               if (!resolved) {
                 return [];
               }
@@ -7959,7 +7982,6 @@ export {
   everyNTurns,
   example,
   explain,
-  extractPlainText,
   fail,
   firstN,
   fragment,