@rudderjs/ai 1.6.0 → 1.6.2

package/dist/tool-helpers.js

@@ -0,0 +1,117 @@
+/**
+ * Detect an async generator (the value returned by `async function*` or any
+ * object implementing the AsyncGenerator protocol). We use a structural check
+ * because the executor may not be authored as a literal `async function*`
+ * (e.g. wrapped or returned from a factory).
+ */
+export function isAsyncGenerator(value) {
+    if (value === null || typeof value !== 'object')
+        return false;
+    const v = value;
+    return typeof v.next === 'function'
+        && typeof v.return === 'function'
+        && typeof v[Symbol.asyncIterator] === 'function';
+}
+/**
+ * Uniformly iterate a tool's `execute`, whether it returns a value, a
+ * promise, or an async generator.
+ *
+ * The helper is itself an async generator: each `yield` is a preliminary
+ * tool-update payload (only generator-style executes produce these), and the
+ * generator's `return` value is the final tool result.
+ *
+ * Streaming callers iterate and emit `tool-update` chunks live as updates
+ * arrive. Non-streaming callers iterate and discard yields, capturing only
+ * the final return value — same tool definition works in both modes.
+ */
+export async function* executeMaybeStreaming(tool, args, ctx) {
+    const execute = tool.execute;
+    if (!execute) {
+        throw new Error('Tool has no execute function');
+    }
+    const ret = execute(args, ctx);
+    if (isAsyncGenerator(ret)) {
+        while (true) {
+            const step = await ret.next();
+            if (step.done)
+                return step.value;
+            yield step.value;
+        }
+    }
+    return await ret;
+}
+/**
+ * Validate a tool call's arguments against the tool's `inputSchema`. On
+ * success, the parsed value is returned — zod transforms (`.transform`,
+ * `.default`, type coercion) are applied, so `execute` receives the
+ * canonical shape the schema describes. On failure, a structured error
+ * suitable for feeding back to the model is returned.
+ */
+export function validateToolArgs(tool, args) {
+    const parsed = tool.definition.inputSchema.safeParse(args);
+    if (parsed.success) {
+        return { ok: true, value: parsed.data };
+    }
+    return {
+        ok: false,
+        error: {
+            error: 'invalid_arguments',
+            message: `Tool "${tool.definition.name}" received arguments that did not match its inputSchema.`,
+            issues: parsed.error.issues.map(i => ({
+                path: i.path.map(seg => String(seg)).join('.'),
+                message: i.message,
+            })),
+        },
+    };
+}
+/**
+ * Default stringification used for the `tool` role message content when a
+ * tool has no `toModelOutput` transform: pass through strings, JSON-encode
+ * everything else.
+ */
+export function defaultStringify(value) {
+    return typeof value === 'string' ? value : JSON.stringify(value);
+}
+/**
+ * Convert a tool's structured `result` into the string the **model** will
+ * see on its next step. Honors `tool.toModelOutput` when present, falling
+ * back to {@link defaultStringify}.
+ *
+ * Per R6 in the ai-loop-parity plan: a throwing `toModelOutput` MUST NOT
+ * crash the loop. We swallow the error, route it through `onError`
+ * middleware so it stays observable, and use the default stringification
+ * as a safety net.
+ */
+export async function applyToModelOutput(tool, result, onError) {
+    if (tool.toModelOutput) {
+        try {
+            return await tool.toModelOutput(result);
+        }
+        catch (err) {
+            if (onError)
+                await onError(err);
+        }
+    }
+    return defaultStringify(result);
+}
+/**
+ * Resolve `needsApproval` for a tool call, taking into account the
+ * client-supplied `approvedToolCallIds` / `rejectedToolCallIds` lists.
+ *
+ * Returns:
+ * - `'allow'`     — execute the tool normally (default; also when approved)
+ * - `'pending'`   — needsApproval is truthy and the call has not been approved
+ * - `'rejected'`  — the call appears in `rejectedToolCallIds`
+ */
+export async function evaluateApproval(tool, tc, options) {
+    const needs = tool.definition.needsApproval;
+    const requires = typeof needs === 'function' ? await needs(tc.arguments) : !!needs;
+    if (!requires)
+        return 'allow';
+    if (options?.rejectedToolCallIds?.includes(tc.id))
+        return 'rejected';
+    if (options?.approvedToolCallIds?.includes(tc.id))
+        return 'allow';
+    return 'pending';
+}
+//# sourceMappingURL=tool-helpers.js.map

package/dist/tool-helpers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-helpers.js","sourceRoot":"","sources":["../src/tool-helpers.ts"],"names":[],"mappings":"AAEA;;;;;GAKG;AACH,MAAM,UAAU,gBAAgB,CAAC,KAAc;IAC7C,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAA;IAC7D,MAAM,CAAC,GAAG,KAA+E,CAAA;IACzF,OAAO,OAAO,CAAC,CAAC,IAAI,KAAK,UAAU;WAC9B,OAAO,CAAC,CAAC,MAAM,KAAK,UAAU;WAC9B,OAAO,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,UAAU,CAAA;AACpD,CAAC;AAED;;;;;;;;;;;GAWG;AACH,MAAM,CAAC,KAAK,SAAS,CAAC,CAAC,qBAAqB,CAC1C,IAAU,EACV,IAA6B,EAC7B,GAAoB;IAEpB,MAAM,OAAO,GAAG,IAAI,CAAC,OAER,CAAA;IACb,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAA;IACjD,CAAC;IACD,MAAM,GAAG,GAAG,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAA;IAC9B,IAAI,gBAAgB,CAAC,GAAG,CAAC,EAAE,CAAC;QAC1B,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAA;YAC7B,IAAI,IAAI,CAAC,IAAI;gBAAE,OAAO,IAAI,CAAC,KAAK,CAAA;YAChC,MAAM,IAAI,CAAC,KAAK,CAAA;QAClB,CAAC;IACH,CAAC;IACD,OAAO,MAAM,GAAG,CAAA;AAClB,CAAC;AAcD;;;;;;GAMG;AACH,MAAM,UAAU,gBAAgB,CAC9B,IAAU,EACV,IAA6B;IAE7B,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,SAAS,CAAC,IAAI,CAAC,CAAA;IAC1D,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,CAAC,IAA+B,EAAE,CAAA;IACpE,CAAC;IACD,OAAO;QACL,EAAE,EAAE,KAAK;QACT,KAAK,EAAE;YACL,KAAK,EAAE,mBAAmB;YAC1B,OAAO,EAAE,SAAS,IAAI,CAAC,UAAU,CAAC,IAAI,0DAA0D;YAChG,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACpC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;gBAC9C,OAAO,EAAE,CAAC,CAAC,OAAO;aACnB,CAAC,CAAC;SACJ;KACF,CAAA;AACH,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,gBAAgB,CAAC,KAAc;IAC7C,OAAO,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAA;AAClE,CAAC;AAED;;;;;;;;;GASG;AACH,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,IAAU,EACV,MAAe,EACf,OAAgD;IAEhD,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;QACvB,IAAI,CAAC;YACH,OAAO,MAAO,IAAI,CAAC,aAA0D,CAAC,MAAM,CAAC,CAAA;QACvF,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,OAAO;gBAAE,MAAM,OAAO,CAAC,GAAG,CAAC,CAAA;QACjC,CAAC;IACH,CAAC;IACD,OAAO,gBAAgB,CAAC,MAAM,CAAC,CAAA;AACjC,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,IAAU,EACV,EAAY,EACZ,OAAuC;IAEvC,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,aAAa,CAAA;IAC3C,MAAM,QAAQ,GAAG,OAAO,KAAK,KAAK,UAAU,CAAC,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAA;IAClF,IAAI,CAAC,QAAQ;QAAE,OAAO,OAAO,CAAA;IAE7B,IAAI,OAAO,EAAE,mBAAmB,EAAE,QAAQ,CAAC,EAAE,CAAC,EAAE,CAAC;QAAE,OAAO,UAAU,CAAA;IACpE,IAAI,OAAO,EAAE,mBAAmB,EAAE,QAAQ,CAAC,EAAE,CAAC,EAAE,CAAC;QAAE,OAAO,OAAO,CAAA;IACjE,OAAO,SAAS,CAAA;AAClB,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rudderjs/ai",
-  "version": "1.6.0",
+  "version": "1.6.2",
   "description": "AI engine — providers, agents, tools, streaming, middleware",
   "rudderjs": {
     "provider": "AiProvider",
@@ -76,7 +76,7 @@
   "peerDependencies": {
     "@modelcontextprotocol/sdk": "^1.13.0",
     "@rudderjs/core": "^1.1.3",
-    "@rudderjs/orm": "^1.9.0"
+    "@rudderjs/orm": "^1.9.1"
   },
   "peerDependenciesMeta": {
     "@rudderjs/core": {
@@ -102,7 +102,7 @@
     "typescript": "^5.4.0",
     "@rudderjs/core": "^1.1.3",
     "@rudderjs/console": "^1.0.1",
-    "@rudderjs/orm": "^1.9.0"
+    "@rudderjs/orm": "^1.9.1"
   },
   "author": "Suleiman Shahbari",
   "scripts": {