@deepagents/context 0.13.0 → 0.14.0

package/dist/index.js

@@ -247,7 +247,7 @@ function assistantText(content, options) {
     parts: [{ type: "text", text: content }]
   });
 }
-var LAZY_ID = Symbol("lazy-id");
+var LAZY_ID = Symbol.for("@deepagents/context:lazy-id");
 function isLazyFragment(fragment2) {
   return LAZY_ID in fragment2;
 }
@@ -1102,7 +1102,7 @@ var ContextEngine = class {
   async #createBranchFrom(messageId, switchTo) {
     const branches = await this.#store.listBranches(this.#chatId);
     const samePrefix = branches.filter(
-      (b) => b.name === this.#branchName || b.name.startsWith(`${this.#branchName}-v`)
+      (it) => it.name === this.#branchName || it.name.startsWith(`${this.#branchName}-v`)
     );
     const newBranchName = `${this.#branchName}-v${samePrefix.length + 1}`;
     const newBranch = {
@@ -1130,6 +1130,15 @@ var ContextEngine = class {
       createdAt: newBranch.createdAt
     };
   }
+  /**
+   * Rewind to a message without clearing pending messages.
+   * Used internally when saving an update to an existing message.
+   */
+  async #rewindForUpdate(messageId) {
+    const pendingBackup = [...this.#pendingMessages];
+    await this.rewind(messageId);
+    this.#pendingMessages = pendingBackup;
+  }
   /**
    * Get the current chat ID.
    */
@@ -1214,7 +1223,18 @@ var ContextEngine = class {
         messages.push(message(msg.data).codec?.decode());
       }
     }
+    for (let i = 0; i < this.#pendingMessages.length; i++) {
+      const fragment2 = this.#pendingMessages[i];
+      if (isLazyFragment(fragment2)) {
+        this.#pendingMessages[i] = await this.#resolveLazyFragment(fragment2);
+      }
+    }
     for (const fragment2 of this.#pendingMessages) {
+      if (!fragment2.codec) {
+        throw new Error(
+          `Fragment "${fragment2.name}" is missing codec. Lazy fragments must be resolved before decode.`
+        );
+      }
       const decoded = fragment2.codec.decode();
       messages.push(decoded);
     }
@@ -1245,9 +1265,24 @@ var ContextEngine = class {
         this.#pendingMessages[i] = await this.#resolveLazyFragment(fragment2);
       }
     }
+    for (const fragment2 of this.#pendingMessages) {
+      if (fragment2.id) {
+        const existing = await this.#store.getMessage(fragment2.id);
+        if (existing && existing.parentId) {
+          await this.#rewindForUpdate(existing.parentId);
+          fragment2.id = crypto.randomUUID();
+          break;
+        }
+      }
+    }
     let parentId = this.#branch.headMessageId;
     const now = Date.now();
     for (const fragment2 of this.#pendingMessages) {
+      if (!fragment2.codec) {
+        throw new Error(
+          `Fragment "${fragment2.name}" is missing codec. Lazy fragments must be resolved before encode.`
+        );
+      }
       const messageData = {
         id: fragment2.id ?? crypto.randomUUID(),
         chatId: this.#chatId,
@@ -1607,33 +1642,28 @@ var ContextEngine = class {
     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.
+   * Extract skill mounts from available_skills fragments.
+   * Returns unified mount array where entries with `name` are individual skills.
    *
    * @example
    * ```ts
    * const context = new ContextEngine({ store, chatId, userId })
    *   .set(skills({ paths: [{ host: './skills', sandbox: '/skills' }] }));
    *
-   * const mounts = context.getSkillMounts();
-   * // [{ host: './skills', sandbox: '/skills' }]
+   * const { mounts } = context.getSkillMounts();
+   * // mounts: [{ name: 'bi-dashboards', host: './skills/bi-dashboards/SKILL.md', sandbox: '/skills/bi-dashboards/SKILL.md' }]
+   *
+   * // Extract skills only (entries with name)
+   * const skills = mounts.filter(m => m.name);
    * ```
    */
   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 });
-          }
-        }
+      if (fragment2.name === "available_skills" && fragment2.metadata?.mounts) {
+        return { mounts: fragment2.metadata.mounts };
       }
     }
-    return mounts;
+    return { mounts: [] };
   }
   /**
    * Inspect the full context state for debugging.
@@ -1870,11 +1900,14 @@ function pass(part) {
 function fail(feedback) {
   return { type: "fail", feedback };
 }
+function stop(part) {
+  return { type: "stop", part };
+}
 function runGuardrailChain(part, guardrails, context) {
   let currentPart = part;
   for (const guardrail2 of guardrails) {
     const result = guardrail2.handle(currentPart, context);
-    if (result.type === "fail") {
+    if (result.type === "fail" || result.type === "stop") {
       return result;
     }
     currentPart = result.part;
@@ -1920,6 +1953,15 @@ var errorRecoveryGuardrail = {
     if (errorText.includes("not in request.tools") || errorText.includes("tool") && errorText.includes("not found")) {
       const toolMatch = errorText.match(/tool '([^']+)'/);
       const toolName = toolMatch ? toolMatch[1] : "unknown";
+      const matchingSkill = context.availableSkills.find(
+        (skill) => skill.name === toolName
+      );
+      if (matchingSkill) {
+        return logAndFail(
+          `Skill confused as tool: ${toolName}`,
+          `"${toolName}" is a skill, not a tool. Read the skill at ${matchingSkill.sandbox} to use it.`
+        );
+      }
       if (context.availableTools.length > 0) {
         return logAndFail(
           `Unregistered tool: ${toolName}`,
@@ -1937,17 +1979,37 @@ var errorRecoveryGuardrail = {
         "I generated malformed JSON for the tool arguments. Let me format my tool call properly with valid JSON."
       );
     }
+    if (errorText.includes("validation failed") && errorText.includes("did not match schema")) {
+      const toolMatch = errorText.match(/parameters for tool (\w+)/);
+      const toolName = toolMatch ? toolMatch[1] : "unknown";
+      const schemaErrors = errorText.match(/errors: \[([^\]]+)\]/)?.[1] || "";
+      return logAndFail(
+        `Schema validation: ${toolName}`,
+        `I called "${toolName}" with invalid parameters. Schema errors: ${schemaErrors}. Let me fix the parameters and try again.`
+      );
+    }
     if (errorText.includes("Parsing failed")) {
       return logAndFail(
         "Parsing failed",
         "My response format was invalid. Let me try again with a properly formatted response."
       );
     }
-    console.dir({ part }, { depth: null });
-    return logAndFail(
-      "Unknown error",
-      `An error occurred: ${errorText}. Let me try a different approach.`
+    if (errorText.includes("Failed to call a function") || errorText.includes("failed_generation")) {
+      if (context.availableTools.length > 0) {
+        return logAndFail(
+          "Failed function call",
+          `My function call was malformed. Available tools: ${context.availableTools.join(", ")}. Let me format my tool call correctly.`
+        );
+      }
+      return logAndFail(
+        "Failed function call (no tools)",
+        "My function call was malformed. Let me respond with plain text instead."
+      );
+    }
+    console.log(
+      `${prefix} ${chalk.yellow("Unknown error - stopping without retry")}`
     );
+    return stop(part);
   }
 };
 
@@ -2713,6 +2775,9 @@ async function createContainerTool(options = {}) {
   };
 }
 
+// packages/context/src/lib/skills/fragments.ts
+import dedent from "dedent";
+
 // packages/context/src/lib/skills/loader.ts
 import * as fs from "node:fs";
 import * as path2 from "node:path";
@@ -2793,7 +2858,14 @@ function skills(options) {
       (s) => !options.exclude.includes(s.name)
     );
   }
-  const skillFragments = filteredSkills.map((skill) => {
+  if (filteredSkills.length === 0) {
+    return {
+      name: "available_skills",
+      data: [],
+      metadata: { mounts: [] }
+    };
+  }
+  const mounts = filteredSkills.map((skill) => {
     const originalPath = skill.skillMdPath;
     let sandboxPath = originalPath;
     for (const [host, sandbox] of pathMapping) {
@@ -2803,14 +2875,21 @@ function skills(options) {
         break;
       }
     }
+    return {
+      name: skill.name,
+      description: skill.description,
+      host: originalPath,
+      sandbox: sandboxPath
+    };
+  });
+  const skillFragments = mounts.map((skill) => {
     return {
       name: "skill",
       data: {
         name: skill.name,
-        path: sandboxPath,
+        path: skill.sandbox,
         description: skill.description
-      },
-      metadata: { originalPath }
+      }
     };
   });
   return {
@@ -2823,100 +2902,57 @@ function skills(options) {
       ...skillFragments
     ],
     metadata: {
-      paths: options.paths
-      // Store original path mappings for getSkillMounts()
+      mounts
     }
   };
 }
-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.
+var SKILLS_INSTRUCTIONS = dedent`A skill is a set of local instructions to follow that is stored in a \`SKILL.md\` file. Below is the list of skills that can be used. Each entry includes a name, description, and file path so you can open the source for full instructions when using a specific skill.
 
-To use a skill:
-1. Identify if the user's request matches a skill's description
-2. Read the SKILL.md file at the skill's path to load full instructions
-3. Follow the skill's guidance to complete the task
-
-Skills are only loaded when relevant - don't read skill files unless needed.`;
+### How to use skills
+- Discovery: The list below shows the skills available in this session (name + description + file path). Skill bodies live on disk at the listed paths.
+- Trigger rules: If the user names a skill (with \`$SkillName\` or plain text) OR the task clearly matches a skill's description shown below, you must use that skill for that turn before doing anything else. Multiple mentions mean use them all. Do not carry skills across turns unless re-mentioned.
+- Missing/blocked: If a named skill isn't in the list or the path can't be read, say so briefly and continue with the best fallback.
+- How to use a skill (progressive disclosure):
+  1) After deciding to use a skill, open its \`SKILL.md\`. Read only enough to follow the workflow.
+  2) If \`SKILL.md\` points to extra folders such as \`references/\`, load only the specific files needed for the request; don't bulk-load everything.
+  3) If \`scripts/\` exist, prefer running or patching them instead of retyping large code blocks.
+  4) If \`assets/\` or templates exist, reuse them instead of recreating from scratch.
+- Coordination and sequencing:
+  - If multiple skills apply, choose the minimal set that covers the request and state the order you'll use them.
+  - Announce which skill(s) you're using and why (one short line). If you skip an obvious skill, say why.
+- Context hygiene:
+  - Keep context small: summarize long sections instead of pasting them; only load extra files when needed.
+  - Avoid deep reference-chasing: prefer opening only files directly linked from \`SKILL.md\` unless you're blocked.
+  - When variants exist (frameworks, providers, domains), pick only the relevant reference file(s) and note that choice.
+- Safety and fallback: If a skill can't be applied cleanly (missing files, unclear instructions), state the issue, pick the next-best approach, and continue.`;
 
 // packages/context/src/lib/store/sqlite.store.ts
 import { DatabaseSync } from "node:sqlite";
-var STORE_DDL = `
--- Chats table
--- createdAt/updatedAt: DEFAULT for insert, inline SET for updates
-CREATE TABLE IF NOT EXISTS chats (
-  id TEXT PRIMARY KEY,
-  userId TEXT NOT NULL,
-  title TEXT,
-  metadata TEXT,
-  createdAt INTEGER NOT NULL DEFAULT (strftime('%s', 'now') * 1000),
-  updatedAt INTEGER NOT NULL DEFAULT (strftime('%s', 'now') * 1000)
-);
-
-CREATE INDEX IF NOT EXISTS idx_chats_updatedAt ON chats(updatedAt);
-CREATE INDEX IF NOT EXISTS idx_chats_userId ON chats(userId);
-
--- Messages table (nodes in the DAG)
-CREATE TABLE IF NOT EXISTS messages (
-  id TEXT PRIMARY KEY,
-  chatId TEXT NOT NULL,
-  parentId TEXT,
-  name TEXT NOT NULL,
-  type TEXT,
-  data TEXT NOT NULL,
-  createdAt INTEGER NOT NULL,
-  FOREIGN KEY (chatId) REFERENCES chats(id) ON DELETE CASCADE,
-  FOREIGN KEY (parentId) REFERENCES messages(id)
-);
-
-CREATE INDEX IF NOT EXISTS idx_messages_chatId ON messages(chatId);
-CREATE INDEX IF NOT EXISTS idx_messages_parentId ON messages(parentId);
 
--- Branches table (pointers to head messages)
-CREATE TABLE IF NOT EXISTS branches (
-  id TEXT PRIMARY KEY,
-  chatId TEXT NOT NULL,
-  name TEXT NOT NULL,
-  headMessageId TEXT,
-  isActive INTEGER NOT NULL DEFAULT 0,
-  createdAt INTEGER NOT NULL,
-  FOREIGN KEY (chatId) REFERENCES chats(id) ON DELETE CASCADE,
-  FOREIGN KEY (headMessageId) REFERENCES messages(id),
-  UNIQUE(chatId, name)
-);
+// packages/context/src/lib/store/ddl.sqlite.sql
+var ddl_sqlite_default = "-- Context Store DDL for SQLite\n-- This schema implements a DAG-based message history with branching and checkpoints.\n\n-- Performance PRAGMAs (session-level, run on each connection)\nPRAGMA journal_mode = WAL;\nPRAGMA synchronous = NORMAL;\nPRAGMA cache_size = -64000;\nPRAGMA temp_store = MEMORY;\nPRAGMA mmap_size = 268435456;\n\n-- Integrity\nPRAGMA foreign_keys = ON;\n\n-- Chats table\n-- createdAt/updatedAt: DEFAULT for insert, inline SET for updates\nCREATE TABLE IF NOT EXISTS chats (\n  id TEXT PRIMARY KEY,\n  userId TEXT NOT NULL,\n  title TEXT,\n  metadata TEXT,\n  createdAt INTEGER NOT NULL DEFAULT (strftime('%s', 'now') * 1000),\n  updatedAt INTEGER NOT NULL DEFAULT (strftime('%s', 'now') * 1000)\n);\n\nCREATE INDEX IF NOT EXISTS idx_chats_updatedAt ON chats(updatedAt);\nCREATE INDEX IF NOT EXISTS idx_chats_userId ON chats(userId);\n-- Composite index for listChats(): WHERE userId = ? ORDER BY updatedAt DESC\nCREATE INDEX IF NOT EXISTS idx_chats_userId_updatedAt ON chats(userId, updatedAt DESC);\n\n-- Messages table (nodes in the DAG)\nCREATE TABLE IF NOT EXISTS messages (\n  id TEXT PRIMARY KEY,\n  chatId TEXT NOT NULL,\n  parentId TEXT,\n  name TEXT NOT NULL,\n  type TEXT,\n  data TEXT NOT NULL,\n  createdAt INTEGER NOT NULL,\n  FOREIGN KEY (chatId) REFERENCES chats(id) ON DELETE CASCADE,\n  FOREIGN KEY (parentId) REFERENCES messages(id)\n);\n\nCREATE INDEX IF NOT EXISTS idx_messages_chatId ON messages(chatId);\nCREATE INDEX IF NOT EXISTS idx_messages_parentId ON messages(parentId);\n-- Composite index for recursive CTE parent traversal in getMessageChain()\nCREATE INDEX IF NOT EXISTS idx_messages_chatId_parentId ON messages(chatId, parentId);\n\n-- Branches table (pointers to head messages)\nCREATE TABLE IF NOT EXISTS branches (\n  id TEXT PRIMARY KEY,\n  chatId TEXT NOT NULL,\n  name TEXT NOT NULL,\n  headMessageId TEXT,\n  isActive INTEGER NOT NULL DEFAULT 0,\n  createdAt INTEGER NOT NULL,\n  FOREIGN KEY (chatId) REFERENCES chats(id) ON DELETE CASCADE,\n  FOREIGN KEY (headMessageId) REFERENCES messages(id),\n  UNIQUE(chatId, name)\n);\n\nCREATE INDEX IF NOT EXISTS idx_branches_chatId ON branches(chatId);\n-- Composite index for getActiveBranch(): WHERE chatId = ? AND isActive = 1\nCREATE INDEX IF NOT EXISTS idx_branches_chatId_isActive ON branches(chatId, isActive);\n\n-- Checkpoints table (pointers to message nodes)\nCREATE TABLE IF NOT EXISTS checkpoints (\n  id TEXT PRIMARY KEY,\n  chatId TEXT NOT NULL,\n  name TEXT NOT NULL,\n  messageId TEXT NOT NULL,\n  createdAt INTEGER NOT NULL,\n  FOREIGN KEY (chatId) REFERENCES chats(id) ON DELETE CASCADE,\n  FOREIGN KEY (messageId) REFERENCES messages(id),\n  UNIQUE(chatId, name)\n);\n\nCREATE INDEX IF NOT EXISTS idx_checkpoints_chatId ON checkpoints(chatId);\n\n-- FTS5 virtual table for full-text search\n-- messageId/chatId/name are UNINDEXED (stored but not searchable, used for filtering/joining)\n-- Only 'content' is indexed for full-text search\nCREATE VIRTUAL TABLE IF NOT EXISTS messages_fts USING fts5(\n  messageId UNINDEXED,\n  chatId UNINDEXED,\n  name UNINDEXED,\n  content,\n  tokenize='porter unicode61'\n);\n";
 
-CREATE INDEX IF NOT EXISTS idx_branches_chatId ON branches(chatId);
-
--- Checkpoints table (pointers to message nodes)
-CREATE TABLE IF NOT EXISTS checkpoints (
-  id TEXT PRIMARY KEY,
-  chatId TEXT NOT NULL,
-  name TEXT NOT NULL,
-  messageId TEXT NOT NULL,
-  createdAt INTEGER NOT NULL,
-  FOREIGN KEY (chatId) REFERENCES chats(id) ON DELETE CASCADE,
-  FOREIGN KEY (messageId) REFERENCES messages(id),
-  UNIQUE(chatId, name)
-);
-
-CREATE INDEX IF NOT EXISTS idx_checkpoints_chatId ON checkpoints(chatId);
-
--- FTS5 virtual table for full-text search
--- messageId/chatId/name are UNINDEXED (stored but not searchable, used for filtering/joining)
--- Only 'content' is indexed for full-text search
-CREATE VIRTUAL TABLE IF NOT EXISTS messages_fts USING fts5(
-  messageId UNINDEXED,
-  chatId UNINDEXED,
-  name UNINDEXED,
-  content,
-  tokenize='porter unicode61'
-);
-`;
+// packages/context/src/lib/store/sqlite.store.ts
 var SqliteContextStore = class extends ContextStore {
   #db;
+  #statements = /* @__PURE__ */ new Map();
+  /**
+   * Get or create a prepared statement.
+   * Statements are cached for the lifetime of the store to avoid
+   * repeated SQL parsing and compilation overhead.
+   */
+  #stmt(sql) {
+    let stmt = this.#statements.get(sql);
+    if (!stmt) {
+      stmt = this.#db.prepare(sql);
+      this.#statements.set(sql, stmt);
+    }
+    return stmt;
+  }
   constructor(path3) {
     super();
     this.#db = new DatabaseSync(path3);
-    this.#db.exec("PRAGMA foreign_keys = ON");
-    this.#db.exec(STORE_DDL);
+    this.#db.exec(ddl_sqlite_default);
   }
   /**
    * Execute a function within a transaction.
@@ -2937,11 +2973,12 @@ var SqliteContextStore = class extends ContextStore {
   // Chat Operations
   // ==========================================================================
   async createChat(chat) {
-    this.#useTransaction(() => {
-      this.#db.prepare(
+    return this.#useTransaction(() => {
+      const row = this.#db.prepare(
         `INSERT INTO chats (id, userId, title, metadata)
-           VALUES (?, ?, ?, ?)`
-      ).run(
+           VALUES (?, ?, ?, ?)
+           RETURNING *`
+      ).get(
         chat.id,
         chat.userId,
         chat.title ?? null,
@@ -2951,6 +2988,14 @@ var SqliteContextStore = class extends ContextStore {
         `INSERT INTO branches (id, chatId, name, headMessageId, isActive, createdAt)
            VALUES (?, ?, 'main', NULL, 1, ?)`
       ).run(crypto.randomUUID(), chat.id, Date.now());
+      return {
+        id: row.id,
+        userId: row.userId,
+        title: row.title ?? void 0,
+        metadata: row.metadata ? JSON.parse(row.metadata) : void 0,
+        createdAt: row.createdAt,
+        updatedAt: row.updatedAt
+      };
     });
   }
   async upsertChat(chat) {
@@ -3093,21 +3138,16 @@ var SqliteContextStore = class extends ContextStore {
   // Message Operations (Graph Nodes)
   // ==========================================================================
   async addMessage(message2) {
-    this.#db.prepare(
+    if (message2.parentId === message2.id) {
+      throw new Error(`Message ${message2.id} cannot be its own parent`);
+    }
+    this.#stmt(
       `INSERT INTO messages (id, chatId, parentId, name, type, data, createdAt)
-         VALUES (
-           ?1,
-           ?2,
-           CASE WHEN ?3 = ?1 THEN (SELECT parentId FROM messages WHERE id = ?1) ELSE ?3 END,
-           ?4,
-           ?5,
-           ?6,
-           ?7
-         )
-         ON CONFLICT(id) DO UPDATE SET
-           name = excluded.name,
-           type = excluded.type,
-           data = excluded.data`
+       VALUES (?, ?, ?, ?, ?, ?, ?)
+       ON CONFLICT(id) DO UPDATE SET
+         name = excluded.name,
+         type = excluded.type,
+         data = excluded.data`
     ).run(
       message2.id,
       message2.chatId,
@@ -3118,14 +3158,16 @@ var SqliteContextStore = class extends ContextStore {
       message2.createdAt
     );
     const content = typeof message2.data === "string" ? message2.data : JSON.stringify(message2.data);
-    this.#db.prepare(`DELETE FROM messages_fts WHERE messageId = ?`).run(message2.id);
-    this.#db.prepare(
+    this.#stmt(`DELETE FROM messages_fts WHERE messageId = ?`).run(message2.id);
+    this.#stmt(
       `INSERT INTO messages_fts(messageId, chatId, name, content)
-         VALUES (?, ?, ?, ?)`
+       VALUES (?, ?, ?, ?)`
     ).run(message2.id, message2.chatId, message2.name, content);
   }
   async getMessage(messageId) {
-    const row = this.#db.prepare("SELECT * FROM messages WHERE id = ?").get(messageId);
+    const row = this.#stmt("SELECT * FROM messages WHERE id = ?").get(
+      messageId
+    );
     if (!row) {
       return void 0;
     }
@@ -3140,15 +3182,16 @@ var SqliteContextStore = class extends ContextStore {
     };
   }
   async getMessageChain(headId) {
-    const rows = this.#db.prepare(
+    const rows = this.#stmt(
       `WITH RECURSIVE chain AS (
-          SELECT *, 0 as depth FROM messages WHERE id = ?
-          UNION ALL
-          SELECT m.*, c.depth + 1 FROM messages m
-          INNER JOIN chain c ON m.id = c.parentId
-        )
-        SELECT * FROM chain
-        ORDER BY depth DESC`
+        SELECT *, 0 as depth FROM messages WHERE id = ?
+        UNION ALL
+        SELECT m.*, c.depth + 1 FROM messages m
+        INNER JOIN chain c ON m.id = c.parentId
+        WHERE c.depth < 100000
+      )
+      SELECT * FROM chain
+      ORDER BY depth DESC`
     ).all(headId);
     return rows.map((row) => ({
       id: row.id,
@@ -3161,7 +3204,7 @@ var SqliteContextStore = class extends ContextStore {
     }));
   }
   async hasChildren(messageId) {
-    const row = this.#db.prepare(
+    const row = this.#stmt(
       "SELECT EXISTS(SELECT 1 FROM messages WHERE parentId = ?) as hasChildren"
     ).get(messageId);
     return row.hasChildren === 1;
@@ -3208,7 +3251,9 @@ var SqliteContextStore = class extends ContextStore {
     };
   }
   async getActiveBranch(chatId) {
-    const row = this.#db.prepare("SELECT * FROM branches WHERE chatId = ? AND isActive = 1").get(chatId);
+    const row = this.#stmt(
+      "SELECT * FROM branches WHERE chatId = ? AND isActive = 1"
+    ).get(chatId);
     if (!row) {
       return void 0;
     }
@@ -3226,45 +3271,43 @@ var SqliteContextStore = class extends ContextStore {
     this.#db.prepare("UPDATE branches SET isActive = 1 WHERE id = ?").run(branchId);
   }
   async updateBranchHead(branchId, messageId) {
-    this.#db.prepare("UPDATE branches SET headMessageId = ? WHERE id = ?").run(messageId, branchId);
+    this.#stmt("UPDATE branches SET headMessageId = ? WHERE id = ?").run(
+      messageId,
+      branchId
+    );
   }
   async listBranches(chatId) {
-    const branches = this.#db.prepare(
+    const rows = this.#db.prepare(
       `SELECT
           b.id,
           b.name,
           b.headMessageId,
           b.isActive,
-          b.createdAt
+          b.createdAt,
+          COALESCE(
+            (
+              WITH RECURSIVE chain AS (
+                SELECT id, parentId FROM messages WHERE id = b.headMessageId
+                UNION ALL
+                SELECT m.id, m.parentId FROM messages m
+                INNER JOIN chain c ON m.id = c.parentId
+              )
+              SELECT COUNT(*) FROM chain
+            ),
+            0
+          ) as messageCount
         FROM branches b
         WHERE b.chatId = ?
         ORDER BY b.createdAt ASC`
     ).all(chatId);
-    const result = [];
-    for (const branch of branches) {
-      let messageCount = 0;
-      if (branch.headMessageId) {
-        const countRow = this.#db.prepare(
-          `WITH RECURSIVE chain AS (
-              SELECT id, parentId FROM messages WHERE id = ?
-              UNION ALL
-              SELECT m.id, m.parentId FROM messages m
-              INNER JOIN chain c ON m.id = c.parentId
-            )
-            SELECT COUNT(*) as count FROM chain`
-        ).get(branch.headMessageId);
-        messageCount = countRow.count;
-      }
-      result.push({
-        id: branch.id,
-        name: branch.name,
-        headMessageId: branch.headMessageId,
-        isActive: branch.isActive === 1,
-        messageCount,
-        createdAt: branch.createdAt
-      });
-    }
-    return result;
+    return rows.map((row) => ({
+      id: row.id,
+      name: row.name,
+      headMessageId: row.headMessageId,
+      isActive: row.isActive === 1,
+      messageCount: row.messageCount,
+      createdAt: row.createdAt
+    }));
   }
   // ==========================================================================
   // Checkpoint Operations
@@ -3417,152 +3460,1450 @@ var InMemoryContextStore = class extends SqliteContextStore {
   }
 };
 
-// packages/context/src/lib/visualize.ts
-function visualizeGraph(data) {
-  if (data.nodes.length === 0) {
-    return `[chat: ${data.chatId}]
+// packages/context/src/lib/store/postgres.store.ts
+import { createRequire } from "node:module";
 
-(empty)`;
-  }
-  const childrenByParentId = /* @__PURE__ */ new Map();
-  const branchHeads = /* @__PURE__ */ new Map();
-  const checkpointsByMessageId = /* @__PURE__ */ new Map();
-  for (const node of data.nodes) {
-    const children = childrenByParentId.get(node.parentId) ?? [];
-    children.push(node);
-    childrenByParentId.set(node.parentId, children);
-  }
-  for (const branch of data.branches) {
-    if (branch.headMessageId) {
-      const heads = branchHeads.get(branch.headMessageId) ?? [];
-      heads.push(branch.isActive ? `${branch.name} *` : branch.name);
-      branchHeads.set(branch.headMessageId, heads);
-    }
-  }
-  for (const checkpoint of data.checkpoints) {
-    const cps = checkpointsByMessageId.get(checkpoint.messageId) ?? [];
-    cps.push(checkpoint.name);
-    checkpointsByMessageId.set(checkpoint.messageId, cps);
-  }
-  const roots = childrenByParentId.get(null) ?? [];
-  const lines = [`[chat: ${data.chatId}]`, ""];
-  function renderNode(node, prefix, isLast, isRoot) {
-    const connector = isRoot ? "" : isLast ? "\u2514\u2500\u2500 " : "\u251C\u2500\u2500 ";
-    const contentPreview = node.content.replace(/\n/g, " ");
-    let line = `${prefix}${connector}${node.id.slice(0, 8)} (${node.role}): "${contentPreview}"`;
-    const branches = branchHeads.get(node.id);
-    if (branches) {
-      line += ` <- [${branches.join(", ")}]`;
-    }
-    const checkpoints = checkpointsByMessageId.get(node.id);
-    if (checkpoints) {
-      line += ` {${checkpoints.join(", ")}}`;
-    }
-    lines.push(line);
-    const children = childrenByParentId.get(node.id) ?? [];
-    const childPrefix = isRoot ? "" : prefix + (isLast ? "    " : "\u2502   ");
-    for (let i = 0; i < children.length; i++) {
-      renderNode(children[i], childPrefix, i === children.length - 1, false);
-    }
-  }
-  for (let i = 0; i < roots.length; i++) {
-    renderNode(roots[i], "", i === roots.length - 1, true);
-  }
-  lines.push("");
-  lines.push("Legend: * = active branch, {...} = checkpoint");
-  return lines.join("\n");
-}
+// packages/context/src/lib/store/ddl.postgres.sql
+var ddl_postgres_default = "-- Context Store DDL for PostgreSQL\n-- This schema implements a DAG-based message history with branching and checkpoints.\n\n-- Chats table\n-- createdAt/updatedAt: DEFAULT for insert, inline SET for updates\nCREATE TABLE IF NOT EXISTS chats (\n  id TEXT PRIMARY KEY,\n  userId TEXT NOT NULL,\n  title TEXT,\n  metadata JSONB,\n  createdAt BIGINT NOT NULL DEFAULT (EXTRACT(EPOCH FROM NOW()) * 1000)::BIGINT,\n  updatedAt BIGINT NOT NULL DEFAULT (EXTRACT(EPOCH FROM NOW()) * 1000)::BIGINT\n);\n\nCREATE INDEX IF NOT EXISTS idx_chats_updatedAt ON chats(updatedAt);\nCREATE INDEX IF NOT EXISTS idx_chats_userId ON chats(userId);\nCREATE INDEX IF NOT EXISTS idx_chats_metadata ON chats USING GIN (metadata);\n\n-- Messages table (nodes in the DAG)\nCREATE TABLE IF NOT EXISTS messages (\n  id TEXT PRIMARY KEY,\n  chatId TEXT NOT NULL,\n  parentId TEXT,\n  name TEXT NOT NULL,\n  type TEXT,\n  data JSONB NOT NULL,\n  createdAt BIGINT NOT NULL,\n  FOREIGN KEY (chatId) REFERENCES chats(id) ON DELETE CASCADE,\n  FOREIGN KEY (parentId) REFERENCES messages(id)\n);\n\nCREATE INDEX IF NOT EXISTS idx_messages_chatId ON messages(chatId);\nCREATE INDEX IF NOT EXISTS idx_messages_parentId ON messages(parentId);\n\n-- Branches table (pointers to head messages)\nCREATE TABLE IF NOT EXISTS branches (\n  id TEXT PRIMARY KEY,\n  chatId TEXT NOT NULL,\n  name TEXT NOT NULL,\n  headMessageId TEXT,\n  isActive BOOLEAN NOT NULL DEFAULT FALSE,\n  createdAt BIGINT NOT NULL,\n  FOREIGN KEY (chatId) REFERENCES chats(id) ON DELETE CASCADE,\n  FOREIGN KEY (headMessageId) REFERENCES messages(id),\n  UNIQUE(chatId, name)\n);\n\nCREATE INDEX IF NOT EXISTS idx_branches_chatId ON branches(chatId);\n\n-- Checkpoints table (pointers to message nodes)\nCREATE TABLE IF NOT EXISTS checkpoints (\n  id TEXT PRIMARY KEY,\n  chatId TEXT NOT NULL,\n  name TEXT NOT NULL,\n  messageId TEXT NOT NULL,\n  createdAt BIGINT NOT NULL,\n  FOREIGN KEY (chatId) REFERENCES chats(id) ON DELETE CASCADE,\n  FOREIGN KEY (messageId) REFERENCES messages(id),\n  UNIQUE(chatId, name)\n);\n\nCREATE INDEX IF NOT EXISTS idx_checkpoints_chatId ON checkpoints(chatId);\n\n-- Full-text search using tsvector + GIN index\nCREATE TABLE IF NOT EXISTS messages_fts (\n  messageId TEXT PRIMARY KEY REFERENCES messages(id) ON DELETE CASCADE,\n  chatId TEXT NOT NULL,\n  name TEXT NOT NULL,\n  content TEXT NOT NULL,\n  content_vector TSVECTOR\n);\n\nCREATE INDEX IF NOT EXISTS idx_messages_fts_vector ON messages_fts USING GIN(content_vector);\nCREATE INDEX IF NOT EXISTS idx_messages_fts_chatId ON messages_fts(chatId);\n\n-- Trigger to automatically update tsvector on insert/update\nCREATE OR REPLACE FUNCTION messages_fts_update_vector() RETURNS TRIGGER AS $$\nBEGIN\n  NEW.content_vector := to_tsvector('english', NEW.content);\n  RETURN NEW;\nEND;\n$$ LANGUAGE plpgsql;\n\nDROP TRIGGER IF EXISTS messages_fts_vector_update ON messages_fts;\nCREATE TRIGGER messages_fts_vector_update\n  BEFORE INSERT OR UPDATE ON messages_fts\n  FOR EACH ROW\n  EXECUTE FUNCTION messages_fts_update_vector();\n";
 
-// packages/context/src/lib/agent.ts
-import { groq } from "@ai-sdk/groq";
-import {
-  NoSuchToolError,
-  Output,
-  convertToModelMessages,
-  createUIMessageStream,
-  generateId as generateId2,
-  generateText,
-  smoothStream,
-  stepCountIs,
-  streamText
-} from "ai";
-import chalk2 from "chalk";
-import "zod";
-import "@deepagents/agent";
-var Agent = class _Agent {
-  #options;
-  #guardrails = [];
-  tools;
+// packages/context/src/lib/store/postgres.store.ts
+var PostgresContextStore = class _PostgresContextStore extends ContextStore {
+  #pool;
+  #initialized;
   constructor(options) {
-    this.#options = options;
-    this.tools = options.tools || {};
-    this.#guardrails = options.guardrails || [];
+    super();
+    const pg = _PostgresContextStore.#requirePg();
+    this.#pool = typeof options.pool === "string" ? new pg.Pool({ connectionString: options.pool }) : new pg.Pool(options.pool);
+    this.#initialized = this.#initialize();
   }
-  async generate(contextVariables, config) {
-    if (!this.#options.context) {
-      throw new Error(`Agent ${this.#options.name} is missing a context.`);
-    }
-    if (!this.#options.model) {
-      throw new Error(`Agent ${this.#options.name} is missing a model.`);
+  static #requirePg() {
+    try {
+      const require2 = createRequire(import.meta.url);
+      return require2("pg");
+    } catch {
+      throw new Error(
+        'PostgresContextStore requires the "pg" package. Install it with: npm install pg'
+      );
     }
-    const { messages, systemPrompt } = await this.#options.context.resolve({
-      renderer: new XmlRenderer()
-    });
-    return generateText({
-      abortSignal: config?.abortSignal,
-      providerOptions: this.#options.providerOptions,
-      model: this.#options.model,
-      system: systemPrompt,
-      messages: await convertToModelMessages(messages),
-      stopWhen: stepCountIs(25),
-      tools: this.#options.tools,
-      experimental_context: contextVariables,
-      experimental_repairToolCall: repairToolCall,
-      toolChoice: this.#options.toolChoice,
-      onStepFinish: (step) => {
-        const toolCall = step.toolCalls.at(-1);
-        if (toolCall) {
-          console.log(
-            `Debug: ${chalk2.yellow("ToolCalled")}: ${toolCall.toolName}(${JSON.stringify(toolCall.input)})`
-          );
-        }
-      }
-    });
+  }
+  async #initialize() {
+    await this.#pool.query(ddl_postgres_default);
   }
   /**
-   * Stream a response from the agent.
-   *
-   * When guardrails are configured, `toUIMessageStream()` is wrapped to provide
-   * self-correction behavior. Direct access to fullStream/textStream bypasses guardrails.
-   *
-   * @example
-   * ```typescript
-   * const stream = await agent.stream({});
-   *
-   * // With guardrails - use toUIMessageStream for protection
-   * await printer.readableStream(stream.toUIMessageStream());
-   *
-   * // Or use printer.stdout which uses toUIMessageStream internally
-   * await printer.stdout(stream);
-   * ```
+   * Ensure initialization is complete before any operation.
    */
-  async stream(contextVariables, config) {
-    if (!this.#options.context) {
-      throw new Error(`Agent ${this.#options.name} is missing a context.`);
-    }
-    if (!this.#options.model) {
-      throw new Error(`Agent ${this.#options.name} is missing a model.`);
-    }
-    const result = await this.#createRawStream(contextVariables, config);
-    if (this.#guardrails.length === 0) {
+  async #ensureInitialized() {
+    await this.#initialized;
+  }
+  /**
+   * Execute a function within a transaction.
+   * Automatically commits on success or rolls back on error.
+   */
+  async #useTransaction(fn) {
+    await this.#ensureInitialized();
+    const client = await this.#pool.connect();
+    try {
+      await client.query("BEGIN");
+      const result = await fn(client);
+      await client.query("COMMIT");
       return result;
+    } catch (error) {
+      await client.query("ROLLBACK");
+      throw error;
+    } finally {
+      client.release();
     }
-    return this.#wrapWithGuardrails(result, contextVariables, config);
   }
   /**
-   * Create a raw stream without guardrail processing.
+   * Execute a query using the pool (no transaction).
    */
-  async #createRawStream(contextVariables, config) {
+  async #query(sql, params) {
+    await this.#ensureInitialized();
+    const result = await this.#pool.query(sql, params);
+    return result.rows;
+  }
+  /**
+   * Close the pool connection.
+   * Call this when done with the store.
+   */
+  async close() {
+    await this.#pool.end();
+  }
+  // ==========================================================================
+  // Chat Operations
+  // ==========================================================================
+  async createChat(chat) {
+    return this.#useTransaction(async (client) => {
+      const result = await client.query(
+        `INSERT INTO chats (id, userId, title, metadata)
+         VALUES ($1, $2, $3, $4)
+         RETURNING *`,
+        [
+          chat.id,
+          chat.userId,
+          chat.title ?? null,
+          chat.metadata ? JSON.stringify(chat.metadata) : null
+        ]
+      );
+      const row = result.rows[0];
+      await client.query(
+        `INSERT INTO branches (id, chatId, name, headMessageId, isActive, createdAt)
+         VALUES ($1, $2, 'main', NULL, TRUE, $3)`,
+        [crypto.randomUUID(), chat.id, Date.now()]
+      );
+      return {
+        id: row.id,
+        userId: row.userid,
+        title: row.title ?? void 0,
+        metadata: row.metadata ?? void 0,
+        createdAt: Number(row.createdat),
+        updatedAt: Number(row.updatedat)
+      };
+    });
+  }
+  async upsertChat(chat) {
+    return this.#useTransaction(async (client) => {
+      const result = await client.query(
+        `INSERT INTO chats (id, userId, title, metadata)
+         VALUES ($1, $2, $3, $4)
+         ON CONFLICT(id) DO UPDATE SET id = EXCLUDED.id
+         RETURNING *`,
+        [
+          chat.id,
+          chat.userId,
+          chat.title ?? null,
+          chat.metadata ? JSON.stringify(chat.metadata) : null
+        ]
+      );
+      const row = result.rows[0];
+      await client.query(
+        `INSERT INTO branches (id, chatId, name, headMessageId, isActive, createdAt)
+         VALUES ($1, $2, 'main', NULL, TRUE, $3)
+         ON CONFLICT(chatId, name) DO NOTHING`,
+        [crypto.randomUUID(), chat.id, Date.now()]
+      );
+      return {
+        id: row.id,
+        userId: row.userid,
+        title: row.title ?? void 0,
+        metadata: row.metadata ?? void 0,
+        createdAt: Number(row.createdat),
+        updatedAt: Number(row.updatedat)
+      };
+    });
+  }
+  async getChat(chatId) {
+    const rows = await this.#query("SELECT * FROM chats WHERE id = $1", [chatId]);
+    if (rows.length === 0) {
+      return void 0;
+    }
+    const row = rows[0];
+    return {
+      id: row.id,
+      userId: row.userid,
+      title: row.title ?? void 0,
+      metadata: row.metadata ?? void 0,
+      createdAt: Number(row.createdat),
+      updatedAt: Number(row.updatedat)
+    };
+  }
+  async updateChat(chatId, updates) {
+    const setClauses = [
+      "updatedAt = (EXTRACT(EPOCH FROM NOW()) * 1000)::BIGINT"
+    ];
+    const params = [];
+    let paramIndex = 1;
+    if (updates.title !== void 0) {
+      setClauses.push(`title = $${paramIndex++}`);
+      params.push(updates.title ?? null);
+    }
+    if (updates.metadata !== void 0) {
+      setClauses.push(`metadata = $${paramIndex++}`);
+      params.push(JSON.stringify(updates.metadata));
+    }
+    params.push(chatId);
+    const rows = await this.#query(
+      `UPDATE chats SET ${setClauses.join(", ")} WHERE id = $${paramIndex} RETURNING *`,
+      params
+    );
+    const row = rows[0];
+    return {
+      id: row.id,
+      userId: row.userid,
+      title: row.title ?? void 0,
+      metadata: row.metadata ?? void 0,
+      createdAt: Number(row.createdat),
+      updatedAt: Number(row.updatedat)
+    };
+  }
+  async listChats(options) {
+    const params = [];
+    const whereClauses = [];
+    let paramIndex = 1;
+    if (options?.userId) {
+      whereClauses.push(`c.userId = $${paramIndex++}`);
+      params.push(options.userId);
+    }
+    if (options?.metadata) {
+      const keyParam = paramIndex++;
+      const valueParam = paramIndex++;
+      whereClauses.push(`c.metadata->$${keyParam} = $${valueParam}::jsonb`);
+      params.push(options.metadata.key);
+      params.push(JSON.stringify(options.metadata.value));
+    }
+    const whereClause = whereClauses.length > 0 ? `WHERE ${whereClauses.join(" AND ")}` : "";
+    let limitClause = "";
+    if (options?.limit !== void 0) {
+      limitClause = ` LIMIT $${paramIndex++}`;
+      params.push(options.limit);
+      if (options.offset !== void 0) {
+        limitClause += ` OFFSET $${paramIndex++}`;
+        params.push(options.offset);
+      }
+    }
+    const rows = await this.#query(
+      `SELECT
+        c.id,
+        c.userId,
+        c.title,
+        c.metadata,
+        c.createdAt,
+        c.updatedAt,
+        COUNT(DISTINCT m.id) as messageCount,
+        COUNT(DISTINCT b.id) as branchCount
+      FROM chats c
+      LEFT JOIN messages m ON m.chatId = c.id
+      LEFT JOIN branches b ON b.chatId = c.id
+      ${whereClause}
+      GROUP BY c.id
+      ORDER BY c.updatedAt DESC${limitClause}`,
+      params
+    );
+    return rows.map((row) => ({
+      id: row.id,
+      userId: row.userid,
+      title: row.title ?? void 0,
+      metadata: row.metadata ?? void 0,
+      messageCount: Number(row.messagecount),
+      branchCount: Number(row.branchcount),
+      createdAt: Number(row.createdat),
+      updatedAt: Number(row.updatedat)
+    }));
+  }
+  async deleteChat(chatId, options) {
+    return this.#useTransaction(async (client) => {
+      let sql = "DELETE FROM chats WHERE id = $1";
+      const params = [chatId];
+      if (options?.userId !== void 0) {
+        sql += " AND userId = $2";
+        params.push(options.userId);
+      }
+      const result = await client.query(sql, params);
+      return (result.rowCount ?? 0) > 0;
+    });
+  }
+  // ==========================================================================
+  // Message Operations (Graph Nodes)
+  // ==========================================================================
+  async addMessage(message2) {
+    if (message2.parentId === message2.id) {
+      throw new Error(`Message ${message2.id} cannot be its own parent`);
+    }
+    await this.#useTransaction(async (client) => {
+      await client.query(
+        `INSERT INTO messages (id, chatId, parentId, name, type, data, createdAt)
+         VALUES ($1, $2, $3, $4, $5, $6, $7)
+         ON CONFLICT(id) DO UPDATE SET
+           name = EXCLUDED.name,
+           type = EXCLUDED.type,
+           data = EXCLUDED.data`,
+        [
+          message2.id,
+          message2.chatId,
+          message2.parentId,
+          message2.name,
+          message2.type ?? null,
+          JSON.stringify(message2.data),
+          message2.createdAt
+        ]
+      );
+      const content = typeof message2.data === "string" ? message2.data : JSON.stringify(message2.data);
+      await client.query(
+        `INSERT INTO messages_fts (messageId, chatId, name, content)
+         VALUES ($1, $2, $3, $4)
+         ON CONFLICT(messageId) DO UPDATE SET
+           chatId = EXCLUDED.chatId,
+           name = EXCLUDED.name,
+           content = EXCLUDED.content`,
+        [message2.id, message2.chatId, message2.name, content]
+      );
+    });
+  }
+  async getMessage(messageId) {
+    const rows = await this.#query("SELECT * FROM messages WHERE id = $1", [messageId]);
+    if (rows.length === 0) {
+      return void 0;
+    }
+    const row = rows[0];
+    return {
+      id: row.id,
+      chatId: row.chatid,
+      parentId: row.parentid,
+      name: row.name,
+      type: row.type ?? void 0,
+      data: row.data,
+      createdAt: Number(row.createdat)
+    };
+  }
+  async getMessageChain(headId) {
+    const rows = await this.#query(
+      `WITH RECURSIVE chain AS (
+        SELECT *, 0 as depth FROM messages WHERE id = $1
+        UNION ALL
+        SELECT m.*, c.depth + 1 FROM messages m
+        INNER JOIN chain c ON m.id = c.parentId
+        WHERE c.depth < 10000
+      )
+      SELECT * FROM chain
+      ORDER BY depth DESC`,
+      [headId]
+    );
+    return rows.map((row) => ({
+      id: row.id,
+      chatId: row.chatid,
+      parentId: row.parentid,
+      name: row.name,
+      type: row.type ?? void 0,
+      data: row.data,
+      createdAt: Number(row.createdat)
+    }));
+  }
+  async hasChildren(messageId) {
+    const rows = await this.#query(
+      "SELECT EXISTS(SELECT 1 FROM messages WHERE parentId = $1) as exists",
+      [messageId]
+    );
+    return rows[0].exists;
+  }
+  async getMessages(chatId) {
+    const chat = await this.getChat(chatId);
+    if (!chat) {
+      throw new Error(`Chat "${chatId}" not found`);
+    }
+    const activeBranch = await this.getActiveBranch(chatId);
+    if (!activeBranch?.headMessageId) {
+      return [];
+    }
+    return this.getMessageChain(activeBranch.headMessageId);
+  }
+  // ==========================================================================
+  // Branch Operations
+  // ==========================================================================
+  async createBranch(branch) {
+    await this.#query(
+      `INSERT INTO branches (id, chatId, name, headMessageId, isActive, createdAt)
+       VALUES ($1, $2, $3, $4, $5, $6)`,
+      [
+        branch.id,
+        branch.chatId,
+        branch.name,
+        branch.headMessageId,
+        branch.isActive,
+        branch.createdAt
+      ]
+    );
+  }
+  async getBranch(chatId, name) {
+    const rows = await this.#query("SELECT * FROM branches WHERE chatId = $1 AND name = $2", [
+      chatId,
+      name
+    ]);
+    if (rows.length === 0) {
+      return void 0;
+    }
+    const row = rows[0];
+    return {
+      id: row.id,
+      chatId: row.chatid,
+      name: row.name,
+      headMessageId: row.headmessageid,
+      isActive: row.isactive,
+      createdAt: Number(row.createdat)
+    };
+  }
+  async getActiveBranch(chatId) {
+    const rows = await this.#query("SELECT * FROM branches WHERE chatId = $1 AND isActive = TRUE", [
+      chatId
+    ]);
+    if (rows.length === 0) {
+      return void 0;
+    }
+    const row = rows[0];
+    return {
+      id: row.id,
+      chatId: row.chatid,
+      name: row.name,
+      headMessageId: row.headmessageid,
+      isActive: true,
+      createdAt: Number(row.createdat)
+    };
+  }
+  async setActiveBranch(chatId, branchId) {
+    await this.#useTransaction(async (client) => {
+      await client.query(
+        "UPDATE branches SET isActive = FALSE WHERE chatId = $1",
+        [chatId]
+      );
+      await client.query("UPDATE branches SET isActive = TRUE WHERE id = $1", [
+        branchId
+      ]);
+    });
+  }
+  async updateBranchHead(branchId, messageId) {
+    await this.#query("UPDATE branches SET headMessageId = $1 WHERE id = $2", [
+      messageId,
+      branchId
+    ]);
+  }
+  async listBranches(chatId) {
+    const branches = await this.#query(
+      `SELECT
+        id,
+        name,
+        headMessageId,
+        isActive,
+        createdAt
+      FROM branches
+      WHERE chatId = $1
+      ORDER BY createdAt ASC`,
+      [chatId]
+    );
+    const result = [];
+    for (const branch of branches) {
+      let messageCount = 0;
+      if (branch.headmessageid) {
+        const countRows = await this.#query(
+          `WITH RECURSIVE chain AS (
+            SELECT id, parentId FROM messages WHERE id = $1
+            UNION ALL
+            SELECT m.id, m.parentId FROM messages m
+            INNER JOIN chain c ON m.id = c.parentId
+          )
+          SELECT COUNT(*) as count FROM chain`,
+          [branch.headmessageid]
+        );
+        messageCount = Number(countRows[0].count);
+      }
+      result.push({
+        id: branch.id,
+        name: branch.name,
+        headMessageId: branch.headmessageid,
+        isActive: branch.isactive,
+        messageCount,
+        createdAt: Number(branch.createdat)
+      });
+    }
+    return result;
+  }
+  // ==========================================================================
+  // Checkpoint Operations
+  // ==========================================================================
+  async createCheckpoint(checkpoint) {
+    await this.#query(
+      `INSERT INTO checkpoints (id, chatId, name, messageId, createdAt)
+       VALUES ($1, $2, $3, $4, $5)
+       ON CONFLICT(chatId, name) DO UPDATE SET
+         messageId = EXCLUDED.messageId,
+         createdAt = EXCLUDED.createdAt`,
+      [
+        checkpoint.id,
+        checkpoint.chatId,
+        checkpoint.name,
+        checkpoint.messageId,
+        checkpoint.createdAt
+      ]
+    );
+  }
+  async getCheckpoint(chatId, name) {
+    const rows = await this.#query("SELECT * FROM checkpoints WHERE chatId = $1 AND name = $2", [
+      chatId,
+      name
+    ]);
+    if (rows.length === 0) {
+      return void 0;
+    }
+    const row = rows[0];
+    return {
+      id: row.id,
+      chatId: row.chatid,
+      name: row.name,
+      messageId: row.messageid,
+      createdAt: Number(row.createdat)
+    };
+  }
+  async listCheckpoints(chatId) {
+    const rows = await this.#query(
+      `SELECT id, name, messageId, createdAt
+       FROM checkpoints
+       WHERE chatId = $1
+       ORDER BY createdAt DESC`,
+      [chatId]
+    );
+    return rows.map((row) => ({
+      id: row.id,
+      name: row.name,
+      messageId: row.messageid,
+      createdAt: Number(row.createdat)
+    }));
+  }
+  async deleteCheckpoint(chatId, name) {
+    await this.#query(
+      "DELETE FROM checkpoints WHERE chatId = $1 AND name = $2",
+      [chatId, name]
+    );
+  }
+  // ==========================================================================
+  // Search Operations
+  // ==========================================================================
+  async searchMessages(chatId, query, options) {
+    const limit = options?.limit ?? 20;
+    const roles = options?.roles;
+    let sql = `
+      SELECT
+        m.id,
+        m.chatId,
+        m.parentId,
+        m.name,
+        m.type,
+        m.data,
+        m.createdAt,
+        ts_rank(fts.content_vector, plainto_tsquery('english', $2)) as rank,
+        ts_headline('english', fts.content, plainto_tsquery('english', $2),
+          'StartSel=<mark>, StopSel=</mark>, MaxWords=32, MinWords=5, MaxFragments=1') as snippet
+      FROM messages_fts fts
+      JOIN messages m ON m.id = fts.messageId
+      WHERE fts.content_vector @@ plainto_tsquery('english', $2)
+        AND fts.chatId = $1
+    `;
+    const params = [chatId, query];
+    let paramIndex = 3;
+    if (roles && roles.length > 0) {
+      const placeholders = roles.map(() => `$${paramIndex++}`).join(", ");
+      sql += ` AND fts.name IN (${placeholders})`;
+      params.push(...roles);
+    }
+    sql += ` ORDER BY rank DESC LIMIT $${paramIndex}`;
+    params.push(limit);
+    const rows = await this.#query(sql, params);
+    return rows.map((row) => ({
+      message: {
+        id: row.id,
+        chatId: row.chatid,
+        parentId: row.parentid,
+        name: row.name,
+        type: row.type ?? void 0,
+        data: row.data,
+        createdAt: Number(row.createdat)
+      },
+      rank: row.rank,
+      snippet: row.snippet
+    }));
+  }
+  // ==========================================================================
+  // Visualization Operations
+  // ==========================================================================
+  async getGraph(chatId) {
+    const messageRows = await this.#query(
+      `SELECT id, parentId, name, data, createdAt
+       FROM messages
+       WHERE chatId = $1
+       ORDER BY createdAt ASC`,
+      [chatId]
+    );
+    const nodes = messageRows.map((row) => {
+      const data = row.data;
+      const content = typeof data === "string" ? data : JSON.stringify(data);
+      return {
+        id: row.id,
+        parentId: row.parentid,
+        role: row.name,
+        content: content.length > 50 ? content.slice(0, 50) + "..." : content,
+        createdAt: Number(row.createdat)
+      };
+    });
+    const branchRows = await this.#query(
+      `SELECT name, headMessageId, isActive
+       FROM branches
+       WHERE chatId = $1
+       ORDER BY createdAt ASC`,
+      [chatId]
+    );
+    const branches = branchRows.map((row) => ({
+      name: row.name,
+      headMessageId: row.headmessageid,
+      isActive: row.isactive
+    }));
+    const checkpointRows = await this.#query(
+      `SELECT name, messageId
+       FROM checkpoints
+       WHERE chatId = $1
+       ORDER BY createdAt ASC`,
+      [chatId]
+    );
+    const checkpoints = checkpointRows.map((row) => ({
+      name: row.name,
+      messageId: row.messageid
+    }));
+    return {
+      chatId,
+      nodes,
+      branches,
+      checkpoints
+    };
+  }
+};
+
+// packages/context/src/lib/store/sqlserver.store.ts
+import { createRequire as createRequire2 } from "node:module";
+
+// packages/context/src/lib/store/ddl.sqlserver.sql
+var ddl_sqlserver_default = "-- Context Store DDL for SQL Server\n-- This schema implements a DAG-based message history with branching and checkpoints.\n\n-- Chats table\n-- createdAt/updatedAt: DEFAULT for insert, inline SET for updates\nIF OBJECT_ID('chats', 'U') IS NULL\nBEGIN\n  CREATE TABLE chats (\n    id NVARCHAR(255) PRIMARY KEY,\n    userId NVARCHAR(255) NOT NULL,\n    title NVARCHAR(MAX),\n    metadata NVARCHAR(MAX),\n    createdAt BIGINT NOT NULL DEFAULT DATEDIFF_BIG(ms, '1970-01-01', GETUTCDATE()),\n    updatedAt BIGINT NOT NULL DEFAULT DATEDIFF_BIG(ms, '1970-01-01', GETUTCDATE())\n  );\nEND;\n\nIF NOT EXISTS (SELECT * FROM sys.indexes WHERE name = 'idx_chats_updatedAt' AND object_id = OBJECT_ID('chats'))\n  CREATE INDEX idx_chats_updatedAt ON chats(updatedAt);\n\nIF NOT EXISTS (SELECT * FROM sys.indexes WHERE name = 'idx_chats_userId' AND object_id = OBJECT_ID('chats'))\n  CREATE INDEX idx_chats_userId ON chats(userId);\n\n-- Messages table (nodes in the DAG)\nIF OBJECT_ID('messages', 'U') IS NULL\nBEGIN\n  CREATE TABLE messages (\n    id NVARCHAR(255) PRIMARY KEY,\n    chatId NVARCHAR(255) NOT NULL,\n    parentId NVARCHAR(255),\n    name NVARCHAR(255) NOT NULL,\n    type NVARCHAR(255),\n    data NVARCHAR(MAX) NOT NULL,\n    createdAt BIGINT NOT NULL,\n    FOREIGN KEY (chatId) REFERENCES chats(id) ON DELETE CASCADE,\n    FOREIGN KEY (parentId) REFERENCES messages(id)\n  );\nEND;\n\nIF NOT EXISTS (SELECT * FROM sys.indexes WHERE name = 'idx_messages_chatId' AND object_id = OBJECT_ID('messages'))\n  CREATE INDEX idx_messages_chatId ON messages(chatId);\n\nIF NOT EXISTS (SELECT * FROM sys.indexes WHERE name = 'idx_messages_parentId' AND object_id = OBJECT_ID('messages'))\n  CREATE INDEX idx_messages_parentId ON messages(parentId);\n\nIF NOT EXISTS (SELECT * FROM sys.indexes WHERE name = 'idx_messages_chatId_parentId' AND object_id = OBJECT_ID('messages'))\n  CREATE INDEX idx_messages_chatId_parentId ON messages(chatId, parentId);\n\n-- Branches table (pointers to head messages)\nIF OBJECT_ID('branches', 'U') IS NULL\nBEGIN\n  CREATE TABLE branches (\n    id NVARCHAR(255) PRIMARY KEY,\n    chatId NVARCHAR(255) NOT NULL,\n    name NVARCHAR(255) NOT NULL,\n    headMessageId NVARCHAR(255),\n    isActive BIT NOT NULL DEFAULT 0,\n    createdAt BIGINT NOT NULL,\n    FOREIGN KEY (chatId) REFERENCES chats(id) ON DELETE CASCADE,\n    FOREIGN KEY (headMessageId) REFERENCES messages(id),\n    CONSTRAINT UQ_branches_chatId_name UNIQUE(chatId, name)\n  );\nEND;\n\nIF NOT EXISTS (SELECT * FROM sys.indexes WHERE name = 'idx_branches_chatId' AND object_id = OBJECT_ID('branches'))\n  CREATE INDEX idx_branches_chatId ON branches(chatId);\n\nIF NOT EXISTS (SELECT * FROM sys.indexes WHERE name = 'idx_branches_chatId_isActive' AND object_id = OBJECT_ID('branches'))\n  CREATE INDEX idx_branches_chatId_isActive ON branches(chatId, isActive);\n\n-- Checkpoints table (pointers to message nodes)\nIF OBJECT_ID('checkpoints', 'U') IS NULL\nBEGIN\n  CREATE TABLE checkpoints (\n    id NVARCHAR(255) PRIMARY KEY,\n    chatId NVARCHAR(255) NOT NULL,\n    name NVARCHAR(255) NOT NULL,\n    messageId NVARCHAR(255) NOT NULL,\n    createdAt BIGINT NOT NULL,\n    FOREIGN KEY (chatId) REFERENCES chats(id) ON DELETE CASCADE,\n    FOREIGN KEY (messageId) REFERENCES messages(id),\n    CONSTRAINT UQ_checkpoints_chatId_name UNIQUE(chatId, name)\n  );\nEND;\n\nIF NOT EXISTS (SELECT * FROM sys.indexes WHERE name = 'idx_checkpoints_chatId' AND object_id = OBJECT_ID('checkpoints'))\n  CREATE INDEX idx_checkpoints_chatId ON checkpoints(chatId);\n\n-- Full-text search table\nIF OBJECT_ID('messages_fts', 'U') IS NULL\nBEGIN\n  CREATE TABLE messages_fts (\n    messageId NVARCHAR(255) NOT NULL,\n    chatId NVARCHAR(255) NOT NULL,\n    name NVARCHAR(255) NOT NULL,\n    content NVARCHAR(MAX) NOT NULL,\n    CONSTRAINT PK_messages_fts PRIMARY KEY (messageId),\n    FOREIGN KEY (messageId) REFERENCES messages(id) ON DELETE CASCADE\n  );\nEND;\n\nIF NOT EXISTS (SELECT * FROM sys.indexes WHERE name = 'idx_messages_fts_chatId' AND object_id = OBJECT_ID('messages_fts'))\n  CREATE INDEX idx_messages_fts_chatId ON messages_fts(chatId);\n\n-- Full-text catalog and index (only if FTS is installed)\n-- FTS is optional - search will gracefully degrade without it\nIF SERVERPROPERTY('IsFullTextInstalled') = 1\nBEGIN\n  -- Create catalog if not exists\n  IF NOT EXISTS (SELECT * FROM sys.fulltext_catalogs WHERE name = 'context_store_catalog')\n    CREATE FULLTEXT CATALOG context_store_catalog AS DEFAULT;\n\n  -- Create full-text index on messages_fts.content\n  -- Note: This requires the table to have a unique index, which PK provides\n  IF NOT EXISTS (SELECT * FROM sys.fulltext_indexes WHERE object_id = OBJECT_ID('messages_fts'))\n  BEGIN\n    CREATE FULLTEXT INDEX ON messages_fts(content)\n      KEY INDEX PK_messages_fts\n      ON context_store_catalog\n      WITH STOPLIST = SYSTEM;\n  END;\nEND;\n";
+
+// packages/context/src/lib/store/sqlserver.store.ts
+var SqlServerContextStore = class _SqlServerContextStore extends ContextStore {
+  #pool;
+  #initialized;
+  constructor(options) {
+    super();
+    const mssql = _SqlServerContextStore.#requireMssql();
+    this.#pool = typeof options.pool === "string" ? new mssql.ConnectionPool(options.pool) : new mssql.ConnectionPool(options.pool);
+    this.#initialized = this.#initialize();
+  }
+  static #requireMssql() {
+    try {
+      const require2 = createRequire2(import.meta.url);
+      return require2("mssql");
+    } catch {
+      throw new Error(
+        'SqlServerContextStore requires the "mssql" package. Install it with: npm install mssql'
+      );
+    }
+  }
+  async #initialize() {
+    await this.#pool.connect();
+    const batches = ddl_sqlserver_default.split(/\bGO\b/i).filter((b) => b.trim());
+    for (const batch of batches) {
+      if (batch.trim()) {
+        await this.#pool.request().batch(batch);
+      }
+    }
+  }
+  /**
+   * Ensure initialization is complete before any operation.
+   */
+  async #ensureInitialized() {
+    await this.#initialized;
+  }
+  /**
+   * Execute a function within a transaction.
+   * Automatically commits on success or rolls back on error.
+   */
+  async #useTransaction(fn) {
+    await this.#ensureInitialized();
+    const mssql = _SqlServerContextStore.#requireMssql();
+    const transaction = new mssql.Transaction(this.#pool);
+    try {
+      await transaction.begin();
+      const result = await fn(transaction);
+      await transaction.commit();
+      return result;
+    } catch (error) {
+      await transaction.rollback();
+      throw error;
+    }
+  }
+  /**
+   * Execute a query using the pool (no transaction).
+   * Converts positional params to SQL Server named params (@p0, @p1, ...).
+   */
+  async #query(sql, params) {
+    await this.#ensureInitialized();
+    const request = this.#pool.request();
+    params?.forEach((value, index) => {
+      request.input(`p${index}`, value);
+    });
+    const result = await request.query(sql);
+    return result.recordset;
+  }
+  /**
+   * Close the pool connection.
+   * Call this when done with the store.
+   */
+  async close() {
+    await this.#pool.close();
+  }
+  // ==========================================================================
+  // Chat Operations
+  // ==========================================================================
+  async createChat(chat) {
+    return this.#useTransaction(async (transaction) => {
+      const mssql = _SqlServerContextStore.#requireMssql();
+      const request = transaction.request();
+      request.input("p0", mssql.NVarChar, chat.id);
+      request.input("p1", mssql.NVarChar, chat.userId);
+      request.input("p2", mssql.NVarChar, chat.title ?? null);
+      request.input(
+        "p3",
+        mssql.NVarChar,
+        chat.metadata ? JSON.stringify(chat.metadata) : null
+      );
+      const result = await request.query(`
+        INSERT INTO chats (id, userId, title, metadata)
+        OUTPUT INSERTED.*
+        VALUES (@p0, @p1, @p2, @p3)
+      `);
+      const row = result.recordset[0];
+      const branchRequest = transaction.request();
+      branchRequest.input("p0", mssql.NVarChar, crypto.randomUUID());
+      branchRequest.input("p1", mssql.NVarChar, chat.id);
+      branchRequest.input("p2", mssql.BigInt, Date.now());
+      await branchRequest.query(`
+        INSERT INTO branches (id, chatId, name, headMessageId, isActive, createdAt)
+        VALUES (@p0, @p1, 'main', NULL, 1, @p2)
+      `);
+      return {
+        id: row.id,
+        userId: row.userId,
+        title: row.title ?? void 0,
+        metadata: row.metadata ? JSON.parse(row.metadata) : void 0,
+        createdAt: Number(row.createdAt),
+        updatedAt: Number(row.updatedAt)
+      };
+    });
+  }
+  async upsertChat(chat) {
+    return this.#useTransaction(async (transaction) => {
+      const mssql = _SqlServerContextStore.#requireMssql();
+      const request = transaction.request();
+      request.input("p0", mssql.NVarChar, chat.id);
+      request.input("p1", mssql.NVarChar, chat.userId);
+      request.input("p2", mssql.NVarChar, chat.title ?? null);
+      request.input(
+        "p3",
+        mssql.NVarChar,
+        chat.metadata ? JSON.stringify(chat.metadata) : null
+      );
+      request.input("p4", mssql.BigInt, BigInt(Date.now()));
+      const result = await request.query(`
+        MERGE chats AS target
+        USING (SELECT @p0 AS id, @p1 AS userId, @p2 AS title, @p3 AS metadata) AS source
+        ON target.id = source.id
+        WHEN MATCHED THEN
+          UPDATE SET id = target.id
+        WHEN NOT MATCHED THEN
+          INSERT (id, userId, title, metadata, createdAt, updatedAt)
+          VALUES (source.id, source.userId, source.title, source.metadata, @p4, @p4)
+        OUTPUT INSERTED.*;
+      `);
+      const row = result.recordset[0];
+      const branchRequest = transaction.request();
+      branchRequest.input("p0", mssql.NVarChar, crypto.randomUUID());
+      branchRequest.input("p1", mssql.NVarChar, chat.id);
+      branchRequest.input("p2", mssql.BigInt, Date.now());
+      await branchRequest.query(`
+        IF NOT EXISTS (SELECT 1 FROM branches WHERE chatId = @p1 AND name = 'main')
+        BEGIN
+          INSERT INTO branches (id, chatId, name, headMessageId, isActive, createdAt)
+          VALUES (@p0, @p1, 'main', NULL, 1, @p2)
+        END
+      `);
+      return {
+        id: row.id,
+        userId: row.userId,
+        title: row.title ?? void 0,
+        metadata: row.metadata ? JSON.parse(row.metadata) : void 0,
+        createdAt: Number(row.createdAt),
+        updatedAt: Number(row.updatedAt)
+      };
+    });
+  }
+  async getChat(chatId) {
+    const rows = await this.#query("SELECT * FROM chats WHERE id = @p0", [chatId]);
+    if (rows.length === 0) {
+      return void 0;
+    }
+    const row = rows[0];
+    return {
+      id: row.id,
+      userId: row.userId,
+      title: row.title ?? void 0,
+      metadata: row.metadata ? JSON.parse(row.metadata) : void 0,
+      createdAt: Number(row.createdAt),
+      updatedAt: Number(row.updatedAt)
+    };
+  }
+  async updateChat(chatId, updates) {
+    const setClauses = [
+      "updatedAt = DATEDIFF_BIG(ms, '1970-01-01', GETUTCDATE())"
+    ];
+    const params = [];
+    let paramIndex = 0;
+    if (updates.title !== void 0) {
+      setClauses.push(`title = @p${paramIndex++}`);
+      params.push(updates.title ?? null);
+    }
+    if (updates.metadata !== void 0) {
+      setClauses.push(`metadata = @p${paramIndex++}`);
+      params.push(JSON.stringify(updates.metadata));
+    }
+    params.push(chatId);
+    const rows = await this.#query(
+      `UPDATE chats SET ${setClauses.join(", ")} OUTPUT INSERTED.* WHERE id = @p${paramIndex}`,
+      params
+    );
+    const row = rows[0];
+    return {
+      id: row.id,
+      userId: row.userId,
+      title: row.title ?? void 0,
+      metadata: row.metadata ? JSON.parse(row.metadata) : void 0,
+      createdAt: Number(row.createdAt),
+      updatedAt: Number(row.updatedAt)
+    };
+  }
+  async listChats(options) {
+    const params = [];
+    const whereClauses = [];
+    let paramIndex = 0;
+    if (options?.userId) {
+      whereClauses.push(`c.userId = @p${paramIndex++}`);
+      params.push(options.userId);
+    }
+    if (options?.metadata) {
+      whereClauses.push(
+        `JSON_VALUE(c.metadata, '$.' + @p${paramIndex}) = @p${paramIndex + 1}`
+      );
+      params.push(options.metadata.key);
+      params.push(String(options.metadata.value));
+      paramIndex += 2;
+    }
+    const whereClause = whereClauses.length > 0 ? `WHERE ${whereClauses.join(" AND ")}` : "";
+    let paginationClause = "";
+    if (options?.limit !== void 0) {
+      paginationClause = ` OFFSET @p${paramIndex} ROWS FETCH NEXT @p${paramIndex + 1} ROWS ONLY`;
+      params.push(options.offset ?? 0);
+      params.push(options.limit);
+    }
+    const rows = await this.#query(
+      `SELECT
+        c.id,
+        c.userId,
+        c.title,
+        c.metadata,
+        c.createdAt,
+        c.updatedAt,
+        COUNT(DISTINCT m.id) as messageCount,
+        COUNT(DISTINCT b.id) as branchCount
+      FROM chats c
+      LEFT JOIN messages m ON m.chatId = c.id
+      LEFT JOIN branches b ON b.chatId = c.id
+      ${whereClause}
+      GROUP BY c.id, c.userId, c.title, c.metadata, c.createdAt, c.updatedAt
+      ORDER BY c.updatedAt DESC${paginationClause}`,
+      params
+    );
+    return rows.map((row) => ({
+      id: row.id,
+      userId: row.userId,
+      title: row.title ?? void 0,
+      metadata: row.metadata ? JSON.parse(row.metadata) : void 0,
+      messageCount: Number(row.messageCount),
+      branchCount: Number(row.branchCount),
+      createdAt: Number(row.createdAt),
+      updatedAt: Number(row.updatedAt)
+    }));
+  }
+  async deleteChat(chatId, options) {
+    return this.#useTransaction(async (transaction) => {
+      const mssql = _SqlServerContextStore.#requireMssql();
+      const request = transaction.request();
+      request.input("p0", mssql.NVarChar, chatId);
+      let sql = "DELETE FROM chats WHERE id = @p0";
+      if (options?.userId !== void 0) {
+        request.input("p1", mssql.NVarChar, options.userId);
+        sql += " AND userId = @p1";
+      }
+      const result = await request.query(sql);
+      return (result.rowsAffected[0] ?? 0) > 0;
+    });
+  }
+  // ==========================================================================
+  // Message Operations (Graph Nodes)
+  // ==========================================================================
+  async addMessage(message2) {
+    if (message2.parentId === message2.id) {
+      throw new Error(`Message ${message2.id} cannot be its own parent`);
+    }
+    await this.#useTransaction(async (transaction) => {
+      const mssql = _SqlServerContextStore.#requireMssql();
+      const request = transaction.request();
+      request.input("p0", mssql.NVarChar, message2.id);
+      request.input("p1", mssql.NVarChar, message2.chatId);
+      request.input("p2", mssql.NVarChar, message2.parentId);
+      request.input("p3", mssql.NVarChar, message2.name);
+      request.input("p4", mssql.NVarChar, message2.type ?? null);
+      request.input("p5", mssql.NVarChar, JSON.stringify(message2.data));
+      request.input("p6", mssql.BigInt, message2.createdAt);
+      await request.query(`
+        MERGE messages AS target
+        USING (SELECT @p0 AS id) AS source
+        ON target.id = source.id
+        WHEN MATCHED THEN
+          UPDATE SET name = @p3, type = @p4, data = @p5
+        WHEN NOT MATCHED THEN
+          INSERT (id, chatId, parentId, name, type, data, createdAt)
+          VALUES (@p0, @p1, @p2, @p3, @p4, @p5, @p6);
+      `);
+      const content = typeof message2.data === "string" ? message2.data : JSON.stringify(message2.data);
+      const ftsRequest = transaction.request();
+      ftsRequest.input("p0", mssql.NVarChar, message2.id);
+      ftsRequest.input("p1", mssql.NVarChar, message2.chatId);
+      ftsRequest.input("p2", mssql.NVarChar, message2.name);
+      ftsRequest.input("p3", mssql.NVarChar, content);
+      await ftsRequest.query(`
+        MERGE messages_fts AS target
+        USING (SELECT @p0 AS messageId) AS source
+        ON target.messageId = source.messageId
+        WHEN MATCHED THEN
+          UPDATE SET chatId = @p1, name = @p2, content = @p3
+        WHEN NOT MATCHED THEN
+          INSERT (messageId, chatId, name, content)
+          VALUES (@p0, @p1, @p2, @p3);
+      `);
+    });
+  }
+  async getMessage(messageId) {
+    const rows = await this.#query("SELECT * FROM messages WHERE id = @p0", [messageId]);
+    if (rows.length === 0) {
+      return void 0;
+    }
+    const row = rows[0];
+    return {
+      id: row.id,
+      chatId: row.chatId,
+      parentId: row.parentId,
+      name: row.name,
+      type: row.type ?? void 0,
+      data: JSON.parse(row.data),
+      createdAt: Number(row.createdAt)
+    };
+  }
+  async getMessageChain(headId) {
+    const rows = await this.#query(
+      `WITH chain AS (
+        SELECT *, 0 as depth FROM messages WHERE id = @p0
+        UNION ALL
+        SELECT m.*, c.depth + 1 FROM messages m
+        INNER JOIN chain c ON m.id = c.parentId
+        WHERE c.depth < 10000
+      )
+      SELECT * FROM chain
+      ORDER BY depth DESC`,
+      [headId]
+    );
+    return rows.map((row) => ({
+      id: row.id,
+      chatId: row.chatId,
+      parentId: row.parentId,
+      name: row.name,
+      type: row.type ?? void 0,
+      data: JSON.parse(row.data),
+      createdAt: Number(row.createdAt)
+    }));
+  }
+  async hasChildren(messageId) {
+    const rows = await this.#query(
+      `SELECT CASE WHEN EXISTS(SELECT 1 FROM messages WHERE parentId = @p0) THEN 1 ELSE 0 END as hasChildren`,
+      [messageId]
+    );
+    return rows[0].hasChildren === 1;
+  }
+  async getMessages(chatId) {
+    const chat = await this.getChat(chatId);
+    if (!chat) {
+      throw new Error(`Chat "${chatId}" not found`);
+    }
+    const activeBranch = await this.getActiveBranch(chatId);
+    if (!activeBranch?.headMessageId) {
+      return [];
+    }
+    return this.getMessageChain(activeBranch.headMessageId);
+  }
+  // ==========================================================================
+  // Branch Operations
+  // ==========================================================================
+  async createBranch(branch) {
+    await this.#query(
+      `INSERT INTO branches (id, chatId, name, headMessageId, isActive, createdAt)
+       VALUES (@p0, @p1, @p2, @p3, @p4, @p5)`,
+      [
+        branch.id,
+        branch.chatId,
+        branch.name,
+        branch.headMessageId,
+        branch.isActive ? 1 : 0,
+        branch.createdAt
+      ]
+    );
+  }
+  async getBranch(chatId, name) {
+    const rows = await this.#query("SELECT * FROM branches WHERE chatId = @p0 AND name = @p1", [
+      chatId,
+      name
+    ]);
+    if (rows.length === 0) {
+      return void 0;
+    }
+    const row = rows[0];
+    return {
+      id: row.id,
+      chatId: row.chatId,
+      name: row.name,
+      headMessageId: row.headMessageId,
+      isActive: row.isActive === true || row.isActive === 1,
+      createdAt: Number(row.createdAt)
+    };
+  }
+  async getActiveBranch(chatId) {
+    const rows = await this.#query("SELECT * FROM branches WHERE chatId = @p0 AND isActive = 1", [chatId]);
+    if (rows.length === 0) {
+      return void 0;
+    }
+    const row = rows[0];
+    return {
+      id: row.id,
+      chatId: row.chatId,
+      name: row.name,
+      headMessageId: row.headMessageId,
+      isActive: true,
+      createdAt: Number(row.createdAt)
+    };
+  }
+  async setActiveBranch(chatId, branchId) {
+    await this.#useTransaction(async (transaction) => {
+      const mssql = _SqlServerContextStore.#requireMssql();
+      const deactivateRequest = transaction.request();
+      deactivateRequest.input("p0", mssql.NVarChar, chatId);
+      await deactivateRequest.query(
+        "UPDATE branches SET isActive = 0 WHERE chatId = @p0"
+      );
+      const activateRequest = transaction.request();
+      activateRequest.input("p0", mssql.NVarChar, branchId);
+      await activateRequest.query(
+        "UPDATE branches SET isActive = 1 WHERE id = @p0"
+      );
+    });
+  }
+  async updateBranchHead(branchId, messageId) {
+    await this.#query(
+      "UPDATE branches SET headMessageId = @p0 WHERE id = @p1",
+      [messageId, branchId]
+    );
+  }
+  async listBranches(chatId) {
+    const branches = await this.#query(
+      `SELECT
+        id,
+        name,
+        headMessageId,
+        isActive,
+        createdAt
+      FROM branches
+      WHERE chatId = @p0
+      ORDER BY createdAt ASC`,
+      [chatId]
+    );
+    const result = [];
+    for (const branch of branches) {
+      let messageCount = 0;
+      if (branch.headMessageId) {
+        const countRows = await this.#query(
+          `WITH chain AS (
+            SELECT id, parentId FROM messages WHERE id = @p0
+            UNION ALL
+            SELECT m.id, m.parentId FROM messages m
+            INNER JOIN chain c ON m.id = c.parentId
+          )
+          SELECT COUNT(*) as count FROM chain`,
+          [branch.headMessageId]
+        );
+        messageCount = Number(countRows[0].count);
+      }
+      result.push({
+        id: branch.id,
+        name: branch.name,
+        headMessageId: branch.headMessageId,
+        isActive: branch.isActive === true || branch.isActive === 1,
+        messageCount,
+        createdAt: Number(branch.createdAt)
+      });
+    }
+    return result;
+  }
+  // ==========================================================================
+  // Checkpoint Operations
+  // ==========================================================================
+  async createCheckpoint(checkpoint) {
+    await this.#useTransaction(async (transaction) => {
+      const mssql = _SqlServerContextStore.#requireMssql();
+      const request = transaction.request();
+      request.input("p0", mssql.NVarChar, checkpoint.id);
+      request.input("p1", mssql.NVarChar, checkpoint.chatId);
+      request.input("p2", mssql.NVarChar, checkpoint.name);
+      request.input("p3", mssql.NVarChar, checkpoint.messageId);
+      request.input("p4", mssql.BigInt, checkpoint.createdAt);
+      await request.query(`
+        MERGE checkpoints AS target
+        USING (SELECT @p1 AS chatId, @p2 AS name) AS source
+        ON target.chatId = source.chatId AND target.name = source.name
+        WHEN MATCHED THEN
+          UPDATE SET messageId = @p3, createdAt = @p4
+        WHEN NOT MATCHED THEN
+          INSERT (id, chatId, name, messageId, createdAt)
+          VALUES (@p0, @p1, @p2, @p3, @p4);
+      `);
+    });
+  }
+  async getCheckpoint(chatId, name) {
+    const rows = await this.#query("SELECT * FROM checkpoints WHERE chatId = @p0 AND name = @p1", [
+      chatId,
+      name
+    ]);
+    if (rows.length === 0) {
+      return void 0;
+    }
+    const row = rows[0];
+    return {
+      id: row.id,
+      chatId: row.chatId,
+      name: row.name,
+      messageId: row.messageId,
+      createdAt: Number(row.createdAt)
+    };
+  }
+  async listCheckpoints(chatId) {
+    const rows = await this.#query(
+      `SELECT id, name, messageId, createdAt
+       FROM checkpoints
+       WHERE chatId = @p0
+       ORDER BY createdAt DESC`,
+      [chatId]
+    );
+    return rows.map((row) => ({
+      id: row.id,
+      name: row.name,
+      messageId: row.messageId,
+      createdAt: Number(row.createdAt)
+    }));
+  }
+  async deleteCheckpoint(chatId, name) {
+    await this.#query(
+      "DELETE FROM checkpoints WHERE chatId = @p0 AND name = @p1",
+      [chatId, name]
+    );
+  }
+  // ==========================================================================
+  // Search Operations
+  // ==========================================================================
+  async searchMessages(chatId, query, options) {
+    const limit = options?.limit ?? 20;
+    const roles = options?.roles;
+    const ftsCheck = await this.#query(
+      `SELECT CAST(SERVERPROPERTY('IsFullTextInstalled') AS INT) as ftsInstalled`
+    );
+    const ftsAvailable = ftsCheck[0]?.ftsInstalled === 1;
+    if (ftsAvailable) {
+      let sql = `
+        SELECT
+          m.id,
+          m.chatId,
+          m.parentId,
+          m.name,
+          m.type,
+          m.data,
+          m.createdAt,
+          ct.RANK as rank,
+          SUBSTRING(fts.content, 1, 200) as snippet
+        FROM messages_fts fts
+        INNER JOIN CONTAINSTABLE(messages_fts, content, @p0) ct
+          ON fts.messageId = ct.[KEY]
+        INNER JOIN messages m ON m.id = fts.messageId
+        WHERE fts.chatId = @p1
+      `;
+      const params = [query, chatId];
+      let paramIndex = 2;
+      if (roles && roles.length > 0) {
+        const placeholders = roles.map(() => `@p${paramIndex++}`).join(", ");
+        sql += ` AND fts.name IN (${placeholders})`;
+        params.push(...roles);
+      }
+      sql += ` ORDER BY ct.RANK DESC OFFSET 0 ROWS FETCH NEXT @p${paramIndex} ROWS ONLY`;
+      params.push(limit);
+      const rows = await this.#query(sql, params);
+      return rows.map((row) => ({
+        message: {
+          id: row.id,
+          chatId: row.chatId,
+          parentId: row.parentId,
+          name: row.name,
+          type: row.type ?? void 0,
+          data: JSON.parse(row.data),
+          createdAt: Number(row.createdAt)
+        },
+        rank: row.rank,
+        snippet: row.snippet
+      }));
+    } else {
+      let sql = `
+        SELECT
+          m.id,
+          m.chatId,
+          m.parentId,
+          m.name,
+          m.type,
+          m.data,
+          m.createdAt,
+          1 as rank,
+          SUBSTRING(fts.content, 1, 200) as snippet
+        FROM messages_fts fts
+        INNER JOIN messages m ON m.id = fts.messageId
+        WHERE fts.chatId = @p0 AND fts.content LIKE '%' + @p1 + '%'
+      `;
+      const params = [chatId, query];
+      let paramIndex = 2;
+      if (roles && roles.length > 0) {
+        const placeholders = roles.map(() => `@p${paramIndex++}`).join(", ");
+        sql += ` AND fts.name IN (${placeholders})`;
+        params.push(...roles);
+      }
+      sql += ` ORDER BY m.createdAt DESC OFFSET 0 ROWS FETCH NEXT @p${paramIndex} ROWS ONLY`;
+      params.push(limit);
+      const rows = await this.#query(sql, params);
+      return rows.map((row) => ({
+        message: {
+          id: row.id,
+          chatId: row.chatId,
+          parentId: row.parentId,
+          name: row.name,
+          type: row.type ?? void 0,
+          data: JSON.parse(row.data),
+          createdAt: Number(row.createdAt)
+        },
+        rank: row.rank,
+        snippet: row.snippet
+      }));
+    }
+  }
+  // ==========================================================================
+  // Visualization Operations
+  // ==========================================================================
+  async getGraph(chatId) {
+    const messageRows = await this.#query(
+      `SELECT id, parentId, name, data, createdAt
+       FROM messages
+       WHERE chatId = @p0
+       ORDER BY createdAt ASC`,
+      [chatId]
+    );
+    const nodes = messageRows.map((row) => {
+      const data = JSON.parse(row.data);
+      const content = typeof data === "string" ? data : JSON.stringify(data);
+      return {
+        id: row.id,
+        parentId: row.parentId,
+        role: row.name,
+        content: content.length > 50 ? content.slice(0, 50) + "..." : content,
+        createdAt: Number(row.createdAt)
+      };
+    });
+    const branchRows = await this.#query(
+      `SELECT name, headMessageId, isActive
+       FROM branches
+       WHERE chatId = @p0
+       ORDER BY createdAt ASC`,
+      [chatId]
+    );
+    const branches = branchRows.map((row) => ({
+      name: row.name,
+      headMessageId: row.headMessageId,
+      isActive: row.isActive === true || row.isActive === 1
+    }));
+    const checkpointRows = await this.#query(
+      `SELECT name, messageId
+       FROM checkpoints
+       WHERE chatId = @p0
+       ORDER BY createdAt ASC`,
+      [chatId]
+    );
+    const checkpoints = checkpointRows.map((row) => ({
+      name: row.name,
+      messageId: row.messageId
+    }));
+    return {
+      chatId,
+      nodes,
+      branches,
+      checkpoints
+    };
+  }
+};
+
+// packages/context/src/lib/visualize.ts
+function visualizeGraph(data) {
+  if (data.nodes.length === 0) {
+    return `[chat: ${data.chatId}]
+
+(empty)`;
+  }
+  const childrenByParentId = /* @__PURE__ */ new Map();
+  const branchHeads = /* @__PURE__ */ new Map();
+  const checkpointsByMessageId = /* @__PURE__ */ new Map();
+  for (const node of data.nodes) {
+    const children = childrenByParentId.get(node.parentId) ?? [];
+    children.push(node);
+    childrenByParentId.set(node.parentId, children);
+  }
+  for (const branch of data.branches) {
+    if (branch.headMessageId) {
+      const heads = branchHeads.get(branch.headMessageId) ?? [];
+      heads.push(branch.isActive ? `${branch.name} *` : branch.name);
+      branchHeads.set(branch.headMessageId, heads);
+    }
+  }
+  for (const checkpoint of data.checkpoints) {
+    const cps = checkpointsByMessageId.get(checkpoint.messageId) ?? [];
+    cps.push(checkpoint.name);
+    checkpointsByMessageId.set(checkpoint.messageId, cps);
+  }
+  const roots = childrenByParentId.get(null) ?? [];
+  const lines = [`[chat: ${data.chatId}]`, ""];
+  function renderNode(node, prefix, isLast, isRoot) {
+    const connector = isRoot ? "" : isLast ? "\u2514\u2500\u2500 " : "\u251C\u2500\u2500 ";
+    const contentPreview = node.content.replace(/\n/g, " ");
+    let line = `${prefix}${connector}${node.id.slice(0, 8)} (${node.role}): "${contentPreview}"`;
+    const branches = branchHeads.get(node.id);
+    if (branches) {
+      line += ` <- [${branches.join(", ")}]`;
+    }
+    const checkpoints = checkpointsByMessageId.get(node.id);
+    if (checkpoints) {
+      line += ` {${checkpoints.join(", ")}}`;
+    }
+    lines.push(line);
+    const children = childrenByParentId.get(node.id) ?? [];
+    const childPrefix = isRoot ? "" : prefix + (isLast ? "    " : "\u2502   ");
+    for (let i = 0; i < children.length; i++) {
+      renderNode(children[i], childPrefix, i === children.length - 1, false);
+    }
+  }
+  for (let i = 0; i < roots.length; i++) {
+    renderNode(roots[i], "", i === roots.length - 1, true);
+  }
+  lines.push("");
+  lines.push("Legend: * = active branch, {...} = checkpoint");
+  return lines.join("\n");
+}
+
+// packages/context/src/lib/agent.ts
+import { groq } from "@ai-sdk/groq";
+import {
+  NoSuchToolError,
+  Output,
+  convertToModelMessages,
+  createUIMessageStream,
+  generateId as generateId2,
+  generateText,
+  smoothStream,
+  stepCountIs,
+  streamText
+} from "ai";
+import chalk2 from "chalk";
+import "zod";
+import "@deepagents/agent";
+var Agent = class _Agent {
+  #options;
+  #guardrails = [];
+  tools;
+  constructor(options) {
+    this.#options = options;
+    this.tools = options.tools || {};
+    this.#guardrails = options.guardrails || [];
+  }
+  async generate(contextVariables, config) {
+    if (!this.#options.context) {
+      throw new Error(`Agent ${this.#options.name} is missing a context.`);
+    }
+    if (!this.#options.model) {
+      throw new Error(`Agent ${this.#options.name} is missing a model.`);
+    }
+    const { messages, systemPrompt } = await this.#options.context.resolve({
+      renderer: new XmlRenderer()
+    });
+    return generateText({
+      abortSignal: config?.abortSignal,
+      providerOptions: this.#options.providerOptions,
+      model: this.#options.model,
+      system: systemPrompt,
+      messages: await convertToModelMessages(messages),
+      stopWhen: stepCountIs(25),
+      tools: this.#options.tools,
+      experimental_context: contextVariables,
+      experimental_repairToolCall: repairToolCall,
+      toolChoice: this.#options.toolChoice,
+      onStepFinish: (step) => {
+        const toolCall = step.toolCalls.at(-1);
+        if (toolCall) {
+          console.log(
+            `Debug: ${chalk2.yellow("ToolCalled")}: ${toolCall.toolName}(${JSON.stringify(toolCall.input)})`
+          );
+        }
+      }
+    });
+  }
+  /**
+   * Stream a response from the agent.
+   *
+   * When guardrails are configured, `toUIMessageStream()` is wrapped to provide
+   * self-correction behavior. Direct access to fullStream/textStream bypasses guardrails.
+   *
+   * @example
+   * ```typescript
+   * const stream = await agent.stream({});
+   *
+   * // With guardrails - use toUIMessageStream for protection
+   * await printer.readableStream(stream.toUIMessageStream());
+   *
+   * // Or use printer.stdout which uses toUIMessageStream internally
+   * await printer.stdout(stream);
+   * ```
+   */
+  async stream(contextVariables, config) {
+    if (!this.#options.context) {
+      throw new Error(`Agent ${this.#options.name} is missing a context.`);
+    }
+    if (!this.#options.model) {
+      throw new Error(`Agent ${this.#options.name} is missing a model.`);
+    }
+    const result = await this.#createRawStream(contextVariables, config);
+    if (this.#guardrails.length === 0) {
+      return result;
+    }
+    return this.#wrapWithGuardrails(result, contextVariables, config);
+  }
+  /**
+   * Create a raw stream without guardrail processing.
+   */
+  async #createRawStream(contextVariables, config) {
     const { messages, systemPrompt } = await this.#options.context.resolve({
       renderer: new XmlRenderer()
     });
@@ -3607,8 +4948,10 @@ var Agent = class _Agent {
         execute: async ({ writer }) => {
           let currentResult = result;
           let attempt = 0;
+          const { mounts } = context.getSkillMounts();
           const guardrailContext = {
-            availableTools: Object.keys(this.tools)
+            availableTools: Object.keys(this.tools),
+            availableSkills: mounts
           };
           while (attempt < maxRetries) {
             if (config?.abortSignal?.aborted) {
@@ -3636,10 +4979,20 @@ var Agent = class _Agent {
                 );
                 break;
               }
+              if (checkResult.type === "stop") {
+                console.log(
+                  chalk2.red(
+                    `[${this.#options.name}] Guardrail stopped - unrecoverable error, no retry`
+                  )
+                );
+                writer.write(part);
+                writer.write({ type: "finish" });
+                return;
+              }
               if (checkResult.part.type === "text-delta") {
                 accumulatedText += checkResult.part.delta;
               }
-              writer.write(checkResult.part);
+              writer.write(part);
             }
             if (!guardrailFailed) {
               writer.write({ type: "finish" });
@@ -3806,7 +5159,9 @@ export {
   ModelsRegistry,
   MountPathError,
   PackageInstallError,
+  PostgresContextStore,
   RuntimeStrategy,
+  SqlServerContextStore,
   SqliteContextStore,
   TomlRenderer,
   ToonRenderer,
@@ -3854,6 +5209,7 @@ export {
   role,
   runGuardrailChain,
   skills,
+  stop,
   structuredOutput,
   styleGuide,
   term,