@agent-native/core 0.109.0 → 0.109.2

package/dist/server/agent-chat-plugin.js

@@ -3,6 +3,7 @@ import nodePath from "node:path";
 import { createError, defineEventHandler, setResponseStatus, setResponseHeader, getMethod, getQuery, getHeader, getRequestIP, } from "h3";
 import { appendA2AArtifactLinks, buildA2ARecoverableArtifactMessage, } from "../a2a/artifact-response.js";
 import { hasConfiguredA2ASecret, isLoopbackAddress, isTrustedLocalRuntime, } from "../a2a/auth-policy.js";
+import { sanitizeA2ACorrelationId, sanitizeA2ACorrelationMetadata, } from "../a2a/correlation.js";
 import { applyAgentTextEventToBuffer } from "../a2a/response-text.js";
 import { createA2AApproval, updateTaskStatusMessage, } from "../a2a/task-store.js";
 import { canUpdateAgentAppModelDefaultSettings, normalizeAgentAppModelDefaultAppId, readAgentAppModelDefaultSettings, resetAgentAppModelDefaultSettings, writeAgentAppModelDefaultSettings, } from "../agent/app-model-defaults.js";
@@ -18,6 +19,7 @@ import { attachToolSearch } from "../agent/tool-search.js";
 import { readAppStateForCurrentTab } from "../application-state/script-helpers.js";
 import { createThread, forkThread, getThread, registerChatThreadsShareable, resolveThreadAccess, listThreads, searchThreads, renameThread, createThreadShareLink, getThreadByShareToken, getThreadShareState, revokeThreadShareLink, setThreadArchived, setThreadPinned, setThreadScope, updateThreadData, withThreadDataLock, deleteThread, setThreadQueuedMessages, } from "../chat-threads/store.js";
 import { createDbAdminAgentTools } from "../db-admin/agent-tools.js";
+import { isTransientDatabaseError } from "../db/client.js";
 import { verifyInternalToken, extractBearerToken, } from "../integrations/internal-token.js";
 import { McpClientManager, loadMcpConfig, autoDetectMcpConfig, mcpToolsToActionEntries, syncMcpActionEntries, mountMcpServersRoutes, mountMcpHubRoutes, buildMergedConfig, startMcpConfigRefresh, getHubStatus, isHubServeEnabled, } from "../mcp-client/index.js";
 import { setProgressPreListHook } from "../progress/store.js";
@@ -38,6 +40,47 @@ import { getModelFamilyOverlay } from "./prompts/index.js";
 import { mountRealtimeVoiceRoutes } from "./realtime-voice.js";
 import { runWithRequestContext, getRequestContext, getRequestOrgId, getRequestUserEmail, getRequestRunContext, ensureRequestRunContext, } from "./request-context.js";
 export { handleSharedThreadRequest };
+function withTransientDatabaseFallback(route, handler) {
+    return defineEventHandler(async (event) => {
+        try {
+            return await handler(event);
+        }
+        catch (error) {
+            if (!isTransientDatabaseError(error))
+                throw error;
+            const method = getMethod(event);
+            const retrySafe = method === "GET" || method === "HEAD";
+            const databaseError = error;
+            captureError(new Error("Transient database failure in agent chat"), {
+                route,
+                method,
+                tags: {
+                    source: "agent-chat",
+                    failureClass: "transient-database",
+                },
+                extra: {
+                    databaseErrorName: error instanceof Error ? error.name : typeof error,
+                    databaseErrorCode: typeof databaseError.code === "string"
+                        ? databaseError.code
+                        : undefined,
+                    databaseCauseCode: typeof databaseError.cause?.code === "string"
+                        ? databaseError.cause.code
+                        : undefined,
+                },
+            });
+            setResponseStatus(event, 503);
+            if (retrySafe)
+                setResponseHeader(event, "Retry-After", "2");
+            return {
+                error: retrySafe
+                    ? "The database is temporarily unavailable. Please retry."
+                    : "The database became unavailable while processing this request. Refresh before deciding whether to retry.",
+                code: "database_unavailable",
+                retryable: retrySafe,
+            };
+        }
+    });
+}
 // Lazy fs — loaded via dynamic import() on first use.
 // This avoids require() which bundlers convert to createRequire(import.meta.url)
 // that crashes on CF Workers where import.meta.url is undefined.
@@ -869,6 +912,7 @@ export function createAgentChatPlugin(options) {
                 : undefined;
             const { mountA2A } = await import("../a2a/server.js");
             mountA2A(nitroApp, {
+                appId: options?.appId,
                 name: options?.appId
                     ? options.appId.charAt(0).toUpperCase() + options.appId.slice(1)
                     : "Agent",
@@ -877,7 +921,7 @@ export function createAgentChatPlugin(options) {
                 publicSkillsOnly: true,
                 streaming: true,
                 durableBackgroundRuns: options?.durableBackgroundRuns,
-                executeReadOnlyAction: async ({ action, input }) => {
+                executeReadOnlyAction: async ({ action, input, invocationId }) => {
                     const actions = filterDirectA2AActions(mcpFullActions ?? allScripts, options ?? {});
                     const entry = actions[action];
                     if (!entry) {
@@ -890,10 +934,12 @@ export function createAgentChatPlugin(options) {
                         actions: { [action]: entry },
                         name: action,
                         input,
-                        callId: `a2a-action-${crypto.randomUUID()}`,
+                        callId: `a2a-action-${invocationId}`,
                         ownerEmail: getRequestUserEmail(),
                         orgId: getRequestOrgId() ?? null,
                         caller: "a2a",
+                        networkProtocol: "a2a",
+                        networkId: invocationId,
                     });
                     if (result.status === "approval_required") {
                         return {
@@ -1199,6 +1245,10 @@ export function createAgentChatPlugin(options) {
                     const recoverableArtifactSecret = await resolveA2ARecoverableArtifactSecret();
                     const recoverableArtifactStatusWriter = createSerializedA2ATaskStatusWriter(context.taskId);
                     const controller = new AbortController();
+                    const correlation = sanitizeA2ACorrelationMetadata(context.metadata);
+                    const telemetryThreadId = sanitizeA2ACorrelationId(context.contextId) ??
+                        correlation.callerThreadId ??
+                        context.taskId;
                     console.log(`[A2A] Starting agent loop: ${a2aToolSurface.tools.length}/${a2aToolSurface.availableTools.length} initial tools, prompt ${systemPrompt.length} chars`);
                     await runA2AAgentLoop({
                         engine: a2aEngine,
@@ -1216,6 +1266,11 @@ export function createAgentChatPlugin(options) {
                         orgId: getRequestOrgId() ?? null,
                         approvedToolCalls: context.approvedActions?.map((approved) => toolCallCacheKey(approved.tool, approved.input)),
                         executionMode: "act",
+                        runId: context.taskId,
+                        networkProtocol: "a2a",
+                        networkId: context.taskId,
+                        threadId: context.taskId,
+                        turnId: context.taskId,
                         send: (event) => {
                             a2aEvents.push(event);
                             if (event.type === "tool_start") {
@@ -1268,6 +1323,25 @@ export function createAgentChatPlugin(options) {
                     }, {
                         backgroundFunction: options?.durableBackgroundRuns === true &&
                             isInBackgroundFunctionRuntime(),
+                    }, {
+                        telemetry: {
+                            runId: context.taskId,
+                            threadId: telemetryThreadId,
+                            userId: userEmail,
+                            delegation: {
+                                protocol: "a2a",
+                                taskId: context.taskId,
+                                ...(correlation.callerApp
+                                    ? { callerApp: correlation.callerApp }
+                                    : {}),
+                                ...(correlation.parentRunId
+                                    ? { parentRunId: correlation.parentRunId }
+                                    : {}),
+                                ...(correlation.parentTurnId
+                                    ? { parentTurnId: correlation.parentTurnId }
+                                    : {}),
+                            },
+                        },
                     });
                     // The continuation can observe terminal output immediately, so make
                     // its latest mutation checkpoint durable first.
@@ -1391,6 +1465,7 @@ export function createAgentChatPlugin(options) {
                         : {}),
                     askAgent: async (message) => {
                         const ownerEmail = getRequestUserEmail();
+                        const mcpRunId = crypto.randomUUID();
                         const { getOwnerActiveApiKey } = await import("../agent/production-agent.js");
                         const ownerApiKey = ownerEmail
                             ? await getOwnerActiveApiKey(ownerEmail)
@@ -1501,9 +1576,14 @@ export function createAgentChatPlugin(options) {
                                 },
                             ],
                             actions: mcpActions,
-                            ownerEmail: getRequestUserEmail(),
+                            ownerEmail,
                             orgId: getRequestOrgId() ?? null,
                             executionMode: "act",
+                            runId: mcpRunId,
+                            networkProtocol: "mcp",
+                            networkId: mcpRunId,
+                            threadId: mcpRunId,
+                            turnId: mcpRunId,
                             send: (event) => {
                                 accumulatedText = applyAgentTextEventToBuffer(accumulatedText, event);
                             },
@@ -1514,6 +1594,13 @@ export function createAgentChatPlugin(options) {
                         }, {
                             backgroundFunction: options?.durableBackgroundRuns === true &&
                                 isInBackgroundFunctionRuntime(),
+                        }, {
+                            telemetry: {
+                                runId: mcpRunId,
+                                threadId: mcpRunId,
+                                userId: ownerEmail ?? null,
+                                delegation: { protocol: "mcp" },
+                            },
                         });
                         return accumulatedText || "(no response)";
                     },
@@ -3370,7 +3457,7 @@ Non-code requests are still fine on this surface: read data, navigate the UI, su
             }));
             // ─── Run management endpoints (for hot-reload resilience) ─────────────
             // GET /runs/active?threadId=X — check if there's an active run for a thread
-            getH3App(nitroApp).use(`${routePath}/runs`, defineEventHandler(async (event) => {
+            getH3App(nitroApp).use(`${routePath}/runs`, withTransientDatabaseFallback(`${routePath}/runs`, async (event) => {
                 // Auth check — ensure the user is authenticated
                 const owner = await getOwnerFromEvent(event);
                 const method = getMethod(event);
@@ -3795,7 +3882,7 @@ Non-code requests are still fine on this surface: read data, navigate the UI, su
                 return { threadId: null, tail: [] };
             };
             const buildShareUrl = (event, token) => `${getOrigin(event)}${routePath}/shared/${encodeURIComponent(token)}`;
-            getH3App(nitroApp).use(`${routePath}/threads`, defineEventHandler(async (event) => {
+            getH3App(nitroApp).use(`${routePath}/threads`, withTransientDatabaseFallback(`${routePath}/threads`, async (event) => {
                 const owner = await getOwnerFromEvent(event);
                 const orgId = await getOrgIdFromEvent(event);
                 const method = getMethod(event);
@@ -4306,7 +4393,7 @@ Non-code requests are still fine on this surface: read data, navigate the UI, su
             // This is mounted last because h3's use() is prefix-based, meaning /_agent-native/agent-chat
             // also matches /_agent-native/agent-chat/threads/... — we skip sub-path requests here so the
             // earlier-mounted handlers (mode, save-key, files, skills, mentions, threads) handle them.
-            getH3App(nitroApp).use(routePath, defineEventHandler(async (event) => {
+            getH3App(nitroApp).use(routePath, withTransientDatabaseFallback(routePath, async (event) => {
                 // Skip sub-path requests — they're handled by earlier-mounted handlers
                 const url = event.node?.req?.url || event.path || "";
                 const afterBase = url.slice(url.indexOf(routePath) + routePath.length);