@byok-sdk/client 0.10.2 → 0.11.0

package/dist/bin/byok-approval-mcp.js

@@ -334,6 +334,205 @@ function createControlClient(socket, reader, opts) {
     }
   };
 }
+var AGENT_MESSAGE_TOOL_NAME = "send_agent_message";
+
+// src/bin/agent-message-mcp-server.ts
+async function handleAgentMessageMcpRequest(request, deps) {
+  const id = request.id;
+  if (request.method === "initialize") {
+    const params = request.params ?? {};
+    return {
+      jsonrpc: "2.0",
+      id,
+      result: {
+        protocolVersion: typeof params.protocolVersion === "string" ? params.protocolVersion : "2024-11-05",
+        capabilities: { tools: {} },
+        serverInfo: { name: "byok-agent-message-mcp", version: "0.0.1" }
+      }
+    };
+  }
+  if (request.method === "notifications/initialized") return void 0;
+  if (request.method === "tools/list") {
+    return {
+      jsonrpc: "2.0",
+      id,
+      result: { tools: [{
+        name: AGENT_MESSAGE_TOOL_NAME,
+        description: "Send the single user-visible Agent reply for this exact task. Routing identity is supplied by the authenticated task context.",
+        inputSchema: {
+          type: "object",
+          additionalProperties: false,
+          required: ["body"],
+          properties: {
+            body: { type: "string", minLength: 1 },
+            contentType: { type: "string", enum: ["text/plain", "text/markdown"] }
+          }
+        }
+      }] }
+    };
+  }
+  if (request.method === "tools/call") {
+    const params = request.params ?? {};
+    if (params.name !== AGENT_MESSAGE_TOOL_NAME) return { jsonrpc: "2.0", id, error: { code: -32602, message: "unknown Agent message tool" } };
+    const args = params.arguments;
+    if (args === null || typeof args !== "object" || Array.isArray(args)) return { jsonrpc: "2.0", id, error: { code: -32602, message: "message input must be an object" } };
+    const record3 = args;
+    if (Object.keys(record3).some((key) => key !== "body" && key !== "contentType") || typeof record3.body !== "string" || record3.body.length === 0) {
+      return { jsonrpc: "2.0", id, error: { code: -32602, message: "message input accepts only non-empty body and optional contentType" } };
+    }
+    const contentType = record3.contentType ?? "text/markdown";
+    if (contentType !== "text/plain" && contentType !== "text/markdown") return { jsonrpc: "2.0", id, error: { code: -32602, message: "unsupported contentType" } };
+    try {
+      const receipt = await deps.publish({ contentType, body: record3.body });
+      return { jsonrpc: "2.0", id, result: { content: [{ type: "text", text: JSON.stringify(receipt) }] } };
+    } catch (error) {
+      return { jsonrpc: "2.0", id, error: { code: -32e3, message: error instanceof Error ? error.message : String(error) } };
+    }
+  }
+  return id === void 0 ? void 0 : { jsonrpc: "2.0", id, error: { code: -32601, message: `unknown method: ${String(request.method)}` } };
+}
+function serveAgentMessageMcpOverStdio(input) {
+  const reader = createInterface({ input: input.stdin ?? process.stdin, terminal: false });
+  const output = input.stdout ?? process.stdout;
+  reader.on("line", (line) => {
+    const trimmed = line.trim();
+    if (trimmed.length === 0) return;
+    void (async () => {
+      let request;
+      try {
+        request = JSON.parse(trimmed);
+      } catch {
+        return;
+      }
+      const response = await handleAgentMessageMcpRequest(request, input.deps);
+      if (response !== void 0) output.write(`${JSON.stringify(response)}
+`);
+    })();
+  });
+}
+var AGENT_MEMORY_RECALL_TOOL_NAME = "memory_recall";
+var AGENT_MEMORY_SAVE_TOOL_NAME = "memory_save";
+function record(value) {
+  return value !== null && typeof value === "object" && !Array.isArray(value) ? value : void 0;
+}
+function invalid(id, message) {
+  return { jsonrpc: "2.0", id, error: { code: -32602, message } };
+}
+function success(id, value) {
+  return { jsonrpc: "2.0", id, result: { content: [{ type: "text", text: JSON.stringify(value) }] } };
+}
+async function handleAgentMemoryMcpRequest(request, deps) {
+  const id = request.id;
+  if (request.method === "initialize") {
+    const params2 = record(request.params) ?? {};
+    return { jsonrpc: "2.0", id, result: { protocolVersion: typeof params2.protocolVersion === "string" ? params2.protocolVersion : "2024-11-05", capabilities: { tools: {} }, serverInfo: { name: "byok-agent-memory-mcp", version: "0.0.1" } } };
+  }
+  if (request.method === "notifications/initialized") return void 0;
+  if (request.method === "tools/list") return {
+    jsonrpc: "2.0",
+    id,
+    result: { tools: [
+      { name: AGENT_MEMORY_RECALL_TOOL_NAME, description: "Recall one SDK-owned memory file for this exact active Agent task. Identity and memory root are never model parameters.", inputSchema: { type: "object", additionalProperties: false, required: ["path"], properties: { path: { type: "string" }, ifRevision: { type: "string", pattern: "^sha256:[a-f0-9]{64}$" } } } },
+      { name: AGENT_MEMORY_SAVE_TOOL_NAME, description: "Atomically replace or delete one SDK-owned memory file with exact sha256 compare-and-swap.", inputSchema: { type: "object", additionalProperties: false, required: ["op", "path", "expectedRevision"], properties: { op: { type: "string", enum: ["replace", "delete"] }, path: { type: "string" }, expectedRevision: { type: "string", pattern: "^sha256:[a-f0-9]{64}$" }, content: { type: "string" } } } }
+    ] }
+  };
+  if (request.method !== "tools/call") return id === void 0 ? void 0 : { jsonrpc: "2.0", id, error: { code: -32601, message: `unknown method: ${String(request.method)}` } };
+  const params = record(request.params);
+  const args = record(params?.arguments);
+  if (!params || !args || typeof params.name !== "string") return invalid(id, "memory tool input must be an object");
+  try {
+    if (params.name === AGENT_MEMORY_RECALL_TOOL_NAME) {
+      if (Object.keys(args).some((key) => key !== "path" && key !== "ifRevision") || typeof args.path !== "string" || args.ifRevision !== void 0 && typeof args.ifRevision !== "string") return invalid(id, "memory_recall accepts only path and optional ifRevision");
+      return success(id, await deps.recall({ path: args.path, ...args.ifRevision === void 0 ? {} : { ifRevision: args.ifRevision } }));
+    }
+    if (params.name === AGENT_MEMORY_SAVE_TOOL_NAME) {
+      if (Object.keys(args).some((key) => key !== "op" && key !== "path" && key !== "expectedRevision" && key !== "content") || args.op !== "replace" && args.op !== "delete" || typeof args.path !== "string" || typeof args.expectedRevision !== "string" || args.op === "replace" && typeof args.content !== "string" || args.op === "delete" && args.content !== void 0) return invalid(id, "memory_save requires replace|delete, path, expectedRevision, and content only for replace");
+      const content = args.content;
+      return success(id, await deps.save({ op: args.op, path: args.path, expectedRevision: args.expectedRevision, ...typeof content === "string" ? { content } : {} }));
+    }
+    return invalid(id, "unknown Agent memory tool");
+  } catch (error) {
+    return { jsonrpc: "2.0", id, error: { code: -32e3, message: error instanceof Error ? error.message : String(error) } };
+  }
+}
+function serveAgentMemoryMcpOverStdio(input) {
+  const reader = createInterface({ input: input.stdin ?? process.stdin, terminal: false });
+  const output = input.stdout ?? process.stdout;
+  reader.on("line", (line) => {
+    const trimmed = line.trim();
+    if (!trimmed) return;
+    void (async () => {
+      let request;
+      try {
+        request = JSON.parse(trimmed);
+      } catch {
+        return;
+      }
+      const response = await handleAgentMemoryMcpRequest(request, input.deps);
+      if (response !== void 0) output.write(`${JSON.stringify(response)}
+`);
+    })();
+  });
+}
+var TEAM_POST_TOOL_NAME = "post_team_message";
+var TEAM_READ_TOOL_NAME = "read_team_messages";
+var TEAM_ACK_TOOL_NAME = "ack_team_messages";
+var record2 = (value) => value !== null && typeof value === "object" && !Array.isArray(value) ? value : void 0;
+var invalid2 = (id, message) => ({ jsonrpc: "2.0", id, error: { code: -32602, message } });
+var success2 = (id, value) => ({ jsonrpc: "2.0", id, result: { content: [{ type: "text", text: JSON.stringify(value) }] } });
+async function handleTeamMcpRequest(request, deps) {
+  const { id } = request;
+  if (request.method === "initialize") {
+    const params2 = record2(request.params) ?? {};
+    return { jsonrpc: "2.0", id, result: { protocolVersion: typeof params2.protocolVersion === "string" ? params2.protocolVersion : "2024-11-05", capabilities: { tools: {} }, serverInfo: { name: "byok-agent-team-mcp", version: "0.0.1" } } };
+  }
+  if (request.method === "notifications/initialized") return void 0;
+  if (request.method === "tools/list") return { jsonrpc: "2.0", id, result: { tools: [
+    { name: TEAM_POST_TOOL_NAME, description: "Post one broadcast message as this exact leased team member. Sender and workspace are daemon-owned.", inputSchema: { type: "object", additionalProperties: false, required: ["body"], properties: { body: { type: "string" }, contentType: { type: "string" } } } },
+    { name: TEAM_READ_TOOL_NAME, description: "Read ordered team messages visible to this exact leased member.", inputSchema: { type: "object", additionalProperties: false, properties: { afterSeq: { type: "integer", minimum: 0 } } } },
+    { name: TEAM_ACK_TOOL_NAME, description: "Durably acknowledge messages through an already delivered sequence for this exact leased member.", inputSchema: { type: "object", additionalProperties: false, required: ["throughSeq"], properties: { throughSeq: { type: "integer", minimum: 0 } } } }
+  ] } };
+  if (request.method !== "tools/call") return id === void 0 ? void 0 : { jsonrpc: "2.0", id, error: { code: -32601, message: `unknown method: ${String(request.method)}` } };
+  const params = record2(request.params);
+  const args = record2(params?.arguments);
+  if (!params || !args || typeof params.name !== "string") return invalid2(id, "team tool input must be an object");
+  try {
+    if (params.name === TEAM_POST_TOOL_NAME) {
+      if (Object.keys(args).some((key) => key !== "body" && key !== "contentType") || typeof args.body !== "string" || args.contentType !== void 0 && typeof args.contentType !== "string") return invalid2(id, "post_team_message accepts body and optional contentType only");
+      return success2(id, await deps.post({ body: args.body, ...typeof args.contentType === "string" ? { contentType: args.contentType } : {} }));
+    }
+    if (params.name === TEAM_READ_TOOL_NAME) {
+      if (Object.keys(args).some((key) => key !== "afterSeq") || args.afterSeq !== void 0 && (!Number.isSafeInteger(args.afterSeq) || Number(args.afterSeq) < 0)) return invalid2(id, "read_team_messages accepts optional non-negative integer afterSeq only");
+      return success2(id, await deps.read(args.afterSeq === void 0 ? {} : { afterSeq: Number(args.afterSeq) }));
+    }
+    if (params.name === TEAM_ACK_TOOL_NAME) {
+      if (Object.keys(args).some((key) => key !== "throughSeq") || !Number.isSafeInteger(args.throughSeq) || Number(args.throughSeq) < 0) return invalid2(id, "ack_team_messages requires non-negative integer throughSeq only");
+      return success2(id, await deps.ack({ throughSeq: Number(args.throughSeq) }));
+    }
+    return invalid2(id, "unknown team tool");
+  } catch (error) {
+    return { jsonrpc: "2.0", id, error: { code: -32e3, message: error instanceof Error ? error.message : String(error) } };
+  }
+}
+function serveTeamMcpOverStdio(input) {
+  const reader = createInterface({ input: input.stdin ?? process.stdin, terminal: false });
+  const output = input.stdout ?? process.stdout;
+  reader.on("line", (line) => {
+    const trimmed = line.trim();
+    if (!trimmed) return;
+    void (async () => {
+      let request;
+      try {
+        request = JSON.parse(trimmed);
+      } catch {
+        return;
+      }
+      const response = await handleTeamMcpRequest(request, input.deps);
+      if (response !== void 0) output.write(`${JSON.stringify(response)}
+`);
+    })();
+  });
+}
 var APPROVAL_TOOL_NAME = "approval_prompt";
 var APPROVAL_SUMMARY_MAX_CHARS = 500;
 function errorMessage2(err) {
@@ -430,67 +629,135 @@ function serveApprovalMcpOverStdio(opts) {
   });
 }
 
-// src/bin/byok-approval-mcp.ts
-function errorMessage3(err) {
-  return err instanceof Error ? err.message : String(err);
-}
-var REQUEST_TIMEOUT_SLOP_MS = 5e3;
-function requireEnv(name) {
+// src/bin/sdk-reserved-helper-runners.ts
+var APPROVAL_REQUEST_TIMEOUT_SLOP_MS = 5e3;
+function required(name) {
   const value = process.env[name];
-  if (!value) {
-    process.stderr.write(`byok-approval-mcp: missing required env var ${name} \u2014 refusing to start
-`);
-    process.exit(1);
-  }
+  if (!value) throw new Error(`missing required environment variable ${name}`);
   return value;
 }
-async function main() {
-  const storeDir = requireEnv("BYOK_STORE_DIR");
-  const productId = requireEnv("BYOK_PRODUCT_ID");
-  const taskId = requireEnv("BYOK_TASK_ID");
+function waitForInputClose(input = process.stdin) {
+  if (input.readableEnded || input.destroyed) return Promise.resolve();
+  return new Promise((resolve) => {
+    const done = () => {
+      input.off("end", done);
+      input.off("close", done);
+      resolve();
+    };
+    input.once("end", done);
+    input.once("close", done);
+  });
+}
+async function runAgentMessageMcp() {
+  const storeDir = required("BYOK_STORE_DIR");
+  const productId = required("BYOK_PRODUCT_ID");
+  const contextToken = required("BYOK_AGENT_MESSAGE_CONTEXT");
+  let clientPromise;
+  const client = async () => {
+    if (!clientPromise) clientPromise = connectControlClient({ storeDir, productId }).then((result) => {
+      if (!result.ok) throw new Error(result.reason);
+      return result.client;
+    });
+    return clientPromise;
+  };
+  const deps = {
+    publish: async (input) => (await client()).request("agent_messages.publish", { contextToken, ...input })
+  };
+  serveAgentMessageMcpOverStdio({ deps });
+  await waitForInputClose();
+}
+async function runAgentMemoryMcp() {
+  const storeDir = required("BYOK_STORE_DIR");
+  const productId = required("BYOK_PRODUCT_ID");
+  const contextToken = required("BYOK_AGENT_MEMORY_CONTEXT");
+  let clientPromise;
+  const client = async () => {
+    if (!clientPromise) clientPromise = connectControlClient({ storeDir, productId }).then((result) => {
+      if (!result.ok) throw new Error(result.reason);
+      return result.client;
+    });
+    return clientPromise;
+  };
+  const deps = {
+    recall: async (input) => (await client()).request("agent_memory.recall", { contextToken, ...input }),
+    save: async (input) => (await client()).request("agent_memory.save", { contextToken, ...input })
+  };
+  serveAgentMemoryMcpOverStdio({ deps });
+  await waitForInputClose();
+}
+async function runApprovalMcp() {
+  const storeDir = required("BYOK_STORE_DIR");
+  const productId = required("BYOK_PRODUCT_ID");
+  const taskId = required("BYOK_TASK_ID");
   const timeoutMs = Number(process.env.BYOK_APPROVAL_TIMEOUT_MS ?? "600000");
   if (!Number.isFinite(timeoutMs) || timeoutMs <= 0) {
-    process.stderr.write("byok-approval-mcp: invalid BYOK_APPROVAL_TIMEOUT_MS env var \u2014 refusing to start\n");
-    process.exit(1);
+    throw new Error("invalid BYOK_APPROVAL_TIMEOUT_MS environment variable");
   }
   let clientPromise;
-  function getClient() {
-    if (!clientPromise) {
-      clientPromise = connectControlClient({ storeDir, productId, requestTimeoutMs: timeoutMs + REQUEST_TIMEOUT_SLOP_MS }).then(
-        (conn) => {
-          if (!conn.ok) {
-            clientPromise = void 0;
-            throw new Error(conn.reason);
-          }
-          return conn.client;
-        }
-      );
-      clientPromise.catch(() => {
-        clientPromise = void 0;
-      });
-    }
+  const client = async () => {
+    if (!clientPromise) clientPromise = connectControlClient({
+      storeDir,
+      productId,
+      requestTimeoutMs: timeoutMs + APPROVAL_REQUEST_TIMEOUT_SLOP_MS
+    }).then((result) => {
+      if (!result.ok) throw new Error(result.reason);
+      return result.client;
+    });
     return clientPromise;
-  }
+  };
   const deps = {
-    requestApproval: async (tId, summary) => {
-      let client;
+    requestApproval: async (requestedTaskId, summary) => {
       try {
-        client = await getClient();
-      } catch (err) {
-        throw new Error(`could not connect to the daemon control socket: ${errorMessage3(err)}`);
-      }
-      try {
-        return await client.request("approvals.request", { taskId: tId, summary });
-      } catch (err) {
+        return await (await client()).request("approvals.request", { taskId: requestedTaskId, summary });
+      } catch (error) {
         clientPromise = void 0;
-        throw err;
+        throw error;
       }
     }
   };
   serveApprovalMcpOverStdio({ taskId, deps });
+  await waitForInputClose();
 }
-main().catch((err) => {
-  process.stderr.write(`byok-approval-mcp: fatal error: ${errorMessage3(err)}
+async function runAgentTeamMcp() {
+  const storeDir = required("BYOK_STORE_DIR");
+  const productId = required("BYOK_PRODUCT_ID");
+  const lease = required("BYOK_TEAM_MEMBER_CONTEXT");
+  let clientPromise;
+  const client = async () => {
+    if (!clientPromise) clientPromise = connectControlClient({ storeDir, productId }).then((result) => {
+      if (!result.ok) throw new Error(result.reason);
+      return result.client;
+    });
+    return clientPromise;
+  };
+  const deps = {
+    post: async (input) => (await client()).request("team_messages.post", { context: lease, ...input }),
+    read: async (input) => (await client()).request("team_messages.read", { context: lease, ...input }),
+    ack: async (input) => (await client()).request("team_messages.ack", { context: lease, ...input })
+  };
+  serveTeamMcpOverStdio({ deps });
+  await waitForInputClose();
+}
+async function runSdkReservedHelper(kind) {
+  switch (kind) {
+    case "agent-message-mcp":
+      await runAgentMessageMcp();
+      return;
+    case "agent-memory-mcp":
+      await runAgentMemoryMcp();
+      return;
+    case "approval-mcp":
+      await runApprovalMcp();
+      return;
+    case "agent-team-mcp":
+      await runAgentTeamMcp();
+      return;
+  }
+}
+
+// src/bin/byok-approval-mcp.ts
+runSdkReservedHelper("approval-mcp").catch((error) => {
+  process.stderr.write(`byok-approval-mcp: ${error instanceof Error ? error.message : String(error)}
 `);
   process.exit(1);
 });