@deepagents/context 0.12.0 → 0.12.1

package/dist/index.js

@@ -222,7 +222,7 @@ function message(content) {
   } : content;
   return {
     id: message2.id,
-    name: "message",
+    name: message2.role,
     data: "content",
     type: "message",
     persist: true,
@@ -244,6 +244,22 @@ function assistantText(content, options) {
     parts: [{ type: "text", text: content }]
   });
 }
+var LAZY_ID = Symbol("lazy-id");
+function isLazyFragment(fragment2) {
+  return LAZY_ID in fragment2;
+}
+function lastAssistantMessage(content) {
+  return {
+    name: "assistant",
+    type: "message",
+    persist: true,
+    data: "content",
+    [LAZY_ID]: {
+      type: "last-assistant",
+      content
+    }
+  };
+}
 
 // packages/context/src/lib/renderers/abstract.renderer.ts
 import pluralize from "pluralize";
@@ -1208,6 +1224,12 @@ var ContextEngine = class {
     if (this.#pendingMessages.length === 0) {
       return;
     }
+    for (let i = 0; i < this.#pendingMessages.length; i++) {
+      const fragment2 = this.#pendingMessages[i];
+      if (isLazyFragment(fragment2)) {
+        this.#pendingMessages[i] = await this.#resolveLazyFragment(fragment2);
+      }
+    }
     let parentId = this.#branch.headMessageId;
     const now = Date.now();
     for (const fragment2 of this.#pendingMessages) {
@@ -1227,6 +1249,39 @@ var ContextEngine = class {
     this.#branch.headMessageId = parentId;
     this.#pendingMessages = [];
   }
+  /**
+   * Resolve a lazy fragment by finding the appropriate ID.
+   */
+  async #resolveLazyFragment(fragment2) {
+    const lazy = fragment2[LAZY_ID];
+    if (lazy.type === "last-assistant") {
+      const lastId = await this.#getLastAssistantId();
+      return assistantText(lazy.content, { id: lastId ?? crypto.randomUUID() });
+    }
+    throw new Error(`Unknown lazy fragment type: ${lazy.type}`);
+  }
+  /**
+   * Find the most recent assistant message ID (pending or persisted).
+   */
+  async #getLastAssistantId() {
+    for (let i = this.#pendingMessages.length - 1; i >= 0; i--) {
+      const msg = this.#pendingMessages[i];
+      if (msg.name === "assistant" && !isLazyFragment(msg)) {
+        return msg.id;
+      }
+    }
+    if (this.#branch?.headMessageId) {
+      const chain = await this.#store.getMessageChain(
+        this.#branch.headMessageId
+      );
+      for (let i = chain.length - 1; i >= 0; i--) {
+        if (chain[i].name === "assistant") {
+          return chain[i].id;
+        }
+      }
+    }
+    return void 0;
+  }
   /**
    * Estimate token count and cost for the full context.
    *
@@ -1506,6 +1561,35 @@ var ContextEngine = class {
   consolidate() {
     return void 0;
   }
+  /**
+   * Extract skill path mappings from available_skills fragments.
+   * Returns array of { host, sandbox } for mounting in sandbox filesystem.
+   *
+   * Reads the original `paths` configuration stored in fragment metadata
+   * by the skills() fragment helper.
+   *
+   * @example
+   * ```ts
+   * const context = new ContextEngine({ store, chatId, userId })
+   *   .set(skills({ paths: [{ host: './skills', sandbox: '/skills' }] }));
+   *
+   * const mounts = context.getSkillMounts();
+   * // [{ host: './skills', sandbox: '/skills' }]
+   * ```
+   */
+  getSkillMounts() {
+    const mounts = [];
+    for (const fragment2 of this.#fragments) {
+      if (fragment2.name === "available_skills" && fragment2.metadata && Array.isArray(fragment2.metadata.paths)) {
+        for (const mapping of fragment2.metadata.paths) {
+          if (typeof mapping === "object" && mapping !== null && typeof mapping.host === "string" && typeof mapping.sandbox === "string") {
+            mounts.push({ host: mapping.host, sandbox: mapping.sandbox });
+          }
+        }
+      }
+    }
+    return mounts;
+  }
   /**
    * Inspect the full context state for debugging.
    * Returns a JSON-serializable object with context information.
@@ -1816,7 +1900,7 @@ var errorRecoveryGuardrail = {
     }
     return logAndFail(
       "Unknown error",
-      `An error occurred: ${errorText.slice(0, 100)}. Let me try a different approach.`
+      `An error occurred: ${errorText}. Let me try a different approach.`
     );
   }
 };
@@ -2642,9 +2726,13 @@ function discoverSkillsInDirectory(directory) {
 
 // packages/context/src/lib/skills/fragments.ts
 function skills(options) {
+  const pathMapping = /* @__PURE__ */ new Map();
+  for (const { host, sandbox } of options.paths) {
+    pathMapping.set(host, sandbox);
+  }
   const skillsMap = /* @__PURE__ */ new Map();
-  for (const dir of options.paths) {
-    const discovered = discoverSkillsInDirectory(dir);
+  for (const { host } of options.paths) {
+    const discovered = discoverSkillsInDirectory(host);
     for (const skill of discovered) {
       skillsMap.set(skill.name, skill);
     }
@@ -2659,14 +2747,26 @@ function skills(options) {
       (s) => !options.exclude.includes(s.name)
     );
   }
-  const skillFragments = filteredSkills.map((skill) => ({
-    name: "skill",
-    data: {
-      name: skill.name,
-      path: skill.skillMdPath,
-      description: skill.description
+  const skillFragments = filteredSkills.map((skill) => {
+    const originalPath = skill.skillMdPath;
+    let sandboxPath = originalPath;
+    for (const [host, sandbox] of pathMapping) {
+      if (originalPath.startsWith(host)) {
+        const relativePath = originalPath.slice(host.length);
+        sandboxPath = sandbox + relativePath;
+        break;
+      }
     }
-  }));
+    return {
+      name: "skill",
+      data: {
+        name: skill.name,
+        path: sandboxPath,
+        description: skill.description
+      },
+      metadata: { originalPath }
+    };
+  });
   return {
     name: "available_skills",
     data: [
@@ -2675,7 +2775,11 @@ function skills(options) {
         data: SKILLS_INSTRUCTIONS
       },
       ...skillFragments
-    ]
+    ],
+    metadata: {
+      paths: options.paths
+      // Store original path mappings for getSkillMounts()
+    }
   };
 }
 var SKILLS_INSTRUCTIONS = `When a user's request matches one of the skills listed below, read the skill's SKILL.md file to get detailed instructions before proceeding. Skills provide specialized knowledge and workflows for specific tasks.
@@ -2933,18 +3037,19 @@ var SqliteContextStore = class extends ContextStore {
   // Message Operations (Graph Nodes)
   // ==========================================================================
   async addMessage(message2) {
+    const existingParent = message2.parentId === message2.id ? this.#db.prepare("SELECT parentId FROM messages WHERE id = ?").get(message2.id) : void 0;
+    const parentId = message2.parentId === message2.id ? existingParent?.parentId ?? null : message2.parentId;
     this.#db.prepare(
       `INSERT INTO messages (id, chatId, parentId, name, type, data, createdAt)
          VALUES (?, ?, ?, ?, ?, ?, ?)
          ON CONFLICT(id) DO UPDATE SET
-           parentId = excluded.parentId,
            name = excluded.name,
            type = excluded.type,
            data = excluded.data`
     ).run(
       message2.id,
       message2.chatId,
-      message2.parentId,
+      parentId,
       message2.name,
       message2.type ?? null,
       JSON.stringify(message2.data),
@@ -3487,13 +3592,9 @@ var Agent = class _Agent {
               writer.write({ type: "finish" });
               return;
             }
-            writer.write({
-              type: "text-delta",
-              id: generateId2(),
-              delta: ` ${failureFeedback}`
-            });
+            writeText(writer, failureFeedback);
             const selfCorrectionText = accumulatedText + " " + failureFeedback;
-            context.set(assistantText(selfCorrectionText));
+            context.set(lastAssistantMessage(selfCorrectionText));
             await context.save();
             currentResult = await this.#createRawStream(
               contextVariables,
@@ -3598,6 +3699,22 @@ var repairToolCall = async ({
   });
   return { ...toolCall, input: JSON.stringify(output) };
 };
+function writeText(writer, text) {
+  const feedbackPartId = generateId2();
+  writer.write({
+    id: feedbackPartId,
+    type: "text-start"
+  });
+  writer.write({
+    id: feedbackPartId,
+    type: "text-delta",
+    delta: ` ${text}`
+  });
+  writer.write({
+    id: feedbackPartId,
+    type: "text-end"
+  });
+}
 
 // packages/context/src/lib/render.ts
 function render(tag, ...fragments) {
@@ -3622,6 +3739,7 @@ export {
   DockerfileBuildError,
   DockerfileStrategy,
   InMemoryContextStore,
+  LAZY_ID,
   MarkdownRenderer,
   ModelsRegistry,
   MountPathError,
@@ -3658,7 +3776,9 @@ export {
   isDockerfileOptions,
   isFragment,
   isFragmentObject,
+  isLazyFragment,
   isMessageFragment,
+  lastAssistantMessage,
   loadSkillMetadata,
   message,
   parseFrontmatter,