@deepagents/context 0.23.0 → 0.25.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";
 
@@ -5881,10 +5932,13 @@ function clampFloat(value, min, max) {
 import { DatabaseSync as DatabaseSync2 } from "node:sqlite";
 
 // packages/context/src/lib/stream/ddl.stream.sqlite.sql
-var ddl_stream_sqlite_default = "PRAGMA journal_mode = WAL;\nPRAGMA synchronous = NORMAL;\nPRAGMA foreign_keys = ON;\n\nCREATE TABLE IF NOT EXISTS streams (\n  id TEXT PRIMARY KEY,\n  status TEXT NOT NULL DEFAULT 'queued'\n    CHECK(status IN ('queued','running','completed','failed','cancelled')),\n  createdAt INTEGER NOT NULL,\n  startedAt INTEGER,\n  finishedAt INTEGER,\n  cancelRequestedAt INTEGER,\n  error TEXT\n);\n\nCREATE TABLE IF NOT EXISTS stream_chunks (\n  streamId TEXT NOT NULL,\n  seq INTEGER NOT NULL,\n  data TEXT NOT NULL,\n  createdAt INTEGER NOT NULL,\n  PRIMARY KEY (streamId, seq),\n  FOREIGN KEY (streamId) REFERENCES streams(id) ON DELETE CASCADE\n);\n";
+var ddl_stream_sqlite_default = "PRAGMA journal_mode = WAL;\nPRAGMA synchronous = NORMAL;\nPRAGMA foreign_keys = ON;\n\nCREATE TABLE IF NOT EXISTS streams (\n  id TEXT PRIMARY KEY,\n  status TEXT NOT NULL DEFAULT 'queued'\n    CHECK(status IN ('queued','running','completed','failed','cancelled')),\n  createdAt INTEGER NOT NULL,\n  startedAt INTEGER,\n  finishedAt INTEGER,\n  cancelRequestedAt INTEGER,\n  error TEXT\n);\n\nCREATE TABLE IF NOT EXISTS stream_chunks (\n  streamId TEXT NOT NULL,\n  seq INTEGER NOT NULL,\n  data TEXT NOT NULL,\n  createdAt INTEGER NOT NULL,\n  PRIMARY KEY (streamId, seq),\n  FOREIGN KEY (streamId) REFERENCES streams(id) ON DELETE CASCADE\n);\n\n-- Supports ordered listing across all streams.\nCREATE INDEX IF NOT EXISTS idx_streams_created_at_id\n  ON streams(createdAt, id);\n\n-- Supports status-filtered ordered listing (e.g. running streams).\nCREATE INDEX IF NOT EXISTS idx_streams_status_created_at_id\n  ON streams(status, createdAt, id);\n";
 
 // packages/context/src/lib/stream/stream-store.ts
 var StreamStore = class {
+  async listRunningStreamIds() {
+    return this.listStreamIds({ status: "running" });
+  }
 };
 
 // packages/context/src/lib/stream/sqlite.stream-store.ts
@@ -5983,6 +6037,17 @@ var SqliteStreamStore = class extends StreamStore {
     );
     return row?.status;
   }
+  async listStreamIds(options) {
+    let sql = "SELECT id FROM streams";
+    const params = [];
+    if (options?.status) {
+      sql += " WHERE status = ?";
+      params.push(options.status);
+    }
+    sql += " ORDER BY createdAt ASC, id ASC";
+    const rows = this.#stmt(sql).all(...params);
+    return rows.map((row) => row.id);
+  }
   async updateStreamStatus(streamId, status, options) {
     const now = Date.now();
     switch (status) {
@@ -6140,6 +6205,9 @@ var StreamManager = class {
   async cancel(streamId) {
     await this.#store.updateStreamStatus(streamId, "cancelled");
   }
+  async listStreamIds(options) {
+    return this.#store.listStreamIds(options);
+  }
   async persist(stream, streamId, options) {
     const existing = await this.#store.getStream(streamId);
     if (existing && isTerminal(existing.status)) {
@@ -6176,6 +6244,12 @@ var StreamManager = class {
             streamId,
             latencyMs
           });
+          if (options?.onCancelDetected) {
+            try {
+              await options.onCancelDetected({ streamId, latencyMs });
+            } catch {
+            }
+          }
           ac.abort();
           break;
         }
@@ -6464,6 +6538,7 @@ function visualizeGraph(data) {
   return lines.join("\n");
 }
 export {
+  BM25SkillClassifier,
   BinaryInstallError,
   ComposeStartError,
   ComposeStrategy,
@@ -6547,6 +6622,7 @@ export {
   role,
   runGuardrailChain,
   skills,
+  skillsReminder,
   soul,
   stop,
   stripReminders,