@deepagents/context 0.38.1 → 0.39.0

package/dist/index.d.ts

@@ -13,8 +13,11 @@ export * from './lib/fragments/reasoning.ts';
 export * from './lib/fragments/reminders/classifier.ts';
 export * from './lib/fragments/reminders/combinators.ts';
 export * from './lib/fragments/reminders/content-predicates.ts';
+export * from './lib/fragments/reminders/message-predicates.ts';
 export * from './lib/fragments/reminders/temporal/index.ts';
+export * from './lib/fragments/reminders/tool-predicates.ts';
 export * from './lib/fragments/reminders/turn-predicates.ts';
+export * from './lib/fragments/reminders/usage-predicates.ts';
 export * from './lib/fragments/socratic.ts';
 export * from './lib/fragments/user.ts';
 export * from './lib/guardrail.ts';

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,kBAAkB,CAAC;AACjC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,eAAe,CAAC;AAC9B,cAAc,sBAAsB,CAAC;AACrC,cAAc,iCAAiC,CAAC;AAChD,cAAc,qCAAqC,CAAC;AACpD,cAAc,iBAAiB,CAAC;AAChC,cAAc,mBAAmB,CAAC;AAClC,cAAc,oBAAoB,CAAC;AACnC,cAAc,2BAA2B,CAAC;AAC1C,cAAc,iCAAiC,CAAC;AAChD,cAAc,8BAA8B,CAAC;AAC7C,cAAc,yCAAyC,CAAC;AACxD,cAAc,0CAA0C,CAAC;AACzD,cAAc,iDAAiD,CAAC;AAChE,cAAc,6CAA6C,CAAC;AAC5D,cAAc,8CAA8C,CAAC;AAC7D,cAAc,6BAA6B,CAAC;AAC5C,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,iBAAiB,CAAC;AAChC,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,gBAAgB,CAAC;AAC/B,cAAc,oBAAoB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,kBAAkB,CAAC;AACjC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,eAAe,CAAC;AAC9B,cAAc,sBAAsB,CAAC;AACrC,cAAc,iCAAiC,CAAC;AAChD,cAAc,qCAAqC,CAAC;AACpD,cAAc,iBAAiB,CAAC;AAChC,cAAc,mBAAmB,CAAC;AAClC,cAAc,oBAAoB,CAAC;AACnC,cAAc,2BAA2B,CAAC;AAC1C,cAAc,iCAAiC,CAAC;AAChD,cAAc,8BAA8B,CAAC;AAC7C,cAAc,yCAAyC,CAAC;AACxD,cAAc,0CAA0C,CAAC;AACzD,cAAc,iDAAiD,CAAC;AAChE,cAAc,iDAAiD,CAAC;AAChE,cAAc,6CAA6C,CAAC;AAC5D,cAAc,8CAA8C,CAAC;AAC7D,cAAc,8CAA8C,CAAC;AAC7D,cAAc,+CAA+C,CAAC;AAC9D,cAAc,6BAA6B,CAAC;AAC5C,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,iBAAiB,CAAC;AAChC,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,gBAAgB,CAAC;AAC/B,cAAc,oBAAoB,CAAC"}

package/dist/index.js

@@ -1402,27 +1402,27 @@ import {
 } from "ai";
 import chalk from "chalk";
 function createRepairToolCall(model, abortSignal) {
-  return async ({ toolCall, tools, inputSchema, error }) => {
+  return async ({ toolCall: toolCall2, tools, inputSchema, error }) => {
     if (NoSuchToolError.isInstance(error)) {
       return null;
     }
     console.log(
-      `Debug: ${chalk.yellow("RepairingToolCall")}: ${toolCall.toolName}`
+      `Debug: ${chalk.yellow("RepairingToolCall")}: ${toolCall2.toolName}`
     );
-    const tool3 = tools[toolCall.toolName];
+    const tool3 = tools[toolCall2.toolName];
     const { output } = await generateText({
       abortSignal,
       model,
       output: Output.object({ schema: tool3.inputSchema }),
       prompt: [
-        `The model tried to call the tool "${toolCall.toolName}" with the following inputs:`,
-        JSON.stringify(toolCall.input),
+        `The model tried to call the tool "${toolCall2.toolName}" with the following inputs:`,
+        JSON.stringify(toolCall2.input),
         `The tool accepts the following schema:`,
-        JSON.stringify(inputSchema(toolCall)),
+        JSON.stringify(inputSchema(toolCall2)),
         "Please fix the inputs."
       ].join("\n")
     });
-    return { ...toolCall, input: JSON.stringify(output) };
+    return { ...toolCall2, input: JSON.stringify(output) };
   };
 }
 
@@ -1470,10 +1470,10 @@ var Agent = class _Agent {
       ),
       toolChoice: this.#options.toolChoice,
       onStepFinish: (step) => {
-        const toolCall = step.toolCalls.at(-1);
-        if (toolCall) {
+        const toolCall2 = step.toolCalls.at(-1);
+        if (toolCall2) {
           console.log(
-            `Debug: ${chalk2.yellow("ToolCalled")}: ${toolCall.toolName}(${JSON.stringify(toolCall.input)})`
+            `Debug: ${chalk2.yellow("ToolCalled")}: ${toolCall2.toolName}(${JSON.stringify(toolCall2.input)})`
           );
         }
       }
@@ -1544,10 +1544,10 @@ var Agent = class _Agent {
       experimental_context: contextVariables,
       toolChoice: this.#options.toolChoice,
       onStepFinish: (step) => {
-        const toolCall = step.toolCalls.at(-1);
-        if (toolCall) {
+        const toolCall2 = step.toolCalls.at(-1);
+        if (toolCall2) {
           console.log(
-            `Debug: (${runId}) ${chalk2.bold.yellow("ToolCalled")}: ${toolCall.toolName}(${JSON.stringify(toolCall.input)})`
+            `Debug: (${runId}) ${chalk2.bold.yellow("ToolCalled")}: ${toolCall2.toolName}(${JSON.stringify(toolCall2.input)})`
           );
         }
       }
@@ -2744,6 +2744,7 @@ var ContextEngine = class _ContextEngine {
     let lastMessageAt;
     let lastMessage;
     let lastAssistantMessage;
+    const lastAssistantMessages = [];
     if (this.#branch?.headMessageId) {
       const chain = await this.#store.getMessageChain(
         this.#branch.headMessageId
@@ -2751,7 +2752,9 @@ var ContextEngine = class _ContextEngine {
       for (const msg of chain) {
         messageCount++;
         if (msg.name === "assistant") {
-          lastAssistantMessage = msg.data;
+          const assistantMsg = msg.data;
+          lastAssistantMessage = assistantMsg;
+          lastAssistantMessages.push(assistantMsg);
         }
         if (msg.name !== "user") {
           continue;
@@ -2770,7 +2773,8 @@ var ContextEngine = class _ContextEngine {
       messageCount,
       lastMessageAt,
       lastMessage,
-      lastAssistantMessage
+      lastAssistantMessage,
+      lastAssistantMessages
     };
   }
   async getTurnCount() {
@@ -2989,7 +2993,8 @@ var ContextEngine = class _ContextEngine {
             messageCount,
             lastMessageAt,
             lastMessage,
-            lastAssistantMessage
+            lastAssistantMessage,
+            lastAssistantMessages
           } = await this.#getChainContext();
           const original = lastUserFragment.codec.encode();
           const plainText = extractPlainText(original);
@@ -3007,7 +3012,8 @@ var ContextEngine = class _ContextEngine {
             branch: this.#branchName,
             elapsed,
             messageCount,
-            lastAssistantMessage
+            lastAssistantMessage,
+            lastAssistantMessages
           };
           const configs = conditionalReminders.map(
             (it) => it.metadata.reminder
@@ -4515,6 +4521,32 @@ function or(...predicates) {
 function not(predicate) {
   return async (ctx) => !await predicate(ctx);
 }
+function withinLastN(n, predicate) {
+  return async (ctx) => {
+    if (n <= 0) return false;
+    const candidates = (ctx.lastAssistantMessages ?? []).slice(-n);
+    for (const message2 of candidates) {
+      if (await predicate({ ...ctx, lastAssistantMessage: message2 })) {
+        return true;
+      }
+    }
+    return false;
+  };
+}
+function everyOfLastN(n, predicate) {
+  return async (ctx) => {
+    if (n <= 0) return false;
+    const history = ctx.lastAssistantMessages ?? [];
+    if (history.length < n) return false;
+    const candidates = history.slice(-n);
+    for (const message2 of candidates) {
+      if (!await predicate({ ...ctx, lastAssistantMessage: message2 })) {
+        return false;
+      }
+    }
+    return true;
+  };
+}
 
 // packages/context/src/lib/fragments/reminders/content-predicates.ts
 function contentMatches(topics, options) {
@@ -4540,6 +4572,37 @@ function contentPattern(pattern) {
   };
 }
 
+// packages/context/src/lib/fragments/reminders/message-predicates.ts
+function assertCountSpec(spec) {
+  const hasEq = spec.eq !== void 0;
+  const hasRange = spec.gte !== void 0 || spec.lte !== void 0;
+  if (!hasEq && !hasRange) {
+    throw new Error("CountSpec must include at least one of gte/lte/eq");
+  }
+  if (hasEq && hasRange) {
+    throw new Error("CountSpec.eq cannot be combined with gte/lte");
+  }
+}
+function checkCount(count, spec) {
+  if (spec.eq !== void 0) return count === spec.eq;
+  if (spec.gte !== void 0 && count < spec.gte) return false;
+  if (spec.lte !== void 0 && count > spec.lte) return false;
+  return true;
+}
+function lastAssistantLength(spec) {
+  assertCountSpec(spec);
+  return (ctx) => {
+    const message2 = ctx.lastAssistantMessage;
+    const text = message2 ? extractPlainText(message2) : "";
+    return checkCount(text.length, spec);
+  };
+}
+
+// packages/context/src/lib/fragments/reminders/temporal/elapsed.ts
+function elapsedExceeds(ms) {
+  return (ctx) => (ctx.elapsed ?? 0) >= ms;
+}
+
 // packages/context/src/lib/fragments/reminders/temporal/predicates.ts
 var LOCALE_METADATA_KEY = "locale";
 function getLocaleFromMessage(message2) {
@@ -4811,6 +4874,73 @@ function temporalReminder(options) {
   ];
 }
 
+// packages/context/src/lib/fragments/reminders/tool-predicates.ts
+import { isStaticToolUIPart } from "ai";
+var COMPLETED_STATES = /* @__PURE__ */ new Set([
+  "input-available",
+  "output-available",
+  "output-error"
+]);
+function matchesName(spec, name) {
+  return typeof spec === "function" ? spec(name) : spec === name;
+}
+function toolNameOf(part) {
+  return part.type.slice("tool-".length);
+}
+function toolPartsOf(message2) {
+  if (!message2) return [];
+  return message2.parts.filter(isStaticToolUIPart);
+}
+function toolCall(options) {
+  return (ctx) => {
+    const parts = toolPartsOf(ctx.lastAssistantMessage);
+    return parts.some((part) => {
+      if (options.state) {
+        if (part.state !== options.state) return false;
+      } else if (!COMPLETED_STATES.has(part.state)) {
+        return false;
+      }
+      if (options.name !== void 0 && !matchesName(options.name, toolNameOf(part))) {
+        return false;
+      }
+      if (options.input && !options.input(part.input)) {
+        return false;
+      }
+      if (options.output) {
+        if (part.state !== "output-available") return false;
+        if (!options.output(part.output))
+          return false;
+      }
+      if (options.errorText) {
+        if (part.state !== "output-error") return false;
+        const text = part.errorText ?? "";
+        if (!options.errorText(text)) return false;
+      }
+      return true;
+    });
+  };
+}
+function toolCalled(name) {
+  return toolCall({ name });
+}
+function toolFailed(name) {
+  return toolCall({ name, state: "output-error" });
+}
+function anyToolCalled() {
+  return (ctx) => toolPartsOf(ctx.lastAssistantMessage).some(
+    (part) => COMPLETED_STATES.has(part.state)
+  );
+}
+function toolCallCount(name, spec) {
+  assertCountSpec(spec);
+  return (ctx) => {
+    const count = toolPartsOf(ctx.lastAssistantMessage).filter(
+      (part) => COMPLETED_STATES.has(part.state) && matchesName(name, toolNameOf(part))
+    ).length;
+    return checkCount(count, spec);
+  };
+}
+
 // packages/context/src/lib/fragments/reminders/turn-predicates.ts
 function everyNTurns(n) {
   return ({ turn }) => turn % n === 0;
@@ -4825,6 +4955,11 @@ function afterTurn(n) {
   return ({ turn }) => turn > n;
 }
 
+// packages/context/src/lib/fragments/reminders/usage-predicates.ts
+function usageExceeds(totalTokens) {
+  return (ctx) => (ctx.usage?.totalTokens ?? 0) >= totalTokens;
+}
+
 // packages/context/src/lib/fragments/socratic.ts
 function socraticPrompting() {
   return [
@@ -9346,17 +9481,20 @@ export {
   alias,
   analogy,
   and,
+  anyToolCalled,
   applyInlineReminder,
   applyPartReminder,
   applyReminderToMessage,
   asStaticWordPartText,
   asStaticWordText,
+  assertCountSpec,
   assistant,
   assistantText,
   buildSchemaSubcommand,
   buildSubcommandRepair,
   chainHooks,
   chat,
+  checkCount,
   clarification,
   classifies,
   contentIncludes,
@@ -9379,10 +9517,12 @@ export {
   defaultTokenizer,
   defineSubcommandGroup,
   discoverSkillsInDirectory,
+  elapsedExceeds,
   encodeSerializedValue,
   errorRecoveryGuardrail,
   estimate,
   everyNTurns,
+  everyOfLastN,
   example,
   executorContext,
   explain,
@@ -9412,6 +9552,7 @@ export {
   isInstallable,
   isMessageFragment,
   isRecord,
+  lastAssistantLength,
   loadSkillMetadata,
   localeReminder,
   mapGenerateErrorToCode,
@@ -9467,13 +9608,19 @@ export {
   timeReminder,
   toFragment,
   toMessageFragment,
+  toolCall,
+  toolCallCount,
+  toolCalled,
+  toolFailed,
   uploadSkills,
+  usageExceeds,
   useAgentOsSandbox,
   useBashMeta,
   useSandbox,
   user,
   visualizeGraph,
   weekChanged,
+  withinLastN,
   workflow,
   yearChanged,
   yearReminder