@rei-standard/amsg-server 2.6.0-next.2 → 2.6.0-next.3

package/dist/{chunk-GN5QTMCL.cjs → chunk-L7WMM33Q.cjs}

@@ -106,10 +106,10 @@ function validateScheduleMessagePayload(payload) {
       };
     }
     if (hasMessages) {
-      const err = validateLlmMessagesArray(payload.messages);
-      if (err) {
+      const err2 = validateLlmMessagesArray(payload.messages);
+      if (err2) {
         return {
-          error: { code: "INVALID_PARAMETERS", message: err, details: { invalidFields: ["messages"] } },
+          error: { code: "INVALID_PARAMETERS", message: err2, details: { invalidFields: ["messages"] } },
           hasCompletePrompt: false,
           hasMessages: true
         };
@@ -399,8 +399,270 @@ function isUniqueViolation(error) {
 
 
 
-var DEFAULT_SPLIT_REGEX = /([。！？!?]+)/;
+
+// src/server/lib/llm.js
+async function callLlm(payload, options = {}) {
+  const requireContent = options.requireContent !== false;
+  const timeoutMs = typeof options.timeoutMs === "number" && Number.isFinite(options.timeoutMs) && options.timeoutMs > 0 ? options.timeoutMs : 3e5;
+  const normalizedApiUrl = normalizeAiApiUrl(payload.apiUrl);
+  const requestBody = buildAiRequestBody(payload);
+  const aiResponse = await fetch(normalizedApiUrl, {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+      "Authorization": `Bearer ${payload.apiKey}`
+    },
+    body: JSON.stringify(requestBody),
+    signal: AbortSignal.timeout(timeoutMs)
+  });
+  if (!aiResponse.ok) {
+    if (aiResponse.status === 405) {
+      throw new Error(
+        `AI API error: 405 Method Not Allowed. apiUrl must point to a full chat endpoint (for example: /chat/completions). Received: ${normalizedApiUrl}`
+      );
+    }
+    throw new Error(
+      `AI API error: ${aiResponse.status} ${aiResponse.statusText || "Unknown Error"}. Request URL: ${normalizedApiUrl}`
+    );
+  }
+  const aiData = await aiResponse.json();
+  const rawContent = _optionalChain([aiData, 'optionalAccess', _2 => _2.choices, 'optionalAccess', _3 => _3[0], 'optionalAccess', _4 => _4.message, 'optionalAccess', _5 => _5.content]);
+  if (requireContent && (typeof rawContent !== "string" || !rawContent.trim())) {
+    throw new Error("AI API error: response missing choices[0].message.content");
+  }
+  return { response: aiData, content: typeof rawContent === "string" ? rawContent.trim() : "" };
+}
+function buildAiRequestBody(payload) {
+  const llmMessages = Array.isArray(payload.messages) && payload.messages.length > 0 ? payload.messages : [{ role: "user", content: payload.completePrompt }];
+  const requestBody = {
+    model: payload.primaryModel,
+    messages: llmMessages
+  };
+  if (payload.temperature !== void 0 && payload.temperature !== null) {
+    requestBody.temperature = payload.temperature;
+  } else if (!Array.isArray(payload.messages)) {
+    requestBody.temperature = 0.8;
+  }
+  if (payload.maxTokens === void 0 || payload.maxTokens === null) {
+    return requestBody;
+  }
+  if (!Number.isInteger(payload.maxTokens) || payload.maxTokens <= 0) {
+    throw new Error("Invalid maxTokens: maxTokens must be a positive integer when provided.");
+  }
+  requestBody.max_tokens = payload.maxTokens;
+  return requestBody;
+}
+function normalizeAiApiUrl(apiUrl) {
+  if (typeof apiUrl !== "string" || !apiUrl.trim()) {
+    throw new Error(
+      "Invalid apiUrl: apiUrl is required. Please provide a chat endpoint URL (for example: https://api.openai.com or https://api.openai.com/v1/chat/completions)."
+    );
+  }
+  const trimmedApiUrl = apiUrl.trim();
+  let parsedUrl;
+  try {
+    parsedUrl = new URL(trimmedApiUrl);
+  } catch (e3) {
+    throw new Error(
+      `Invalid apiUrl: "${apiUrl}". Please provide a valid absolute URL.`
+    );
+  }
+  let path = parsedUrl.pathname.replace(/\/+$/, "") || "/";
+  if (/\/chat\/completions$/.test(path)) {
+  } else if (path === "/") {
+    path = "/v1/chat/completions";
+  } else if (/\/v\d+$/.test(path)) {
+    path = `${path}/chat/completions`;
+  }
+  parsedUrl.pathname = path;
+  return parsedUrl.toString();
+}
+
+// src/server/lib/agentic-fire.js
+
+
+
+
+
+
+var DEFAULT_MAX_TOOL_ITERATIONS = 5;
+var DEFAULT_TOTAL_TIMEOUT_MS = 24e4;
 var SLEEP_BETWEEN_MESSAGES_MS = 1500;
+var CREDENTIAL_PAYLOAD_KEYS = /* @__PURE__ */ new Set(["apiKey", "pushSubscription"]);
+var defaultSleep = (ms) => new Promise((resolve) => setTimeout(resolve, ms));
+function taskNeedsLlm(decryptedPayload) {
+  const type = decryptedPayload.messageType;
+  if (type === "prompted" || type === "auto") return true;
+  if (type === "instant") {
+    return !!(decryptedPayload.apiUrl && decryptedPayload.apiKey && decryptedPayload.primaryModel);
+  }
+  return false;
+}
+function buildHookTask(task, decryptedPayload) {
+  const safe = {};
+  for (const [key, value] of Object.entries(decryptedPayload)) {
+    if (!CREDENTIAL_PAYLOAD_KEYS.has(key)) safe[key] = value;
+  }
+  return Object.freeze({
+    ...safe,
+    id: _nullishCoalesce(task.id, () => ( null)),
+    uuid: _nullishCoalesce(task.uuid, () => ( null)),
+    nextSendAt: _nullishCoalesce(task.next_send_at, () => ( null)),
+    retryCount: _nullishCoalesce(task.retry_count, () => ( 0))
+  });
+}
+function normalizeBeforeFireResult(result) {
+  if (Array.isArray(result)) {
+    return { messages: result };
+  }
+  if (result && typeof result === "object" && Array.isArray(result.messages)) {
+    return {
+      messages: result.messages,
+      maxToolIterations: result.maxToolIterations,
+      totalTimeoutMs: result.totalTimeoutMs
+    };
+  }
+  throw new TypeError(
+    "AGENTIC_BAD_BEFORE_FIRE: onBeforeFire must return ChatMessage[] | { messages, maxToolIterations?, totalTimeoutMs? } | null"
+  );
+}
+function firstPositiveInt(values, fallback) {
+  for (const v of values) {
+    if (Number.isInteger(v) && v > 0) return v;
+  }
+  return fallback;
+}
+function firstPositiveNumber(values, fallback) {
+  for (const v of values) {
+    if (typeof v === "number" && Number.isFinite(v) && v > 0) return v;
+  }
+  return fallback;
+}
+async function runAgenticFire({ task, decryptedPayload, userKey, ctx }) {
+  const hooks = ctx.hooks;
+  if (typeof hooks.onLLMOutput !== "function") {
+    throw new Error("AGENTIC_CONFIG_ERROR: hooks.onBeforeFire requires hooks.onLLMOutput to classify LLM rounds");
+  }
+  const nowFn = typeof ctx._agenticNow === "function" ? ctx._agenticNow : Date.now;
+  const sleep = typeof ctx._agenticSleep === "function" ? ctx._agenticSleep : defaultSleep;
+  const readState = async (namespace) => {
+    if (typeof namespace !== "string" || !namespace.trim()) {
+      throw new TypeError("readState(namespace) requires a non-empty string");
+    }
+    if (!ctx.db || typeof ctx.db.getClientState !== "function") return [];
+    const rows = await ctx.db.getClientState(task.user_id, namespace);
+    return Promise.all(rows.map(async (row) => ({
+      namespace: row.namespace,
+      key: row.key,
+      value: await decryptFromStorage(row.value, userKey),
+      updatedAt: row.updated_at
+    })));
+  };
+  const fireCtx = Object.freeze({
+    task: buildHookTask(task, decryptedPayload),
+    userId: task.user_id,
+    readState,
+    now: new Date(nowFn())
+  });
+  const before = await hooks.onBeforeFire(fireCtx);
+  if (before == null) return { handled: false };
+  const normalized = normalizeBeforeFireResult(before);
+  const maxToolIterations = firstPositiveInt(
+    [normalized.maxToolIterations, ctx.maxToolIterations],
+    DEFAULT_MAX_TOOL_ITERATIONS
+  );
+  const totalTimeoutMs = firstPositiveNumber(
+    [normalized.totalTimeoutMs, ctx.totalTimeoutMs],
+    DEFAULT_TOTAL_TIMEOUT_MS
+  );
+  const deadline = nowFn() + totalTimeoutMs;
+  const sessionId = task.id != null ? `sess_task_${task.id}` : `sess_${randomUUID()}`;
+  let messages = normalized.messages.slice();
+  for (let iteration = 0; iteration < maxToolIterations; iteration++) {
+    if (nowFn() >= deadline) {
+      throw new Error(`AGENTIC_TOTAL_TIMEOUT: fire chain exceeded ${totalTimeoutMs}ms after ${iteration} LLM round(s)`);
+    }
+    const roundTimeoutMs = Math.max(1, Math.min(3e5, deadline - nowFn()));
+    const { response: llmResponse } = await callLlm(
+      { ...decryptedPayload, messages },
+      { requireContent: false, timeoutMs: roundTimeoutMs }
+    );
+    const assistantMessage = _amsgshared.extractAssistantMessage.call(void 0, llmResponse);
+    messages = [...messages, assistantMessage];
+    const sessionCtx = _amsgshared.buildSessionContext.call(void 0, {
+      sessionId,
+      messages,
+      llmResponse,
+      iteration,
+      contactName: decryptedPayload.contactName,
+      avatarUrl: decryptedPayload.avatarUrl || void 0,
+      charId: decryptedPayload.charId,
+      metadata: decryptedPayload.metadata
+    });
+    const decision = await hooks.onLLMOutput(sessionCtx);
+    _amsgshared.assertValidDecision.call(void 0, decision, { inlineToolCalls: true });
+    if (decision.decision === "continue") {
+      messages = decision.nextHistory.slice();
+      continue;
+    }
+    if (decision.decision === "skip-push") {
+      return { handled: true, result: { success: true, messagesSent: 0, status: "skipped", iterations: iteration + 1 } };
+    }
+    if (decision.decision === "finish") {
+      const messagesSent = await sendHookPushPayloads(decision.pushPayloads, decryptedPayload, ctx, sessionId, task, sleep);
+      return { handled: true, result: { success: true, messagesSent, status: "finished", iterations: iteration + 1 } };
+    }
+    const toolCalls = _amsgshared.extractToolCallsFromDecision.call(void 0, decision);
+    if (toolCalls.length === 0) {
+      throw new Error("AGENTIC_EMPTY_TOOL_REQUEST: tool-request decision carried no toolCalls (neither decision.toolCalls nor pushPayloads[].toolCalls)");
+    }
+    if (typeof hooks.executeToolCalls !== "function") {
+      throw new Error("AGENTIC_CONFIG_ERROR: onLLMOutput returned tool-request but hooks.executeToolCalls is not configured");
+    }
+    if (iteration === maxToolIterations - 1) {
+      break;
+    }
+    let toolResults;
+    try {
+      toolResults = await hooks.executeToolCalls(toolCalls, sessionCtx);
+      if (!Array.isArray(toolResults)) {
+        throw new TypeError('executeToolCalls must resolve to an array of { tool_call_id, role: "tool", content }');
+      }
+    } catch (error) {
+      toolResults = toolCalls.map((toolCall) => ({
+        tool_call_id: toolCall && typeof toolCall === "object" && typeof toolCall.id === "string" ? toolCall.id : "",
+        role: "tool",
+        content: `Tool execution failed: ${_nullishCoalesce(_optionalChain([error, 'optionalAccess', _6 => _6.message]), () => ( String(error)))}`
+      }));
+    }
+    const assistantWithTools = Array.isArray(assistantMessage.tool_calls) && assistantMessage.tool_calls.length > 0 ? assistantMessage : { ...assistantMessage, tool_calls: toolCalls };
+    messages = [...messages.slice(0, -1), assistantWithTools, ...toolResults];
+  }
+  throw new Error(`AGENTIC_LOOP_EXCEEDED: no finish/skip-push decision within ${maxToolIterations} LLM round(s)`);
+}
+async function sendHookPushPayloads(pushPayloads, decryptedPayload, ctx, sessionId, task, sleep) {
+  if (!ctx.vapid || !ctx.vapid.email || !ctx.vapid.publicKey || !ctx.vapid.privateKey) {
+    throw new Error("VAPID configuration missing - push notifications cannot be sent");
+  }
+  const pushSubscription = decryptedPayload.pushSubscription;
+  const total = pushPayloads.length;
+  const messageIdBase = task.id != null ? `msg_task_${task.id}` : `msg_${randomUUID()}`;
+  for (let i = 0; i < total; i++) {
+    const push = { ...pushPayloads[i] };
+    if (typeof push.messageId !== "string" || !push.messageId) push.messageId = `${messageIdBase}_hook_${i}`;
+    if (typeof push.sessionId !== "string" || !push.sessionId) push.sessionId = sessionId;
+    if (typeof push.timestamp !== "string" || !push.timestamp) push.timestamp = (/* @__PURE__ */ new Date()).toISOString();
+    push.messageIndex = i + 1;
+    push.totalMessages = total;
+    await ctx.webpush.sendNotification(pushSubscription, JSON.stringify(push));
+    if (i < total - 1) await sleep(SLEEP_BETWEEN_MESSAGES_MS);
+  }
+  return total;
+}
+
+// src/server/lib/message-processor.js
+var DEFAULT_SPLIT_REGEX = /([。！？!?]+)/;
+var SLEEP_BETWEEN_MESSAGES_MS2 = 1500;
 function splitOnceByRegex(chunk, regex) {
   const out = chunk.split(regex).reduce((acc, part, i, arr) => {
     if (i % 2 === 0 && part.trim()) {
@@ -428,6 +690,10 @@ async function processSingleMessage(task, ctx, providedMasterKey) {
     }
     const userKey = await deriveUserEncryptionKey(task.user_id, masterKey);
     const decryptedPayload = JSON.parse(await decryptFromStorage(task.encrypted_payload, userKey));
+    if (ctx.hooks && typeof ctx.hooks.onBeforeFire === "function" && taskNeedsLlm(decryptedPayload)) {
+      const agentic = await runAgenticFire({ task, decryptedPayload, userKey, ctx });
+      if (agentic.handled) return agentic.result;
+    }
     let messageContent;
     let llmResponse = null;
     if (decryptedPayload.messageType === "fixed") {
@@ -435,7 +701,7 @@ async function processSingleMessage(task, ctx, providedMasterKey) {
     } else if (decryptedPayload.messageType === "instant") {
       const hasPrompt = !!decryptedPayload.completePrompt || Array.isArray(decryptedPayload.messages) && decryptedPayload.messages.length > 0;
       if (hasPrompt && decryptedPayload.apiUrl && decryptedPayload.apiKey && decryptedPayload.primaryModel) {
-        const aiResult = await _callAI(decryptedPayload);
+        const aiResult = await callLlm(decryptedPayload);
         messageContent = aiResult.content;
         llmResponse = aiResult.response;
       } else if (decryptedPayload.userMessage) {
@@ -444,7 +710,7 @@ async function processSingleMessage(task, ctx, providedMasterKey) {
         throw new Error("Invalid instant message: no content source available");
       }
     } else if (decryptedPayload.messageType === "prompted" || decryptedPayload.messageType === "auto") {
-      const aiResult = await _callAI(decryptedPayload);
+      const aiResult = await callLlm(decryptedPayload);
       messageContent = aiResult.content;
       llmResponse = aiResult.response;
     } else {
@@ -480,7 +746,7 @@ async function processSingleMessage(task, ctx, providedMasterKey) {
         metadata
       });
       await ctx.webpush.sendNotification(pushSubscription, JSON.stringify(reasoningPush));
-      await new Promise((resolve) => setTimeout(resolve, SLEEP_BETWEEN_MESSAGES_MS));
+      await new Promise((resolve) => setTimeout(resolve, SLEEP_BETWEEN_MESSAGES_MS2));
     }
     for (let i = 0; i < messages.length; i++) {
       const contentPush = _amsgshared.buildContentPush.call(void 0, {
@@ -501,7 +767,7 @@ async function processSingleMessage(task, ctx, providedMasterKey) {
       });
       await ctx.webpush.sendNotification(pushSubscription, JSON.stringify(contentPush));
       if (i < messages.length - 1) {
-        await new Promise((resolve) => setTimeout(resolve, SLEEP_BETWEEN_MESSAGES_MS));
+        await new Promise((resolve) => setTimeout(resolve, SLEEP_BETWEEN_MESSAGES_MS2));
       }
     }
     return { success: true, messagesSent: messages.length };
@@ -571,80 +837,6 @@ async function processMessagesByUuid(uuid, ctx, maxRetries = 2, userId, provided
     return { success: true, messagesSent: result.messagesSent, retriesUsed: retryCount };
   }
 }
-async function _callAI(payload) {
-  const normalizedApiUrl = normalizeAiApiUrl(payload.apiUrl);
-  const requestBody = buildAiRequestBody(payload);
-  const aiResponse = await fetch(normalizedApiUrl, {
-    method: "POST",
-    headers: {
-      "Content-Type": "application/json",
-      "Authorization": `Bearer ${payload.apiKey}`
-    },
-    body: JSON.stringify(requestBody),
-    signal: AbortSignal.timeout(3e5)
-  });
-  if (!aiResponse.ok) {
-    if (aiResponse.status === 405) {
-      throw new Error(
-        `AI API error: 405 Method Not Allowed. apiUrl must point to a full chat endpoint (for example: /chat/completions). Received: ${normalizedApiUrl}`
-      );
-    }
-    throw new Error(
-      `AI API error: ${aiResponse.status} ${aiResponse.statusText || "Unknown Error"}. Request URL: ${normalizedApiUrl}`
-    );
-  }
-  const aiData = await aiResponse.json();
-  const content = _optionalChain([aiData, 'optionalAccess', _2 => _2.choices, 'optionalAccess', _3 => _3[0], 'optionalAccess', _4 => _4.message, 'optionalAccess', _5 => _5.content]);
-  if (typeof content !== "string" || !content.trim()) {
-    throw new Error("AI API error: response missing choices[0].message.content");
-  }
-  return { response: aiData, content: content.trim() };
-}
-function buildAiRequestBody(payload) {
-  const llmMessages = Array.isArray(payload.messages) && payload.messages.length > 0 ? payload.messages : [{ role: "user", content: payload.completePrompt }];
-  const requestBody = {
-    model: payload.primaryModel,
-    messages: llmMessages
-  };
-  if (payload.temperature !== void 0 && payload.temperature !== null) {
-    requestBody.temperature = payload.temperature;
-  } else if (!Array.isArray(payload.messages)) {
-    requestBody.temperature = 0.8;
-  }
-  if (payload.maxTokens === void 0 || payload.maxTokens === null) {
-    return requestBody;
-  }
-  if (!Number.isInteger(payload.maxTokens) || payload.maxTokens <= 0) {
-    throw new Error("Invalid maxTokens: maxTokens must be a positive integer when provided.");
-  }
-  requestBody.max_tokens = payload.maxTokens;
-  return requestBody;
-}
-function normalizeAiApiUrl(apiUrl) {
-  if (typeof apiUrl !== "string" || !apiUrl.trim()) {
-    throw new Error(
-      "Invalid apiUrl: apiUrl is required. Please provide a chat endpoint URL (for example: https://api.openai.com or https://api.openai.com/v1/chat/completions)."
-    );
-  }
-  const trimmedApiUrl = apiUrl.trim();
-  let parsedUrl;
-  try {
-    parsedUrl = new URL(trimmedApiUrl);
-  } catch (e3) {
-    throw new Error(
-      `Invalid apiUrl: "${apiUrl}". Please provide a valid absolute URL.`
-    );
-  }
-  let path = parsedUrl.pathname.replace(/\/+$/, "") || "/";
-  if (/\/chat\/completions$/.test(path)) {
-  } else if (path === "/") {
-    path = "/v1/chat/completions";
-  } else if (/\/v\d+$/.test(path)) {
-    path = `${path}/chat/completions`;
-  }
-  parsedUrl.pathname = path;
-  return parsedUrl.toString();
-}
 
 // src/server/handlers/schedule-message.js
 function createScheduleMessageHandler(ctx) {
@@ -1224,6 +1416,16 @@ var SQLITE_INDEXES = [
     critical: true
   }
 ];
+var CLIENT_STATE_TABLE_SQL = `
+  CREATE TABLE IF NOT EXISTS client_state (
+    user_id TEXT NOT NULL,
+    namespace TEXT NOT NULL,
+    key TEXT NOT NULL,
+    value TEXT NOT NULL,
+    updated_at INTEGER NOT NULL,
+    PRIMARY KEY (user_id, namespace, key)
+  )
+`;
 
 // src/server/adapters/d1.js
 var UPDATABLE_COLUMNS = /* @__PURE__ */ new Set([
@@ -1256,6 +1458,7 @@ var D1Adapter = class {
   }
   async initSchema() {
     await this._db.prepare(SQLITE_TABLE_SQL).run();
+    await this._db.prepare(CLIENT_STATE_TABLE_SQL).run();
     const indexResults = [];
     for (const index of SQLITE_INDEXES) {
       try {
@@ -1282,6 +1485,7 @@ var D1Adapter = class {
   }
   async dropSchema() {
     await this._db.prepare("DROP TABLE IF EXISTS scheduled_messages").run();
+    await this._db.prepare("DROP TABLE IF EXISTS client_state").run();
   }
   async createTask(params) {
     const now = this._now();
@@ -1410,6 +1614,77 @@ var D1Adapter = class {
     ).bind(uuid, userId).first();
     return row ? row.status : null;
   }
+  // ── client_state (single-user cloud state mirror) ──────────────────────
+  /**
+   * Batch upsert. Last-write-wins per (namespace, key): an entry older
+   * than the stored row (updatedAt strictly lower) is skipped; equal or
+   * newer overwrites. Values arrive pre-encrypted (the handler encrypts).
+   *
+   * Uses D1's batch() — one network round trip for the whole set (implicit
+   * transaction). The client calls this endpoint inside its few-seconds
+   * background window, so N sequential round trips could eat the whole
+   * window. Bindings without batch() (e.g. the sqlite test shim, custom
+   * adapters) fall back to a sequential loop.
+   *
+   * @param {string} userId
+   * @param {Array<{ namespace: string, key: string, value: string, updatedAt: number }>} entries
+   * @returns {Promise<{ upserted: number, skipped: number }>}
+   */
+  async upsertClientState(userId, entries) {
+    const UPSERT_SQL = `INSERT INTO client_state (user_id, namespace, key, value, updated_at)
+       VALUES (?, ?, ?, ?, ?)
+       ON CONFLICT (user_id, namespace, key) DO UPDATE SET
+         value = excluded.value,
+         updated_at = excluded.updated_at
+       WHERE excluded.updated_at >= client_state.updated_at`;
+    let results;
+    if (typeof this._db.batch === "function") {
+      const statements = entries.map(
+        (entry) => this._db.prepare(UPSERT_SQL).bind(userId, entry.namespace, entry.key, entry.value, entry.updatedAt)
+      );
+      results = await this._db.batch(statements);
+    } else {
+      results = [];
+      for (const entry of entries) {
+        results.push(
+          await this._db.prepare(UPSERT_SQL).bind(userId, entry.namespace, entry.key, entry.value, entry.updatedAt).run()
+        );
+      }
+    }
+    let upserted = 0;
+    let skipped = 0;
+    for (const res of results) {
+      if (res.meta.changes > 0) upserted++;
+      else skipped++;
+    }
+    return { upserted, skipped };
+  }
+  /**
+   * All entries of one namespace (values still encrypted).
+   *
+   * @param {string} userId
+   * @param {string} namespace
+   * @returns {Promise<Array<{ namespace: string, key: string, value: string, updated_at: number }>>}
+   */
+  async getClientState(userId, namespace) {
+    const res = await this._db.prepare(
+      `SELECT namespace, key, value, updated_at
+       FROM client_state
+       WHERE user_id = ? AND namespace = ?
+       ORDER BY key ASC`
+    ).bind(userId, namespace).all();
+    return res.results || [];
+  }
+  /**
+   * Wipe every entry of this user.
+   *
+   * @param {string} userId
+   * @returns {Promise<number>} rows deleted
+   */
+  async clearClientState(userId) {
+    const res = await this._db.prepare("DELETE FROM client_state WHERE user_id = ?").bind(userId).run();
+    return res.meta.changes || 0;
+  }
 };
 function createD1Adapter(db) {
   if (!db || typeof db.prepare !== "function") {
@@ -1523,6 +1798,131 @@ function createVapidPublicKeyHandler(ctx) {
   return { GET };
 }
 
+// src/server/handlers/client-state.js
+var MAX_STATE_VALUE_BYTES = 200 * 1024;
+var MAX_STATE_ENTRIES_PER_REQUEST = 200;
+var MAX_NAMESPACE_CHARS = 128;
+var MAX_KEY_CHARS = 256;
+var utf82 = new TextEncoder();
+function err(status, code, message, details) {
+  const error = details === void 0 ? { code, message } : { code, message, details };
+  return { status, body: { success: false, error } };
+}
+function requireUserId(headers) {
+  const userId = getHeader(headers, "x-user-id");
+  if (!userId) return { error: err(400, "USER_ID_REQUIRED", "\u7F3A\u5C11\u7528\u6237\u6807\u8BC6\u7B26") };
+  if (!isValidUUIDv4(userId)) return { error: err(400, "INVALID_USER_ID_FORMAT", "X-User-Id \u5FC5\u987B\u662F UUID v4 \u683C\u5F0F") };
+  return { userId };
+}
+function validateEntry(entry, index) {
+  if (!isPlainObject(entry)) return err(400, "INVALID_STATE_ENTRY", `entries[${index}] \u5FC5\u987B\u662F\u5BF9\u8C61`);
+  if (typeof entry.namespace !== "string" || !entry.namespace.trim() || entry.namespace.length > MAX_NAMESPACE_CHARS) {
+    return err(400, "INVALID_STATE_NAMESPACE", `entries[${index}].namespace \u5FC5\u987B\u662F 1-${MAX_NAMESPACE_CHARS} \u5B57\u7B26\u7684\u5B57\u7B26\u4E32`);
+  }
+  if (typeof entry.key !== "string" || !entry.key.trim() || entry.key.length > MAX_KEY_CHARS) {
+    return err(400, "INVALID_STATE_KEY", `entries[${index}].key \u5FC5\u987B\u662F 1-${MAX_KEY_CHARS} \u5B57\u7B26\u7684\u5B57\u7B26\u4E32`);
+  }
+  if (typeof entry.value !== "string") {
+    return err(400, "INVALID_STATE_VALUE", `entries[${index}].value \u5FC5\u987B\u662F\u5B57\u7B26\u4E32\uFF08\u5BBF\u4E3B\u81EA\u884C\u5E8F\u5217\u5316\uFF09`);
+  }
+  const bytes = utf82.encode(entry.value).length;
+  if (bytes > MAX_STATE_VALUE_BYTES) {
+    return err(413, "STATE_VALUE_TOO_LARGE", `entries[${index}].value \u8D85\u8FC7\u5355\u6761\u4E0A\u9650`, { index, key: entry.key, bytes, maxBytes: MAX_STATE_VALUE_BYTES });
+  }
+  if (!Number.isInteger(entry.updatedAt) || entry.updatedAt <= 0) {
+    return err(400, "INVALID_STATE_UPDATED_AT", `entries[${index}].updatedAt \u5FC5\u987B\u662F\u6B63\u6574\u6570\uFF08epoch \u6BEB\u79D2\uFF09`);
+  }
+  return null;
+}
+function createClientStateHandler(ctx) {
+  async function PUT(headers, body) {
+    const tenantResult = await ctx.tenantManager.resolveTenant(headers);
+    if (!tenantResult.ok) return tenantResult.error;
+    const { db, masterKey } = tenantResult.context;
+    if (getHeader(headers, "x-payload-encrypted") !== "true") {
+      return err(400, "ENCRYPTION_REQUIRED", "\u8BF7\u6C42\u4F53\u5FC5\u987B\u52A0\u5BC6");
+    }
+    const gate = requireUserId(headers);
+    if (gate.error) return gate.error;
+    const { userId } = gate;
+    if (getHeader(headers, "x-encryption-version") !== "1") {
+      return err(400, "UNSUPPORTED_ENCRYPTION_VERSION", "\u52A0\u5BC6\u7248\u672C\u4E0D\u652F\u6301");
+    }
+    const parsedBody = parseEncryptedBody(body);
+    if (!parsedBody.ok) return { status: 400, body: { success: false, error: parsedBody.error } };
+    const userKey = await deriveUserEncryptionKey(userId, masterKey);
+    let payload;
+    try {
+      payload = await decryptPayload(parsedBody.data, userKey);
+    } catch (error) {
+      if (error instanceof SyntaxError) {
+        return err(400, "INVALID_PAYLOAD_FORMAT", "\u89E3\u5BC6\u540E\u7684\u6570\u636E\u4E0D\u662F\u6709\u6548 JSON");
+      }
+      return err(400, "DECRYPTION_FAILED", "\u8BF7\u6C42\u4F53\u89E3\u5BC6\u5931\u8D25");
+    }
+    if (!isPlainObject(payload)) return err(400, "INVALID_PAYLOAD_FORMAT", "\u89E3\u5BC6\u540E\u7684\u6570\u636E\u5FC5\u987B\u662F JSON \u5BF9\u8C61");
+    const entries = payload.entries;
+    if (!Array.isArray(entries) || entries.length === 0) {
+      return err(400, "INVALID_STATE_ENTRIES", "entries \u5FC5\u987B\u662F\u975E\u7A7A\u6570\u7EC4");
+    }
+    if (entries.length > MAX_STATE_ENTRIES_PER_REQUEST) {
+      return err(400, "TOO_MANY_STATE_ENTRIES", `\u5355\u6B21\u6700\u591A ${MAX_STATE_ENTRIES_PER_REQUEST} \u6761`, { count: entries.length });
+    }
+    for (let i = 0; i < entries.length; i++) {
+      const invalid = validateEntry(entries[i], i);
+      if (invalid) return invalid;
+    }
+    if (typeof db.upsertClientState !== "function") {
+      return err(501, "CLIENT_STATE_NOT_SUPPORTED", "\u5F53\u524D\u6570\u636E\u5E93\u9002\u914D\u5668\u4E0D\u652F\u6301 client_state");
+    }
+    const encryptedEntries = await Promise.all(entries.map(async (entry) => ({
+      namespace: entry.namespace,
+      key: entry.key,
+      value: await encryptForStorage(entry.value, userKey),
+      updatedAt: entry.updatedAt
+    })));
+    const { upserted, skipped } = await db.upsertClientState(userId, encryptedEntries);
+    return { status: 200, body: { success: true, data: { upserted, skipped } } };
+  }
+  async function GET(url, headers) {
+    const tenantResult = await ctx.tenantManager.resolveTenant(headers);
+    if (!tenantResult.ok) return tenantResult.error;
+    const { db, masterKey } = tenantResult.context;
+    const gate = requireUserId(headers);
+    if (gate.error) return gate.error;
+    const { userId } = gate;
+    const namespace = new URL(url, "https://dummy").searchParams.get("namespace") || "";
+    if (!namespace.trim()) return err(400, "NAMESPACE_REQUIRED", "\u5FC5\u987B\u63D0\u4F9B namespace \u67E5\u8BE2\u53C2\u6570");
+    if (typeof db.getClientState !== "function") {
+      return err(501, "CLIENT_STATE_NOT_SUPPORTED", "\u5F53\u524D\u6570\u636E\u5E93\u9002\u914D\u5668\u4E0D\u652F\u6301 client_state");
+    }
+    const userKey = await deriveUserEncryptionKey(userId, masterKey);
+    const rows = await db.getClientState(userId, namespace);
+    const decrypted = await Promise.all(rows.map(async (row) => ({
+      namespace: row.namespace,
+      key: row.key,
+      value: await decryptFromStorage(row.value, userKey),
+      updatedAt: row.updated_at
+    })));
+    const encryptedResponse = await encryptPayload({ namespace, entries: decrypted }, userKey);
+    return { status: 200, body: { success: true, encrypted: true, version: 1, data: encryptedResponse } };
+  }
+  async function DELETE(url, headers) {
+    const tenantResult = await ctx.tenantManager.resolveTenant(headers);
+    if (!tenantResult.ok) return tenantResult.error;
+    const { db } = tenantResult.context;
+    const gate = requireUserId(headers);
+    if (gate.error) return gate.error;
+    const { userId } = gate;
+    if (typeof db.clearClientState !== "function") {
+      return err(501, "CLIENT_STATE_NOT_SUPPORTED", "\u5F53\u524D\u6570\u636E\u5E93\u9002\u914D\u5668\u4E0D\u652F\u6301 client_state");
+    }
+    const deleted = await db.clearClientState(userId);
+    return { status: 200, body: { success: true, data: { deleted } } };
+  }
+  return { PUT, GET, DELETE };
+}
+
 // src/server/single-user.js
 function createSingleUserServer(config) {
   if (!config || !config.db) throw new Error("[amsg-server single-user] config.db is required");
@@ -1540,7 +1940,13 @@ function createSingleUserServer(config) {
       privateKey: vapid.privateKey || ""
     },
     webpush: config.webpush || null,
-    tenantManager
+    tenantManager,
+    // Fire-time hooks (optional): the in-server instant path fires through
+    // processMessagesByUuid with this ctx, so instant-type tasks can take
+    // the agentic path too.
+    hooks: config.hooks || null,
+    maxToolIterations: config.maxToolIterations,
+    totalTimeoutMs: config.totalTimeoutMs
   };
   return {
     ctx,
@@ -1551,7 +1957,8 @@ function createSingleUserServer(config) {
       updateMessage: createUpdateMessageHandler(ctx),
       cancelMessage: createCancelMessageHandler(ctx),
       messages: createMessagesHandler(ctx),
-      vapidPublicKey: createVapidPublicKeyHandler(ctx)
+      vapidPublicKey: createVapidPublicKeyHandler(ctx),
+      clientState: createClientStateHandler(ctx)
     }
   };
 }
@@ -1605,12 +2012,12 @@ async function sendWebPush({ subscription, payload, vapid, ttl, fetch: fetchImpl
   });
   if (!res.ok) {
     const text = await safeReadText(res);
-    const err = new Error(
+    const err2 = new Error(
       `Web Push delivery failed: ${res.status} ${res.statusText || ""}${text ? ` \u2014 ${text}` : ""}`
     );
-    err.code = "PUSH_SEND_FAILED";
-    err.statusCode = res.status;
-    throw err;
+    err2.code = "PUSH_SEND_FAILED";
+    err2.statusCode = res.status;
+    throw err2;
   }
   return {
     statusCode: res.status,
@@ -1817,6 +2224,12 @@ function createSingleUserCloudflareWorker(buildConfig) {
         result = await server.handlers.cancelMessage.DELETE(url, headers);
       } else if (method === "GET" && pathname.endsWith("/vapid-public-key")) {
         result = await server.handlers.vapidPublicKey.GET(url, headers);
+      } else if (method === "PUT" && pathname.endsWith("/client-state")) {
+        result = await server.handlers.clientState.PUT(headers, await request.text());
+      } else if (method === "GET" && pathname.endsWith("/client-state")) {
+        result = await server.handlers.clientState.GET(url, headers);
+      } else if (method === "DELETE" && pathname.endsWith("/client-state")) {
+        result = await server.handlers.clientState.DELETE(url, headers);
       } else {
         result = { status: 404, body: { success: false, error: { code: "NOT_FOUND", message: "Unknown route" } } };
       }
@@ -1834,7 +2247,17 @@ function createSingleUserCloudflareWorker(buildConfig) {
       return;
     }
     try {
-      await runScheduledTick({ db: cfg.db, masterKey: cfg.masterKey, vapid, webpush: cfg.webpush });
+      await runScheduledTick({
+        db: cfg.db,
+        masterKey: cfg.masterKey,
+        vapid,
+        webpush: cfg.webpush,
+        // Fire-time hooks (optional; see lib/agentic-fire.js). runScheduledTick
+        // spreads its ctx into processSingleMessage, so these ride along.
+        hooks: cfg.hooks || null,
+        maxToolIterations: cfg.maxToolIterations,
+        totalTimeoutMs: cfg.totalTimeoutMs
+      });
     } catch (error) {
       console.error("[amsg single-user] scheduled(): tick failed:", error && error.message);
     }