@deepagents/context 0.17.0 → 0.18.0

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,
@@ -12,8 +10,7 @@ import {
   streamText
 } 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 +166,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 +222,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 +362,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 +384,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 +395,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 +1562,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 +1675,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 +1727,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.
@@ -5699,6 +5676,47 @@ var SqliteStreamStore = class extends StreamStore {
   async deleteStream(streamId) {
     this.#stmt("DELETE FROM streams WHERE id = ?").run(streamId);
   }
+  async reopenStream(streamId) {
+    return this.#transaction(() => {
+      const row = this.#stmt("SELECT * FROM streams WHERE id = ?").get(
+        streamId
+      );
+      if (!row) {
+        throw new Error(`Stream "${streamId}" not found`);
+      }
+      if (row.status !== "completed" && row.status !== "failed" && row.status !== "cancelled") {
+        throw new Error(
+          `Cannot reopen stream "${streamId}" with status "${row.status}". Only terminal streams can be reopened.`
+        );
+      }
+      this.#stmt("DELETE FROM streams WHERE id = ?").run(streamId);
+      const now = Date.now();
+      this.#stmt(
+        `INSERT INTO streams (id, status, createdAt, startedAt, finishedAt, cancelRequestedAt, error)
+         VALUES (?, ?, ?, ?, ?, ?, ?)`
+      ).run(streamId, "queued", now, null, null, null, null);
+      return {
+        id: streamId,
+        status: "queued",
+        createdAt: now,
+        startedAt: null,
+        finishedAt: null,
+        cancelRequestedAt: null,
+        error: null
+      };
+    });
+  }
+  #transaction(callback) {
+    try {
+      this.#db.exec("BEGIN IMMEDIATE");
+      const result = callback();
+      this.#db.exec("COMMIT");
+      return result;
+    } catch (error) {
+      this.#db.exec("ROLLBACK");
+      throw error;
+    }
+  }
 };
 
 // packages/context/src/lib/stream/stream-manager.ts
@@ -5823,6 +5841,10 @@ var StreamManager = class {
       }
     });
   }
+  async reopen(streamId) {
+    const stream = await this.#store.reopenStream(streamId);
+    return { stream, created: true };
+  }
   async cleanup(streamId) {
     await this.#store.deleteStream(streamId);
   }