@rudderjs/ai 1.17.3 → 1.18.0

package/dist/similarity-search.d.ts

@@ -1,163 +0,0 @@
-/**
- * `similaritySearch({ model, column, embedWith, ... })` — agent-tool
- * factory that wraps an `@rudderjs/orm` Model + a vector column into a
- * drop-in `Tool` an agent can call to retrieve semantically similar
- * rows (#B7 Phase 2).
- *
- * Composes B7 Phase 1's ORM primitives (`whereVectorSimilarTo` +
- * `selectVectorDistance`) with `AI.embed()`. The model emits a
- * natural-language `query`; the tool embeds it, runs a vector search,
- * and returns the top-K rows ranked by similarity.
- *
- * @example
- * ```ts
- * import { similaritySearch } from '@rudderjs/ai'
- * import { Document } from './app/Models/Document.js'
- *
- * class KnowledgeAgent extends Agent {
- *   tools() {
- *     return [
- *       similaritySearch({
- *         model:         Document,
- *         column:        'embedding',
- *         embedWith:     'openai/text-embedding-3-small',
- *         minSimilarity: 0.7,
- *         limit:         10,
- *       }),
- *     ]
- *   }
- * }
- * ```
- *
- * Phase 2 limitations (lifted in Phase 2.5):
- * - **Standalone queries only.** No `scope`/`.where()` chaining yet —
- *   the underlying `whereVectorSimilarTo` query must run alone. Agents
- *   see every row in the corpus that matches the vector. Phase 2.5
- *   adds a `scope: (q) => q.where(...)` callback.
- * - **Cosine-centric similarity.** `similarity` is reported as
- *   `1 - distance` regardless of `metric`. For non-cosine metrics
- *   (`l2`, `inner-product`) the value is internally consistent with
- *   the `minSimilarity` filter applied by the adapter but isn't a
- *   meaningful normalized score on its own.
- */
-import type { ServerToolBuilder } from './tool.js';
-/**
- * Structural type for the model class similaritySearch accepts.
- *
- * Declared locally instead of importing `Model` from `@rudderjs/orm`
- * so the main entry stays free of orm runtime — the tool calls
- * `model.query()` and never references the `@rudderjs/orm` package
- * itself. The user's app brings its own Model class.
- */
-export interface SimilaritySearchModel<TInstance> {
-    readonly name: string;
-    query(): SimilaritySearchQueryBuilder<TInstance>;
-}
-/**
- * WhereOperator strings the `scope` callback may pass to `.where()`.
- * Mirrors `@rudderjs/contracts`'s `WhereOperator`. Duplicated here so the
- * main entry has no compile-time `@rudderjs/contracts` dep.
- */
-export type SimilaritySearchWhereOperator = '=' | '!=' | '>' | '>=' | '<' | '<=' | 'LIKE' | 'NOT LIKE' | 'IN' | 'NOT IN';
-/**
- * Structural type for the QueryBuilder methods similaritySearch needs.
- * Mirrors a subset of `@rudderjs/contracts`'s `QueryBuilder<T>` so apps
- * writing a `scope` callback get autocomplete on the methods that actually
- * compose with `whereVectorSimilarTo` (#B7 Phase 2.5):
- *
- * - `.where()` / `.orWhere()` chains compose into the SQL with positional
- *   parameter binding.
- * - `.with()` / `.orderBy()` / `.whereGroup()` / `.whereHas()` still throw
- *   when paired with a vector clause — out of scope for `scope`.
- */
-export interface SimilaritySearchQueryBuilder<TInstance> {
-    where(column: string, value: unknown): SimilaritySearchQueryBuilder<TInstance>;
-    where(column: string, operator: SimilaritySearchWhereOperator, value: unknown): SimilaritySearchQueryBuilder<TInstance>;
-    orWhere(column: string, value: unknown): SimilaritySearchQueryBuilder<TInstance>;
-    orWhere(column: string, operator: SimilaritySearchWhereOperator, value: unknown): SimilaritySearchQueryBuilder<TInstance>;
-    withTrashed?(): SimilaritySearchQueryBuilder<TInstance>;
-    onlyTrashed?(): SimilaritySearchQueryBuilder<TInstance>;
-    whereVectorSimilarTo?(column: string, query: number[] | string, opts?: {
-        metric?: 'cosine' | 'l2' | 'inner-product';
-        minSimilarity?: number;
-        embedWith?: string;
-    }): SimilaritySearchQueryBuilder<TInstance>;
-    selectVectorDistance?(column: string, query: number[], alias: string): SimilaritySearchQueryBuilder<TInstance>;
-    limit(n: number): SimilaritySearchQueryBuilder<TInstance>;
-    get(): Promise<TInstance[]>;
-}
-export interface SimilarityHit<TInstance> {
-    readonly row: TInstance;
-    /**
-     * Higher = closer. For `metric: 'cosine'` (default) this is
-     * `1 - distance` ∈ [-1, 1]. For other metrics see the JSDoc on
-     * {@link similaritySearch}.
-     */
-    readonly similarity: number;
-}
-export interface SimilaritySearchOptions<TInstance> {
-    /** The Model class whose rows will be searched. */
-    model: SimilaritySearchModel<TInstance>;
-    /** The column on `model` declared with the `vector({ dimensions })` cast. */
-    column: string;
-    /**
-     * Embedding model id (`<provider>/<model>`). Required — fails loud
-     * at factory construction time if missing so apps don't accidentally
-     * route to whatever `AiRegistry.getDefault()` happens to be.
-     */
-    embedWith: string;
-    /** Default `'cosine'`. */
-    metric?: 'cosine' | 'l2' | 'inner-product';
-    /**
-     * Drops rows whose `1 - distance` falls below this threshold. Cosine
-     * range is `[-1, 1]` so values near `0.7–0.9` are typical for
-     * "relevant" documents.
-     */
-    minSimilarity?: number;
-    /** Default `10`. */
-    limit?: number;
-    /**
-     * Pre-filter the query before the vector clause attaches. Receives a
-     * fresh `model.query()` builder; return the chained builder. Use for
-     * tenancy (`.where('tenantId', userId)`), publication state
-     * (`.where('published', true)`), or any flat predicate that should
-     * narrow the search corpus.
-     *
-     * **Phase 2.5 limits:** flat `.where()` / `.orWhere()` chains compose
-     * into the SQL. `whereGroup` / `whereHas` / `with()` still throw at
-     * terminal time — keep the scope flat.
-     *
-     * @example
-     * ```ts
-     * similaritySearch({
-     *   model:     KnowledgeArticle,
-     *   column:    'embedding',
-     *   embedWith: 'openai/text-embedding-3-small',
-     *   scope:     q => q.where('tenantId', currentTenant).where('published', true),
-     * })
-     * ```
-     */
-    scope?: (q: SimilaritySearchQueryBuilder<TInstance>) => SimilaritySearchQueryBuilder<TInstance>;
-    /**
-     * Override the tool name. Default
-     * `similarity_search_<model_name_lowercase>`.
-     */
-    name?: string;
-    /** Override the tool description. */
-    description?: string;
-    /**
-     * Custom string projection for each hit, replaces the default
-     * `(0.87) {"id":1,"content":"..."}` shape that the model sees on its
-     * next step. The structured array still flows to the UI via the
-     * `tool-result` chunk.
-     */
-    projectResult?: (row: TInstance, similarity: number) => string;
-}
-/**
- * Build a `Tool` that embeds a natural-language query and returns the
- * top-K similar rows from `model.column`.
- */
-export declare function similaritySearch<TInstance>(opts: SimilaritySearchOptions<TInstance>): ServerToolBuilder<{
-    query: string;
-}, SimilarityHit<TInstance>[]>;
-//# sourceMappingURL=similarity-search.d.ts.map

package/dist/similarity-search.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"similarity-search.d.ts","sourceRoot":"","sources":["../src/similarity-search.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAyCG;AAKH,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,WAAW,CAAA;AAElD;;;;;;;GAOG;AACH,MAAM,WAAW,qBAAqB,CAAC,SAAS;IAC9C,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAA;IACrB,KAAK,IAAI,4BAA4B,CAAC,SAAS,CAAC,CAAA;CACjD;AAED;;;;GAIG;AACH,MAAM,MAAM,6BAA6B,GACrC,GAAG,GAAG,IAAI,GAAG,GAAG,GAAG,IAAI,GAAG,GAAG,GAAG,IAAI,GACpC,MAAM,GAAG,UAAU,GACnB,IAAI,GAAG,QAAQ,CAAA;AAEnB;;;;;;;;;;GAUG;AACH,MAAM,WAAW,4BAA4B,CAAC,SAAS;IACrD,KAAK,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,GAAG,4BAA4B,CAAC,SAAS,CAAC,CAAA;IAC9E,KAAK,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,6BAA6B,EAAE,KAAK,EAAE,OAAO,GAAG,4BAA4B,CAAC,SAAS,CAAC,CAAA;IACvH,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,GAAG,4BAA4B,CAAC,SAAS,CAAC,CAAA;IAChF,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,6BAA6B,EAAE,KAAK,EAAE,OAAO,GAAG,4BAA4B,CAAC,SAAS,CAAC,CAAA;IACzH,WAAW,CAAC,IAAI,4BAA4B,CAAC,SAAS,CAAC,CAAA;IACvD,WAAW,CAAC,IAAI,4BAA4B,CAAC,SAAS,CAAC,CAAA;IACvD,oBAAoB,CAAC,CACnB,MAAM,EAAE,MAAM,EACd,KAAK,EAAG,MAAM,EAAE,GAAG,MAAM,EACzB,IAAI,CAAC,EAAG;QAAE,MAAM,CAAC,EAAE,QAAQ,GAAG,IAAI,GAAG,eAAe,CAAC;QAAC,aAAa,CAAC,EAAE,MAAM,CAAC;QAAC,SAAS,CAAC,EAAE,MAAM,CAAA;KAAE,GACjG,4BAA4B,CAAC,SAAS,CAAC,CAAA;IAC1C,oBAAoB,CAAC,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,MAAM,GAAG,4BAA4B,CAAC,SAAS,CAAC,CAAA;IAC9G,KAAK,CAAC,CAAC,EAAE,MAAM,GAAG,4BAA4B,CAAC,SAAS,CAAC,CAAA;IACzD,GAAG,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC,CAAA;CAC5B;AAED,MAAM,WAAW,aAAa,CAAC,SAAS;IACtC,QAAQ,CAAC,GAAG,EAAE,SAAS,CAAA;IACvB;;;;OAIG;IACH,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAA;CAC5B;AAED,MAAM,WAAW,uBAAuB,CAAC,SAAS;IAChD,mDAAmD;IACnD,KAAK,EAAE,qBAAqB,CAAC,SAAS,CAAC,CAAA;IAEvC,6EAA6E;IAC7E,MAAM,EAAE,MAAM,CAAA;IAEd;;;;OAIG;IACH,SAAS,EAAE,MAAM,CAAA;IAEjB,0BAA0B;IAC1B,MAAM,CAAC,EAAE,QAAQ,GAAG,IAAI,GAAG,eAAe,CAAA;IAE1C;;;;OAIG;IACH,aAAa,CAAC,EAAE,MAAM,CAAA;IAEtB,oBAAoB;IACpB,KAAK,CAAC,EAAE,MAAM,CAAA;IAEd;;;;;;;;;;;;;;;;;;;;OAoBG;IACH,KAAK,CAAC,EAAE,CAAC,CAAC,EAAE,4BAA4B,CAAC,SAAS,CAAC,KAAK,4BAA4B,CAAC,SAAS,CAAC,CAAA;IAE/F;;;OAGG;IACH,IAAI,CAAC,EAAE,MAAM,CAAA;IAEb,qCAAqC;IACrC,WAAW,CAAC,EAAE,MAAM,CAAA;IAEpB;;;;;OAKG;IACH,aAAa,CAAC,EAAE,CAAC,GAAG,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,KAAK,MAAM,CAAA;CAC/D;AAKD;;;GAGG;AACH,wBAAgB,gBAAgB,CAAC,SAAS,EACxC,IAAI,EAAE,uBAAuB,CAAC,SAAS,CAAC,GACvC,iBAAiB,CAAC;IAAE,KAAK,EAAE,MAAM,CAAA;CAAE,EAAE,aAAa,CAAC,SAAS,CAAC,EAAE,CAAC,CA0FlE"}

package/dist/similarity-search.js

@@ -1,147 +0,0 @@
-/**
- * `similaritySearch({ model, column, embedWith, ... })` — agent-tool
- * factory that wraps an `@rudderjs/orm` Model + a vector column into a
- * drop-in `Tool` an agent can call to retrieve semantically similar
- * rows (#B7 Phase 2).
- *
- * Composes B7 Phase 1's ORM primitives (`whereVectorSimilarTo` +
- * `selectVectorDistance`) with `AI.embed()`. The model emits a
- * natural-language `query`; the tool embeds it, runs a vector search,
- * and returns the top-K rows ranked by similarity.
- *
- * @example
- * ```ts
- * import { similaritySearch } from '@rudderjs/ai'
- * import { Document } from './app/Models/Document.js'
- *
- * class KnowledgeAgent extends Agent {
- *   tools() {
- *     return [
- *       similaritySearch({
- *         model:         Document,
- *         column:        'embedding',
- *         embedWith:     'openai/text-embedding-3-small',
- *         minSimilarity: 0.7,
- *         limit:         10,
- *       }),
- *     ]
- *   }
- * }
- * ```
- *
- * Phase 2 limitations (lifted in Phase 2.5):
- * - **Standalone queries only.** No `scope`/`.where()` chaining yet —
- *   the underlying `whereVectorSimilarTo` query must run alone. Agents
- *   see every row in the corpus that matches the vector. Phase 2.5
- *   adds a `scope: (q) => q.where(...)` callback.
- * - **Cosine-centric similarity.** `similarity` is reported as
- *   `1 - distance` regardless of `metric`. For non-cosine metrics
- *   (`l2`, `inner-product`) the value is internally consistent with
- *   the `minSimilarity` filter applied by the adapter but isn't a
- *   meaningful normalized score on its own.
- */
-import { z } from 'zod';
-import { toolDefinition } from './tool.js';
-import { AI } from './facade.js';
-/** Internal alias used to read distance back off each row. */
-const SIMILARITY_DISTANCE_ALIAS = '__rudderjs_similarity_distance__';
-/**
- * Build a `Tool` that embeds a natural-language query and returns the
- * top-K similar rows from `model.column`.
- */
-export function similaritySearch(opts) {
-    const { model, column, embedWith, metric = 'cosine', minSimilarity, limit = 10, name, description, projectResult, scope, } = opts;
-    if (!embedWith || typeof embedWith !== 'string') {
-        throw new Error('[Rudder AI] similaritySearch requires opts.embedWith (e.g. "openai/text-embedding-3-small"). ' +
-            'No default — fail loud so embeddings never silently route through whichever provider happens to be the AI default.');
-    }
-    if (!column || typeof column !== 'string') {
-        throw new Error('[Rudder AI] similaritySearch requires opts.column — the Model column declared with vector({ dimensions }).');
-    }
-    if (!model || typeof model.query !== 'function') {
-        throw new Error('[Rudder AI] similaritySearch requires opts.model — a Model class with a static query() method.');
-    }
-    if (limit <= 0 || !Number.isFinite(limit)) {
-        throw new Error(`[Rudder AI] similaritySearch limit must be a positive finite number; got ${String(limit)}.`);
-    }
-    const toolName = name ?? `similarity_search_${model.name.toLowerCase()}`;
-    const toolDescription = description ?? `Semantic search over ${model.name} records. Pass a natural-language \`query\` string; the most similar rows are returned.`;
-    return toolDefinition({
-        name: toolName,
-        description: toolDescription,
-        inputSchema: z.object({
-            query: z.string().min(1).describe('Natural-language query to search for.'),
-        }),
-    })
-        .server(async ({ query }) => {
-        const embedResult = await AI.embed(query, { model: embedWith });
-        const vector = embedResult.embeddings[0];
-        if (!vector || vector.length === 0) {
-            throw new Error(`[Rudder AI] similaritySearch: AI.embed("${query}", { model: "${embedWith}" }) returned no embedding.`);
-        }
-        const baseQb = model.query();
-        const scopedQb = scope ? scope(baseQb) : baseQb;
-        const whereVec = scopedQb.whereVectorSimilarTo;
-        const selectDist = scopedQb.selectVectorDistance;
-        if (typeof whereVec !== 'function' || typeof selectDist !== 'function') {
-            throw new Error(`[Rudder AI] similaritySearch: ${model.name}'s ORM adapter does not implement vector queries. ` +
-                'Use @rudderjs/orm-prisma against a Postgres + pgvector connection.');
-        }
-        const vectorOpts = { metric };
-        if (minSimilarity !== undefined)
-            vectorOpts.minSimilarity = minSimilarity;
-        const filtered = whereVec.call(scopedQb, column, vector, vectorOpts);
-        const projected = selectDist.call(filtered, column, vector, SIMILARITY_DISTANCE_ALIAS);
-        const rows = await projected.limit(limit).get();
-        return rows.map((row) => {
-            const distance = readDistance(row);
-            return { row, similarity: 1 - distance };
-        });
-    })
-        .modelOutput((results) => {
-        if (results.length === 0) {
-            return `No similar ${model.name} records found.`;
-        }
-        if (projectResult) {
-            return results
-                .map(({ row, similarity }) => projectResult(row, similarity))
-                .join('\n');
-        }
-        return results
-            .map(({ row, similarity }) => {
-            const json = serializeRow(row);
-            return `(${similarity.toFixed(2)}) ${json}`;
-        })
-            .join('\n');
-    });
-}
-function readDistance(row) {
-    if (row === null || typeof row !== 'object')
-        return 0;
-    const raw = row[SIMILARITY_DISTANCE_ALIAS];
-    const n = typeof raw === 'number' ? raw : Number(raw);
-    return Number.isFinite(n) ? n : 0;
-}
-function serializeRow(row) {
-    if (row === null || typeof row !== 'object')
-        return JSON.stringify(row);
-    const candidate = row;
-    const toJSON = candidate.toJSON;
-    const data = typeof toJSON === 'function'
-        ? toJSON.call(candidate)
-        : sanitizeRow(row);
-    return JSON.stringify(data);
-}
-/**
- * Strip the internal distance alias from rows whose `toJSON` we can't
- * call (plain objects, fakes). `toJSON()`-bearing Models already
- * filter their own internal fields.
- */
-function sanitizeRow(row) {
-    if (!(SIMILARITY_DISTANCE_ALIAS in row))
-        return row;
-    const { [SIMILARITY_DISTANCE_ALIAS]: _, ...rest } = row;
-    void _;
-    return rest;
-}
-//# sourceMappingURL=similarity-search.js.map

package/dist/similarity-search.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"similarity-search.js","sourceRoot":"","sources":["../src/similarity-search.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAyCG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AACvB,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAA;AAC1C,OAAO,EAAE,EAAE,EAAE,MAAM,aAAa,CAAA;AAoIhC,8DAA8D;AAC9D,MAAM,yBAAyB,GAAG,kCAAkC,CAAA;AAEpE;;;GAGG;AACH,MAAM,UAAU,gBAAgB,CAC9B,IAAwC;IAExC,MAAM,EACJ,KAAK,EACL,MAAM,EACN,SAAS,EACT,MAAM,GAAG,QAAQ,EACjB,aAAa,EACb,KAAK,GAAG,EAAE,EACV,IAAI,EACJ,WAAW,EACX,aAAa,EACb,KAAK,GACN,GAAG,IAAI,CAAA;IAER,IAAI,CAAC,SAAS,IAAI,OAAO,SAAS,KAAK,QAAQ,EAAE,CAAC;QAChD,MAAM,IAAI,KAAK,CACb,+FAA+F;YAC/F,oHAAoH,CACrH,CAAA;IACH,CAAC;IACD,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC1C,MAAM,IAAI,KAAK,CAAC,4GAA4G,CAAC,CAAA;IAC/H,CAAC;IACD,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,CAAC,KAAK,KAAK,UAAU,EAAE,CAAC;QAChD,MAAM,IAAI,KAAK,CAAC,gGAAgG,CAAC,CAAA;IACnH,CAAC;IACD,IAAI,KAAK,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;QAC1C,MAAM,IAAI,KAAK,CAAC,4EAA4E,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;IAC/G,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,IAAI,qBAAqB,KAAK,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,CAAA;IACxE,MAAM,eAAe,GAAG,WAAW,IAAI,wBAAwB,KAAK,CAAC,IAAI,yFAAyF,CAAA;IAElK,OAAO,cAAc,CAAC;QACpB,IAAI,EAAS,QAAQ;QACrB,WAAW,EAAE,eAAe;QAC5B,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;YACpB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,uCAAuC,CAAC;SAC3E,CAAC;KACH,CAAC;SACC,MAAM,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,EAAuC,EAAE;QAC/D,MAAM,WAAW,GAAG,MAAM,EAAE,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAA;QAC/D,MAAM,MAAM,GAAG,WAAW,CAAC,UAAU,CAAC,CAAC,CAAC,CAAA;QACxC,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnC,MAAM,IAAI,KAAK,CACb,2CAA2C,KAAK,gBAAgB,SAAS,6BAA6B,CACvG,CAAA;QACH,CAAC;QAED,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,EAAE,CAAA;QAC5B,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAA;QAC/C,MAAM,QAAQ,GAAG,QAAQ,CAAC,oBAAoB,CAAA;QAC9C,MAAM,UAAU,GAAG,QAAQ,CAAC,oBAAoB,CAAA;QAChD,IAAI,OAAO,QAAQ,KAAK,UAAU,IAAI,OAAO,UAAU,KAAK,UAAU,EAAE,CAAC;YACvE,MAAM,IAAI,KAAK,CACb,iCAAiC,KAAK,CAAC,IAAI,oDAAoD;gBAC/F,oEAAoE,CACrE,CAAA;QACH,CAAC;QAED,MAAM,UAAU,GAA0E,EAAE,MAAM,EAAE,CAAA;QACpG,IAAI,aAAa,KAAK,SAAS;YAAE,UAAU,CAAC,aAAa,GAAG,aAAa,CAAA;QAEzE,MAAM,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,CAAC,CAAA;QACpE,MAAM,SAAS,GAAG,UAAU,CAAC,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,yBAAyB,CAAC,CAAA;QACtF,MAAM,IAAI,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,CAAA;QAE/C,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAA4B,EAAE;YAChD,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,CAAA;YAClC,OAAO,EAAE,GAAG,EAAE,UAAU,EAAE,CAAC,GAAG,QAAQ,EAAE,CAAA;QAC1C,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC;SACD,WAAW,CAAC,CAAC,OAAO,EAAE,EAAE;QACvB,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,OAAO,cAAc,KAAK,CAAC,IAAI,iBAAiB,CAAA;QAClD,CAAC;QAED,IAAI,aAAa,EAAE,CAAC;YAClB,OAAO,OAAO;iBACX,GAAG,CAAC,CAAC,EAAE,GAAG,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,aAAa,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;iBAC5D,IAAI,CAAC,IAAI,CAAC,CAAA;QACf,CAAC;QAED,OAAO,OAAO;aACX,GAAG,CAAC,CAAC,EAAE,GAAG,EAAE,UAAU,EAAE,EAAE,EAAE;YAC3B,MAAM,IAAI,GAAG,YAAY,CAAC,GAAG,CAAC,CAAA;YAC9B,OAAO,IAAI,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAA;QAC7C,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC,CAAA;IACf,CAAC,CAAC,CAAA;AACN,CAAC;AAED,SAAS,YAAY,CAAC,GAAY;IAChC,IAAI,GAAG,KAAK,IAAI,IAAI,OAAO,GAAG,KAAK,QAAQ;QAAE,OAAO,CAAC,CAAA;IACrD,MAAM,GAAG,GAAI,GAA+B,CAAC,yBAAyB,CAAC,CAAA;IACvE,MAAM,CAAC,GAAG,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAA;IACrD,OAAO,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;AACnC,CAAC;AAED,SAAS,YAAY,CAAC,GAAY;IAChC,IAAI,GAAG,KAAK,IAAI,IAAI,OAAO,GAAG,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAA;IACvE,MAAM,SAAS,GAAG,GAAiC,CAAA;IACnD,MAAM,MAAM,GAAG,SAAS,CAAC,MAAM,CAAA;IAC/B,MAAM,IAAI,GAAG,OAAO,MAAM,KAAK,UAAU;QACvC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC;QACxB,CAAC,CAAC,WAAW,CAAC,GAA8B,CAAC,CAAA;IAC/C,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAA;AAC7B,CAAC;AAED;;;;GAIG;AACH,SAAS,WAAW,CAAC,GAA4B;IAC/C,IAAI,CAAC,CAAC,yBAAyB,IAAI,GAAG,CAAC;QAAE,OAAO,GAAG,CAAA;IACnD,MAAM,EAAE,CAAC,yBAAyB,CAAC,EAAE,CAAC,EAAE,GAAG,IAAI,EAAE,GAAG,GAAG,CAAA;IACvD,KAAK,CAAC,CAAA;IACN,OAAO,IAAI,CAAA;AACb,CAAC"}

package/dist/sub-agent-run-store.d.ts

@@ -1,157 +0,0 @@
-import type { AiMessage, ToolCall } from './types.js';
-/**
- * Discriminator for the kind of pause a snapshot represents. Determines
- * what payload `Agent.resumeAsTool` expects on continuation:
- *
- * - `'client_tool'` — resume must carry one tool-result per id in
- *   `pendingToolCallIds`. This is the original v1.4 behaviour and the
- *   default when the field is absent.
- * - `'approval'` — resume must carry `approvedToolCallIds` and/or
- *   `rejectedToolCallIds` covering the single id in `pendingToolCallIds`
- *   (the inner approval-gated tool call).
- */
-export type SubAgentPauseKind = 'client_tool' | 'approval';
-/**
- * Snapshot of a paused sub-agent run, persisted between an
- * {@link Agent.asTool} pause and an {@link Agent.resumeAsTool} resume.
- *
- * The shape is intentionally simple — `messages` is the full inner
- * conversation up to the pause point (system prompt + user input +
- * every interleaved tool result), so resume only needs to append the
- * incoming client-tool results (or inject approval decisions) and
- * re-enter the loop in `messages` mode.
- */
-export interface SubAgentRunSnapshot {
-    /** Inner-agent message history at suspend time. */
-    messages: AiMessage[];
-    /**
-     * Tool-call ids the sub-agent is waiting on.
-     *
-     * - `pauseKind === 'client_tool'` (default): one entry per client tool
-     *   the inner loop surfaced; resume appends one result per id.
-     * - `pauseKind === 'approval'`: a single entry for the approval-gated
-     *   tool call; resume injects the id into `approvedToolCallIds`
-     *   (or `rejectedToolCallIds`).
-     */
-    pendingToolCallIds: string[];
-    /** Total steps the inner agent has executed across all suspends so far. */
-    stepsSoFar: number;
-    /** Total prompt+completion tokens accumulated across all suspends. */
-    tokensSoFar: number;
-    /**
-     * Discriminator for the resume contract. Defaults to `'client_tool'`
-     * when absent so older v1.4 snapshots remain readable on disk/redis
-     * after the host upgrades to a version that knows about approval pauses.
-     */
-    pauseKind?: SubAgentPauseKind;
-    /**
-     * Approval pauses only. The full pending tool-call payload (name + args
-     * + id) so a renderer can show "approve `delete_user(id=42)`?" without
-     * round-tripping back to the inner agent. Mirrors the structure of
-     * `AgentResponse.pendingApprovalToolCall`.
-     */
-    pendingApprovalToolCall?: {
-        toolCall: ToolCall;
-        isClientTool: boolean;
-    };
-    /**
-     * Opaque metadata the host can pass through. The framework treats
-     * this as JSON and never reads it — useful for hosts that need to
-     * rehydrate context (e.g. `{ resourceSlug, recordId, fieldScope, userId }`)
-     * around the resume call.
-     */
-    meta?: unknown;
-}
-/**
- * Pluggable persistence backend for paused sub-agent runs. The framework
- * ships two reference implementations:
- *
- * - {@link InMemorySubAgentRunStore} — a `Map`-backed store. Single-process
- *   only; fine for unit tests and small dev setups, lossy across worker
- *   processes and restarts.
- * - {@link CachedSubAgentRunStore} — lazy adapter on top of `@rudderjs/cache`.
- *   Cross-process / cross-restart when the cache is configured with redis
- *   or any non-memory driver.
- *
- * Hosts may implement their own (Redis directly, Prisma, etc.) by
- * satisfying this interface.
- */
-export interface SubAgentRunStore {
-    /** Persist a snapshot under `subRunId`. Implementations MAY apply a TTL. */
-    store(subRunId: string, snapshot: SubAgentRunSnapshot): Promise<void>;
-    /**
-     * Atomic read + delete. Returns `null` if the id is unknown or the
-     * snapshot has expired. Single-use semantics matter: a forged or
-     * replayed `subRunId` must not return data twice.
-     */
-    consume(subRunId: string): Promise<SubAgentRunSnapshot | null>;
-    /**
-     * Non-destructive read. Returns the snapshot without deleting it, or
-     * `null` if the id is unknown or the snapshot has expired. Optional —
-     * for hosts that need a **validate-then-resume** pre-flight: inspect a
-     * paused snapshot's `meta` (ownership, resource-context, tool-result
-     * coverage) before handing the id to {@link Agent.resumeAsTool} /
-     * {@link Agent.resumeManyAsTool}, which own the single {@link consume}.
-     * The resume paths never call this — so a `load` then resume reads the
-     * snapshot once for validation and consumes it once on resume, with no
-     * `consume` then re-`store` round-trip. Mirrors {@link AgentRunStore.load}.
-     */
-    load?(subRunId: string): Promise<SubAgentRunSnapshot | null>;
-}
-/**
- * `Map`-backed implementation suitable for tests and single-process dev.
- * Loses state across restarts and worker processes — for any multi-worker
- * deployment, use {@link CachedSubAgentRunStore} or a custom backend.
- */
-export declare class InMemorySubAgentRunStore implements SubAgentRunStore {
-    private readonly snapshots;
-    store(subRunId: string, snapshot: SubAgentRunSnapshot): Promise<void>;
-    consume(subRunId: string): Promise<SubAgentRunSnapshot | null>;
-    load(subRunId: string): Promise<SubAgentRunSnapshot | null>;
-    /** Test helper — clears all snapshots without consuming. */
-    clear(): void;
-}
-/**
- * Minimal structural shape of a cache adapter (the methods this store
- * touches). Mirrors `@rudderjs/cache`'s `CacheAdapter` so the dep stays
- * structural — the framework's main entry stays runtime-agnostic.
- */
-interface CacheStoreLike {
-    get<T = unknown>(key: string): Promise<T | null>;
-    set(key: string, value: unknown, ttlSeconds?: number): Promise<void>;
-    forget(key: string): Promise<void>;
-}
-export interface CachedSubAgentRunStoreOptions {
-    /**
-     * Cache adapter to use. When omitted, the registry tries to load
-     * `@rudderjs/cache` lazily and falls back to the registered global
-     * adapter (`CacheRegistry.get()`); throws if neither resolves.
-     */
-    cache?: CacheStoreLike;
-    /** Key namespace prefix. Default `'rudderjs:ai:sub-agent-run:'`. */
-    keyPrefix?: string;
-    /** Time-to-live in seconds. Default 5 minutes. */
-    ttlSeconds?: number;
-}
-/**
- * Sub-agent run store backed by `@rudderjs/cache`. Loads the cache
- * adapter lazily so `@rudderjs/ai`'s main entry stays runtime-agnostic
- * (no static import on the cache package).
- *
- * Default TTL is 5 minutes — long enough for a browser to round-trip a
- * few client tool calls, short enough that abandoned runs garbage-collect
- * promptly and the storage bill stays bounded.
- */
-export declare class CachedSubAgentRunStore implements SubAgentRunStore {
-    private readonly explicitCache?;
-    private readonly keyPrefix;
-    private readonly ttlSeconds;
-    private resolvedCache?;
-    constructor(opts?: CachedSubAgentRunStoreOptions);
-    private getCache;
-    store(subRunId: string, snapshot: SubAgentRunSnapshot): Promise<void>;
-    consume(subRunId: string): Promise<SubAgentRunSnapshot | null>;
-    load(subRunId: string): Promise<SubAgentRunSnapshot | null>;
-}
-export {};
-//# sourceMappingURL=sub-agent-run-store.d.ts.map

package/dist/sub-agent-run-store.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"sub-agent-run-store.d.ts","sourceRoot":"","sources":["../src/sub-agent-run-store.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAA;AAErD;;;;;;;;;;GAUG;AACH,MAAM,MAAM,iBAAiB,GAAG,aAAa,GAAG,UAAU,CAAA;AAE1D;;;;;;;;;GASG;AACH,MAAM,WAAW,mBAAmB;IAClC,mDAAmD;IACnD,QAAQ,EAAY,SAAS,EAAE,CAAA;IAC/B;;;;;;;;OAQG;IACH,kBAAkB,EAAE,MAAM,EAAE,CAAA;IAC5B,2EAA2E;IAC3E,UAAU,EAAU,MAAM,CAAA;IAC1B,sEAAsE;IACtE,WAAW,EAAS,MAAM,CAAA;IAC1B;;;;OAIG;IACH,SAAS,CAAC,EAAU,iBAAiB,CAAA;IACrC;;;;;OAKG;IACH,uBAAuB,CAAC,EAAE;QAAE,QAAQ,EAAE,QAAQ,CAAC;QAAC,YAAY,EAAE,OAAO,CAAA;KAAE,CAAA;IACvE;;;;;OAKG;IACH,IAAI,CAAC,EAAE,OAAO,CAAA;CACf;AAED;;;;;;;;;;;;;GAaG;AACH,MAAM,WAAW,gBAAgB;IAC/B,4EAA4E;IAC5E,KAAK,CAAC,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IACrE;;;;OAIG;IACH,OAAO,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,mBAAmB,GAAG,IAAI,CAAC,CAAA;IAC9D;;;;;;;;;;OAUG;IACH,IAAI,CAAC,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,mBAAmB,GAAG,IAAI,CAAC,CAAA;CAC7D;AAID;;;;GAIG;AACH,qBAAa,wBAAyB,YAAW,gBAAgB;IAC/D,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAyC;IAE7D,KAAK,CAAC,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IAIrE,OAAO,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,mBAAmB,GAAG,IAAI,CAAC;IAO9D,IAAI,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,mBAAmB,GAAG,IAAI,CAAC;IAIjE,4DAA4D;IAC5D,KAAK,IAAI,IAAI;CAGd;AAID;;;;GAIG;AACH,UAAU,cAAc;IACtB,GAAG,CAAC,CAAC,GAAG,OAAO,EAAE,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC,CAAA;IAChD,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IACpE,MAAM,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;CACnC;AAED,MAAM,WAAW,6BAA6B;IAC5C;;;;OAIG;IACH,KAAK,CAAC,EAAM,cAAc,CAAA;IAC1B,oEAAoE;IACpE,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,kDAAkD;IAClD,UAAU,CAAC,EAAE,MAAM,CAAA;CACpB;AAED;;;;;;;;GAQG;AACH,qBAAa,sBAAuB,YAAW,gBAAgB;IAC7D,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAgB;IAC/C,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAa;IACvC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAY;IACvC,OAAO,CAAC,aAAa,CAAC,CAAyB;gBAEnC,IAAI,GAAE,6BAAkC;YAMtC,QAAQ;IAwBhB,KAAK,CAAC,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IAKrE,OAAO,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,mBAAmB,GAAG,IAAI,CAAC;IAS9D,IAAI,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,mBAAmB,GAAG,IAAI,CAAC;CAIlE"}

package/dist/sub-agent-run-store.js

@@ -1,87 +0,0 @@
-// ─── In-memory ─────────────────────────────────────────────
-/**
- * `Map`-backed implementation suitable for tests and single-process dev.
- * Loses state across restarts and worker processes — for any multi-worker
- * deployment, use {@link CachedSubAgentRunStore} or a custom backend.
- */
-export class InMemorySubAgentRunStore {
-    snapshots = new Map();
-    async store(subRunId, snapshot) {
-        this.snapshots.set(subRunId, snapshot);
-    }
-    async consume(subRunId) {
-        const snapshot = this.snapshots.get(subRunId);
-        if (!snapshot)
-            return null;
-        this.snapshots.delete(subRunId);
-        return snapshot;
-    }
-    async load(subRunId) {
-        return this.snapshots.get(subRunId) ?? null;
-    }
-    /** Test helper — clears all snapshots without consuming. */
-    clear() {
-        this.snapshots.clear();
-    }
-}
-/**
- * Sub-agent run store backed by `@rudderjs/cache`. Loads the cache
- * adapter lazily so `@rudderjs/ai`'s main entry stays runtime-agnostic
- * (no static import on the cache package).
- *
- * Default TTL is 5 minutes — long enough for a browser to round-trip a
- * few client tool calls, short enough that abandoned runs garbage-collect
- * promptly and the storage bill stays bounded.
- */
-export class CachedSubAgentRunStore {
-    explicitCache;
-    keyPrefix;
-    ttlSeconds;
-    resolvedCache;
-    constructor(opts = {}) {
-        if (opts.cache)
-            this.explicitCache = opts.cache;
-        this.keyPrefix = opts.keyPrefix ?? 'rudderjs:ai:sub-agent-run:';
-        this.ttlSeconds = opts.ttlSeconds ?? 5 * 60;
-    }
-    async getCache() {
-        if (this.resolvedCache)
-            return this.resolvedCache;
-        if (this.explicitCache) {
-            this.resolvedCache = this.explicitCache;
-            return this.resolvedCache;
-        }
-        // Lazy-import @rudderjs/cache and ask the registry for the active
-        // adapter. This keeps the static import surface zero — the import
-        // only fires when the host actually opts into suspendable sub-agents.
-        // We dodge static module-resolution by using an indirected specifier
-        // so `@rudderjs/cache` doesn't need to be a declared dep of
-        // `@rudderjs/ai` (it stays an optional runtime peer).
-        const cacheSpecifier = '@rudderjs/cache';
-        const mod = await import(/* @vite-ignore */ cacheSpecifier);
-        const adapter = mod.CacheRegistry?.get?.();
-        if (!adapter) {
-            throw new Error('[Rudder AI] CachedSubAgentRunStore needs a cache adapter. Install `@rudderjs/cache`, register a driver, or pass `{ cache }` explicitly.');
-        }
-        this.resolvedCache = adapter;
-        return adapter;
-    }
-    async store(subRunId, snapshot) {
-        const cache = await this.getCache();
-        await cache.set(this.keyPrefix + subRunId, snapshot, this.ttlSeconds);
-    }
-    async consume(subRunId) {
-        const cache = await this.getCache();
-        const key = this.keyPrefix + subRunId;
-        const snapshot = await cache.get(key);
-        if (!snapshot)
-            return null;
-        await cache.forget(key);
-        return snapshot;
-    }
-    async load(subRunId) {
-        const cache = await this.getCache();
-        return cache.get(this.keyPrefix + subRunId);
-    }
-}
-//# sourceMappingURL=sub-agent-run-store.js.map

package/dist/sub-agent-run-store.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"sub-agent-run-store.js","sourceRoot":"","sources":["../src/sub-agent-run-store.ts"],"names":[],"mappings":"AAqGA,8DAA8D;AAE9D;;;;GAIG;AACH,MAAM,OAAO,wBAAwB;IAClB,SAAS,GAAG,IAAI,GAAG,EAA+B,CAAA;IAEnE,KAAK,CAAC,KAAK,CAAC,QAAgB,EAAE,QAA6B;QACzD,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAA;IACxC,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,QAAgB;QAC5B,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;QAC7C,IAAI,CAAC,QAAQ;YAAE,OAAO,IAAI,CAAA;QAC1B,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAA;QAC/B,OAAO,QAAQ,CAAA;IACjB,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,QAAgB;QACzB,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAA;IAC7C,CAAC;IAED,4DAA4D;IAC5D,KAAK;QACH,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAA;IACxB,CAAC;CACF;AA4BD;;;;;;;;GAQG;AACH,MAAM,OAAO,sBAAsB;IAChB,aAAa,CAAiB;IAC9B,SAAS,CAAa;IACtB,UAAU,CAAY;IAC/B,aAAa,CAA0B;IAE/C,YAAY,OAAsC,EAAE;QAClD,IAAI,IAAI,CAAC,KAAK;YAAE,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,KAAK,CAAA;QAC/C,IAAI,CAAC,SAAS,GAAI,IAAI,CAAC,SAAS,IAAK,4BAA4B,CAAA;QACjE,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,IAAI,CAAC,GAAG,EAAE,CAAA;IAC7C,CAAC;IAEO,KAAK,CAAC,QAAQ;QACpB,IAAI,IAAI,CAAC,aAAa;YAAE,OAAO,IAAI,CAAC,aAAa,CAAA;QACjD,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACvB,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa,CAAA;YACvC,OAAO,IAAI,CAAC,aAAa,CAAA;QAC3B,CAAC;QACD,kEAAkE;QAClE,kEAAkE;QAClE,sEAAsE;QACtE,qEAAqE;QACrE,4DAA4D;QAC5D,sDAAsD;QACtD,MAAM,cAAc,GAAG,iBAAiB,CAAA;QACxC,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,cAAc,CAEzD,CAAA;QACD,MAAM,OAAO,GAAG,GAAG,CAAC,aAAa,EAAE,GAAG,EAAE,EAAE,CAAA;QAC1C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,yIAAyI,CAAC,CAAA;QAC5J,CAAC;QACD,IAAI,CAAC,aAAa,GAAG,OAAO,CAAA;QAC5B,OAAO,OAAO,CAAA;IAChB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,QAAgB,EAAE,QAA6B;QACzD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAA;QACnC,MAAM,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,GAAG,QAAQ,EAAE,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,CAAA;IACvE,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,QAAgB;QAC5B,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAA;QACnC,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAA;QACrC,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,CAAsB,GAAG,CAAC,CAAA;QAC1D,IAAI,CAAC,QAAQ;YAAE,OAAO,IAAI,CAAA;QAC1B,MAAM,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAA;QACvB,OAAO,QAAQ,CAAA;IACjB,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,QAAgB;QACzB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAA;QACnC,OAAO,KAAK,CAAC,GAAG,CAAsB,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC,CAAA;IAClE,CAAC;CACF"}

package/dist/tool-execution.d.ts

@@ -1,16 +0,0 @@
-import type { LoopContext } from './agent.js';
-import type { AiMessage, StreamChunk, ToolCall, ToolResult } from './types.js';
-/**
- * Execute the tool phase for a single agent step. Yields the same
- * `StreamChunk` sequence (`tool-call` → `tool-update*` → `tool-result`) that
- * the streaming caller surfaces to consumers. Non-streaming callers iterate
- * via `.next()` and discard yields — the side effects (message pushes,
- * pending-state mutations on `loopCtx`) are identical regardless of whether
- * the chunks reach a consumer.
- *
- * Returns the step's `ToolResult[]`. The caller passes the assistant message
- * to push before iteration so the AgentStep shape (response.message) and the
- * final `messages` array stay in sync with the loop variant.
- */
-export declare function executeToolPhase(loopCtx: LoopContext, toolCalls: ToolCall[], assistantMessage: AiMessage): AsyncGenerator<StreamChunk, ToolResult[], void>;
-//# sourceMappingURL=tool-execution.d.ts.map

package/dist/tool-execution.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"tool-execution.d.ts","sourceRoot":"","sources":["../src/tool-execution.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,YAAY,CAAA;AAkB7C,OAAO,KAAK,EAAE,SAAS,EAAW,WAAW,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,YAAY,CAAA;AAEvF;;;;;;;;;;;GAWG;AACH,wBAAuB,gBAAgB,CACrC,OAAO,EAAW,WAAW,EAC7B,SAAS,EAAS,QAAQ,EAAE,EAC5B,gBAAgB,EAAE,SAAS,GAC1B,cAAc,CAAC,WAAW,EAAE,UAAU,EAAE,EAAE,IAAI,CAAC,CA8BjD"}