@deepagents/context 0.17.0 → 0.17.1

package/dist/index.js

@@ -1,7 +1,5 @@
 // packages/context/src/lib/agent.ts
-import { groq } from "@ai-sdk/groq";
 import {
-  NoSuchToolError,
   Output,
   convertToModelMessages,
   createUIMessageStream,
@@ -13,7 +11,7 @@ import {
 } from "ai";
 import chalk from "chalk";
 import "zod";
-import "@deepagents/agent";
+import { createRepairToolCall } from "@deepagents/agent";
 
 // packages/context/src/lib/fragments.ts
 import { generateId } from "ai";
@@ -169,7 +167,7 @@ var Agent = class _Agent {
       stopWhen: stepCountIs(25),
       tools: this.#options.tools,
       experimental_context: contextVariables,
-      experimental_repairToolCall: repairToolCall,
+      experimental_repairToolCall: createRepairToolCall(this.#options.model),
       toolChoice: this.#options.toolChoice,
       onStepFinish: (step) => {
         const toolCall = step.toolCalls.at(-1);
@@ -225,7 +223,7 @@ var Agent = class _Agent {
       model: this.#options.model,
       system: systemPrompt,
       messages: await convertToModelMessages(messages),
-      experimental_repairToolCall: repairToolCall,
+      experimental_repairToolCall: createRepairToolCall(this.#options.model),
       stopWhen: stepCountIs(50),
       experimental_transform: config?.transform ?? smoothStream(),
       tools: this.#options.tools,
@@ -365,7 +363,7 @@ function structuredOutput(options) {
         system: systemPrompt,
         messages: await convertToModelMessages(messages),
         stopWhen: stepCountIs(25),
-        experimental_repairToolCall: repairToolCall,
+        experimental_repairToolCall: createRepairToolCall(options.model),
         experimental_context: contextVariables,
         output: Output.object({ schema: options.schema }),
         tools: options.tools
@@ -387,7 +385,7 @@ function structuredOutput(options) {
         providerOptions: options.providerOptions,
         model: options.model,
         system: systemPrompt,
-        experimental_repairToolCall: repairToolCall,
+        experimental_repairToolCall: createRepairToolCall(options.model),
         messages: await convertToModelMessages(messages),
         stopWhen: stepCountIs(50),
         experimental_transform: config?.transform ?? smoothStream(),
@@ -398,34 +396,6 @@ function structuredOutput(options) {
     }
   };
 }
-var repairToolCall = async ({
-  toolCall,
-  tools,
-  inputSchema,
-  error
-}) => {
-  console.log(
-    `Debug: ${chalk.yellow("RepairingToolCall")}: ${chalk.bgYellow(toolCall.toolName)}`,
-    error.name,
-    JSON.stringify(toolCall)
-  );
-  if (NoSuchToolError.isInstance(error)) {
-    return null;
-  }
-  const tool = tools[toolCall.toolName];
-  const { output } = await generateText({
-    model: groq("openai/gpt-oss-20b"),
-    output: Output.object({ schema: tool.inputSchema }),
-    prompt: [
-      `The model tried to call the tool "${toolCall.toolName}" with the following inputs:`,
-      JSON.stringify(toolCall.input),
-      `The tool accepts the following schema:`,
-      JSON.stringify(inputSchema(toolCall)),
-      "Please fix the inputs."
-    ].join("\n")
-  });
-  return { ...toolCall, input: JSON.stringify(output) };
-};
 function writeText(writer, text) {
   const feedbackPartId = generateId2();
   writer.write({
@@ -1593,6 +1563,13 @@ var ContextEngine = class {
   get branch() {
     return this.#branchName;
   }
+  /**
+   * Get the current branch head message ID.
+   * Returns undefined if no messages have been saved yet.
+   */
+  get headMessageId() {
+    return this.#branch?.headMessageId ?? void 0;
+  }
   /**
    * Get metadata for the current chat.
    * Returns null if the chat hasn't been initialized yet.
@@ -1699,7 +1676,7 @@ var ContextEngine = class {
   async save(options) {
     await this.#ensureInitialized();
     if (this.#pendingMessages.length === 0) {
-      return;
+      return { headMessageId: this.#branch?.headMessageId ?? void 0 };
     }
     const shouldBranch = options?.branch ?? true;
     for (let i = 0; i < this.#pendingMessages.length; i++) {
@@ -1751,6 +1728,7 @@ var ContextEngine = class {
     await this.#store.updateBranchHead(this.#branch.id, parentId);
     this.#branch.headMessageId = parentId;
     this.#pendingMessages = [];
+    return { headMessageId: this.#branch.headMessageId ?? void 0 };
   }
   /**
    * Resolve a lazy fragment by finding the appropriate ID.