zidane 6.4.2 → 6.4.4

package/dist/{providers-BSrJx0Tn.js → providers-z1vU0b2Q.js}

@@ -1,6 +1,6 @@
-import { m as openaiCompat } from "./openai-compat-0MJq4-g9.js";
+import { m as openaiCompat } from "./openai-compat-B3c_hzwf.js";
 import { s as resolveOAuthApiKey } from "./oauth-bkEOybqm.js";
-import { r as generatePkce } from "./xai-CpPbCGWO.js";
+import { r as generatePkce } from "./xai-BPwMQSuF.js";
 //#region src/chat/oauth-page/cursor.ts
 /** pi-ai / zidane OAuth provider id. Also the credentials-file key. */
 const CURSOR_OAUTH_PROVIDER_ID = "cursor";
@@ -163,6 +163,102 @@ function cursor(params) {
 	};
 }
 //#endregion
-export { createCursorOAuthProvider as n, cursor as t };
+//#region src/providers/mistral.ts
+const BASE_URL$1 = "https://api.mistral.ai/v1";
+function getApiKey$1(params) {
+	if (typeof params?.apiKey === "string" && params.apiKey.length > 0) return params.apiKey;
+	if (process.env.MISTRAL_API_KEY) return process.env.MISTRAL_API_KEY;
+	throw new Error("No Mistral API key found. Pass `apiKey` or set MISTRAL_API_KEY in your environment.");
+}
+/**
+* Model ids that accept `reasoning_effort`. Mistral rejects unknown request
+* fields on some routes, so the knob is sent only for the families that
+* document it (`mistral-small-*`, `mistral-medium-3.5` / `-2604`, and the
+* deprecated-but-live `magistral-*`); `codestral` / `devstral` / `ministral`
+* requests stay byte-identical to what they were without this planner.
+*/
+const REASONING_MODELS = /^(?:magistral-|mistral-small-|mistral-medium-(?:3\.5|2604))/;
+/**
+* Map zidane's thinking level to Mistral's `reasoning_effort`. Only the two
+* documented values are emitted — `'none'` (thinking chunk omitted, `content`
+* stays a plain string) and `'high'` (full thinking chunk). `adaptive` sends
+* nothing so the model's own default applies.
+*
+* Exported for tests.
+*/
+function planMistralReasoning(ctx) {
+	if (!REASONING_MODELS.test(ctx.model)) return void 0;
+	if (ctx.thinking === void 0 || ctx.thinking === "adaptive") return void 0;
+	return { reasoning_effort: ctx.thinking === "off" ? "none" : "high" };
+}
+/**
+* Mistral provider (La Plateforme).
+*
+* Thin wrapper around {@link openaiCompat} with Mistral-specific defaults
+* (base URL, default model, `reasoning_effort` planner). `supportsReasoning`
+* stays off: Mistral carries thinking inside `content` chunks rather than a
+* `reasoning_content` field, so there is nothing to echo back upstream —
+* thinking streams to the UI and is dropped from subsequent requests.
+*/
+function mistral(params) {
+	return openaiCompat({
+		name: "mistral",
+		apiKey: getApiKey$1(params),
+		baseURL: params?.baseURL || BASE_URL$1,
+		defaultModel: params?.defaultModel || "mistral-medium-latest",
+		capabilities: params?.capabilities ?? {
+			vision: true,
+			imageInToolResult: false
+		},
+		planReasoningRequest: planMistralReasoning,
+		extraHeaders: params?.extraHeaders
+	});
+}
+//#endregion
+//#region src/providers/zai.ts
+const BASE_URL = "https://api.z.ai/api/paas/v4";
+function getApiKey(params) {
+	if (typeof params?.apiKey === "string" && params.apiKey.length > 0) return params.apiKey;
+	if (process.env.ZAI_API_KEY) return process.env.ZAI_API_KEY;
+	throw new Error("No Z.ai API key found. Pass `apiKey` or set ZAI_API_KEY in your environment.");
+}
+/**
+* Map zidane's thinking level to Z.ai's `extra_body.thinking` request shape.
+* GLM has no depth knob — thinking is a binary switch — so any non-`off`
+* level enables it and `off` explicitly disables it (GLM-5.x/4.7 think by
+* default, so a silent omission would not turn thinking off).
+*
+* Exported for tests.
+*/
+function planZaiReasoning(ctx) {
+	return { thinking: { type: !!ctx.thinking && ctx.thinking !== "off" ? "enabled" : "disabled" } };
+}
+/**
+* Z.ai (Zhipu) GLM provider.
+*
+* Thin wrapper around {@link openaiCompat} pinned to the `reasoning_content`
+* round-trip mode: thinking arrives in the `reasoning_content` delta field and
+* is echoed back on assistant messages under the same name, keeping GLM's
+* interleaved/preserved thinking coherent across tool calls.
+*/
+function zai(params) {
+	return openaiCompat({
+		name: "zai",
+		apiKey: getApiKey(params),
+		baseURL: params?.baseURL || BASE_URL,
+		defaultModel: params?.defaultModel || "glm-4.7-flash",
+		capabilities: params?.capabilities ?? {
+			vision: false,
+			imageInToolResult: false
+		},
+		supportsReasoning: true,
+		reasoningMode: "reasoning_content",
+		reasoningContentField: "reasoning_content",
+		planReasoningRequest: planZaiReasoning,
+		extraHeaders: params?.extraHeaders
+	});
+}
+//#endregion
+export { cursor as a, planMistralReasoning as i, zai as n, createCursorOAuthProvider as o, mistral as r, planZaiReasoning as t };
 
-//# sourceMappingURL=providers-BSrJx0Tn.js.map
+//# sourceMappingURL=providers-z1vU0b2Q.js.map

package/dist/providers-z1vU0b2Q.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"providers-z1vU0b2Q.js","names":["BASE_URL","getApiKey"],"sources":["../src/chat/oauth-page/cursor.ts","../src/providers/cursor.ts","../src/providers/mistral.ts","../src/providers/zai.ts"],"sourcesContent":["/**\n * Cursor OAuth flow.\n *\n * Unlike Anthropic / OpenAI Codex (loopback-redirect PKCE), Cursor uses a\n * **poll-based** PKCE flow with no local callback server:\n *\n *   1. Generate a PKCE verifier + challenge and a random UUID.\n *   2. Open the browser to `cursor.com/loginDeepControl?challenge&uuid&mode&redirectTarget`.\n *   3. Poll `api2.cursor.sh/auth/poll?uuid&verifier` until tokens come back\n *      (HTTP 404 = \"not ready yet\", keep polling with backoff).\n *   4. Refresh via `POST api2.cursor.sh/auth/exchange_user_api_key` with\n *      `Authorization: Bearer <refresh>`.\n *\n * Token expiry is derived from the access token's JWT `exp` claim (minus a\n * 5-minute safety margin), falling back to +1h when the claim is missing.\n *\n * Flow adapted from https://github.com/ndraiman/pi-cursor-provider, itself\n * derived from https://github.com/ephraimduncan/opencode-cursor.\n *\n * Cursor's auth + inference protocol is unofficial and undocumented — re-verify\n * the endpoints/params below if login starts failing.\n */\n\nimport type {\n  OAuthCredentials,\n  OAuthLoginCallbacks,\n} from '@earendil-works/pi-ai/oauth'\nimport type { OAuthProviderInterface } from '../../providers/oauth-registry'\nimport { generatePkce } from './pkce'\n\n/** pi-ai / zidane OAuth provider id. Also the credentials-file key. */\nexport const CURSOR_OAUTH_PROVIDER_ID = 'cursor'\n\nconst CURSOR_LOGIN_URL = 'https://cursor.com/loginDeepControl'\nconst CURSOR_POLL_URL = 'https://api2.cursor.sh/auth/poll'\nconst CURSOR_REFRESH_URL = 'https://api2.cursor.sh/auth/exchange_user_api_key'\n\nconst POLL_MAX_ATTEMPTS = 150\nconst POLL_BASE_DELAY_MS = 1000\nconst POLL_MAX_DELAY_MS = 10_000\nconst POLL_BACKOFF_MULTIPLIER = 1.2\nconst POLL_MAX_CONSECUTIVE_ERRORS = 3\n\n/** Fallback lifetime when the JWT carries no `exp`. */\nconst DEFAULT_TOKEN_TTL_MS = 3600 * 1000\n/** Refresh slightly early so requests don't race expiry. */\nconst TOKEN_EXPIRY_SKEW_MS = 5 * 60 * 1000\n\ninterface CursorTokenResponse {\n  accessToken: string\n  refreshToken: string\n}\n\n/**\n * Derive expiry (epoch ms) from a JWT's `exp` claim, minus a safety skew.\n * Returns `now + 1h` for malformed tokens so a missing claim never wedges\n * refresh into an immediate-expire loop.\n */\nexport function cursorTokenExpiry(token: string): number {\n  try {\n    const parts = token.split('.')\n    const payload = parts[1]\n    if (parts.length !== 3 || !payload)\n      return Date.now() + DEFAULT_TOKEN_TTL_MS\n\n    const json = JSON.parse(atob(payload.replace(/-/g, '+').replace(/_/g, '/'))) as { exp?: unknown }\n    if (typeof json.exp === 'number')\n      return json.exp * 1000 - TOKEN_EXPIRY_SKEW_MS\n  }\n  catch {\n    // fall through to default\n  }\n  return Date.now() + DEFAULT_TOKEN_TTL_MS\n}\n\nfunction delay(ms: number): Promise<void> {\n  return new Promise(r => setTimeout(r, ms))\n}\n\n/**\n * Poll Cursor's auth endpoint until login completes. `404` means the user\n * hasn't finished the browser flow yet; anything else non-2xx is a hard error.\n */\nasync function pollForTokens(\n  uuid: string,\n  verifier: string,\n  callbacks: OAuthLoginCallbacks,\n): Promise<CursorTokenResponse> {\n  let wait = POLL_BASE_DELAY_MS\n  let consecutiveErrors = 0\n\n  for (let attempt = 0; attempt < POLL_MAX_ATTEMPTS; attempt++) {\n    if (callbacks.signal?.aborted)\n      throw new Error('Cursor login aborted')\n\n    await delay(wait)\n\n    try {\n      const url = `${CURSOR_POLL_URL}?uuid=${encodeURIComponent(uuid)}&verifier=${encodeURIComponent(verifier)}`\n      const response = await fetch(url, { signal: callbacks.signal })\n\n      if (response.status === 404) {\n        consecutiveErrors = 0\n        wait = Math.min(wait * POLL_BACKOFF_MULTIPLIER, POLL_MAX_DELAY_MS)\n        continue\n      }\n\n      if (response.ok) {\n        const data = await response.json() as Partial<CursorTokenResponse>\n        if (!data.accessToken || !data.refreshToken)\n          throw new Error(`Cursor poll response missing tokens: ${JSON.stringify(data)}`)\n        return { accessToken: data.accessToken, refreshToken: data.refreshToken }\n      }\n\n      throw new Error(`Cursor poll failed (${response.status})`)\n    }\n    catch (err) {\n      if (callbacks.signal?.aborted)\n        throw new Error('Cursor login aborted')\n      consecutiveErrors++\n      if (consecutiveErrors >= POLL_MAX_CONSECUTIVE_ERRORS)\n        throw err instanceof Error ? err : new Error(String(err))\n      callbacks.onProgress?.('Waiting for Cursor login to complete…')\n    }\n  }\n\n  throw new Error('Cursor authentication timed out')\n}\n\nexport async function loginCursor(callbacks: OAuthLoginCallbacks): Promise<OAuthCredentials> {\n  const { verifier, challenge } = await generatePkce()\n  const uuid = crypto.randomUUID()\n\n  const authUrl = new URL(CURSOR_LOGIN_URL)\n  authUrl.searchParams.set('challenge', challenge)\n  authUrl.searchParams.set('uuid', uuid)\n  authUrl.searchParams.set('mode', 'login')\n  authUrl.searchParams.set('redirectTarget', 'cli')\n\n  callbacks.onAuth({\n    url: authUrl.toString(),\n    instructions: 'A browser window should open. Complete login in Cursor to finish.',\n  })\n\n  const tokens = await pollForTokens(uuid, verifier, callbacks)\n\n  return {\n    access: tokens.accessToken,\n    refresh: tokens.refreshToken,\n    expires: cursorTokenExpiry(tokens.accessToken),\n  }\n}\n\nexport async function refreshCursorToken(credentials: OAuthCredentials): Promise<OAuthCredentials> {\n  const response = await fetch(CURSOR_REFRESH_URL, {\n    method: 'POST',\n    headers: {\n      'Authorization': `Bearer ${credentials.refresh}`,\n      'Content-Type': 'application/json',\n    },\n    body: '{}',\n  })\n\n  if (!response.ok) {\n    const text = await response.text().catch(() => '')\n    throw new Error(`Cursor token refresh failed (${response.status}): ${text || response.statusText}`)\n  }\n\n  const data = await response.json() as Partial<CursorTokenResponse>\n  if (!data.accessToken)\n    throw new Error(`Cursor token refresh response missing access token: ${JSON.stringify(data)}`)\n\n  return {\n    access: data.accessToken,\n    // Cursor may omit a rotated refresh token — keep the existing one.\n    refresh: data.refreshToken || credentials.refresh,\n    expires: cursorTokenExpiry(data.accessToken),\n  }\n}\n\n/**\n * Build a Cursor `OAuthProviderInterface`. Shape matches Anthropic / Codex so\n * it drops into `BUILTIN_PROVIDERS` and `src/auth.ts` unchanged.\n *\n * No `renderPage` parameter: Cursor's poll flow has no local callback server,\n * so there is no post-redirect page to theme.\n */\nexport function createCursorOAuthProvider(): OAuthProviderInterface {\n  return {\n    id: CURSOR_OAUTH_PROVIDER_ID,\n    name: 'Cursor',\n    usesCallbackServer: false,\n    login: loginCursor,\n    refreshToken: refreshCursorToken,\n    getApiKey: credentials => credentials.access,\n  }\n}\n","import type { Provider, StreamCallbacks } from '.'\nimport { CURSOR_OAUTH_PROVIDER_ID } from '../chat/oauth-page/cursor'\nimport { resolveOAuthApiKey } from './oauth'\nimport { openaiCompat } from './openai-compat'\n\n/**\n * Cursor provider (OAuth).\n *\n * Auth is fully wired: `bun run auth --cursor` (or the TUI wizard) logs in via\n * Cursor's poll-based PKCE flow and persists credentials; this provider resolves\n * + refreshes the OAuth token on every turn via {@link resolveOAuthApiKey}.\n *\n * Inference is **not** wired yet. Cursor does not expose an OpenAI-compatible\n * HTTP endpoint — it speaks a protobuf/Connect agent protocol over HTTP/2 to\n * `api2.cursor.sh`. Implementing `stream()` requires porting that transport\n * (generated protobuf schemas + an HTTP/2 streaming client) in-process, which\n * is intentionally left as a follow-up. Until then `stream()` resolves the\n * token (proving auth works) and then throws a clear, actionable error.\n *\n * The `openaiCompat` base supplies the message-format helpers (`formatTools`,\n * `userMessage`, `toolResultsMessage`, …) so the provider satisfies the full\n * `Provider` contract and slots into the registry; only `stream` is overridden.\n */\nexport interface CursorParams {\n  /** Bypass OAuth resolution with an explicit token (mainly for tests). */\n  apiKey?: string\n  defaultModel?: string\n  /**\n   * Extra HTTP headers for the eventual Cursor transport. Accepted for API\n   * parity with the other providers and forwarded to the openai-compat base,\n   * but inert until Cursor inference is implemented (the `stream` override\n   * currently throws before any request is made).\n   */\n  extraHeaders?: Record<string, string>\n}\n\nconst DEFAULT_MODEL = 'claude-4.6-sonnet'\n\nconst NOT_IMPLEMENTED_MESSAGE\n  = 'Cursor OAuth login works, but inference over Cursor is not implemented yet. '\n    + 'Cursor uses a protobuf/HTTP-2 agent protocol (not an OpenAI-compatible API), '\n    + 'so the streaming transport still needs to be ported. Use another provider for now.'\n\nexport function cursor(params?: CursorParams): Provider {\n  const defaultModel = params?.defaultModel || DEFAULT_MODEL\n\n  // Base only provides the format/serialization methods; `baseURL` is never\n  // hit because we override `stream` below.\n  const base = openaiCompat({\n    name: 'cursor',\n    apiKey: params?.apiKey ?? 'oauth',\n    baseURL: 'https://api2.cursor.sh',\n    defaultModel,\n    capabilities: { vision: true, imageInToolResult: false },\n    extraHeaders: params?.extraHeaders,\n  })\n\n  return {\n    ...base,\n    meta: { ...base.meta, defaultModel },\n    async stream(options, callbacks: StreamCallbacks) {\n      // Resolve (and lazily refresh) the OAuth token so auth failures surface\n      // here with the right remediation, distinct from the not-implemented path.\n      await resolveOAuthApiKey(\n        {\n          provider: 'cursor',\n          providerId: CURSOR_OAUTH_PROVIDER_ID,\n          params,\n          missingError: 'No Cursor credentials found. Run `bun run auth --cursor` first.',\n          refreshError: reason => `Cursor OAuth token refresh failed. Run \\`bun run auth --cursor\\` again. ${reason}`,\n        },\n        callbacks,\n      )\n      void options\n      throw new Error(NOT_IMPLEMENTED_MESSAGE)\n    },\n  }\n}\n","import type { Provider, ProviderCapabilities } from '.'\nimport type { ThinkingLevel } from '../types'\nimport { openaiCompat } from './openai-compat'\n\n// Mistral La Plateforme — OpenAI-compatible `/v1/chat/completions`. Model ids\n// are plain slugs (`mistral-medium-latest`, `devstral-2512`, …). Reasoning is\n// NOT a separate wire field here: with `reasoning_effort: 'high'` the response\n// `content` becomes a list of `thinking` / `text` chunks (handled by\n// `openaiCompat`'s content-chunk path), and with `'none'` it stays a plain\n// string. See https://docs.mistral.ai/studio-api/conversations/reasoning.\nconst BASE_URL = 'https://api.mistral.ai/v1'\n\nexport interface MistralParams {\n  apiKey?: string\n  defaultModel?: string\n  /**\n   * Override the base URL — e.g. point at a regional endpoint or a gateway in\n   * front of La Plateforme.\n   */\n  baseURL?: string\n  /**\n   * Provider capability flags. The `mistral-medium` / `mistral-large` /\n   * `pixtral` families accept `image_url` parts — default:\n   * `{ vision: true, imageInToolResult: false }`. Set `vision: false` when\n   * pinning a text-only model (`codestral`, `devstral`, `ministral`).\n   */\n  capabilities?: ProviderCapabilities\n  /**\n   * Extra HTTP headers sent on every request. Forwarded to the underlying\n   * {@link openaiCompat} client.\n   */\n  extraHeaders?: Record<string, string>\n}\n\nfunction getApiKey(params?: MistralParams): string {\n  // Resolution order aligned with the other providers: explicit > env > fail.\n  if (typeof params?.apiKey === 'string' && params.apiKey.length > 0)\n    return params.apiKey\n\n  if (process.env.MISTRAL_API_KEY)\n    return process.env.MISTRAL_API_KEY\n\n  throw new Error('No Mistral API key found. Pass `apiKey` or set MISTRAL_API_KEY in your environment.')\n}\n\n/**\n * Model ids that accept `reasoning_effort`. Mistral rejects unknown request\n * fields on some routes, so the knob is sent only for the families that\n * document it (`mistral-small-*`, `mistral-medium-3.5` / `-2604`, and the\n * deprecated-but-live `magistral-*`); `codestral` / `devstral` / `ministral`\n * requests stay byte-identical to what they were without this planner.\n */\nconst REASONING_MODELS = /^(?:magistral-|mistral-small-|mistral-medium-(?:3\\.5|2604))/\n\n/**\n * Map zidane's thinking level to Mistral's `reasoning_effort`. Only the two\n * documented values are emitted — `'none'` (thinking chunk omitted, `content`\n * stays a plain string) and `'high'` (full thinking chunk). `adaptive` sends\n * nothing so the model's own default applies.\n *\n * Exported for tests.\n */\nexport function planMistralReasoning(\n  ctx: { model: string, thinking?: ThinkingLevel },\n): Record<string, unknown> | undefined {\n  if (!REASONING_MODELS.test(ctx.model))\n    return undefined\n  if (ctx.thinking === undefined || ctx.thinking === 'adaptive')\n    return undefined\n  return { reasoning_effort: ctx.thinking === 'off' ? 'none' : 'high' }\n}\n\n/**\n * Mistral provider (La Plateforme).\n *\n * Thin wrapper around {@link openaiCompat} with Mistral-specific defaults\n * (base URL, default model, `reasoning_effort` planner). `supportsReasoning`\n * stays off: Mistral carries thinking inside `content` chunks rather than a\n * `reasoning_content` field, so there is nothing to echo back upstream —\n * thinking streams to the UI and is dropped from subsequent requests.\n */\nexport function mistral(params?: MistralParams): Provider {\n  return openaiCompat({\n    name: 'mistral',\n    apiKey: getApiKey(params),\n    baseURL: params?.baseURL || BASE_URL,\n    defaultModel: params?.defaultModel || 'mistral-medium-latest',\n    capabilities: params?.capabilities ?? { vision: true, imageInToolResult: false },\n    planReasoningRequest: planMistralReasoning,\n    extraHeaders: params?.extraHeaders,\n  })\n}\n","import type { Provider, ProviderCapabilities } from '.'\nimport type { ThinkingLevel } from '../types'\nimport { openaiCompat } from './openai-compat'\n\n// Z.ai (Zhipu) GLM Open Platform — OpenAI-compatible endpoint. Model ids are\n// plain lowercase slugs (`glm-4.7`, `glm-4.7-flash`, …), reasoning streams in\n// the `reasoning_content` delta field, and thinking is toggled via an\n// `extra_body.thinking` object rather than a `reasoning_effort` string.\n// See https://docs.z.ai/guides/develop/http/introduction and\n// https://docs.z.ai/guides/capabilities/thinking-mode.\nconst BASE_URL = 'https://api.z.ai/api/paas/v4'\n\nexport interface ZaiParams {\n  apiKey?: string\n  defaultModel?: string\n  /**\n   * Override the base URL — e.g. point at the dedicated Coding Plan endpoint\n   * instead of the standard API endpoint.\n   */\n  baseURL?: string\n  /**\n   * Provider capability flags. GLM text models are text-only — default:\n   * `{ vision: false, imageInToolResult: false }`.\n   */\n  capabilities?: ProviderCapabilities\n  /**\n   * Extra HTTP headers sent on every request. Forwarded to the underlying\n   * {@link openaiCompat} client.\n   */\n  extraHeaders?: Record<string, string>\n}\n\nfunction getApiKey(params?: ZaiParams): string {\n  // Resolution order aligned with the other providers: explicit > env > fail.\n  if (typeof params?.apiKey === 'string' && params.apiKey.length > 0)\n    return params.apiKey\n\n  if (process.env.ZAI_API_KEY)\n    return process.env.ZAI_API_KEY\n\n  throw new Error('No Z.ai API key found. Pass `apiKey` or set ZAI_API_KEY in your environment.')\n}\n\n/**\n * Map zidane's thinking level to Z.ai's `extra_body.thinking` request shape.\n * GLM has no depth knob — thinking is a binary switch — so any non-`off`\n * level enables it and `off` explicitly disables it (GLM-5.x/4.7 think by\n * default, so a silent omission would not turn thinking off).\n *\n * Exported for tests.\n */\nexport function planZaiReasoning(\n  ctx: { thinking?: ThinkingLevel },\n): Record<string, unknown> {\n  const enabled = !!ctx.thinking && ctx.thinking !== 'off'\n  return { thinking: { type: enabled ? 'enabled' : 'disabled' } }\n}\n\n/**\n * Z.ai (Zhipu) GLM provider.\n *\n * Thin wrapper around {@link openaiCompat} pinned to the `reasoning_content`\n * round-trip mode: thinking arrives in the `reasoning_content` delta field and\n * is echoed back on assistant messages under the same name, keeping GLM's\n * interleaved/preserved thinking coherent across tool calls.\n */\nexport function zai(params?: ZaiParams): Provider {\n  return openaiCompat({\n    name: 'zai',\n    apiKey: getApiKey(params),\n    baseURL: params?.baseURL || BASE_URL,\n    defaultModel: params?.defaultModel || 'glm-4.7-flash',\n    capabilities: params?.capabilities ?? { vision: false, imageInToolResult: false },\n    supportsReasoning: true,\n    reasoningMode: 'reasoning_content',\n    reasoningContentField: 'reasoning_content',\n    planReasoningRequest: planZaiReasoning,\n    extraHeaders: params?.extraHeaders,\n  })\n}\n"],"mappings":";;;;;AA+BA,MAAa,2BAA2B;AAExC,MAAM,mBAAmB;AACzB,MAAM,kBAAkB;AACxB,MAAM,qBAAqB;AAE3B,MAAM,oBAAoB;AAC1B,MAAM,qBAAqB;AAC3B,MAAM,oBAAoB;AAC1B,MAAM,0BAA0B;AAChC,MAAM,8BAA8B;;AAGpC,MAAM,uBAAuB,OAAO;;AAEpC,MAAM,uBAAuB,MAAS;;;;;;AAYtC,SAAgB,kBAAkB,OAAuB;CACvD,IAAI;EACF,MAAM,QAAQ,MAAM,MAAM,GAAG;EAC7B,MAAM,UAAU,MAAM;EACtB,IAAI,MAAM,WAAW,KAAK,CAAC,SACzB,OAAO,KAAK,IAAI,IAAI;EAEtB,MAAM,OAAO,KAAK,MAAM,KAAK,QAAQ,QAAQ,MAAM,GAAG,CAAC,CAAC,QAAQ,MAAM,GAAG,CAAC,CAAC;EAC3E,IAAI,OAAO,KAAK,QAAQ,UACtB,OAAO,KAAK,MAAM,MAAO;CAC7B,QACM,CAEN;CACA,OAAO,KAAK,IAAI,IAAI;AACtB;AAEA,SAAS,MAAM,IAA2B;CACxC,OAAO,IAAI,SAAQ,MAAK,WAAW,GAAG,EAAE,CAAC;AAC3C;;;;;AAMA,eAAe,cACb,MACA,UACA,WAC8B;CAC9B,IAAI,OAAO;CACX,IAAI,oBAAoB;CAExB,KAAK,IAAI,UAAU,GAAG,UAAU,mBAAmB,WAAW;EAC5D,IAAI,UAAU,QAAQ,SACpB,MAAM,IAAI,MAAM,sBAAsB;EAExC,MAAM,MAAM,IAAI;EAEhB,IAAI;GACF,MAAM,MAAM,GAAG,gBAAgB,QAAQ,mBAAmB,IAAI,EAAE,YAAY,mBAAmB,QAAQ;GACvG,MAAM,WAAW,MAAM,MAAM,KAAK,EAAE,QAAQ,UAAU,OAAO,CAAC;GAE9D,IAAI,SAAS,WAAW,KAAK;IAC3B,oBAAoB;IACpB,OAAO,KAAK,IAAI,OAAO,yBAAyB,iBAAiB;IACjE;GACF;GAEA,IAAI,SAAS,IAAI;IACf,MAAM,OAAO,MAAM,SAAS,KAAK;IACjC,IAAI,CAAC,KAAK,eAAe,CAAC,KAAK,cAC7B,MAAM,IAAI,MAAM,wCAAwC,KAAK,UAAU,IAAI,GAAG;IAChF,OAAO;KAAE,aAAa,KAAK;KAAa,cAAc,KAAK;IAAa;GAC1E;GAEA,MAAM,IAAI,MAAM,uBAAuB,SAAS,OAAO,EAAE;EAC3D,SACO,KAAK;GACV,IAAI,UAAU,QAAQ,SACpB,MAAM,IAAI,MAAM,sBAAsB;GACxC;GACA,IAAI,qBAAqB,6BACvB,MAAM,eAAe,QAAQ,MAAM,IAAI,MAAM,OAAO,GAAG,CAAC;GAC1D,UAAU,aAAa,uCAAuC;EAChE;CACF;CAEA,MAAM,IAAI,MAAM,iCAAiC;AACnD;AAEA,eAAsB,YAAY,WAA2D;CAC3F,MAAM,EAAE,UAAU,cAAc,MAAM,aAAa;CACnD,MAAM,OAAO,OAAO,WAAW;CAE/B,MAAM,UAAU,IAAI,IAAI,gBAAgB;CACxC,QAAQ,aAAa,IAAI,aAAa,SAAS;CAC/C,QAAQ,aAAa,IAAI,QAAQ,IAAI;CACrC,QAAQ,aAAa,IAAI,QAAQ,OAAO;CACxC,QAAQ,aAAa,IAAI,kBAAkB,KAAK;CAEhD,UAAU,OAAO;EACf,KAAK,QAAQ,SAAS;EACtB,cAAc;CAChB,CAAC;CAED,MAAM,SAAS,MAAM,cAAc,MAAM,UAAU,SAAS;CAE5D,OAAO;EACL,QAAQ,OAAO;EACf,SAAS,OAAO;EAChB,SAAS,kBAAkB,OAAO,WAAW;CAC/C;AACF;AAEA,eAAsB,mBAAmB,aAA0D;CACjG,MAAM,WAAW,MAAM,MAAM,oBAAoB;EAC/C,QAAQ;EACR,SAAS;GACP,iBAAiB,UAAU,YAAY;GACvC,gBAAgB;EAClB;EACA,MAAM;CACR,CAAC;CAED,IAAI,CAAC,SAAS,IAAI;EAChB,MAAM,OAAO,MAAM,SAAS,KAAK,CAAC,CAAC,YAAY,EAAE;EACjD,MAAM,IAAI,MAAM,gCAAgC,SAAS,OAAO,KAAK,QAAQ,SAAS,YAAY;CACpG;CAEA,MAAM,OAAO,MAAM,SAAS,KAAK;CACjC,IAAI,CAAC,KAAK,aACR,MAAM,IAAI,MAAM,uDAAuD,KAAK,UAAU,IAAI,GAAG;CAE/F,OAAO;EACL,QAAQ,KAAK;EAEb,SAAS,KAAK,gBAAgB,YAAY;EAC1C,SAAS,kBAAkB,KAAK,WAAW;CAC7C;AACF;;;;;;;;AASA,SAAgB,4BAAoD;CAClE,OAAO;EACL,IAAI;EACJ,MAAM;EACN,oBAAoB;EACpB,OAAO;EACP,cAAc;EACd,YAAW,gBAAe,YAAY;CACxC;AACF;;;AChKA,MAAM,gBAAgB;AAEtB,MAAM,0BACF;AAIJ,SAAgB,OAAO,QAAiC;CACtD,MAAM,eAAe,QAAQ,gBAAgB;CAI7C,MAAM,OAAO,aAAa;EACxB,MAAM;EACN,QAAQ,QAAQ,UAAU;EAC1B,SAAS;EACT;EACA,cAAc;GAAE,QAAQ;GAAM,mBAAmB;EAAM;EACvD,cAAc,QAAQ;CACxB,CAAC;CAED,OAAO;EACL,GAAG;EACH,MAAM;GAAE,GAAG,KAAK;GAAM;EAAa;EACnC,MAAM,OAAO,SAAS,WAA4B;GAGhD,MAAM,mBACJ;IACE,UAAU;IACV,YAAY;IACZ;IACA,cAAc;IACd,eAAc,WAAU,2EAA2E;GACrG,GACA,SACF;GAEA,MAAM,IAAI,MAAM,uBAAuB;EACzC;CACF;AACF;;;ACnEA,MAAMA,aAAW;AAwBjB,SAASC,YAAU,QAAgC;CAEjD,IAAI,OAAO,QAAQ,WAAW,YAAY,OAAO,OAAO,SAAS,GAC/D,OAAO,OAAO;CAEhB,IAAI,QAAQ,IAAI,iBACd,OAAO,QAAQ,IAAI;CAErB,MAAM,IAAI,MAAM,qFAAqF;AACvG;;;;;;;;AASA,MAAM,mBAAmB;;;;;;;;;AAUzB,SAAgB,qBACd,KACqC;CACrC,IAAI,CAAC,iBAAiB,KAAK,IAAI,KAAK,GAClC,OAAO,KAAA;CACT,IAAI,IAAI,aAAa,KAAA,KAAa,IAAI,aAAa,YACjD,OAAO,KAAA;CACT,OAAO,EAAE,kBAAkB,IAAI,aAAa,QAAQ,SAAS,OAAO;AACtE;;;;;;;;;;AAWA,SAAgB,QAAQ,QAAkC;CACxD,OAAO,aAAa;EAClB,MAAM;EACN,QAAQA,YAAU,MAAM;EACxB,SAAS,QAAQ,WAAWD;EAC5B,cAAc,QAAQ,gBAAgB;EACtC,cAAc,QAAQ,gBAAgB;GAAE,QAAQ;GAAM,mBAAmB;EAAM;EAC/E,sBAAsB;EACtB,cAAc,QAAQ;CACxB,CAAC;AACH;;;ACjFA,MAAM,WAAW;AAsBjB,SAAS,UAAU,QAA4B;CAE7C,IAAI,OAAO,QAAQ,WAAW,YAAY,OAAO,OAAO,SAAS,GAC/D,OAAO,OAAO;CAEhB,IAAI,QAAQ,IAAI,aACd,OAAO,QAAQ,IAAI;CAErB,MAAM,IAAI,MAAM,8EAA8E;AAChG;;;;;;;;;AAUA,SAAgB,iBACd,KACyB;CAEzB,OAAO,EAAE,UAAU,EAAE,MADL,CAAC,CAAC,IAAI,YAAY,IAAI,aAAa,QACd,YAAY,WAAW,EAAE;AAChE;;;;;;;;;AAUA,SAAgB,IAAI,QAA8B;CAChD,OAAO,aAAa;EAClB,MAAM;EACN,QAAQ,UAAU,MAAM;EACxB,SAAS,QAAQ,WAAW;EAC5B,cAAc,QAAQ,gBAAgB;EACtC,cAAc,QAAQ,gBAAgB;GAAE,QAAQ;GAAO,mBAAmB;EAAM;EAChF,mBAAmB;EACnB,eAAe;EACf,uBAAuB;EACvB,sBAAsB;EACtB,cAAc,QAAQ;CACxB,CAAC;AACH"}

package/dist/providers.d.ts

@@ -1,2 +1,2 @@
-import { $f as cerebras, Ap as readOAuthCredentials, Cp as resolvePromptCacheKey, Df as OpenAICompatParams, Dp as credentialsFromParams, Ef as OpenAICompatHttpError, Ep as credentialsFilePath, Jf as LocalParams, Kf as openai, Lf as mapOAIFinishReason, Mp as resolveOAuthApiKey, Np as writeOAuthCredentials, Op as extractRuntimeCredentials, Pf as classifyOpenAICompatError, Pp as OAuthProviderInterface, Qf as CerebrasParams, Sg as setModelCatalog, Sp as derivePromptCacheKey, Tf as OpenAICompatAuthHeader, Tp as ResolveOAuthApiKeyOptions, Uf as OpenAIParams, Xf as CursorParams, Yf as local, Zf as cursor, _f as sanitizeToolSchema, af as StreamCallbacks, bg as ModelCatalogSource, bp as PromptCacheKeyContext, cf as ToolCall, df as TurnResult, dp as applyAnthropicCacheBreakpoints, ep as BasetenParams, ff as XaiParams, gf as SchemaSanitizeResult, hf as SchemaSanitizeProfile, if as ProviderCapabilities, ip as arcee, jp as registerCustomOAuthProvider, kp as isOAuthAccessToken, lf as ToolResult, mf as SchemaSanitizeOptions, nf as HistoryRepair, np as planBasetenReasoning, of as StreamOptions, pf as xai, rf as Provider, rp as ArceeParams, sf as TokenCountPayload, sp as AnthropicParams, tp as baseten, uf as ToolSpec, up as anthropic, vf as sanitizeToolSpecs, vg as ModelCatalog, wp as OAuthParams, xf as openrouter, xg as getModelCatalog, xp as PromptCacheKeyOption, yf as OpenRouterParams, yg as ModelCatalogEntry, zf as openaiCompat } from "./index-Gy9wi80M.js";
-export { type AnthropicParams, type ArceeParams, type BasetenParams, type CerebrasParams, type CursorParams, HistoryRepair, type LocalParams, type ModelCatalog, type ModelCatalogEntry, type ModelCatalogSource, type OAuthParams, type OAuthProviderInterface, type OpenAICompatAuthHeader, OpenAICompatHttpError, type OpenAICompatParams, type OpenAIParams, type OpenRouterParams, type PromptCacheKeyContext, type PromptCacheKeyOption, Provider, ProviderCapabilities, type ResolveOAuthApiKeyOptions, type SchemaSanitizeOptions, type SchemaSanitizeProfile, type SchemaSanitizeResult, StreamCallbacks, StreamOptions, TokenCountPayload, ToolCall, ToolResult, ToolSpec, TurnResult, type XaiParams, anthropic, applyAnthropicCacheBreakpoints, arcee, baseten, cerebras, classifyOpenAICompatError, credentialsFilePath, credentialsFromParams, cursor, derivePromptCacheKey, extractRuntimeCredentials, getModelCatalog, isOAuthAccessToken, local, mapOAIFinishReason, openai, openaiCompat, openrouter, planBasetenReasoning, readOAuthCredentials, registerCustomOAuthProvider, resolveOAuthApiKey, resolvePromptCacheKey, sanitizeToolSchema, sanitizeToolSpecs, setModelCatalog, writeOAuthCredentials, xai };
+import { $f as planMistralReasoning, Af as OpenAICompatParams, Ap as OAuthParams, Bf as mapOAIFinishReason, Bp as OAuthProviderInterface, Dp as PromptCacheKeyOption, Ep as PromptCacheKeyContext, Fg as ModelCatalogSource, Fp as isOAuthAccessToken, Hf as openaiCompat, Ig as getModelCatalog, Ip as readOAuthCredentials, Kf as OpenAIParams, Lf as classifyOpenAICompatError, Lg as setModelCatalog, Lp as registerCustomOAuthProvider, Mp as credentialsFilePath, Ng as ModelCatalog, Np as credentialsFromParams, Of as OpenAICompatAuthHeader, Op as derivePromptCacheKey, Pg as ModelCatalogEntry, Pp as extractRuntimeCredentials, Qf as mistral, Rp as resolveOAuthApiKey, Sf as OpenRouterParams, Yf as openai, Zf as MistralParams, _f as SchemaSanitizeOptions, _p as applyAnthropicCacheBreakpoints, af as StreamCallbacks, ap as cerebras, bf as sanitizeToolSchema, cf as ToolCall, cp as planBasetenReasoning, df as TurnResult, ep as LocalParams, ff as ZaiParams, gf as xai, gp as anthropic, hf as XaiParams, if as ProviderCapabilities, ip as CerebrasParams, jp as ResolveOAuthApiKeyOptions, kf as OpenAICompatHttpError, kp as resolvePromptCacheKey, lf as ToolResult, lp as ArceeParams, mf as zai, nf as HistoryRepair, np as CursorParams, of as StreamOptions, op as BasetenParams, pf as planZaiReasoning, pp as AnthropicParams, rf as Provider, rp as cursor, sf as TokenCountPayload, sp as baseten, tp as local, uf as ToolSpec, up as arcee, vf as SchemaSanitizeProfile, wf as openrouter, xf as sanitizeToolSpecs, yf as SchemaSanitizeResult, zp as writeOAuthCredentials } from "./index-C-Nd2ZbV.js";
+export { type AnthropicParams, type ArceeParams, type BasetenParams, type CerebrasParams, type CursorParams, HistoryRepair, type LocalParams, type MistralParams, type ModelCatalog, type ModelCatalogEntry, type ModelCatalogSource, type OAuthParams, type OAuthProviderInterface, type OpenAICompatAuthHeader, OpenAICompatHttpError, type OpenAICompatParams, type OpenAIParams, type OpenRouterParams, type PromptCacheKeyContext, type PromptCacheKeyOption, Provider, ProviderCapabilities, type ResolveOAuthApiKeyOptions, type SchemaSanitizeOptions, type SchemaSanitizeProfile, type SchemaSanitizeResult, StreamCallbacks, StreamOptions, TokenCountPayload, ToolCall, ToolResult, ToolSpec, TurnResult, type XaiParams, type ZaiParams, anthropic, applyAnthropicCacheBreakpoints, arcee, baseten, cerebras, classifyOpenAICompatError, credentialsFilePath, credentialsFromParams, cursor, derivePromptCacheKey, extractRuntimeCredentials, getModelCatalog, isOAuthAccessToken, local, mapOAIFinishReason, mistral, openai, openaiCompat, openrouter, planBasetenReasoning, planMistralReasoning, planZaiReasoning, readOAuthCredentials, registerCustomOAuthProvider, resolveOAuthApiKey, resolvePromptCacheKey, sanitizeToolSchema, sanitizeToolSpecs, setModelCatalog, writeOAuthCredentials, xai, zai };

package/dist/providers.js

@@ -1,12 +1,12 @@
-import { E as getModelCatalog, S as resolvePromptCacheKey, b as sanitizeToolSpecs, f as mapOAIFinishReason, j as setModelCatalog, l as classifyOpenAICompatError, m as openaiCompat, n as OpenAICompatHttpError, x as derivePromptCacheKey, y as sanitizeToolSchema } from "./openai-compat-0MJq4-g9.js";
-import { n as anthropic, r as applyAnthropicCacheBreakpoints } from "./anthropic-BJhMEi7x.js";
+import { E as getModelCatalog, S as resolvePromptCacheKey, b as sanitizeToolSpecs, f as mapOAIFinishReason, j as setModelCatalog, l as classifyOpenAICompatError, m as openaiCompat, n as OpenAICompatHttpError, x as derivePromptCacheKey, y as sanitizeToolSchema } from "./openai-compat-B3c_hzwf.js";
+import { n as anthropic, r as applyAnthropicCacheBreakpoints } from "./anthropic-D3lkw5mv.js";
 import { a as readOAuthCredentials, c as writeOAuthCredentials, i as isOAuthAccessToken, n as credentialsFromParams, o as registerCustomOAuthProvider, r as extractRuntimeCredentials, s as resolveOAuthApiKey, t as credentialsFilePath } from "./oauth-bkEOybqm.js";
 import { arcee } from "./providers/arcee.js";
 import { baseten, planBasetenReasoning } from "./providers/baseten.js";
 import { cerebras } from "./providers/cerebras.js";
-import { t as xai } from "./xai-CpPbCGWO.js";
-import { t as cursor } from "./providers-BSrJx0Tn.js";
+import { t as xai } from "./xai-BPwMQSuF.js";
+import { a as cursor, i as planMistralReasoning, n as zai, r as mistral, t as planZaiReasoning } from "./providers-z1vU0b2Q.js";
 import { local } from "./providers/local.js";
 import { openai } from "./providers/openai.js";
 import { openrouter } from "./providers/openrouter.js";
-export { OpenAICompatHttpError, anthropic, applyAnthropicCacheBreakpoints, arcee, baseten, cerebras, classifyOpenAICompatError, credentialsFilePath, credentialsFromParams, cursor, derivePromptCacheKey, extractRuntimeCredentials, getModelCatalog, isOAuthAccessToken, local, mapOAIFinishReason, openai, openaiCompat, openrouter, planBasetenReasoning, readOAuthCredentials, registerCustomOAuthProvider, resolveOAuthApiKey, resolvePromptCacheKey, sanitizeToolSchema, sanitizeToolSpecs, setModelCatalog, writeOAuthCredentials, xai };
+export { OpenAICompatHttpError, anthropic, applyAnthropicCacheBreakpoints, arcee, baseten, cerebras, classifyOpenAICompatError, credentialsFilePath, credentialsFromParams, cursor, derivePromptCacheKey, extractRuntimeCredentials, getModelCatalog, isOAuthAccessToken, local, mapOAIFinishReason, mistral, openai, openaiCompat, openrouter, planBasetenReasoning, planMistralReasoning, planZaiReasoning, readOAuthCredentials, registerCustomOAuthProvider, resolveOAuthApiKey, resolvePromptCacheKey, sanitizeToolSchema, sanitizeToolSpecs, setModelCatalog, writeOAuthCredentials, xai, zai };

package/dist/restate.d.ts

@@ -1,4 +1,4 @@
-import { Fp as AgentBehavior, Ip as AgentClock, Km as ToolContext, Nm as ToolResultContent, Xi as McpToolWrap, Yi as McpToolMeta, gh as SessionStore, km as ToolBatchExecutor, of as StreamOptions, qm as ToolDef, rf as Provider } from "./index-Gy9wi80M.js";
+import { $m as ToolDef, Fm as ToolBatchExecutor, Hp as AgentClock, Qm as ToolContext, Vp as AgentBehavior, Xi as McpToolWrap, Yi as McpToolMeta, jh as SessionStore, mh as ToolStateStore, of as StreamOptions, rf as Provider, zm as ToolResultContent } from "./index-C-Nd2ZbV.js";
 import { g as TaskExitInfo, h as TaskEntry, l as ExecutionHandle, o as DetachedTasksCapability } from "./types-M2DauK2D.js";
 
 //#region src/restate/types.d.ts
@@ -838,6 +838,56 @@ declare function wrapAgentTools(tools: Record<string, ToolDef>, ctx: RestateCont
   exclude?: readonly string[];
 }): Record<string, ToolDef>;
 //#endregion
+//#region src/restate/tool-state.d.ts
+/** Default virtual-object state key holding the tool-state map. */
+declare const DEFAULT_TOOL_STATE_KEY = "zidane_tool_state";
+interface RestateToolStateStoreOptions {
+  /** Override the state key. Defaults to {@link DEFAULT_TOOL_STATE_KEY}. */
+  key?: string;
+  /**
+   * Per-value size budget enforced when a tool stages a write. Defaults to
+   * the library's `DEFAULT_MAX_VALUE_BYTES` (64 KiB), which is far below
+   * Restate's own journal-entry ceiling on purpose: an oversized state value
+   * doesn't fail politely there — the invocation fails with `RT0003`
+   * (MessageSizeLimit) and is retried under the configured policy, i.e. a fat
+   * write becomes a poison pill rather than a dropped one.
+   */
+  maxValueBytes?: number;
+}
+/**
+ * Build a {@link ToolStateStore} over the bound virtual object's state.
+ *
+ * ```ts
+ * const agent = createAgent({
+ *   provider,
+ *   tools: wrapAgentTools({ shell }, ctx),
+ *   session: createSession({ id: ctx.key, store: restateSessionStore(ctx) }),
+ *   toolState: restateToolStateStore(ctx),
+ * })
+ * ```
+ *
+ * Single-writer like the session store: the state belongs to the object key,
+ * so bind one key per conversation and one agent per invocation.
+ *
+ * Build it INSIDE the handler. The returned store closes over this `ctx`, so
+ * an agent constructed once at module scope and reused across invocations
+ * would commit through a context that is already finished, and would never
+ * re-hydrate (the controller memoizes the load for the life of the agent).
+ * Same rule as `restateSessionStore` / `restateProvider` / `restateClock` —
+ * but easier to get wrong here, because `toolState` looks like
+ * agent-lifetime configuration rather than a per-invocation binding.
+ *
+ * The map is a schema you own: one JSON object, no versioning, no migration.
+ * A value written by an earlier deploy is still there after you change its
+ * shape, so either read tolerantly or move to a fresh `key`.
+ *
+ * Enabling this adds a `ctx.get` at run start, which changes the journal
+ * shape for NEW invocations. Existing in-flight invocations replay against
+ * the old shape — drain them (or accept the failed attempts) before rolling
+ * it out.
+ */
+declare function restateToolStateStore(ctx: RestateObjectContextLike, options?: RestateToolStateStoreOptions): ToolStateStore;
+//#endregion
 //#region src/restate/wait-background.d.ts
 interface RestateWaitBackgroundOptions {
   /** The Restate context the awakeable + journal entries bind to. */
@@ -891,5 +941,5 @@ declare function restateWaitBackground(options: RestateWaitBackgroundOptions): (
   signal?: AbortSignal;
 }) => Promise<TaskExitInfo | null>;
 //#endregion
-export { type AwakeableWithTimeoutOptions, type ReplayDoneHooks, type ReplayTracker, type ReplayTrackerOptions, type RestateAwakeable, type RestateAwakeableToolOptions, type RestateClockOptions, type RestateContextLike, type RestateDurableStart, type RestateDurableWait, type RestateMcpToolWrapOptions, type RestateObjectContextLike, type RestatePromiseAllLike, type RestatePromiseLike, type RestatePromiseRaceLike, type RestateProviderOptions, type RestateRunOptions, type RestateSessionStoreOptions, type RestateToolBatchExecutorOptions, type RestateToolOptions, type RestateWaitBackgroundOptions, awakeableWithTimeout, createReplayTracker, deterministicUuid, replayAwareAbort, restateAwakeableTool, restateBehavior, restateClock, restateMcpToolWrap, restateProvider, restateSessionStore, restateTool, restateToolBatchExecutor, restateWaitBackground, shouldRethrowRestateControlError, wrapAgentTools };
+export { type AwakeableWithTimeoutOptions, DEFAULT_TOOL_STATE_KEY, type ReplayDoneHooks, type ReplayTracker, type ReplayTrackerOptions, type RestateAwakeable, type RestateAwakeableToolOptions, type RestateClockOptions, type RestateContextLike, type RestateDurableStart, type RestateDurableWait, type RestateMcpToolWrapOptions, type RestateObjectContextLike, type RestatePromiseAllLike, type RestatePromiseLike, type RestatePromiseRaceLike, type RestateProviderOptions, type RestateRunOptions, type RestateSessionStoreOptions, type RestateToolBatchExecutorOptions, type RestateToolOptions, type RestateToolStateStoreOptions, type RestateWaitBackgroundOptions, awakeableWithTimeout, createReplayTracker, deterministicUuid, replayAwareAbort, restateAwakeableTool, restateBehavior, restateClock, restateMcpToolWrap, restateProvider, restateSessionStore, restateTool, restateToolBatchExecutor, restateToolStateStore, restateWaitBackground, shouldRethrowRestateControlError, wrapAgentTools };
 //# sourceMappingURL=restate.d.ts.map

package/dist/restate.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"restate.d.ts","names":[],"sources":["../src/restate/types.ts","../src/restate/awakeable.ts","../src/restate/awakeable-tool.ts","../src/restate/behavior.ts","../src/restate/clock.ts","../src/restate/errors.ts","../src/restate/replay.ts","../src/restate/mcp.ts","../src/restate/parallel.ts","../src/restate/provider.ts","../src/restate/session.ts","../src/restate/tool.ts","../src/restate/wait-background.ts"],"mappings":";;;;;;;;AAmBA;;;;;;;;;;;AAYO;AAUP;;UAtBiB,iBAAA;EAwBC;EAtBhB,gBAAA;EAqBA;EAnBA,gBAAA;EAoBS;EAlBT,oBAAA;EAkBkB;EAhBlB,gBAAA;EAuCe;EArCf,mBAAA;EAqCiC;EAnCjC,KAAA;AAAA;;;;;;;;UAUe,gBAAA;EACf,EAAA;EACA,OAAA,EAAS,OAAO,CAAC,CAAA;AAAA;;;;;;;;;;;;;;;;;;;;;UAuBF,kBAAA;EACf,GAAA,MACE,IAAA,UACA,EAAA,QAAU,OAAA,CAAQ,CAAA,IAAK,CAAA,EACvB,OAAA,GAAU,iBAAA,KACP,OAAA,CAAQ,CAAA;EACb,IAAA;IAAQ,GAAA,QAAW,OAAA;EAAA;EACnB,IAAA;IAAQ,MAAA;EAAA;EACR,SAAA,WAAoB,gBAAA,CAAiB,CAAA;EA0BP;;;;;;;;;;;;EAb9B,KAAA,GAAQ,MAAA,aAAmB,OAAA;AAAA;;;;;;AAcP;;UAJL,wBAAA,SAAiC,kBAAA;EAChD,GAAA;EACA,GAAA,MAAS,IAAA,aAAiB,OAAA,CAAQ,CAAA;EAClC,GAAA,MAAS,IAAA,UAAc,KAAA,EAAO,CAAA;EAC9B,KAAA,GAAQ,IAAA;AAAA;;;;;AAtEH;AAUP;;;;;UClBiB,sBAAA,kBAAwC,OAAA,YAAmB,OAAA;EAC1E,IAAA,GAAO,MAAA,WAAiB,QAAA,OAAe,OAAA;AAAA;AAAA,UAGxB,2BAAA,kBAA6C,OAAA,YAAmB,OAAA;EDgB7D;AAAA;AAuBpB;;EClCE,SAAA;EDqCoB;;;;;;;EC7BpB,QAAA,GAAW,sBAAA,CAAuB,QAAA;AAAA;;;;;;;;;;;;;;;;;iBAmBpB,oBAAA,qBAAyC,OAAA,YAAmB,OAAA,WAC1E,GAAA,EAAK,kBAAA,EACL,OAAA,GAAS,2BAAA,CAA4B,QAAA,IACpC,gBAAA,CAAiB,CAAA;;;UC5BH,2BAAA;EFgCA;EE9Bf,IAAA;EF8BiC;EE5BjC,WAAA;EF+BY;EE7BZ,WAAA,EAAa,MAAA;EF8BD;EE5BZ,GAAA,EAAK,kBAAA;EF6BA;;;;;EEvBL,MAAA,GAAS,IAAA;IACP,WAAA;IACA,KAAA,EAAO,MAAA;IACP,OAAA,EAAS,WAAA;EAAA,aACE,OAAA;EFiBD;;;;EEZZ,QAAA,IAAY,KAAA,EAAO,CAAA,EAAG,KAAA,EAAO,MAAA,+BAAqC,iBAAA;EFahE;;;;EERF,UAAA,GAAa,iBAAA;EFUM;;;;EELnB,iBAAA;AAAA;;;;;;AFoBkC;AAUpC;;iBEnBgB,oBAAA,cACd,OAAA,EAAS,2BAAA,CAA4B,CAAA,IACpC,OAAA;;;;;;AF7DH;;;;;;;;;;;AAYO;AAUP;;;;;;;;;;AAEoB;AAuBpB;;;;;;;iBG9BgB,eAAA,CAAgB,SAAA,GAAW,aAAA,GAAqB,aAAa;;;;;AHmDzC;AAUpC;;iBItBgB,iBAAA,CAAkB,IAAY;AAAA,UAQ7B,mBAAA;EJgBW;;;;;;;;;;;;;;EID1B,QAAA;IAAuB,aAAA;EAAA;EJGvB;;;AAAoB;;;;AC9EtB;EGoFE,MAAA;AAAA;;;;;;;;;;;;;iBAec,YAAA,CAAa,GAAA,EAAK,kBAAA,EAAoB,OAAA,GAAS,mBAAA,GAA2B,UAAA;;;;;;;AJvG1F;;;iBKZgB,gCAAA,CAAiC,KAAc;;;;;;;ALY/D;;;;;;;;;;;AAYO;AAUP;;;;;;;;;;AAEoB;AAuBpB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAqBoC;AAUpC;;;UM1BiB,eAAA;EACf,QAAA,GAAW,IAAA,iBAAqB,GAAA,EAAK,MAAM;AAAA;AAAA,UAG5B,oBAAA;ENsBmD;;;;;;;EMdlE,KAAA,GAAQ,eAAe;ENgBW;;;;;;;;;AAEd;;;;AC9EtB;;;EK6EE,QAAA;AAAA;AAAA,UAGe,aAAA;EL/EwB;;;;;;EKsFvC,WAAA;ELtFwB;;;;AAAsB;AAGhD;;;;;;EK+FE,UAAA;ELlFiC;;;;;;EKyFjC,MAAA,GAAS,EAAA;ELzFE;;;AAA+B;AAmB5C;EK4EE,QAAA;AAAA;;;;;;iBAQc,mBAAA,CAAoB,OAAA,GAAS,oBAAA,GAA4B,aAAa;;;;;;;;;;;;iBAwDtE,gBAAA,CAAiB,OAAA,EAAS,aAAa,EAAE,KAAA;;;UClIxC,yBAAA;EPkBmB;AAUpC;;;;;EOrBE,aAAA,GAAgB,aAAA;EPqBgC;;;;;;;EObhD,UAAA,GAAa,iBAAA;EPea;;;;;EOT1B,SAAA,IAAa,GAAA,UAAa,IAAA,UAAc,KAAA,EAAO,MAAA,mBAAyB,MAAA;EPUjD;;;;AACH;;EOJpB,OAAA;;AN1EF;;;;;;;EMmFE,QAAA,IAAY,IAAA,EAAM,OAAA,EAAS,IAAA,EAAM,WAAA,KAAgB,OAAA;AAAA;;;;;;;iBAkCnC,kBAAA,CACd,GAAA,EAAK,kBAAA,EACL,OAAA,GAAS,yBAAA,GACR,WAAA;;;UC5Ic,qBAAA,kBAAuC,OAAA,YAAmB,kBAAA;EACzE,GAAA,GAAM,MAAA,WAAiB,QAAA,OAAe,OAAA;AAAA;AAAA,KAG5B,kBAAA,MAAwB,OAAA,CAAQ,CAAA;EAC1C,GAAA,MAAS,MAAA,GAAS,KAAA,GAAQ,CAAA,EAAG,OAAA,eAAsB,CAAA,KAAM,OAAA,CAAQ,CAAA;AAAA;AAAA,UAGlD,+BAAA,kBAAiD,OAAA,YAAmB,kBAAA;ERUnF;;;;EQLA,QAAA,EAAU,qBAAA,CAAsB,QAAA;AAAA;;ARe3B;AAUP;;;;;iBQfgB,wBAAA,kBAA0C,OAAA,YAAmB,kBAAA,WAC3E,OAAA,EAAS,+BAAA,CAAgC,QAAA,IACxC,iBAAA;;;UCFc,sBAAA;ETiBG;AAAA;AAuBpB;;;;;;;;;;;;;;;;;;;ESlBE,UAAA,GAAa,iBAAA;ETqBY;;;;;;;ESbzB,SAAA,IAAa,GAAA,UAAa,IAAA,EAAM,aAAA;ETgBb;;;;;;;ESRnB,eAAA;ETuBQ;;;AAA0B;AAUpC;;;ESzBE,aAAA,GAAgB,aAAA;AAAA;;;;;;;iBASF,eAAA,CACd,KAAA,EAAO,QAAA,EACP,GAAA,EAAK,kBAAA,EACL,OAAA,GAAS,sBAAA,GACR,QAAA;;;UCVc,0BAAA;EVsBA;;;;;;;;;;;;;EURf,KAAA,GAAQ,UAAU;AAAA;;;;;;;;;iBAWJ,mBAAA,CACd,GAAA,EAAK,wBAAA,EACL,OAAA,GAAS,0BAAA,GACR,YAAA;;;;;;;;;;;;;;;;;;;;;;;UC5Cc,mBAAA;EXWb;EWTF,KAAA,GAAQ,KAAA,EAAO,MAAA,mBAAyB,OAAA,EAAS,WAAA,KAAgB,OAAA,CAAQ,KAAA,IAAS,KAAA;EXUrE;;;;;;;;;;;;EWGb,MAAA,IAAU,KAAA,EAAO,KAAA,EAAO,KAAA,EAAO,MAAA,mBAAyB,OAAA,EAAS,WAAA,KAAgB,OAAA,UAAiB,iBAAA;EXahE;AAAA;AAUpC;;;;;;;;;;;;;;;;;EWHE,IAAA,GAAO,kBAAA,CAAmB,KAAA,EAAO,OAAA;EXMxB;;;;;;AACW;;;;AC9EtB;;;;;;;;;;EU4FE,OAAA,IAAW,KAAA,EAAO,KAAA,EAAO,KAAA,EAAO,MAAA,mBAAyB,OAAA,EAAS,WAAA,KAAgB,OAAA;AAAA;;;;;;AV3FpC;AAGhD;;;;;UUsGiB,kBAAA;EVzFJ;;;;;;EUgGX,QAAA,GAAW,KAAA,EAAO,KAAA,EAAO,WAAA,UAAqB,KAAA,EAAO,MAAA,mBAAyB,OAAA,EAAS,WAAA,KAAgB,OAAA;EVhGvG;;;;AAA0C;AAmB5C;EUoFE,MAAA,GAAS,KAAA,EAAO,KAAA,EAAO,OAAA,EAAS,OAAA,SAAgB,KAAA,EAAO,MAAA,mBAAyB,OAAA,EAAS,WAAA,KAAgB,OAAA,UAAiB,iBAAA,eAAgC,iBAAA;EVpFxH;;;;;;;EU4FlC,SAAA,cAAuB,KAAA,EAAO,MAAA,mBAAyB,OAAA,EAAS,WAAA;EVzF/C;;;;;EU+FjB,QAAA,GAAW,sBAAA;EVjGN;;;;;;;;AAEc;;;;AC5BrB;;;;;;ES8IE,MAAA,IAAU,KAAA,EAAO,KAAA,EAAO,KAAA,EAAO,MAAA,mBAAyB,OAAA,EAAS,WAAA;AAAA;AAAA,UASlD,kBAAA;EThIc;;;;;;;ESwI7B,UAAA,GAAa,iBAAA;ETzJb;;;;;;;;ESkKA,SAAA,IAAa,GAAA,UAAa,IAAA,UAAc,KAAA,EAAO,MAAA,mBAAyB,MAAA;ETvJ7D;;;;;;;ES+JX,sBAAA;ETzJkE;;;;;AAUjD;AAWnB;;;;ES+IE,aAAA,GAAgB,uBAAA;ET7If;;;;;ESmJD,YAAA,GAAe,mBAAA,CAAoB,KAAA,EAAO,OAAA;ETpJ1C;;;AACQ;;;;AC5CV;EQwME,aAAA,GAAgB,aAAA;;;;;;;ARxM2D;;;;ACuC7E;;;;AAA8C;AAQ9C;;;;;EO8KE,UAAA;EPtJA;;AAAM;AAeR;;;;;;;EOkJE,kBAAA;AAAA;;;;;iBAsBc,WAAA,qCACd,KAAA,EAAO,OAAA,EACP,GAAA,EAAK,kBAAA,EACL,OAAA,GAAS,kBAAA,CAAmB,KAAA,EAAO,OAAA,IAClC,OAAA;AP5KiG;;;;ACnHpG;;;;ADmHoG,iBOkhBpF,cAAA,CACd,KAAA,EAAO,MAAA,SAAe,OAAA,GACtB,GAAA,EAAK,kBAAA,EACL,IAAA,GAAM,IAAA,CAAK,kBAAA;EAAwC,OAAA;AAAA,IAClD,MAAA,SAAe,OAAA;;;UCnmBD,4BAAA;EZ6BsB;EY3BrC,GAAA,EAAK,kBAAA;EZwCsB;;;;;;EYjC3B,KAAA,GAAQ,MAAA,EAAQ,eAAA,EAAiB,MAAA,aAAmB,OAAA,UAAiB,SAAA,eAAwB,SAAA;EZezE;;;;;;EYRpB,QAAA,GAAW,WAAA,UAAqB,MAAA,UAAgB,MAAA,EAAQ,eAAA,KAAoB,OAAA;EZW5E;;;;;EYLA,MAAA,IAAU,WAAA,UAAqB,MAAA,UAAgB,IAAA,EAAM,YAAA,SAAqB,MAAA,EAAQ,eAAA,KAAoB,OAAA;EZO1F;;;;;EYDZ,SAAA;EZckC;AAAA;AAUpC;;;EYlBE,QAAA,GAAW,sBAAA;EZoBe;EYlB1B,UAAA,GAAa,iBAAA;EZgBmC;EYdhD,WAAA;EZckE;EYZlE,aAAA,GAAgB,aAAA;AAAA;;;;;;iBAiDF,qBAAA,CACd,OAAA,EAAS,4BAAA,IACP,MAAA,EAAQ,eAAA,EAAiB,MAAA,UAAgB,IAAA;EAAS,SAAA;EAAoB,MAAA,GAAS,WAAA;AAAA,MAAkB,OAAA,CAAQ,YAAA"}
+{"version":3,"file":"restate.d.ts","names":[],"sources":["../src/restate/types.ts","../src/restate/awakeable.ts","../src/restate/awakeable-tool.ts","../src/restate/behavior.ts","../src/restate/clock.ts","../src/restate/errors.ts","../src/restate/replay.ts","../src/restate/mcp.ts","../src/restate/parallel.ts","../src/restate/provider.ts","../src/restate/session.ts","../src/restate/tool.ts","../src/restate/tool-state.ts","../src/restate/wait-background.ts"],"mappings":";;;;;;;;AAmBA;;;;;;;;;;;AAYO;AAUP;;UAtBiB,iBAAA;EAwBC;EAtBhB,gBAAA;EAqBA;EAnBA,gBAAA;EAoBS;EAlBT,oBAAA;EAkBkB;EAhBlB,gBAAA;EAuCe;EArCf,mBAAA;EAqCiC;EAnCjC,KAAA;AAAA;;;;;;;;UAUe,gBAAA;EACf,EAAA;EACA,OAAA,EAAS,OAAO,CAAC,CAAA;AAAA;;;;;;;;;;;;;;;;;;;;;UAuBF,kBAAA;EACf,GAAA,MACE,IAAA,UACA,EAAA,QAAU,OAAA,CAAQ,CAAA,IAAK,CAAA,EACvB,OAAA,GAAU,iBAAA,KACP,OAAA,CAAQ,CAAA;EACb,IAAA;IAAQ,GAAA,QAAW,OAAA;EAAA;EACnB,IAAA;IAAQ,MAAA;EAAA;EACR,SAAA,WAAoB,gBAAA,CAAiB,CAAA;EA0BP;;;;;;;;;;;;EAb9B,KAAA,GAAQ,MAAA,aAAmB,OAAA;AAAA;;;;;;AAcP;;UAJL,wBAAA,SAAiC,kBAAA;EAChD,GAAA;EACA,GAAA,MAAS,IAAA,aAAiB,OAAA,CAAQ,CAAA;EAClC,GAAA,MAAS,IAAA,UAAc,KAAA,EAAO,CAAA;EAC9B,KAAA,GAAQ,IAAA;AAAA;;;;;AAtEH;AAUP;;;;;UClBiB,sBAAA,kBAAwC,OAAA,YAAmB,OAAA;EAC1E,IAAA,GAAO,MAAA,WAAiB,QAAA,OAAe,OAAA;AAAA;AAAA,UAGxB,2BAAA,kBAA6C,OAAA,YAAmB,OAAA;EDgB7D;AAAA;AAuBpB;;EClCE,SAAA;EDqCoB;;;;;;;EC7BpB,QAAA,GAAW,sBAAA,CAAuB,QAAA;AAAA;;;;;;;;;;;;;;;;;iBAmBpB,oBAAA,qBAAyC,OAAA,YAAmB,OAAA,WAC1E,GAAA,EAAK,kBAAA,EACL,OAAA,GAAS,2BAAA,CAA4B,QAAA,IACpC,gBAAA,CAAiB,CAAA;;;UC5BH,2BAAA;EFgCA;EE9Bf,IAAA;EF8BiC;EE5BjC,WAAA;EF+BY;EE7BZ,WAAA,EAAa,MAAA;EF8BD;EE5BZ,GAAA,EAAK,kBAAA;EF6BA;;;;;EEvBL,MAAA,GAAS,IAAA;IACP,WAAA;IACA,KAAA,EAAO,MAAA;IACP,OAAA,EAAS,WAAA;EAAA,aACE,OAAA;EFiBD;;;;EEZZ,QAAA,IAAY,KAAA,EAAO,CAAA,EAAG,KAAA,EAAO,MAAA,+BAAqC,iBAAA;EFahE;;;;EERF,UAAA,GAAa,iBAAA;EFUM;;;;EELnB,iBAAA;AAAA;;;;;;AFoBkC;AAUpC;;iBEnBgB,oBAAA,cACd,OAAA,EAAS,2BAAA,CAA4B,CAAA,IACpC,OAAA;;;;;;AF7DH;;;;;;;;;;;AAYO;AAUP;;;;;;;;;;AAEoB;AAuBpB;;;;;;;iBG9BgB,eAAA,CAAgB,SAAA,GAAW,aAAA,GAAqB,aAAa;;;;;AHmDzC;AAUpC;;iBItBgB,iBAAA,CAAkB,IAAY;AAAA,UAQ7B,mBAAA;EJgBW;;;;;;;;;;;;;;EID1B,QAAA;IAAuB,aAAA;EAAA;EJGvB;;;AAAoB;;;;AC9EtB;EGoFE,MAAA;AAAA;;;;;;;;;;;;;iBAec,YAAA,CAAa,GAAA,EAAK,kBAAA,EAAoB,OAAA,GAAS,mBAAA,GAA2B,UAAA;;;;;;;AJvG1F;;;iBKZgB,gCAAA,CAAiC,KAAc;;;;;;;ALY/D;;;;;;;;;;;AAYO;AAUP;;;;;;;;;;AAEoB;AAuBpB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAqBoC;AAUpC;;;UM1BiB,eAAA;EACf,QAAA,GAAW,IAAA,iBAAqB,GAAA,EAAK,MAAM;AAAA;AAAA,UAG5B,oBAAA;ENsBmD;;;;;;;EMdlE,KAAA,GAAQ,eAAe;ENgBW;;;;;;;;;AAEd;;;;AC9EtB;;;EK6EE,QAAA;AAAA;AAAA,UAGe,aAAA;EL/EwB;;;;;;EKsFvC,WAAA;ELtFwB;;;;AAAsB;AAGhD;;;;;;EK+FE,UAAA;ELlFiC;;;;;;EKyFjC,MAAA,GAAS,EAAA;ELzFE;;;AAA+B;AAmB5C;EK4EE,QAAA;AAAA;;;;;;iBAQc,mBAAA,CAAoB,OAAA,GAAS,oBAAA,GAA4B,aAAa;;;;;;;;;;;;iBAwDtE,gBAAA,CAAiB,OAAA,EAAS,aAAa,EAAE,KAAA;;;UClIxC,yBAAA;EPkBmB;AAUpC;;;;;EOrBE,aAAA,GAAgB,aAAA;EPqBgC;;;;;;;EObhD,UAAA,GAAa,iBAAA;EPea;;;;;EOT1B,SAAA,IAAa,GAAA,UAAa,IAAA,UAAc,KAAA,EAAO,MAAA,mBAAyB,MAAA;EPUjD;;;;AACH;;EOJpB,OAAA;;AN1EF;;;;;;;EMmFE,QAAA,IAAY,IAAA,EAAM,OAAA,EAAS,IAAA,EAAM,WAAA,KAAgB,OAAA;AAAA;;;;;;;iBAkCnC,kBAAA,CACd,GAAA,EAAK,kBAAA,EACL,OAAA,GAAS,yBAAA,GACR,WAAA;;;UC5Ic,qBAAA,kBAAuC,OAAA,YAAmB,kBAAA;EACzE,GAAA,GAAM,MAAA,WAAiB,QAAA,OAAe,OAAA;AAAA;AAAA,KAG5B,kBAAA,MAAwB,OAAA,CAAQ,CAAA;EAC1C,GAAA,MAAS,MAAA,GAAS,KAAA,GAAQ,CAAA,EAAG,OAAA,eAAsB,CAAA,KAAM,OAAA,CAAQ,CAAA;AAAA;AAAA,UAGlD,+BAAA,kBAAiD,OAAA,YAAmB,kBAAA;ERUnF;;;;EQLA,QAAA,EAAU,qBAAA,CAAsB,QAAA;AAAA;;ARe3B;AAUP;;;;;iBQfgB,wBAAA,kBAA0C,OAAA,YAAmB,kBAAA,WAC3E,OAAA,EAAS,+BAAA,CAAgC,QAAA,IACxC,iBAAA;;;UCFc,sBAAA;ETiBG;AAAA;AAuBpB;;;;;;;;;;;;;;;;;;;ESlBE,UAAA,GAAa,iBAAA;ETqBY;;;;;;;ESbzB,SAAA,IAAa,GAAA,UAAa,IAAA,EAAM,aAAA;ETgBb;;;;;;;ESRnB,eAAA;ETuBQ;;;AAA0B;AAUpC;;;ESzBE,aAAA,GAAgB,aAAA;AAAA;;;;;;;iBASF,eAAA,CACd,KAAA,EAAO,QAAA,EACP,GAAA,EAAK,kBAAA,EACL,OAAA,GAAS,sBAAA,GACR,QAAA;;;UCVc,0BAAA;EVsBA;;;;;;;;;;;;;EURf,KAAA,GAAQ,UAAU;AAAA;;;;;;;;;iBAWJ,mBAAA,CACd,GAAA,EAAK,wBAAA,EACL,OAAA,GAAS,0BAAA,GACR,YAAA;;;;;;;;;;;;;;;;;;;;;;;UC1Cc,mBAAA;EXSb;EWPF,KAAA,GAAQ,KAAA,EAAO,MAAA,mBAAyB,OAAA,EAAS,WAAA,KAAgB,OAAA,CAAQ,KAAA,IAAS,KAAA;EXQrE;;;;;;;;;;;;EWKb,MAAA,IAAU,KAAA,EAAO,KAAA,EAAO,KAAA,EAAO,MAAA,mBAAyB,OAAA,EAAS,WAAA,KAAgB,OAAA,UAAiB,iBAAA;EXWhE;AAAA;AAUpC;;;;;;;;;;;;;;;;;EWDE,IAAA,GAAO,kBAAA,CAAmB,KAAA,EAAO,OAAA;EXIxB;;;;;;AACW;;;;AC9EtB;;;;;;;;;;EU8FE,OAAA,IAAW,KAAA,EAAO,KAAA,EAAO,KAAA,EAAO,MAAA,mBAAyB,OAAA,EAAS,WAAA,KAAgB,OAAA;AAAA;;;;;;AV7FpC;AAGhD;;;;;UUwGiB,kBAAA;EV3FJ;;;;;;EUkGX,QAAA,GAAW,KAAA,EAAO,KAAA,EAAO,WAAA,UAAqB,KAAA,EAAO,MAAA,mBAAyB,OAAA,EAAS,WAAA,KAAgB,OAAA;EVlGvG;;;;AAA0C;AAmB5C;EUsFE,MAAA,GAAS,KAAA,EAAO,KAAA,EAAO,OAAA,EAAS,OAAA,SAAgB,KAAA,EAAO,MAAA,mBAAyB,OAAA,EAAS,WAAA,KAAgB,OAAA,UAAiB,iBAAA,eAAgC,iBAAA;EVtFxH;;;;;;;EU8FlC,SAAA,cAAuB,KAAA,EAAO,MAAA,mBAAyB,OAAA,EAAS,WAAA;EV3F/C;;;;;EUiGjB,QAAA,GAAW,sBAAA;EVnGN;;;;;;;;AAEc;;;;AC5BrB;;;;;;ESgJE,MAAA,IAAU,KAAA,EAAO,KAAA,EAAO,KAAA,EAAO,MAAA,mBAAyB,OAAA,EAAS,WAAA;AAAA;AAAA,UASlD,kBAAA;ETlIc;;;;;;;ES0I7B,UAAA,GAAa,iBAAA;ET3Jb;;;;;;;;ESoKA,SAAA,IAAa,GAAA,UAAa,IAAA,UAAc,KAAA,EAAO,MAAA,mBAAyB,MAAA;ETzJ7D;;;;;;;ESiKX,sBAAA;ET3JkE;;;;;AAUjD;AAWnB;;;;ESiJE,aAAA,GAAgB,uBAAA;ET/If;;;;;ESqJD,YAAA,GAAe,mBAAA,CAAoB,KAAA,EAAO,OAAA;ETtJ1C;;;AACQ;;;;AC5CV;EQ0ME,aAAA,GAAgB,aAAA;;;;;;;AR1M2D;;;;ACuC7E;;;;AAA8C;AAQ9C;;;;;EOgLE,UAAA;EPxJA;;AAAM;AAeR;;;;;;;EOoJE,kBAAA;AAAA;;;;;iBAsBc,WAAA,qCACd,KAAA,EAAO,OAAA,EACP,GAAA,EAAK,kBAAA,EACL,OAAA,GAAS,kBAAA,CAAmB,KAAA,EAAO,OAAA,IAClC,OAAA;AP9KiG;;;;ACnHpG;;;;ADmHoG,iBO0pBpF,cAAA,CACd,KAAA,EAAO,MAAA,SAAe,OAAA,GACtB,GAAA,EAAK,kBAAA,EACL,IAAA,GAAM,IAAA,CAAK,kBAAA;EAAwC,OAAA;AAAA,IAClD,MAAA,SAAe,OAAA;;;;cCjvBL,sBAAA;AAAA,UAEI,4BAAA;EZ8CY;EY5C3B,GAAA;EZ4CkC;;;;;;;;EYnClC,aAAa;AAAA;;;;;;;;;;;;;;;;AZmCqB;AAUpC;;;;;;;;;;;;;;;;iBYVgB,qBAAA,CACd,GAAA,EAAK,wBAAA,EACL,OAAA,GAAS,4BAAA,GACR,cAAA;;;UC7Cc,4BAAA;Eb6BsB;Ea3BrC,GAAA,EAAK,kBAAA;EbwCsB;;;;;;EajC3B,KAAA,GAAQ,MAAA,EAAQ,eAAA,EAAiB,MAAA,aAAmB,OAAA,UAAiB,SAAA,eAAwB,SAAA;EbezE;;;;;;EaRpB,QAAA,GAAW,WAAA,UAAqB,MAAA,UAAgB,MAAA,EAAQ,eAAA,KAAoB,OAAA;EbW5E;;;;;EaLA,MAAA,IAAU,WAAA,UAAqB,MAAA,UAAgB,IAAA,EAAM,YAAA,SAAqB,MAAA,EAAQ,eAAA,KAAoB,OAAA;EbO1F;;;;;EaDZ,SAAA;EbckC;AAAA;AAUpC;;;EalBE,QAAA,GAAW,sBAAA;EboBe;EalB1B,UAAA,GAAa,iBAAA;EbgBmC;EadhD,WAAA;EbckE;EaZlE,aAAA,GAAgB,aAAA;AAAA;;;;;;iBAiDF,qBAAA,CACd,OAAA,EAAS,4BAAA,IACP,MAAA,EAAQ,eAAA,EAAiB,MAAA,UAAgB,IAAA;EAAS,SAAA;EAAoB,MAAA,GAAS,WAAA;AAAA,MAAkB,OAAA,CAAQ,YAAA"}

package/dist/restate.js

@@ -1,6 +1,6 @@
 import { n as buildAliasMaps } from "./aliasing-lVFbm81n.js";
 import { r as coerceToolResultContent } from "./types-BIarq1qE.js";
-import { a as hydrateReadStateFromSessionAsync } from "./read-state-Dt7UbVsd.js";
+import { i as stagedToolState, l as hydrateReadStateFromSessionAsync, r as isToolStateOps } from "./tool-state-DTs-3ZHH.js";
 import { createHash } from "node:crypto";
 //#region src/restate/awakeable.ts
 /**
@@ -318,26 +318,52 @@ function restateTool(inner, ctx, options = {}) {
 		return ctx.run(nameFor(seq, canonicalName, input, toolCtx.callId), async () => {
 			options.replayTracker?.markLive();
 			await hydrateOnce(toolCtx);
+			const staged = stagedToolState(toolCtx.state);
+			staged?.clear();
 			if (shellBackgroundDisabled && input.run_in_background === true) return SHELL_BACKGROUND_DISABLED_MESSAGE;
 			if (!softErrors) {
 				const out = await inner.execute(input, toolCtx);
-				return coerceBlocks ? coerceToolResultContent(out) : out;
+				return attachToolState(coerceBlocks ? coerceToolResultContent(out) : out, staged);
 			}
 			try {
 				const out = await inner.execute(input, toolCtx);
-				return coerceBlocks ? coerceToolResultContent(out) : out;
+				return attachToolState(coerceBlocks ? coerceToolResultContent(out) : out, staged);
 			} catch (err) {
 				if (isRestateControlOrAbort(err, toolCtx.signal)) throw err;
-				return makeSoftToolError(err);
+				return attachToolState(makeSoftToolError(err), staged);
 			}
 		}, runOpts);
 	};
-	const executeDurably = softErrors ? async (input, toolCtx) => rethrowSoftToolError(await runJournaled(input, toolCtx)) : runJournaled;
-	const durableExecuteImpl = softErrors ? (input, toolCtx) => unwrapDurableSoftError(runJournaled(input, toolCtx)) : runJournaled;
-	const executeTwoPhase = async (input, toolCtx) => {
+	const finishJournaled = async (journaled, staged) => {
+		let value;
+		try {
+			value = await journaled;
+		} catch (err) {
+			staged?.clear();
+			throw err;
+		}
+		const output = applyToolState(value, staged);
+		return softErrors ? rethrowSoftToolError(output) : output;
+	};
+	const executeDurably = (input, toolCtx) => {
+		const staged = stagedToolState(toolCtx.state);
+		if (!softErrors && !staged) return runJournaled(input, toolCtx);
+		return finishJournaled(runJournaled(input, toolCtx), staged);
+	};
+	const durableExecuteImpl = (input, toolCtx) => {
+		const staged = stagedToolState(toolCtx.state);
+		if (!softErrors && !staged) return runJournaled(input, toolCtx);
+		return unwrapDurableToolValue(runJournaled(input, toolCtx), staged, softErrors);
+	};
+	const executeTwoPhaseInner = async (input, toolCtx) => {
 		const start = durableStart;
+		const staged = stagedToolState(toolCtx.state);
 		seq += 1;
 		const entry = nameFor(seq, canonicalName, input, toolCtx.callId);
+		const runFinal = async (body) => applyToolState(await ctx.run(entry, async () => {
+			options.replayTracker?.markLive();
+			return attachToolState(await body(), staged);
+		}, runOpts), staged);
 		const token = await ctx.run(`${entry}-start`, async () => {
 			options.replayTracker?.markLive();
 			await hydrateOnce(toolCtx);
@@ -356,17 +382,11 @@ function restateTool(inner, ctx, options = {}) {
 		const wait = start.wait;
 		if (wait) {
 			if (wait.inline?.(token, input, toolCtx)) {
-				if (!onAbort || !signal) return await ctx.run(entry, () => {
-					options.replayTracker?.markLive();
-					return wait.result(token, null, input, toolCtx);
-				}, runOpts);
+				if (!onAbort || !signal) return await runFinal(() => wait.result(token, null, input, toolCtx));
 				if (signal.aborted) fireAbortCleanup();
 				else signal.addEventListener("abort", fireAbortCleanup, { once: true });
 				try {
-					return await ctx.run(entry, () => {
-						options.replayTracker?.markLive();
-						return wait.result(token, null, input, toolCtx);
-					}, runOpts);
+					return await runFinal(() => wait.result(token, null, input, toolCtx));
 				} finally {
 					signal.removeEventListener("abort", fireAbortCleanup);
 				}
@@ -392,18 +412,12 @@ function restateTool(inner, ctx, options = {}) {
 				}, runOpts);
 				if (marker.awakeableId !== awakeable.id) throw new Error(`restateTool(${canonicalName}): journal drift — registration marker carries awakeable "${marker.awakeableId}" but this replay minted "${awakeable.id}". Refusing to park on an unreachable awakeable.`);
 				const resumed = await parkWithSignal(awakeable.promise, signal, canonicalName);
-				return await ctx.run(entry, () => {
-					options.replayTracker?.markLive();
-					return wait.result(token, resumed, input, toolCtx);
-				}, runOpts);
+				return await runFinal(() => wait.result(token, resumed, input, toolCtx));
 			} finally {
 				signal?.removeEventListener("abort", fireAbortCleanup);
 			}
 		}
-		const attach = () => ctx.run(entry, () => {
-			options.replayTracker?.markLive();
-			return start.attach(token, input, toolCtx);
-		}, runOpts);
+		const attach = () => runFinal(() => start.attach(token, input, toolCtx));
 		if (!onAbort || !signal) return attach();
 		if (signal.aborted) {
 			fireAbortCleanup();
@@ -416,6 +430,15 @@ function restateTool(inner, ctx, options = {}) {
 			signal.removeEventListener("abort", fireAbortCleanup);
 		}
 	};
+	const executeTwoPhase = async (input, toolCtx) => {
+		const staged = stagedToolState(toolCtx.state);
+		try {
+			return await executeTwoPhaseInner(input, toolCtx);
+		} catch (err) {
+			staged?.clear();
+			throw err;
+		}
+	};
 	if (durableStart) {
 		const { durableExecute: _dropped, ...rest } = inner;
 		return {
@@ -431,20 +454,59 @@ function restateTool(inner, ctx, options = {}) {
 		durableExecute: durableExecuteImpl
 	};
 }
+function isToolStateEnvelope(value) {
+	if (typeof value !== "object" || value === null || Array.isArray(value)) return false;
+	const candidate = value;
+	return candidate.__zidaneToolState === 1 && Array.isArray(candidate.ops) && "output" in candidate;
+}
+/**
+* Wrap the output with the staged ops so they cross the journal as DATA.
+* Returns the output untouched when nothing was staged — the no-state path
+* must not change the journaled shape.
+*/
+function attachToolState(output, staged) {
+	const ops = staged?.pending() ?? [];
+	return ops.length === 0 ? output : {
+		__zidaneToolState: 1,
+		ops,
+		output
+	};
+}
 /**
-* Compose the soft-error unwrap into the runtime's mappable durable promise.
-* A journaled sentinel becomes a thrown (rejected) value so the Restate batch
-* scheduler classifies it as a tool failure → `tool_result { isError }`; a real
-* action failure (control error) is rethrown verbatim. Non-mappable promises
-* (a ctx without the SDK's `RestatePromise`) pass through so the loop's durable
-* guard surfaces the "not a mappable durable promise" error instead.
+* Inverse of {@link attachToolState}, run OUTSIDE the journal: re-seed the
+* staging view from the journaled ops so the loop's commit writes the same
+* thing on a live attempt and on every replay. A value with no envelope
+* clears the view — whatever it holds was staged by an attempt no journal
+* entry records, and committing that would be a write replay can't reproduce.
 */
-function unwrapDurableSoftError(promise) {
+function applyToolState(value, staged) {
+	if (isToolStateEnvelope(value)) {
+		if (isToolStateOps(value.ops)) staged?.restore(value.ops);
+		else staged?.clear();
+		return value.output;
+	}
+	staged?.clear();
+	return value;
+}
+/**
+* Compose the state unwrap + soft-error unwrap into the runtime's mappable
+* durable promise. A journaled sentinel becomes a thrown (rejected) value so
+* the Restate batch scheduler classifies it as a tool failure →
+* `tool_result { isError }`; a real action failure (control error) is rethrown
+* verbatim. Non-mappable promises (a ctx without the SDK's `RestatePromise`)
+* pass through so the loop's durable guard surfaces the "not a mappable
+* durable promise" error instead.
+*/
+function unwrapDurableToolValue(promise, staged, softErrors) {
 	const mappable = promise;
 	if (typeof mappable.map !== "function") return promise;
 	return mappable.map((value, failure) => {
-		if (failure !== void 0) throw failure instanceof Error ? failure : new Error(String(failure));
-		return rethrowSoftToolError(value);
+		if (failure !== void 0) {
+			staged?.clear();
+			throw failure instanceof Error ? failure : new Error(String(failure));
+		}
+		const output = applyToolState(value, staged);
+		return softErrors ? rethrowSoftToolError(output) : output;
 	});
 }
 /**
@@ -903,6 +965,62 @@ function maxDefined(values) {
 	return max;
 }
 //#endregion
+//#region src/restate/tool-state.ts
+/** Default virtual-object state key holding the tool-state map. */
+const DEFAULT_TOOL_STATE_KEY = "zidane_tool_state";
+/**
+* Build a {@link ToolStateStore} over the bound virtual object's state.
+*
+* ```ts
+* const agent = createAgent({
+*   provider,
+*   tools: wrapAgentTools({ shell }, ctx),
+*   session: createSession({ id: ctx.key, store: restateSessionStore(ctx) }),
+*   toolState: restateToolStateStore(ctx),
+* })
+* ```
+*
+* Single-writer like the session store: the state belongs to the object key,
+* so bind one key per conversation and one agent per invocation.
+*
+* Build it INSIDE the handler. The returned store closes over this `ctx`, so
+* an agent constructed once at module scope and reused across invocations
+* would commit through a context that is already finished, and would never
+* re-hydrate (the controller memoizes the load for the life of the agent).
+* Same rule as `restateSessionStore` / `restateProvider` / `restateClock` —
+* but easier to get wrong here, because `toolState` looks like
+* agent-lifetime configuration rather than a per-invocation binding.
+*
+* The map is a schema you own: one JSON object, no versioning, no migration.
+* A value written by an earlier deploy is still there after you change its
+* shape, so either read tolerantly or move to a fresh `key`.
+*
+* Enabling this adds a `ctx.get` at run start, which changes the journal
+* shape for NEW invocations. Existing in-flight invocations replay against
+* the old shape — drain them (or accept the failed attempts) before rolling
+* it out.
+*/
+function restateToolStateStore(ctx, options = {}) {
+	const key = options.key ?? "zidane_tool_state";
+	const maxValueBytes = options.maxValueBytes;
+	let cache;
+	return {
+		...maxValueBytes !== void 0 ? { maxValueBytes } : {},
+		load: async () => {
+			cache = await ctx.get(key) ?? {};
+			return { ...cache };
+		},
+		commit: (ops) => {
+			if (ops.length === 0) return;
+			const next = { ...cache ?? {} };
+			for (const op of ops) if (op.delete === true) delete next[op.key];
+			else next[op.key] = op.value;
+			cache = next;
+			ctx.set(key, next);
+		}
+	};
+}
+//#endregion
 //#region src/restate/wait-background.ts
 function entryToExit(entry) {
 	return {
@@ -981,6 +1099,6 @@ function restateWaitBackground(options) {
 	};
 }
 //#endregion
-export { awakeableWithTimeout, createReplayTracker, deterministicUuid, replayAwareAbort, restateAwakeableTool, restateBehavior, restateClock, restateMcpToolWrap, restateProvider, restateSessionStore, restateTool, restateToolBatchExecutor, restateWaitBackground, shouldRethrowRestateControlError, wrapAgentTools };
+export { DEFAULT_TOOL_STATE_KEY, awakeableWithTimeout, createReplayTracker, deterministicUuid, replayAwareAbort, restateAwakeableTool, restateBehavior, restateClock, restateMcpToolWrap, restateProvider, restateSessionStore, restateTool, restateToolBatchExecutor, restateToolStateStore, restateWaitBackground, shouldRethrowRestateControlError, wrapAgentTools };
 
 //# sourceMappingURL=restate.js.map