@haven_ai/mcp 0.4.0-alpha.0 → 0.6.0-alpha.0

package/README.md

@@ -117,6 +117,9 @@ Environment variable form:
 - `haven_discover_tools`
 - `haven_submit_catalog_entry`
 - `haven_list_receipts`
+- `haven_open_task_budget` (#3329)
+- `haven_close_task_budget` (#3329)
+- `haven_submit` (#3329 — relays a local signer signature by `task_budget_id`; a `payment_id` is refused on this runtime, which signs and submits payments inline)
 
 ### `idempotencyKey` is deprecated — send `idempotency_key`
 

package/dist/cli.cjs

@@ -246,7 +246,9 @@ var toolSchemas = {
     amount: v3.z.string().min(1),
     idempotency_key: v3.z.string().optional(),
     /** Legacy spelling, accepted during the #2366 window. Warns; do not use. */
-    idempotencyKey: v3.z.string().optional()
+    idempotencyKey: v3.z.string().optional(),
+    /** #3329: spend against an open task budget instead of the agent's period budget. */
+    task_budget_id: v3.z.string().min(1).optional()
   },
   haven_pay_mcp_tool: {
     merchant_url: v3.z.string().url(),
@@ -269,7 +271,9 @@ var toolSchemas = {
     quote: v3.z.unknown(),
     idempotency_key: v3.z.string().optional(),
     /** Legacy spelling, accepted during the #2366 window. Warns; do not use. */
-    idempotencyKey: v3.z.string().optional()
+    idempotencyKey: v3.z.string().optional(),
+    /** #3329: spend against an open task budget instead of the agent's period budget. */
+    task_budget_id: v3.z.string().min(1).optional()
   },
   haven_pay_x402: {
     url: v3.z.string().url(),
@@ -278,7 +282,9 @@ var toolSchemas = {
     body: v3.z.string().optional(),
     idempotency_key: v3.z.string().optional(),
     /** Legacy spelling, accepted during the #2366 window. Warns; do not use. */
-    idempotencyKey: v3.z.string().optional()
+    idempotencyKey: v3.z.string().optional(),
+    /** #3329: spend against an open task budget instead of the agent's period budget. */
+    task_budget_id: v3.z.string().min(1).optional()
   },
   haven_resume_x402_payment: {
     payment_id: v3.z.string().optional(),
@@ -310,9 +316,59 @@ var toolSchemas = {
   },
   haven_verify_receipt: {
     receipt: v3.z.unknown()
+  },
+  // #3329: a budget for one task that ends by itself — a short-lived child of
+  // the agent's own budget, capped and time-boxed independently of the period
+  // reset.
+  haven_open_task_budget: {
+    max_amount_human: v3.z.string().min(1),
+    ttl_minutes: v3.z.number().int().min(1).max(1440),
+    recipient: v3.z.string().optional(),
+    label: v3.z.string().max(120).optional(),
+    token: v3.z.string().optional()
+  },
+  haven_close_task_budget: {
+    task_budget_id: v3.z.string().min(1)
+  },
+  // #3329: relays a signature from the local signer — either the open/close
+  // signature for a task budget, or (schema parity with the hosted surface)
+  // a direct-payment signature by payment_id. Exactly one of task_budget_id /
+  // payment_id, never both or neither.
+  haven_submit: {
+    task_budget_id: v3.z.string().min(1).optional(),
+    payment_id: v3.z.string().min(1).optional(),
+    signature: v3.z.string().regex(/^0x[0-9a-fA-F]+$/, "signature must be a 0x-prefixed hex string")
   }
   // #3101: keys survive on the type (see the hosted server's contracts.ts).
 };
+var OPEN_TASK_BUDGET_DESCRIPTION = [
+  "Open a budget for one task that ends by itself: a spending cap, good for at most ttl_minutes,",
+  "reserved out of the agent's own budget and separate from its period reset.",
+  'Pass max_amount_human (whole tokens, e.g. "5" for 5 USDC), ttl_minutes (1-1440), and optionally',
+  "recipient (pins every payment against this task budget to one address), label, and token",
+  '(defaults to USDC). Returns { task_budget, next_action: "sign", next_tool, next_arguments } \u2014',
+  "call next_tool with next_arguments EXACTLY as given to get a signature, then relay it with",
+  "haven_submit. Spending anything above the reserved amount, past the deadline, or to a",
+  "different recipient than the one pinned here is declined on the spot \u2014 nothing is queued.",
+  "A recipient pin is checked against where each payment first goes: haven_pay_x402 and",
+  "haven_pay_x402_quote first move funds into the agent's own wallet, so a budget pinned to a",
+  "merchant is declined there \u2014 pin only budgets meant for haven_send."
+].join(" ");
+var CLOSE_TASK_BUDGET_DESCRIPTION = [
+  "End a task budget early, before its deadline, releasing whatever of its cap was unspent back",
+  "to the agent's own budget. Pass task_budget_id. If the budget was never signed (still pending),",
+  'this ends it immediately with { task_budget, status: "closed" } and nothing was ever reserved',
+  'on-chain. Otherwise it returns a signature request: { task_budget, next_action: "sign",',
+  "next_tool, next_arguments } \u2014 call next_tool with next_arguments EXACTLY as given, then relay",
+  "the signature with haven_submit. A task budget past its own deadline closes immediately, the",
+  "same as the pending case."
+].join(" ");
+var SUBMIT_DESCRIPTION = [
+  "Relay a signature from the local signer. Pass exactly one of task_budget_id (from",
+  "haven_open_task_budget or haven_close_task_budget) or payment_id \u2014 never both, never neither \u2014",
+  "plus signature. For a task budget this opens or closes it on-chain and returns { task_budget,",
+  'status }; a close in progress may return { task_budget, status: "closed", close_tx_hash }.'
+].join(" ");
 var toolDescriptions = {
   haven_send: sdk.composeDescription(sdk.toolDescriptions.send),
   haven_pay_mcp_tool: sdk.composeDescription(sdk.toolDescriptions.payMcpTool),
@@ -328,7 +384,10 @@ var toolDescriptions = {
   haven_discover_tools: sdk.composeDescription(sdk.toolDescriptions.discoverTools),
   haven_submit_catalog_entry: sdk.composeDescription(sdk.toolDescriptions.submitCatalogEntry),
   haven_list_receipts: sdk.composeDescription(sdk.toolDescriptions.listReceipts),
-  haven_verify_receipt: sdk.composeDescription(sdk.toolDescriptions.verifyReceipt)
+  haven_verify_receipt: sdk.composeDescription(sdk.toolDescriptions.verifyReceipt),
+  haven_open_task_budget: OPEN_TASK_BUDGET_DESCRIPTION,
+  haven_close_task_budget: CLOSE_TASK_BUDGET_DESCRIPTION,
+  haven_submit: SUBMIT_DESCRIPTION
 };
 function createToolHandlers(haven) {
   return {
@@ -344,7 +403,10 @@ function createToolHandlers(haven) {
             to: args.recipient,
             // #1207: was accepted by the schema but silently dropped — now
             // carried to the backend's replay contract.
-            idempotencyKey: args.idempotencyKey
+            idempotencyKey: args.idempotencyKey,
+            // #3329: spend against an open task budget instead of the
+            // agent's period budget, when the caller names one.
+            ...typeof args.task_budget_id === "string" ? { taskBudgetId: args.task_budget_id } : {}
           });
           return {
             payment_id: result.paymentId,
@@ -442,7 +504,10 @@ function createToolHandlers(haven) {
         );
       }
       return runTool(async () => {
-        const response = await haven.payX402Quote(args.quote, { idempotencyKey: args.idempotencyKey });
+        const response = await haven.payX402Quote(args.quote, {
+          idempotencyKey: args.idempotencyKey,
+          ...typeof args.task_budget_id === "string" ? { taskBudgetId: args.task_budget_id } : {}
+        });
         return responsePayload(response);
       }, warnings);
     },
@@ -451,7 +516,10 @@ function createToolHandlers(haven) {
       if ("success" in pf) return pf;
       const { args, warnings } = pf;
       return runTool(async () => {
-        const response = await haven.fetch(args.url, requestInit(args), { idempotencyKey: args.idempotencyKey });
+        const response = await haven.fetch(args.url, requestInit(args), {
+          idempotencyKey: args.idempotencyKey,
+          ...typeof args.task_budget_id === "string" ? { taskBudgetId: args.task_budget_id } : {}
+        });
         return responsePayload(response);
       }, warnings);
     },
@@ -552,6 +620,73 @@ function createToolHandlers(haven) {
     haven_verify_receipt: async (input) => {
       const args = objectInput("haven_verify_receipt", input);
       return runTool(async () => sdk.verifyPaymentReceipt(args.receipt));
+    },
+    haven_open_task_budget: async (input) => {
+      const args = objectInput("haven_open_task_budget", input);
+      return runTool(async () => {
+        const token = await resolveTaskBudgetToken(haven, typeof args.token === "string" ? args.token : void 0);
+        const maxAmountAtomic = humanToAtomicOrNull(args.max_amount_human, token.decimals);
+        if (maxAmountAtomic === null) {
+          throw new sdk.HavenApiError(
+            `max_amount_human ("${args.max_amount_human}") is not a valid decimal amount for ${token.symbol} (${token.decimals} decimal places). Nothing was reserved.`,
+            400
+          );
+        }
+        const result = await haven.openTaskBudget({
+          tokenAddress: token.address,
+          maxAmountAtomic,
+          ttlSeconds: Number(args.ttl_minutes) * 60,
+          recipientAddress: typeof args.recipient === "string" ? args.recipient : void 0,
+          label: typeof args.label === "string" ? args.label : void 0
+        });
+        return {
+          task_budget: result.taskBudget,
+          next_action: "sign",
+          next_tool: "mcp__haven-signer__haven_sign",
+          next_arguments: { task_budget_id: result.taskBudget.id }
+        };
+      });
+    },
+    haven_close_task_budget: async (input) => {
+      const args = objectInput("haven_close_task_budget", input);
+      return runTool(async () => {
+        const result = await haven.closeTaskBudget(args.task_budget_id);
+        if (result.status === "closed") {
+          return { task_budget: result.taskBudget, status: "closed" };
+        }
+        return {
+          task_budget: result.taskBudget,
+          next_action: "sign",
+          next_tool: "mcp__haven-signer__haven_sign",
+          next_arguments: { task_budget_id: args.task_budget_id }
+        };
+      });
+    },
+    haven_submit: async (input) => {
+      const args = objectInput("haven_submit", input);
+      const hasTaskBudget = typeof args.task_budget_id === "string" && args.task_budget_id.length > 0;
+      const hasPayment = typeof args.payment_id === "string" && args.payment_id.length > 0;
+      if (hasTaskBudget === hasPayment) {
+        return {
+          success: false,
+          code: "INVALID_INPUT",
+          message: "haven_submit takes exactly one of task_budget_id or payment_id, never both or neither. Nothing was relayed."
+        };
+      }
+      return runTool(async () => {
+        if (hasTaskBudget) {
+          const result = await haven.submitTaskBudget(args.task_budget_id, args.signature);
+          return {
+            task_budget: result.taskBudget,
+            status: result.status,
+            ...result.closeTxHash !== void 0 ? { close_tx_hash: result.closeTxHash } : {}
+          };
+        }
+        throw new sdk.HavenApiError(
+          "haven_submit with payment_id is not supported on the local MCP surface: haven_send signs and submits a direct payment in one call, so there is no separate relay step. Use haven_submit with task_budget_id to relay a task-budget open/close signature.",
+          400
+        );
+      });
     }
   };
   async function resumeState(args, rail) {
@@ -643,6 +778,28 @@ async function responsePayload(response) {
     body: parseMaybeJson(text)
   };
 }
+async function resolveTaskBudgetToken(haven, symbolOrAddress) {
+  const wanted = (symbolOrAddress ?? "USDC").toLowerCase();
+  const { allowances } = await haven.getAllowances();
+  const isAddress = /^0x[0-9a-fA-F]{40}$/.test(wanted);
+  const match = isAddress ? allowances.find((a) => a.tokenAddress.toLowerCase() === wanted) : allowances.find((a) => (a.tokenSymbol ?? "").toLowerCase() === wanted);
+  if (!match) {
+    const known = allowances.map((a) => `${a.tokenSymbol} (${a.tokenAddress})`);
+    throw new sdk.HavenApiError(
+      `token "${symbolOrAddress ?? "USDC"}" is not the symbol or address of any allowance this agent holds${known.length > 0 ? ` (it holds: ${known.join(", ")})` : " (it holds none)"}. Nothing was reserved.`,
+      400
+    );
+  }
+  const resolved = sdk.resolveTokenFromAddress(match.tokenAddress);
+  return { address: match.tokenAddress, symbol: match.tokenSymbol, decimals: resolved?.decimals ?? 6 };
+}
+function humanToAtomicOrNull(human, decimals) {
+  if (!/^[0-9]+(\.[0-9]+)?$/.test(human)) return null;
+  const [whole, frac = ""] = human.split(".");
+  if (frac.length > decimals) return null;
+  const atomic = BigInt(whole || "0") * 10n ** BigInt(decimals) + BigInt(frac.padEnd(decimals, "0") || "0");
+  return atomic.toString();
+}
 function parseMaybeJson(text) {
   if (!text) return null;
   try {
@@ -732,6 +889,9 @@ function normalizeError(err) {
       phase: stringOrUndefined(body?.phase),
       nextAction: stringOrUndefined(body?.nextAction) ?? stringOrUndefined(body?.next_action) ?? sdk.AgentPaymentNextAction.StopAndTellUser,
       next_action: stringOrUndefined(body?.nextAction) ?? stringOrUndefined(body?.next_action) ?? sdk.AgentPaymentNextAction.StopAndTellUser,
+      // #3303: a backend refusal that already names why no tool follows (the
+      // 426 `client_outdated` does) keeps that reason at the top level.
+      ...typeof body?.next_tool_omitted_reason === "string" ? { next_tool_omitted_reason: body.next_tool_omitted_reason } : {},
       body: err.body
     };
   }
@@ -784,7 +944,7 @@ function renderConsentBlock(input, hash) {
   ];
   if (input.apiUrl) lines.push(`Haven API: ${input.apiUrl}`);
   if (input.agentId) lines.push(`Agent ID:  ${input.agentId}`);
-  if (input.accountAddress) lines.push(`Haven wallet (Safe): ${input.accountAddress}`);
+  if (input.accountAddress) lines.push(`Haven wallet: ${input.accountAddress}`);
   if (input.delegateAddress) lines.push(`Delegate (local signer): ${input.delegateAddress}`);
   if (typeof input.chainId === "number") lines.push(`Chain ID:  ${input.chainId}`);
   lines.push("");
@@ -943,12 +1103,15 @@ async function resolveHavenClient(options = {}) {
   const client = new sdk.HavenClient({
     apiKey: credentials.apiKey,
     delegateKey: credentials.delegateKey,
-    baseUrl: credentials.apiUrl
+    baseUrl: credentials.apiUrl,
+    // #3303: name this package, not the SDK inside it, so the backend's
+    // update hint and refusal speak about what the user actually installed.
+    clientIdentity: sdk.havenClientIdentity(MCP_NAME, MCP_VERSION)
   });
   return { client, credentials };
 }
 var MCP_NAME = "@haven_ai/mcp";
-var MCP_VERSION = "0.4.0-alpha.0";
+var MCP_VERSION = "0.6.0-alpha.0";
 var MCP_INSTRUCTIONS = [
   "Haven local MCP server: signs in-process with the delegate key it holds on",
   "this machine \u2014 the key never leaves this process. Call haven_get_agent",
@@ -985,7 +1148,7 @@ function buildMcpServer(haven) {
       toolSchemas[name],
       async (args) => haven.withRequestContext(
         { "X-Haven-MCP-Tool": name },
-        async () => toMcpResult(await handlers[name](args))
+        async () => toMcpResult(withClientUpdate(await handlers[name](args), haven.clientUpdate()))
       )
     );
   }
@@ -1035,6 +1198,10 @@ async function runConsentGate(haven, credentials, options) {
     writeAck: options.writeAck
   });
 }
+function withClientUpdate(payload, hint) {
+  if (!hint || payload.client_update) return payload;
+  return { ...payload, client_update: hint };
+}
 function toMcpResult(payload) {
   return {
     isError: !payload.success,