@deepagents/context 0.22.0 → 0.24.0

package/dist/index.js

@@ -412,6 +412,18 @@ import {
   createUIMessageStream as createUIMessageStream2,
   generateId as generateId3
 } from "ai";
+function toMessageFragment(item) {
+  if (isFragment(item) && isMessageFragment(item)) {
+    return item;
+  }
+  return message(item);
+}
+function chatMessageToUIMessage(item) {
+  if (isFragment(item) && isMessageFragment(item)) {
+    return item.codec.decode();
+  }
+  return item;
+}
 async function chat(agent2, messages, options) {
   const context = agent2.context;
   if (!context) {
@@ -422,17 +434,20 @@ async function chat(agent2, messages, options) {
   if (messages.length === 0) {
     throw new Error("messages must not be empty");
   }
-  const lastMessage = messages[messages.length - 1];
+  const lastItem = messages[messages.length - 1];
+  const lastFragment = toMessageFragment(lastItem);
+  const lastUIMessage = chatMessageToUIMessage(lastItem);
   let assistantMsgId;
-  if (lastMessage.role === "assistant") {
-    context.set(message(lastMessage));
+  if (lastUIMessage.role === "assistant") {
+    context.set(lastFragment);
     await context.save({ branch: false });
-    assistantMsgId = lastMessage.id;
+    assistantMsgId = lastUIMessage.id;
   } else {
-    context.set(message(lastMessage));
+    context.set(lastFragment);
     await context.save();
     assistantMsgId = options?.generateMessageId?.() ?? generateId3();
   }
+  const uiMessages = messages.map(chatMessageToUIMessage);
   const streamContextVariables = options?.contextVariables === void 0 ? {} : options.contextVariables;
   const result = await agent2.stream(streamContextVariables, {
     transform: options?.transform
@@ -443,12 +458,12 @@ async function chat(agent2, messages, options) {
     sendFinish: true,
     sendReasoning: true,
     sendSources: true,
-    originalMessages: messages,
+    originalMessages: uiMessages,
     generateMessageId: () => assistantMsgId,
     messageMetadata: options?.messageMetadata
   });
   return createUIMessageStream2({
-    originalMessages: messages,
+    originalMessages: uiMessages,
     generateId: () => assistantMsgId,
     onStepFinish: async ({ responseMessage }) => {
       const normalizedMessage = {
@@ -2364,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) {
@@ -2433,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);
 }
@@ -2497,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
@@ -2511,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,
@@ -3456,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";
 
@@ -4106,7 +4172,7 @@ var SqliteContextStore = class extends ContextStore {
     ).all(chatId);
     const nodes = messageRows.map((row) => {
       const data = JSON.parse(row.data);
-      const content = typeof data === "string" ? data : JSON.stringify(data);
+      const content = typeof data === "string" ? data : Array.isArray(data.parts) ? data.parts.filter((p) => p.type === "text").map((p) => p.text).join(" ") : JSON.stringify(data);
       return {
         id: row.id,
         parentId: row.parentId,
@@ -6161,6 +6227,12 @@ var StreamManager = class {
             streamId,
             latencyMs
           });
+          if (options?.onCancelDetected) {
+            try {
+              await options.onCancelDetected({ streamId, latencyMs });
+            } catch {
+            }
+          }
           ac.abort();
           break;
         }
@@ -6449,6 +6521,7 @@ function visualizeGraph(data) {
   return lines.join("\n");
 }
 export {
+  BM25SkillClassifier,
   BinaryInstallError,
   ComposeStartError,
   ComposeStrategy,
@@ -6485,6 +6558,7 @@ export {
   assistant,
   assistantText,
   chat,
+  chatMessageToUIMessage,
   clarification,
   correction,
   createAdaptivePollingState,
@@ -6531,6 +6605,7 @@ export {
   role,
   runGuardrailChain,
   skills,
+  skillsReminder,
   soul,
   stop,
   stripReminders,
@@ -6538,6 +6613,7 @@ export {
   structuredOutput,
   styleGuide,
   term,
+  toMessageFragment,
   useSandbox,
   user,
   userContext,