@agent-native/dispatch 0.10.3 → 0.10.4

package/src/actions/query-staged-dataset.ts

@@ -0,0 +1,125 @@
+import { defineAction } from "@agent-native/core";
+import { runAggregateQuery } from "@agent-native/core/provider-api/staged-datasets-aggregate";
+import {
+  getStagedDatasetMeta,
+  getStagedDatasetRows,
+} from "@agent-native/core/provider-api/staged-datasets-store";
+import { getCredentialContext } from "@agent-native/core/server/request-context";
+import { z } from "zod";
+import { DISPATCH_APP_ID } from "../server/lib/provider-api.js";
+
+const WhereSchema = z.object({
+  column: z.string().min(1),
+  op: z.enum([
+    "equals",
+    "not_equals",
+    "contains",
+    "not_contains",
+    "gt",
+    "gte",
+    "lt",
+    "lte",
+    "exists",
+    "not_exists",
+  ]),
+  value: z.unknown().optional(),
+});
+
+const AggregateFieldSchema = z.object({
+  column: z.string().min(1).describe("Column to aggregate."),
+  op: z
+    .enum(["sum", "avg", "count", "min", "max", "count_distinct"])
+    .describe("Aggregation function."),
+  as: z
+    .string()
+    .optional()
+    .describe("Output column name. Default: {op}_{column}."),
+});
+
+export default defineAction({
+  description:
+    "Run a filter/aggregate/project query over a staged dataset previously written by provider-api-request (stageAs). Use after staging provider records, messages, documents, issues, events, or search results to count, group, filter, or project rows without re-fetching provider APIs.",
+  schema: z.object({
+    datasetId: z
+      .string()
+      .min(1)
+      .describe(
+        "Dataset id from provider-api-request stageAs result, or from list-staged-datasets.",
+      ),
+    where: z
+      .array(WhereSchema)
+      .optional()
+      .describe(
+        "Row-level filters (AND). Ops: equals, not_equals, contains, not_contains, gt, gte, lt, lte, exists, not_exists.",
+      ),
+    groupBy: z
+      .array(z.string().min(1))
+      .optional()
+      .describe(
+        "Column(s) to group by. Omit for a single aggregate over all rows.",
+      ),
+    aggregate: z
+      .array(AggregateFieldSchema)
+      .optional()
+      .describe(
+        "Aggregation operations. When set, non-group columns are aggregated. Omit to return raw rows.",
+      ),
+    select: z
+      .array(z.string().min(1))
+      .optional()
+      .describe("Column projection when aggregate is empty."),
+    orderBy: z.string().optional().describe("Sort output by this column."),
+    orderDir: z
+      .enum(["asc", "desc"])
+      .optional()
+      .describe("Sort direction (default asc)."),
+    limit: z.coerce
+      .number()
+      .int()
+      .min(1)
+      .max(10_000)
+      .optional()
+      .describe("Maximum rows to return (default all, max 10000)."),
+  }),
+  http: false,
+  readOnly: true,
+  run: async (args) => {
+    const ctx = getCredentialContext();
+    if (!ctx) {
+      throw new Error("No authenticated context for query-staged-dataset.");
+    }
+
+    const meta = await getStagedDatasetMeta({
+      id: args.datasetId,
+      appId: DISPATCH_APP_ID,
+      ownerEmail: ctx.userEmail,
+    });
+    if (!meta) {
+      throw new Error(
+        `Dataset ${args.datasetId} not found (or belongs to a different owner/app).`,
+      );
+    }
+
+    const rows = await getStagedDatasetRows({
+      id: args.datasetId,
+      appId: DISPATCH_APP_ID,
+      ownerEmail: ctx.userEmail,
+    });
+
+    const result = runAggregateQuery(rows, {
+      where: args.where,
+      groupBy: args.groupBy,
+      aggregate: args.aggregate,
+      select: args.select,
+      orderBy: args.orderBy,
+      orderDir: args.orderDir,
+      limit: args.limit,
+    });
+
+    return {
+      dataset: { id: meta.id, name: meta.name, totalRows: meta.rowCount },
+      rowCount: result.length,
+      rows: result,
+    };
+  },
+});

package/src/server/lib/provider-api.ts

@@ -8,6 +8,7 @@ import {
 } from "@agent-native/core/provider-api";
 import { getCredentialContext } from "@agent-native/core/server";
 
+export const DISPATCH_APP_ID = "dispatch";
 export const DISPATCH_PROVIDER_API_IDS = PROVIDER_API_IDS;
 export type DispatchProviderApiId = ProviderApiId;
 export type { ProviderApiMethod, ProviderApiRequestArgs };
@@ -23,7 +24,7 @@ function requireCtx(action: string) {
 }
 
 const runtime = createProviderApiRuntime({
-  appId: "dispatch",
+  appId: DISPATCH_APP_ID,
   localCredentialSource: "dispatch_local",
   getCredentialContext: () => requireCtx("requests"),
   getCustomProviders: async () => {

package/src/server/plugins/agent-chat.ts

@@ -15,6 +15,7 @@ export default createAgentChatPlugin({
   // a build-time-generated `.generated/actions-registry.ts` (the latter is a
   // template-only construct that the Vite plugin emits next to actions/).
   actions: dispatchActions,
+  codeExecution: { production: "sandboxed" },
   systemPrompt: `You are the central dispatch for this workspace.
 
 Default posture:
@@ -33,6 +34,7 @@ Use the standard workspace primitives:
 - If the starter template is used, treat it as scaffolding only: the finished app must be branded as the requested app with its own home screen/navigation/package metadata/manifest, and must not leave visible "Starter", "Blank app", or "New app" UI behind.
 - Treat first-party apps such as Mail, Calendar, Analytics, Brain, Assets, and Dispatch as existing hosted/connected neighbors available through links and A2A/default connected agents. Do not create wrapper apps, child apps, nested routes, or cloned template copies just to give a new app access to them; build only the genuinely new workflow and delegate cross-app work to those existing apps.
 - Integration grants are not provider capability limits. For ad hoc provider inspection, querying, reporting, or troubleshooting, call provider-api-catalog/provider-api-docs, then provider-api-request against the provider's real HTTP API. Use connectionId for a specific shared grant and accountId for a specific OAuth account. Never expose secret values or silently widen app access while doing this.
+- For broad provider searches, joins, classification, corpus counts, or absence claims, fetch every relevant page or an explicitly bounded cohort, stage/save large responses with stageAs/saveToFile/fetchAllPages, and reduce them with query-staged-dataset or run-code. Report source, filters, row counts, pagination, truncation, failed pages, and uncovered gaps.
 
 When a user asks for something like a digest, reminder, routing rule, or saved behavior:
 - First decide whether it should be a resource, a recurring job, a destination, or a delegated task.