@agent-native/core 0.58.1 → 0.58.3

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

@@ -1,6 +1,7 @@
 import { runWithRequestContext, getRequestOrgId, getRequestUserEmail, getRequestRunContext, ensureRequestRunContext, } from "./request-context.js";
 import { getSetting, putSetting } from "../settings/store.js";
 import { createDbAdminAgentTools } from "../db-admin/agent-tools.js";
+import { dbExecToolParameters } from "../scripts/db/tool-schemas.js";
 import { getH3App, markDefaultPluginProvided, trackPluginInit, } from "./framework-request-handler.js";
 import { createProductionAgentHandler, actionsToEngineTools, getActiveRunForThreadAsync, abortRun, subscribeToRun, } from "../agent/production-agent.js";
 import { runAgentLoopDirectWithSoftTimeout } from "../agent/run-loop-with-resume.js";
@@ -33,6 +34,8 @@ import { updateTaskStatusMessage } from "../a2a/task-store.js";
 import { collectFinalResponseTextFromAgentEvents } from "../a2a/response-text.js";
 import { buildRuntimeContextPrompt } from "../agent/runtime-context.js";
 import { buildFrameworkCore, buildFrameworkCoreCompact, FIRST_SESSION_PERSONALIZATION, getModelFamilyOverlay, } from "./prompts/index.js";
+import { ACTION_CHAT_UI_DATA_WIDGET_RENDERER } from "../action-ui.js";
+import { dataWidgetResultSchema } from "../data-widgets/index.js";
 // 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.
@@ -643,6 +646,58 @@ function createUrlTools() {
         },
     };
 }
+function createDataWidgetActionEntries() {
+    return {
+        "render-data-widget": {
+            readOnly: true,
+            parallelSafe: true,
+            chatUI: {
+                renderer: ACTION_CHAT_UI_DATA_WIDGET_RENDERER,
+                title: "Data widget",
+                description: "Render a validated native data table or chart in chat.",
+            },
+            tool: {
+                description: "Render a native Agent-Native chat data widget from compact, real data you already retrieved or the user provided. Use this for in-chat tables, charts, graphs, trends, and compact reports when no domain-specific action already returns a native widget. Never fabricate rows or metrics just to make a chart.",
+                parameters: {
+                    type: "object",
+                    properties: {
+                        widget: {
+                            type: "string",
+                            enum: ["data-table", "data-chart", "data-insights"],
+                            description: "Widget kind. Use data-chart for a chart, data-table for a table, or data-insights for a combined summary/chart/table card.",
+                        },
+                        widgetId: {
+                            type: "string",
+                            description: "Optional stable widget identifier.",
+                        },
+                        title: {
+                            type: "string",
+                            description: "Optional widget title.",
+                        },
+                        summary: {
+                            type: "object",
+                            description: "Optional scalar summary values for data-insights cards.",
+                        },
+                        display: {
+                            type: "object",
+                            description: "Optional display metadata: title, description, primaryAction.",
+                        },
+                        table: {
+                            type: "object",
+                            description: "For data-table/data-insights: { title?, columns: [{ key, label, align? }], rows, totalRows?, sampledRows?, truncated? }.",
+                        },
+                        chartSeries: {
+                            type: "object",
+                            description: "For data-chart/data-insights: { type: 'bar'|'line'|'area', title?, xKey, series: [{ key, label, color? }], data, sampled? }.",
+                        },
+                    },
+                    required: ["widget"],
+                },
+            },
+            run: async (args) => dataWidgetResultSchema.parse(args),
+        },
+    };
+}
 /**
  * Creates db-* tools (db-query, db-exec, db-patch, db-schema) as native tools.
  * These let the agent read and write the app's own SQL database. Scoping to
@@ -702,28 +757,7 @@ async function createDbScriptEntries() {
             }, queryMod.default, { readOnly: true }),
             "db-exec": wrapCliScript({
                 description: "Write to the app's own SQL database ONLY. Runs INSERT / UPDATE / DELETE / REPLACE against the app's internal tables. For multiple related writes, pass `statements` so they run sequentially in one transaction instead of issuing several db-exec calls. Writes are auto-scoped to the current user/org, and `owner_email` / `org_id` are auto-injected on INSERT. Schema changes (CREATE/ALTER/DROP) are blocked. Never use this to backfill missing data for a read/analysis request or to create/modify users, members, roles, permissions, admin flags, or ownership; use a dedicated app action or reviewed code. IMPORTANT: This tool CANNOT write to external data sources like BigQuery, HubSpot, etc. For external services, use the appropriate template action.",
-                parameters: {
-                    type: "object",
-                    properties: {
-                        sql: {
-                            type: "string",
-                            description: "Single INSERT / UPDATE / DELETE / REPLACE statement. Use parameterized placeholders (?) where possible.",
-                        },
-                        args: {
-                            type: "string",
-                            description: 'Optional JSON array of positional bind args for `sql`. Example: \'["published","form-123"]\'',
-                        },
-                        statements: {
-                            type: "string",
-                            description: 'Optional JSON array of write statements to execute in one transaction. Prefer this over multiple db-exec calls. Example: \'[{"sql":"INSERT INTO notes (id,title) VALUES (?,?)","args":["n1","One"]},{"sql":"UPDATE counters SET value = value + 1 WHERE key = ?","args":["notes"]}]\'',
-                        },
-                        format: {
-                            type: "string",
-                            description: 'Output format: "json" or "text" (default: text)',
-                            enum: ["json", "text"],
-                        },
-                    },
-                },
+                parameters: dbExecToolParameters(),
             }, execMod.default),
             "db-patch": wrapCliScript({
                 description: "Surgical patch on a large text/JSON column in the app's SQL database. Two modes: (1) text find/replace via `find`/`replace`/`edits` — best for small edits to documents, slide HTML, etc. (2) structural JSON ops via `json-ops` — STRONGLY PREFERRED when the column is JSON (dashboard configs, form schemas, slide decks) because it avoids all the brace/quote/comma surgery that text find/replace requires. Use `json-ops` to set/remove values at a JSON Pointer path, or to move/insert array items — e.g. reorder dashboard panels, add a filter, rename a field. Targets exactly one row (narrow `where` by primary key). Same per-user/org scoping as db-exec.",
@@ -1951,6 +1985,8 @@ export const _agentChatPromptSectionsForTests = (() => {
         frameworkCore,
         frameworkCoreCompact,
         frameworkContextSections: FRAMEWORK_CONTEXT_SECTIONS,
+        generateActionsPrompt,
+        createDataWidgetActionEntries,
     };
 })();
 /**
@@ -2137,10 +2173,13 @@ function generateActionsPrompt(registry, mode = "tool") {
     if (!registry || Object.keys(registry).length === 0)
         return "";
     const actionEntries = Object.entries(registry);
+    const nativeWidgetNote = (entry) => entry.chatUI && typeof entry.chatUI.renderer === "string"
+        ? ` Native chat widget: \`${entry.chatUI.renderer}\`.`
+        : "";
     if (mode === "tool") {
         const summaryLines = actionEntries.map(([name, entry]) => {
             const desc = compactPromptLine(entry.tool.description, MAX_ACTION_SUMMARY_DESCRIPTION_CHARS);
-            return `- \`${name}\` — ${desc}`;
+            return `- \`${name}\` — ${desc}${nativeWidgetNote(entry)}`;
         });
         return `\n\n## Available Actions
 
@@ -2154,7 +2193,7 @@ ${summaryLines.join("\n")}`;
         const requiredFields = new Set(entry.tool.parameters?.required ?? []);
         // CLI mode: emit `pnpm action <name> --required <type> [--optional <type>]`
         if (!params || Object.keys(params).length === 0) {
-            return `- \`pnpm action ${name}\` — ${desc}`;
+            return `- \`pnpm action ${name}\` — ${desc}${nativeWidgetNote(entry)}`;
         }
         const entries = Object.entries(params);
         // Required first (alphabetical), then optional (alphabetical)
@@ -2181,7 +2220,7 @@ ${summaryLines.join("\n")}`;
         }
         const cmd = ["pnpm action " + name, ...required, ...optional].join(" ");
         const requiredNote = requiredNames.length > 0 ? ` Required: ${requiredNames.join(", ")}.` : "";
-        return `- \`${cmd}\` — ${desc}.${requiredNote}`;
+        return `- \`${cmd}\` — ${desc}.${requiredNote}${nativeWidgetNote(entry)}`;
     });
     return `\n\n## Available Actions
 
@@ -2749,10 +2788,13 @@ export function createAgentChatPlugin(options) {
                 workspaceFilesTool = createWorkspaceFilesTool();
             }
             catch { }
-            let toolActions = {};
+            let toolActions = createDataWidgetActionEntries();
             try {
                 const { createExtensionActionEntries } = await import("../extensions/actions.js");
-                toolActions = createExtensionActionEntries();
+                toolActions = {
+                    ...toolActions,
+                    ...createExtensionActionEntries(),
+                };
             }
             catch { }
             let browserSessionTools = {};