@deepagents/context 0.23.0 → 0.24.0

package/dist/index.js

@@ -2379,7 +2379,7 @@ function policy(input) {
 }
 
 // packages/context/src/lib/fragments/message/user.ts
-import { generateId as generateId4 } from "ai";
+import { generateId as generateId4, isTextUIPart } from "ai";
 var SYSTEM_REMINDER_OPEN_TAG = "<system-reminder>";
 var SYSTEM_REMINDER_CLOSE_TAG = "</system-reminder>";
 function getReminderRanges(metadata) {
@@ -2448,6 +2448,9 @@ 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);
 }
@@ -2512,7 +2515,9 @@ function applyPartReminder(message2, value) {
   };
 }
 function reminder(text, options) {
-  assertReminderText(text);
+  if (typeof text === "string") {
+    assertReminderText(text);
+  }
   return {
     text,
     asPart: options?.asPart ?? false
@@ -2526,16 +2531,22 @@ function user(content, ...reminders) {
   } : { ...content, role: "user", parts: [...content.parts] };
   if (reminders.length > 0) {
     const addedReminders = [];
+    const plainText = extractPlainText(message2);
     for (const item of reminders) {
-      assertReminderText(item.text);
+      const resolvedText = typeof item.text === "function" ? item.text(plainText) : item.text;
+      if (resolvedText.trim().length === 0) {
+        continue;
+      }
       addedReminders.push(
-        item.asPart ? applyPartReminder(message2, item.text) : applyInlineReminder(message2, item.text)
+        item.asPart ? applyPartReminder(message2, resolvedText) : applyInlineReminder(message2, resolvedText)
       );
     }
-    const metadata = isRecord(message2.metadata) ? { ...message2.metadata } : {};
-    const existingReminders = Array.isArray(metadata.reminders) ? metadata.reminders : [];
-    metadata.reminders = [...existingReminders, ...addedReminders];
-    message2.metadata = metadata;
+    if (addedReminders.length > 0) {
+      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,
@@ -3471,6 +3482,46 @@ async function createContainerTool(options = {}) {
   };
 }
 
+// packages/context/src/lib/skills/classifier.ts
+import { Corpus } from "tiny-tfidf";
+var BM25SkillClassifier = class {
+  #corpus;
+  #skillsByName;
+  constructor(skills2) {
+    const names = skills2.map((s) => s.name);
+    const texts = skills2.map((s) => `${s.name} ${s.description}`);
+    this.#corpus = new Corpus(names, texts);
+    this.#skillsByName = new Map(skills2.map((s) => [s.name, s]));
+  }
+  match(query, options) {
+    const topN = options?.topN ?? 5;
+    const threshold = options?.threshold ?? 0;
+    const results = this.#corpus.getResultsForQuery(query);
+    return results.filter(([, score]) => score > threshold).slice(0, topN).map(([name, score]) => ({
+      skill: this.#skillsByName.get(name),
+      score
+    }));
+  }
+};
+function isSkillClassifier(value) {
+  return typeof value === "object" && value !== null && "match" in value && typeof value.match === "function";
+}
+function formatSkillReminder(matches) {
+  const lines = matches.map(
+    (m) => `- ${m.skill.name} (${m.score.toFixed(2)}): ${m.skill.description} [${m.skill.skillMdPath}]`
+  );
+  return `Relevant skills:
+${lines.join("\n")}`;
+}
+function skillsReminder(skillsOrClassifier, options) {
+  const classifier = isSkillClassifier(skillsOrClassifier) ? skillsOrClassifier : new BM25SkillClassifier(skillsOrClassifier);
+  return reminder((content) => {
+    const matches = classifier.match(content, options);
+    if (matches.length === 0) return "";
+    return formatSkillReminder(matches);
+  });
+}
+
 // packages/context/src/lib/skills/fragments.ts
 import dedent from "dedent";
 
@@ -6176,6 +6227,12 @@ var StreamManager = class {
             streamId,
             latencyMs
           });
+          if (options?.onCancelDetected) {
+            try {
+              await options.onCancelDetected({ streamId, latencyMs });
+            } catch {
+            }
+          }
           ac.abort();
           break;
         }
@@ -6464,6 +6521,7 @@ function visualizeGraph(data) {
   return lines.join("\n");
 }
 export {
+  BM25SkillClassifier,
   BinaryInstallError,
   ComposeStartError,
   ComposeStrategy,
@@ -6547,6 +6605,7 @@ export {
   role,
   runGuardrailChain,
   skills,
+  skillsReminder,
   soul,
   stop,
   stripReminders,