ai-pieces-custom 0.0.35 → 0.0.37

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-pieces-custom",
-  "version": "0.0.35",
+  "version": "0.0.37",
   "type": "commonjs",
   "main": "./src/index.js",
   "types": "./src/index.d.ts",

package/src/lib/actions/agents/document-store-tool.d.ts

@@ -1,9 +1,7 @@
 import type { Tool } from 'ai';
 export interface DocumentStoreToolConfig {
-    storeId: string;
-    datasetIds?: string[];
+    storeIds: string[];
     searchType?: 'GRAPH_COMPLETION' | 'SEMANTIC' | 'HYBRID' | 'KEYWORD';
-    apiUrl?: string;
     token: string;
 }
 export declare function buildDocumentStoreTool(config: DocumentStoreToolConfig, token: string): Tool;

package/src/lib/actions/agents/document-store-tool.js

@@ -4,23 +4,31 @@ exports.buildDocumentStoreTool = buildDocumentStoreTool;
 const tslib_1 = require("tslib");
 const zod_1 = require("zod");
 const documentStoreSchema = zod_1.z.object({
-    question: zod_1.z.string().describe('The question or search query to look up in the document store.'),
+    question: zod_1.z
+        .string()
+        .describe('The question or search query to look up in the document store.'),
 });
 function buildDocumentStoreTool(config, token) {
-    var _a;
-    const baseUrl = ((_a = config.apiUrl) !== null && _a !== void 0 ? _a : 'https://dv-knowledge.takween.ai').replace(/\/$/, '');
+    const baseUrl = process.env['KNOWLEDGE_API_BASE'];
     return {
         description: 'Search the knowledge document store for information relevant to the user query. ' +
             'Use this whenever you need to look up facts, documents, or data from the knowledge base. ' +
             'You MUST always provide a "question" argument with the user\'s query.',
         inputSchema: documentStoreSchema,
         execute: (_a) => tslib_1.__awaiter(this, [_a], void 0, function* ({ question }) {
-            var _b, _c, _d, _e, _f, _g;
+            var _b, _c, _d, _e, _f;
             if (!(question === null || question === void 0 ? void 0 : question.trim())) {
                 return { error: 'No question provided to document store search.' };
             }
             try {
-                const body = Object.assign({ store_ids: [config.storeId], question, enable_citation: true, document_retrieval: false, search_type: (_b = config.searchType) !== null && _b !== void 0 ? _b : 'GRAPH_COMPLETION' }, (((_c = config.datasetIds) === null || _c === void 0 ? void 0 : _c.length) ? { dataset_ids: config.datasetIds } : {}));
+                const body = {
+                    store_ids: config.storeIds,
+                    question,
+                    search_type: (_b = config.searchType) !== null && _b !== void 0 ? _b : 'GRAPH_COMPLETION',
+                    enable_citation: true,
+                    document_retrieval: false,
+                    context_only: false,
+                };
                 const res = yield fetch(`${baseUrl}/api/v1/document-stores/search`, {
                     method: 'POST',
                     headers: {
@@ -32,17 +40,20 @@ function buildDocumentStoreTool(config, token) {
                 });
                 if (!res.ok) {
                     const errorBody = yield res.text().catch(() => res.statusText);
-                    return { error: `Document store search failed (HTTP ${res.status}): ${errorBody}` };
+                    return {
+                        error: `Document store search failed (HTTP ${res.status}): ${errorBody}`,
+                    };
                 }
                 const json = yield res.json();
-                const results = (_e = (_d = json === null || json === void 0 ? void 0 : json.data) === null || _d === void 0 ? void 0 : _d.result) !== null && _e !== void 0 ? _e : [];
-                const citations = (_g = (_f = json === null || json === void 0 ? void 0 : json.data) === null || _f === void 0 ? void 0 : _f.citations) !== null && _g !== void 0 ? _g : [];
+                const results = (_d = (_c = json === null || json === void 0 ? void 0 : json.data) === null || _c === void 0 ? void 0 : _c.result) !== null && _d !== void 0 ? _d : [];
+                const citations = (_f = (_e = json === null || json === void 0 ? void 0 : json.data) === null || _e === void 0 ? void 0 : _e.citations) !== null && _f !== void 0 ? _f : [];
                 const resultText = results
                     .map((r) => (typeof r === 'string' ? r : JSON.stringify(r, null, 2)))
                     .join('\n\n');
                 const citationList = citations.map((c) => { var _a; return String((_a = c['file_name']) !== null && _a !== void 0 ? _a : 'unknown'); });
                 return {
-                    answer: resultText.trim() || 'No relevant information found in the document store.',
+                    answer: resultText.trim() ||
+                        'No relevant information found in the document store.',
                     sources: citationList,
                 };
             }

package/src/lib/actions/agents/document-store-tool.js.map

@@ -1 +1 @@
-{"version":3,"file":"document-store-tool.js","sourceRoot":"","sources":["../../../../../../../../../packages/pieces/custom/ai-custom/src/lib/actions/agents/document-store-tool.ts"],"names":[],"mappings":";;AA0BA,wDA4DC;;AAtFD,6BAAwB;AAsBxB,MAAM,mBAAmB,GAAG,OAAC,CAAC,MAAM,CAAC;IACnC,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,gEAAgE,CAAC;CAChG,CAAC,CAAC;AAEH,SAAgB,sBAAsB,CAAC,MAA+B,EAAE,KAAa;;IACnF,MAAM,OAAO,GAAG,CAAC,MAAA,MAAM,CAAC,MAAM,mCAAI,iCAAiC,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IAExF,OAAO;QACL,WAAW,EACT,kFAAkF;YAClF,2FAA2F;YAC3F,uEAAuE;QACzE,WAAW,EAAE,mBAAmB;QAChC,OAAO,EAAE,KAA2C,EAAE,kDAAtC,EAAE,QAAQ,EAAwB;;YAChD,IAAI,CAAC,CAAA,QAAQ,aAAR,QAAQ,uBAAR,QAAQ,CAAE,IAAI,EAAE,CAAA,EAAE,CAAC;gBACtB,OAAO,EAAE,KAAK,EAAE,gDAAgD,EAAE,CAAC;YACrE,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,IAAI,mBACR,SAAS,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,EAC3B,QAAQ,EACR,eAAe,EAAE,IAAI,EACrB,kBAAkB,EAAE,KAAK,EACzB,WAAW,EAAE,MAAA,MAAM,CAAC,UAAU,mCAAI,kBAAkB,IACjD,CAAC,CAAA,MAAA,MAAM,CAAC,UAAU,0CAAE,MAAM,EAAC,CAAC,CAAC,EAAE,WAAW,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CACzE,CAAC;gBAEF,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,OAAO,gCAAgC,EAAE;oBAClE,MAAM,EAAE,MAAM;oBACd,OAAO,EAAE;wBACP,MAAM,EAAE,kBAAkB;wBAC1B,cAAc,EAAE,kBAAkB;wBAClC,aAAa,EAAE,UAAU,KAAK,EAAE;qBACjC;oBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;iBAC3B,CAAC,CAAC;gBAEH,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;oBACZ,MAAM,SAAS,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;oBAC/D,OAAO,EAAE,KAAK,EAAE,sCAAsC,GAAG,CAAC,MAAM,MAAM,SAAS,EAAE,EAAE,CAAC;gBACtF,CAAC;gBAED,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;gBAC9B,MAAM,OAAO,GAAc,MAAA,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,0CAAE,MAAM,mCAAI,EAAE,CAAC;gBACpD,MAAM,SAAS,GAAc,MAAA,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,0CAAE,SAAS,mCAAI,EAAE,CAAC;gBAEzD,MAAM,UAAU,GAAG,OAAO;qBACvB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;qBACpE,IAAI,CAAC,MAAM,CAAC,CAAC;gBAEhB,MAAM,YAAY,GAAI,SAAuC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,WACtE,OAAA,MAAM,CAAC,MAAA,CAAC,CAAC,WAAW,CAAC,mCAAI,SAAS,CAAC,CAAA,EAAA,CACpC,CAAC;gBAEF,OAAO;oBACL,MAAM,EAAE,UAAU,CAAC,IAAI,EAAE,IAAI,sDAAsD;oBACnF,OAAO,EAAE,YAAY;iBACtB,CAAC;YACJ,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,EAAE,KAAK,EAAE,gCAAgC,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;YAClE,CAAC;QACH,CAAC,CAAA;KACF,CAAC;AACJ,CAAC"}
+{"version":3,"file":"document-store-tool.js","sourceRoot":"","sources":["../../../../../../../../../packages/pieces/custom/ai-custom/src/lib/actions/agents/document-store-tool.ts"],"names":[],"mappings":";;AAeA,wDAmEC;;AAlFD,6BAAwB;AASxB,MAAM,mBAAmB,GAAG,OAAC,CAAC,MAAM,CAAC;IACnC,QAAQ,EAAE,OAAC;SACR,MAAM,EAAE;SACR,QAAQ,CAAC,gEAAgE,CAAC;CAC9E,CAAC,CAAC;AAEH,SAAgB,sBAAsB,CACpC,MAA+B,EAC/B,KAAa;IAEb,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;IAElD,OAAO;QACL,WAAW,EACT,kFAAkF;YAClF,2FAA2F;YAC3F,uEAAuE;QACzE,WAAW,EAAE,mBAAmB;QAChC,OAAO,EAAE,KAA2C,EAAE,kDAAtC,EAAE,QAAQ,EAAwB;;YAChD,IAAI,CAAC,CAAA,QAAQ,aAAR,QAAQ,uBAAR,QAAQ,CAAE,IAAI,EAAE,CAAA,EAAE,CAAC;gBACtB,OAAO,EAAE,KAAK,EAAE,gDAAgD,EAAE,CAAC;YACrE,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,IAAI,GAAG;oBACX,SAAS,EAAE,MAAM,CAAC,QAAQ;oBAC1B,QAAQ;oBACR,WAAW,EAAE,MAAA,MAAM,CAAC,UAAU,mCAAI,kBAAkB;oBACpD,eAAe,EAAE,IAAI;oBACrB,kBAAkB,EAAE,KAAK;oBACzB,YAAY,EAAE,KAAK;iBACpB,CAAC;gBAEF,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,OAAO,gCAAgC,EAAE;oBAClE,MAAM,EAAE,MAAM;oBACd,OAAO,EAAE;wBACP,MAAM,EAAE,kBAAkB;wBAC1B,cAAc,EAAE,kBAAkB;wBAClC,aAAa,EAAE,UAAU,KAAK,EAAE;qBACjC;oBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;iBAC3B,CAAC,CAAC;gBAEH,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;oBACZ,MAAM,SAAS,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;oBAC/D,OAAO;wBACL,KAAK,EAAE,sCAAsC,GAAG,CAAC,MAAM,MAAM,SAAS,EAAE;qBACzE,CAAC;gBACJ,CAAC;gBAED,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;gBAC9B,MAAM,OAAO,GAAc,MAAA,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,0CAAE,MAAM,mCAAI,EAAE,CAAC;gBACpD,MAAM,SAAS,GAAc,MAAA,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,0CAAE,SAAS,mCAAI,EAAE,CAAC;gBAEzD,MAAM,UAAU,GAAG,OAAO;qBACvB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;qBACpE,IAAI,CAAC,MAAM,CAAC,CAAC;gBAEhB,MAAM,YAAY,GAAI,SAAuC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,WACtE,OAAA,MAAM,CAAC,MAAA,CAAC,CAAC,WAAW,CAAC,mCAAI,SAAS,CAAC,CAAA,EAAA,CACpC,CAAC;gBAEF,OAAO;oBACL,MAAM,EACJ,UAAU,CAAC,IAAI,EAAE;wBACjB,sDAAsD;oBACxD,OAAO,EAAE,YAAY;iBACtB,CAAC;YACJ,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,EAAE,KAAK,EAAE,gCAAgC,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;YAClE,CAAC;QACH,CAAC,CAAA;KACF,CAAC;AACJ,CAAC"}

package/src/lib/actions/agents/run-agents.d.ts

@@ -3,8 +3,7 @@ export declare const runAgent: import("@activepieces/pieces-framework").IAction<
     prompt: import("@activepieces/pieces-framework").LongTextProperty<true>;
     aiProviderModel: import("@activepieces/pieces-framework").ObjectProperty<true>;
     documentStoreEnabled: import("@activepieces/pieces-framework").CheckboxProperty<false>;
-    documentStoreId: import("@activepieces/pieces-framework").DropdownProperty<string, false, import("@activepieces/pieces-framework").OAuth2Property<import("@activepieces/pieces-framework").OAuth2Props>>;
-    documentStoreDatasetIds: import("@activepieces/pieces-framework").MultiSelectDropdownProperty<string, false, import("@activepieces/pieces-framework").OAuth2Property<import("@activepieces/pieces-framework").OAuth2Props>>;
+    documentStoreIds: import("@activepieces/pieces-framework").MultiSelectDropdownProperty<string, false, import("@activepieces/pieces-framework").OAuth2Property<import("@activepieces/pieces-framework").OAuth2Props>>;
     documentStoreSearchType: import("@activepieces/pieces-framework").StaticDropdownProperty<string, false>;
     agentTools: import("@activepieces/pieces-framework").ArrayProperty<true> | import("@activepieces/pieces-framework").ArrayProperty<false>;
     structuredOutput: import("@activepieces/pieces-framework").ArrayProperty<true> | import("@activepieces/pieces-framework").ArrayProperty<false>;

package/src/lib/actions/agents/run-agents.js

@@ -13,26 +13,43 @@ const tools_1 = require("./tools");
 const index_1 = require("../../../index");
 // ─── Helpers ──────────────────────────────────────────────────────────────────
 const KNOWLEDGE_API_BASE = process.env['KNOWLEDGE_API_BASE'];
-function fetchDocumentStores(token) {
-    return tslib_1.__awaiter(this, void 0, void 0, function* () {
-        var _a, _b;
+/**
+ * Fetch ALL document stores by paginating through every page.
+ * Falls back gracefully — returns whatever was collected so far on any error.
+ */
+function fetchAllDocumentStores(token_1) {
+    return tslib_1.__awaiter(this, arguments, void 0, function* (token, pageSize = 50) {
+        var _a, _b, _c, _d;
+        const all = [];
         try {
-            const res = yield fetch(`${KNOWLEDGE_API_BASE}/api/v1/document-stores`, {
-                headers: { Authorization: `Bearer ${token}` },
-            });
-            if (!res.ok)
-                return [];
-            const json = yield res.json();
-            return (_b = (_a = json === null || json === void 0 ? void 0 : json.data) === null || _a === void 0 ? void 0 : _a.stores) !== null && _b !== void 0 ? _b : [];
+            let page = 1;
+            let totalPages = 1;
+            while (page <= totalPages) {
+                const res = yield fetch(`${KNOWLEDGE_API_BASE}/api/v1/document-stores?page=${page}&pageSize=${pageSize}`, { headers: { Authorization: `Bearer ${token}` } });
+                if (!res.ok)
+                    break;
+                const json = yield res.json();
+                const stores = (_b = (_a = json === null || json === void 0 ? void 0 : json.data) === null || _a === void 0 ? void 0 : _a.stores) !== null && _b !== void 0 ? _b : [];
+                all.push(...stores);
+                if ((_c = json === null || json === void 0 ? void 0 : json.data) === null || _c === void 0 ? void 0 : _c.pagination) {
+                    totalPages = (_d = json.data.pagination.totalPages) !== null && _d !== void 0 ? _d : 1;
+                }
+                else {
+                    break;
+                }
+                if (all.length >= 1000)
+                    break;
+                page++;
+            }
         }
-        catch (_c) {
-            return [];
+        catch (_e) {
+            // Return whatever we have
         }
+        return all;
     });
 }
 /**
  * Recursively strip null/undefined from every array in the output object.
- * AgentTimeline maps over output.steps and crashes on null items.
  */
 function sanitizeAgentOutput(output) {
     if (output === null || output === undefined)
@@ -94,18 +111,17 @@ exports.runAgent = (0, pieces_framework_1.createAction)({
         // ── Document Store ────────────────────────────────────────────────────
         documentStoreEnabled: pieces_framework_1.Property.Checkbox({
             displayName: 'Enable Document Store Search',
-            description: 'When enabled, the agent will ALWAYS search the document store first before answering.',
+            description: 'When enabled, the agent will ALWAYS search the selected document stores first before answering.',
             required: false,
             defaultValue: false,
         }),
-        documentStoreId: pieces_framework_1.Property.Dropdown({
-            displayName: 'Document Store',
-            description: 'Select the document store to search.',
+        documentStoreIds: pieces_framework_1.Property.MultiSelectDropdown({
+            displayName: 'Document Stores',
+            description: 'Select one or more document stores to search. All selected stores are queried in a single call.',
             required: false,
             auth: index_1.httpOauth2Auth,
             refreshers: ['documentStoreEnabled'],
             options: (propsValue, ctx) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
-                console.log(ctx);
                 const documentStoreEnabled = propsValue['documentStoreEnabled'];
                 if (!documentStoreEnabled || !ctx.server.oauthUserToken) {
                     return {
@@ -114,7 +130,7 @@ exports.runAgent = (0, pieces_framework_1.createAction)({
                         placeholder: 'Enable Document Store Search first',
                     };
                 }
-                const stores = yield fetchDocumentStores(ctx.server.oauthUserToken);
+                const stores = yield fetchAllDocumentStores(ctx.server.oauthUserToken);
                 if (!stores.length) {
                     return {
                         disabled: true,
@@ -124,7 +140,7 @@ exports.runAgent = (0, pieces_framework_1.createAction)({
                 }
                 return {
                     disabled: false,
-                    placeholder: 'Select a document store',
+                    placeholder: 'Select one or more document stores',
                     options: stores.map((s) => ({
                         label: `${s.name}${s.description ? ` — ${s.description}` : ''}`,
                         value: s.id,
@@ -132,42 +148,6 @@ exports.runAgent = (0, pieces_framework_1.createAction)({
                 };
             }),
         }),
-        documentStoreDatasetIds: pieces_framework_1.Property.MultiSelectDropdown({
-            displayName: 'Datasets (optional)',
-            description: 'Restrict search to specific datasets. Leave empty to search the entire store.',
-            required: false,
-            auth: index_1.httpOauth2Auth,
-            refreshers: ['documentStoreId', 'documentStoreEnabled'],
-            options: (propsValue, ctx) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
-                var _a;
-                const documentStoreEnabled = propsValue['documentStoreEnabled'];
-                const documentStoreId = propsValue['documentStoreId'];
-                if (!documentStoreEnabled || !documentStoreId || !ctx.server.oauthUserToken) {
-                    return {
-                        disabled: true,
-                        options: [],
-                        placeholder: 'Select a document store first',
-                    };
-                }
-                const stores = yield fetchDocumentStores(ctx.server.oauthUserToken);
-                const store = stores.find((s) => s.id === documentStoreId);
-                if (!((_a = store === null || store === void 0 ? void 0 : store.datasets) === null || _a === void 0 ? void 0 : _a.length)) {
-                    return {
-                        disabled: true,
-                        options: [],
-                        placeholder: 'No datasets found in this store',
-                    };
-                }
-                return {
-                    disabled: false,
-                    placeholder: 'All datasets (leave empty) or pick specific ones',
-                    options: store.datasets.map((d) => ({
-                        label: `${d.name}${d.description ? ` — ${d.description}` : ''}`,
-                        value: d.id,
-                    })),
-                };
-            }),
-        }),
         documentStoreSearchType: pieces_framework_1.Property.StaticDropdown({
             displayName: 'Search Type',
             description: 'How the document store query is executed.',
@@ -219,7 +199,7 @@ exports.runAgent = (0, pieces_framework_1.createAction)({
         return tslib_1.__awaiter(this, void 0, void 0, function* () {
             var _a, e_1, _b, _c;
             var _d;
-            const { prompt, maxSteps, aiProviderModel, documentStoreEnabled, documentStoreId, documentStoreDatasetIds, documentStoreSearchType, } = context.propsValue;
+            const { prompt, maxSteps, aiProviderModel, documentStoreEnabled, documentStoreIds, documentStoreSearchType, } = context.propsValue;
             const agentProviderModel = aiProviderModel;
             const model = yield (0, ai_sdk_1.createAIModel)({
                 modelId: agentProviderModel.model,
@@ -236,33 +216,26 @@ exports.runAgent = (0, pieces_framework_1.createAction)({
             const structuredOutput = hasStructuredOutput
                 ? context.propsValue.structuredOutput
                 : undefined;
-            // Only the standard AgentTool entries — DOCUMENT_STORE is NOT a known
-            // AgentToolType so it must never be passed to outputBuilder.startToolCall
-            // (which looks up by type and returns null causing AgentTimeline to crash).
             const agentTools = ((_d = context.propsValue.agentTools) !== null && _d !== void 0 ? _d : []);
-            // ── Inject document store tool if enabled ────────────────────────────
-            // Keep this separate from agentTools so it never reaches outputBuilder's
-            // tool-type lookup. constructAgentTools handles it via buildDocumentStoreTools.
-            let documentStoreTool;
-            if (documentStoreEnabled && documentStoreId) {
-                documentStoreTool = {
-                    type: 'DOCUMENT_STORE',
-                    toolName: 'search_document_store',
-                    auth: {
-                        storeId: documentStoreId,
-                        datasetIds: Array.isArray(documentStoreDatasetIds) &&
-                            documentStoreDatasetIds.length > 0
-                            ? documentStoreDatasetIds
-                            : undefined,
-                        searchType: documentStoreSearchType !== null && documentStoreSearchType !== void 0 ? documentStoreSearchType : 'GRAPH_COMPLETION',
-                        token: context.server.token,
+            // ── Inject a single document store tool carrying all selected store IDs ──
+            const selectedStoreIds = documentStoreEnabled &&
+                Array.isArray(documentStoreIds) &&
+                documentStoreIds.length > 0
+                ? documentStoreIds
+                : [];
+            const allToolsForConstruct = selectedStoreIds.length > 0
+                ? [
+                    ...agentTools,
+                    {
+                        type: 'DOCUMENT_STORE',
+                        toolName: 'search_document_store',
+                        auth: {
+                            storeIds: selectedStoreIds,
+                            searchType: documentStoreSearchType !== null && documentStoreSearchType !== void 0 ? documentStoreSearchType : 'GRAPH_COMPLETION',
+                            token: context.server.token,
+                        },
                     },
-                };
-            }
-            // allTools includes the DOCUMENT_STORE entry so constructAgentTools can
-            // build it — but startToolCall only ever receives agentTools (no DOCUMENT_STORE).
-            const allToolsForConstruct = documentStoreTool
-                ? [...agentTools, documentStoreTool]
+                ]
                 : agentTools;
             const { mcpClients, tools } = yield (0, tools_1.constructAgentTools)({
                 context,
@@ -274,17 +247,16 @@ exports.runAgent = (0, pieces_framework_1.createAction)({
             const errors = [];
             // ── System prompt ────────────────────────────────────────────────────
             const baseSystem = utils_1.agentUtils.getPrompts(prompt).system;
-            const system = documentStoreEnabled && documentStoreId
+            const system = selectedStoreIds.length > 0
                 ? `${baseSystem}
 
 IMPORTANT INSTRUCTION: You have access to a knowledge base tool called "search_document_store".
+It searches across ${selectedStoreIds.length} document store${selectedStoreIds.length > 1 ? 's' : ''} simultaneously.
 You MUST call this tool FIRST before answering ANY user query, without exception.
 Pass the user's question exactly as the "question" argument.
 Use the retrieved context to formulate your answer.
 Do not answer from your training knowledge alone when this tool is available.`
                 : baseSystem;
-            // Sanitise before every UI update — removes null items from steps arrays
-            // that AgentTimeline would otherwise crash on.
             const safeUpdate = () => tslib_1.__awaiter(this, void 0, void 0, function* () {
                 const raw = outputBuilder.build();
                 yield context.output.update({
@@ -326,8 +298,6 @@ Do not answer from your training knowledge alone when this tool is available.`
                                             toolName: chunk.toolName,
                                             toolCallId: chunk.toolCallId,
                                             input: chunk.input,
-                                            // Pass allToolsForConstruct so the DOCUMENT_STORE entry is
-                                            // visible to getToolMetadata's lookup.
                                             agentTools: allToolsForConstruct,
                                         });
                                     }
@@ -376,7 +346,6 @@ Do not answer from your training knowledge alone when this tool is available.`
                                     });
                                     break;
                                 }
-                                // 'finish' carries no displayable state — ignore silently.
                                 case 'finish':
                                     break;
                                 default:
@@ -391,8 +360,6 @@ Do not answer from your training knowledge alone when this tool is available.`
                                 details: (0, util_1.inspect)(innerError),
                             });
                         }
-                        // safeUpdate is outside the inner try-catch so a UI update failure
-                        // is not misreported as a chunk-processing-error.
                         yield safeUpdate().catch((e) => {
                             console.error('[runAgent] safeUpdate failed:', (0, util_1.inspect)(e));
                         });

package/src/lib/actions/agents/run-agents.js.map

@@ -1 +1 @@
-{"version":3,"file":"run-agents.js","sourceRoot":"","sources":["../../../../../../../../../packages/pieces/custom/ai-custom/src/lib/actions/agents/run-agents.ts"],"names":[],"mappings":";;;;AAAA,qEAIwC;AAExC,iDAU8B;AAC9B,2BAA0D;AAC1D,iEAA4D;AAC5D,gDAAoD;AACpD,+BAA+B;AAC/B,mCAAqC;AACrC,mCAA8C;AAC9C,0CAAgD;AAwBhD,iFAAiF;AAEjF,MAAM,kBAAkB,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;AAE7D,SAAe,mBAAmB,CAAC,KAAa;;;QAC9C,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,kBAAkB,yBAAyB,EAAE;gBACtE,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,KAAK,EAAE,EAAE;aAC9C,CAAC,CAAC;YACH,IAAI,CAAC,GAAG,CAAC,EAAE;gBAAE,OAAO,EAAE,CAAC;YACvB,MAAM,IAAI,GAA2B,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;YACtD,OAAO,MAAA,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,0CAAE,MAAM,mCAAI,EAAE,CAAC;QAClC,CAAC;QAAC,WAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;CAAA;AAED;;;GAGG;AACH,SAAS,mBAAmB,CAAC,MAAe;IAC1C,IAAI,MAAM,KAAK,IAAI,IAAI,MAAM,KAAK,SAAS;QAAE,OAAO,MAAM,CAAC;IAC3D,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAC1B,OAAO,MAAM;aACV,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,IAAI,IAAI,IAAI,KAAK,SAAS,CAAC;aACrD,GAAG,CAAC,mBAAmB,CAAC,CAAC;IAC9B,CAAC;IACD,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC/B,MAAM,MAAM,GAA4B,EAAE,CAAC;QAC3C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CACvC,MAAiC,CAClC,EAAE,CAAC;YACF,MAAM,CAAC,GAAG,CAAC,GAAG,mBAAmB,CAAC,KAAK,CAAC,CAAC;QAC3C,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,iFAAiF;AAEjF,MAAM,mBAAmB,GAA2B;IAClD,IAAI,EAAE,2BAAQ,CAAC,SAAS,CAAC,EAAE,WAAW,EAAE,WAAW,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;IACtE,QAAQ,EAAE,2BAAQ,CAAC,SAAS,CAAC,EAAE,WAAW,EAAE,WAAW,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;IAC1E,aAAa,EAAE,2BAAQ,CAAC,IAAI,CAAC;QAC3B,WAAW,EAAE,gBAAgB;QAC7B,QAAQ,EAAE,KAAK;KAChB,CAAC;IACF,cAAc,EAAE,2BAAQ,CAAC,SAAS,CAAC;QACjC,WAAW,EAAE,kBAAkB;QAC/B,QAAQ,EAAE,KAAK;KAChB,CAAC;IACF,SAAS,EAAE,2BAAQ,CAAC,SAAS,CAAC;QAC5B,WAAW,EAAE,gBAAgB;QAC7B,QAAQ,EAAE,KAAK;KAChB,CAAC;IACF,QAAQ,EAAE,2BAAQ,CAAC,SAAS,CAAC,EAAE,WAAW,EAAE,UAAU,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;IAC1E,IAAI,EAAE,2BAAQ,CAAC,IAAI,CAAC,EAAE,WAAW,EAAE,oBAAoB,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;CAC5E,CAAC;AAEF,iFAAiF;AAEpE,QAAA,QAAQ,GAAG,IAAA,+BAAY,EAAC;IACnC,IAAI,EAAE,kBAAkB;IACxB,WAAW,EAAE,WAAW;IACxB,WAAW,EACT,+HAA+H;IACjI,IAAI,EAAE,sBAAc;IACpB,KAAK,EAAE;QACL,CAAC,WAAW,CAAC,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YAC/B,WAAW,EAAE,YAAY;YACzB,QAAQ,EAAE,KAAK;YACf,WAAW,EACT,8FAA8F;SACjG,CAAC;QAEF,CAAC,wBAAe,CAAC,MAAM,CAAC,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YAC1C,WAAW,EAAE,QAAQ;YACrB,WAAW,EAAE,6CAA6C;YAC1D,QAAQ,EAAE,IAAI;SACf,CAAC;QAEF,CAAC,wBAAe,CAAC,iBAAiB,CAAC,EAAE,2BAAQ,CAAC,MAAM,CAAC;YACnD,WAAW,EAAE,UAAU;YACvB,QAAQ,EAAE,IAAI;SACf,CAAC;QAEF,yEAAyE;QACzE,oBAAoB,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YACtC,WAAW,EAAE,8BAA8B;YAC3C,WAAW,EACT,uFAAuF;YACzF,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,KAAK;SACpB,CAAC;QAEF,eAAe,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YACjC,WAAW,EAAE,gBAAgB;YAC7B,WAAW,EAAE,sCAAsC;YACnD,QAAQ,EAAE,KAAK;YACf,IAAI,EAAE,sBAAc;YACpB,UAAU,EAAE,CAAE,sBAAsB,CAAC;YACrC,OAAO,EAAE,CAAO,UAAU,EAAE,GAAG,EAAE,EAAE;gBACjC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;gBAChB,MAAM,oBAAoB,GAAG,UAAU,CAAC,sBAAsB,CAEjD,CAAC;gBAEd,IAAI,CAAC,oBAAoB,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;oBACxD,OAAO;wBACL,QAAQ,EAAE,IAAI;wBACd,OAAO,EAAE,EAAE;wBACX,WAAW,EAAE,oCAAoC;qBAClD,CAAC;gBACJ,CAAC;gBAED,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,GAAG,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;gBACpE,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;oBACnB,OAAO;wBACL,QAAQ,EAAE,IAAI;wBACd,OAAO,EAAE,EAAE;wBACX,WAAW,EAAE,0BAA0B;qBACxC,CAAC;gBACJ,CAAC;gBAED,OAAO;oBACL,QAAQ,EAAE,KAAK;oBACf,WAAW,EAAE,yBAAyB;oBACtC,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;wBAC1B,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE;wBAC/D,KAAK,EAAE,CAAC,CAAC,EAAE;qBACZ,CAAC,CAAC;iBACJ,CAAC;YACJ,CAAC,CAAA;SACF,CAAC;QAEF,uBAAuB,EAAE,2BAAQ,CAAC,mBAAmB,CAAC;YACpD,WAAW,EAAE,qBAAqB;YAClC,WAAW,EACT,+EAA+E;YACjF,QAAQ,EAAE,KAAK;YACf,IAAI,EAAE,sBAAc;YACpB,UAAU,EAAE,CAAC,iBAAiB,EAAE,sBAAsB,CAAC;YACvD,OAAO,EAAE,CAAO,UAAU,EAAE,GAAG,EAAE,EAAE;;gBACjC,MAAM,oBAAoB,GAAG,UAAU,CAAC,sBAAsB,CAEjD,CAAC;gBACd,MAAM,eAAe,GAAG,UAAU,CAAC,iBAAiB,CAEvC,CAAC;gBAEd,IAAI,CAAC,oBAAoB,IAAI,CAAC,eAAe,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;oBAC5E,OAAO;wBACL,QAAQ,EAAE,IAAI;wBACd,OAAO,EAAE,EAAE;wBACX,WAAW,EAAE,+BAA+B;qBAC7C,CAAC;gBACJ,CAAC;gBAED,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,GAAG,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;gBACpE,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,eAAe,CAAC,CAAC;gBAE3D,IAAI,CAAC,CAAA,MAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,QAAQ,0CAAE,MAAM,CAAA,EAAE,CAAC;oBAC7B,OAAO;wBACL,QAAQ,EAAE,IAAI;wBACd,OAAO,EAAE,EAAE;wBACX,WAAW,EAAE,iCAAiC;qBAC/C,CAAC;gBACJ,CAAC;gBAED,OAAO;oBACL,QAAQ,EAAE,KAAK;oBACf,WAAW,EAAE,kDAAkD;oBAC/D,OAAO,EAAE,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;wBAClC,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE;wBAC/D,KAAK,EAAE,CAAC,CAAC,EAAE;qBACZ,CAAC,CAAC;iBACJ,CAAC;YACJ,CAAC,CAAA;SACF,CAAC;QAEF,uBAAuB,EAAE,2BAAQ,CAAC,cAAc,CAAC;YAC/C,WAAW,EAAE,aAAa;YAC1B,WAAW,EAAE,2CAA2C;YACxD,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,kBAAkB;YAChC,OAAO,EAAE;gBACP,OAAO,EAAE;oBACP;wBACE,KAAK,EAAE,gCAAgC;wBACvC,KAAK,EAAE,kBAAkB;qBAC1B;oBACD,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,UAAU,EAAE;oBACxC,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE;oBACpC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE;iBACvC;aACF;SACF,CAAC;QAEF,yEAAyE;QACzE,CAAC,wBAAe,CAAC,WAAW,CAAC,EAAE,2BAAQ,CAAC,KAAK,CAAC;YAC5C,WAAW,EAAE,aAAa;YAC1B,QAAQ,EAAE,KAAK;YACf,UAAU,EAAE,mBAAmB;SAChC,CAAC;QAEF,CAAC,wBAAe,CAAC,iBAAiB,CAAC,EAAE,2BAAQ,CAAC,KAAK,CAAC;YAClD,WAAW,EAAE,mBAAmB;YAChC,YAAY,EAAE,SAAS;YACvB,QAAQ,EAAE,KAAK;YACf,UAAU,EAAE;gBACV,WAAW,EAAE,2BAAQ,CAAC,SAAS,CAAC;oBAC9B,WAAW,EAAE,cAAc;oBAC3B,QAAQ,EAAE,IAAI;iBACf,CAAC;gBACF,WAAW,EAAE,2BAAQ,CAAC,SAAS,CAAC;oBAC9B,WAAW,EAAE,aAAa;oBAC1B,QAAQ,EAAE,KAAK;iBAChB,CAAC;gBACF,IAAI,EAAE,2BAAQ,CAAC,SAAS,CAAC,EAAE,WAAW,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;aAClE;SACF,CAAC;QAEF,CAAC,wBAAe,CAAC,SAAS,CAAC,EAAE,2BAAQ,CAAC,MAAM,CAAC;YAC3C,WAAW,EAAE,WAAW;YACxB,WAAW,EAAE,2CAA2C;YACxD,QAAQ,EAAE,IAAI;YACd,YAAY,EAAE,EAAE;SACjB,CAAC;KACH;IAED,6EAA6E;IACvE,GAAG,CAAC,OAAO;;;;YACf,MAAM,EACJ,MAAM,EACN,QAAQ,EACR,eAAe,EACf,oBAAoB,EACpB,eAAe,EACf,uBAAuB,EACvB,uBAAuB,GACxB,GAAG,OAAO,CAAC,UAAU,CAAC;YAEvB,MAAM,kBAAkB,GAAG,eAAqC,CAAC;YAEjE,MAAM,KAAK,GAAG,MAAM,IAAA,sBAAa,EAAC;gBAChC,OAAO,EAAE,kBAAkB,CAAC,KAAK;gBACjC,QAAQ,EAAE,kBAAkB,CAAC,QAA0B;gBACvD,WAAW,EAAE,OAAO,CAAC,MAAM,CAAC,KAAK;gBACjC,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,MAAM;gBAC7B,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,EAAE;gBAC7B,MAAM,EAAE,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE;gBAChC,KAAK,EAAE,OAAO,CAAC,GAAG,CAAC,EAAE;aACtB,CAAC,CAAC;YAEH,MAAM,aAAa,GAAG,IAAA,yCAAkB,EAAC,MAAM,CAAC,CAAC;YAEjD,MAAM,mBAAmB,GACvB,CAAC,IAAA,cAAK,EAAC,OAAO,CAAC,UAAU,CAAC,gBAAgB,CAAC;gBAC3C,OAAO,CAAC,UAAU,CAAC,gBAAgB,CAAC,MAAM,GAAG,CAAC,CAAC;YACjD,MAAM,gBAAgB,GAAG,mBAAmB;gBAC1C,CAAC,CAAE,OAAO,CAAC,UAAU,CAAC,gBAAuC;gBAC7D,CAAC,CAAC,SAAS,CAAC;YAEd,sEAAsE;YACtE,0EAA0E;YAC1E,4EAA4E;YAC5E,MAAM,UAAU,GAAG,CAAC,MAAA,OAAO,CAAC,UAAU,CAAC,UAAU,mCAAI,EAAE,CAAgB,CAAC;YAExE,wEAAwE;YACxE,yEAAyE;YACzE,gFAAgF;YAChF,IAAI,iBAAwC,CAAC;YAE7C,IAAI,oBAAoB,IAAI,eAAe,EAAE,CAAC;gBAC5C,iBAAiB,GAAG;oBAClB,IAAI,EAAE,gBAAgB;oBACtB,QAAQ,EAAE,uBAAuB;oBACjC,IAAI,EAAE;wBACJ,OAAO,EAAE,eAAe;wBACxB,UAAU,EACR,KAAK,CAAC,OAAO,CAAC,uBAAuB,CAAC;4BACtC,uBAAuB,CAAC,MAAM,GAAG,CAAC;4BAChC,CAAC,CAAC,uBAAuB;4BACzB,CAAC,CAAC,SAAS;wBACf,UAAU,EAAE,uBAAuB,aAAvB,uBAAuB,cAAvB,uBAAuB,GAAI,kBAAkB;wBACzD,KAAK,EAAE,OAAO,CAAC,MAAM,CAAC,KAAK;qBAC5B;iBACsB,CAAC;YAC5B,CAAC;YAED,wEAAwE;YACxE,kFAAkF;YAClF,MAAM,oBAAoB,GAAG,iBAAiB;gBAC5C,CAAC,CAAC,CAAC,GAAG,UAAU,EAAE,iBAAiB,CAAC;gBACpC,CAAC,CAAC,UAAU,CAAC;YAEf,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,MAAM,IAAA,2BAAmB,EAAC;gBACtD,OAAO;gBACP,UAAU,EAAE,oBAAoB;gBAChC,KAAK;gBACL,aAAa;gBACb,gBAAgB;aACjB,CAAC,CAAC;YAEH,MAAM,MAAM,GAA2D,EAAE,CAAC;YAE1E,wEAAwE;YACxE,MAAM,UAAU,GAAG,kBAAU,CAAC,UAAU,CAAC,MAAgB,CAAC,CAAC,MAAM,CAAC;YAClE,MAAM,MAAM,GACV,oBAAoB,IAAI,eAAe;gBACrC,CAAC,CAAC,GAAG,UAAU;;;;;;8EAMuD;gBACtE,CAAC,CAAC,UAAU,CAAC;YAEjB,yEAAyE;YACzE,+CAA+C;YAC/C,MAAM,UAAU,GAAG,GAAS,EAAE;gBAC5B,MAAM,GAAG,GAAG,aAAa,CAAC,KAAK,EAAE,CAAC;gBAClC,MAAM,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC;oBAC1B,IAAI,EAAE,mBAAmB,CAAC,GAAG,CAAe;iBAC7C,CAAC,CAAC;YACL,CAAC,CAAA,CAAC;YAEF,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,IAAA,eAAU,EAAC;oBACxB,KAAK;oBACL,MAAM;oBACN,MAAM,EAAE,kBAAU,CAAC,UAAU,CAAC,MAAgB,CAAC,CAAC,MAAM;oBACtD,KAAK;oBACL,QAAQ,EAAE;wBACR,IAAA,gBAAW,EAAC,QAAQ,CAAC;wBACrB,IAAA,gBAAW,EAAC,kCAAyB,CAAC;qBACvC;oBACD,QAAQ,EAAE,GAAS,EAAE;wBACnB,MAAM,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;oBAChE,CAAC,CAAA;iBACF,CAAC,CAAC;;oBAEH,KAA0B,eAAA,KAAA,sBAAA,MAAM,CAAC,UAAU,CAAA,IAAA,sDAAE,CAAC;wBAApB,cAAiB;wBAAjB,WAAiB;wBAAhC,MAAM,KAAK,KAAA,CAAA;wBACpB,IAAI,CAAC;4BACH,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;gCACnB,KAAK,YAAY,CAAC,CAAC,CAAC;oCAClB,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;wCACnC,aAAa,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oCACxC,CAAC;oCACD,MAAM;gCACR,CAAC;gCAED,KAAK,WAAW,CAAC,CAAC,CAAC;oCACjB,IAAI,kBAAU,CAAC,wBAAwB,CAAC,KAAK,CAAC,QAAQ,CAAC;wCAAE,SAAS;oCAElE,IAAI,CAAC;wCACH,aAAa,CAAC,aAAa,CAAC;4CAC1B,QAAQ,EAAE,KAAK,CAAC,QAAQ;4CACxB,UAAU,EAAE,KAAK,CAAC,UAAU;4CAC5B,KAAK,EAAE,KAAK,CAAC,KAAgC;4CAC7C,2DAA2D;4CAC3D,uCAAuC;4CACvC,UAAU,EAAE,oBAAoB;yCACjC,CAAC,CAAC;oCACL,CAAC;oCAAC,OAAO,CAAC,EAAE,CAAC;wCACX,OAAO,CAAC,IAAI,CACV,sCAAsC,EACtC,KAAK,CAAC,QAAQ,EACd,CAAC,CACF,CAAC;oCACJ,CAAC;oCACD,MAAM;gCACR,CAAC;gCAED,KAAK,aAAa,CAAC,CAAC,CAAC;oCACnB,IAAI,kBAAU,CAAC,wBAAwB,CAAC,KAAK,CAAC,QAAQ,CAAC;wCAAE,SAAS;oCAElE,IAAI,CAAC;wCACH,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,CAAC;wCAC/B,MAAM,UAAU,GACd,SAAS,KAAK,IAAI,IAAI,OAAO,SAAS,KAAK,QAAQ;4CACjD,CAAC,CAAE,SAAqC;4CACxC,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,SAAS,aAAT,SAAS,cAAT,SAAS,GAAI,EAAE,CAAC,EAAE,CAAC;wCAE1C,aAAa,CAAC,cAAc,CAAC;4CAC3B,UAAU,EAAE,KAAK,CAAC,UAAU;4CAC5B,MAAM,EAAE,UAAU;yCACnB,CAAC,CAAC;oCACL,CAAC;oCAAC,OAAO,CAAC,EAAE,CAAC;wCACX,OAAO,CAAC,IAAI,CACV,uCAAuC,EACvC,KAAK,CAAC,QAAQ,EACd,CAAC,CACF,CAAC;oCACJ,CAAC;oCACD,MAAM;gCACR,CAAC;gCAED,KAAK,YAAY,CAAC,CAAC,CAAC;oCAClB,MAAM,CAAC,IAAI,CAAC;wCACV,IAAI,EAAE,YAAY;wCAClB,OAAO,EAAE,QAAQ,KAAK,CAAC,QAAQ,SAAS;wCACxC,OAAO,EAAE,KAAK,CAAC,KAAK;qCACrB,CAAC,CAAC;oCACH,IAAI,CAAC;wCACH,aAAa,CAAC,YAAY,CAAC,EAAE,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC;oCAC/D,CAAC;oCAAC,OAAO,CAAC,EAAE,CAAC;wCACX,OAAO,CAAC,IAAI,CAAC,qCAAqC,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;oCACzE,CAAC;oCACD,MAAM;gCACR,CAAC;gCAED,KAAK,OAAO,CAAC,CAAC,CAAC;oCACb,MAAM,CAAC,IAAI,CAAC;wCACV,IAAI,EAAE,cAAc;wCACpB,OAAO,EAAE,wBAAwB;wCACjC,OAAO,EAAE,IAAA,cAAO,EAAC,KAAK,CAAC,KAAK,CAAC;qCAC9B,CAAC,CAAC;oCACH,MAAM;gCACR,CAAC;gCAED,2DAA2D;gCAC3D,KAAK,QAAQ;oCACX,MAAM;gCAER;oCACE,MAAM;4BACV,CAAC;wBACH,CAAC;wBAAC,OAAO,UAAU,EAAE,CAAC;4BACpB,OAAO,CAAC,KAAK,CACX,4CAA4C,EAC5C,KAAK,CAAC,IAAI,EACV,IAAA,cAAO,EAAC,UAAU,CAAC,CACpB,CAAC;4BACF,MAAM,CAAC,IAAI,CAAC;gCACV,IAAI,EAAE,wBAAwB;gCAC9B,OAAO,EAAE,wBAAwB,KAAK,CAAC,IAAI,MAAM,MAAM,CACrD,UAAU,CACX,EAAE;gCACH,OAAO,EAAE,IAAA,cAAO,EAAC,UAAU,CAAC;6BAC7B,CAAC,CAAC;wBACL,CAAC;wBAED,mEAAmE;wBACnE,kDAAkD;wBAClD,MAAM,UAAU,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE;4BAC7B,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,IAAA,cAAO,EAAC,CAAC,CAAC,CAAC,CAAC;wBAC7D,CAAC,CAAC,CAAC;oBACL,CAAC;;;;;;;;;gBAED,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACtB,MAAM,YAAY,GAAG,MAAM;yBACxB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,OAAO,EAAE,CAAC;yBACrC,IAAI,CAAC,IAAI,CAAC,CAAC;oBACd,aAAa,CAAC,WAAW,CACvB,gCAAgC,YAAY,EAAE,CAC/C,CAAC;oBACF,aAAa,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,6BAA6B,EAAE,CAAC,CAAC;oBAC/D,MAAM,UAAU,EAAE,CAAC;gBACrB,CAAC;qBAAM,CAAC;oBACN,aAAa,CAAC,SAAS,CAAC,wBAAe,CAAC,SAAS,CAAC,CAAC;gBACrD,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,YAAY,GAAG,8BAA8B,IAAA,cAAO,EAAC,KAAK,CAAC,EAAE,CAAC;gBACpE,aAAa,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;gBAC9C,MAAM,UAAU,EAAE,CAAC;gBACnB,MAAM,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;YAChE,CAAC;YAED,MAAM,WAAW,GAAG,aAAa,CAAC,KAAK,EAAE,CAAC;YAC1C,OAAO,mBAAmB,CAAC,WAAW,CAAuB,CAAC;QAChE,CAAC;KAAA;CACF,CAAC,CAAC"}
+{"version":3,"file":"run-agents.js","sourceRoot":"","sources":["../../../../../../../../../packages/pieces/custom/ai-custom/src/lib/actions/agents/run-agents.ts"],"names":[],"mappings":";;;;AAAA,qEAIwC;AAExC,iDAU8B;AAC9B,2BAA0D;AAC1D,iEAA4D;AAC5D,gDAAoD;AACpD,+BAA+B;AAC/B,mCAAqC;AACrC,mCAA8C;AAC9C,0CAAgD;AA8BhD,iFAAiF;AAEjF,MAAM,kBAAkB,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;AAE7D;;;GAGG;AACH,SAAe,sBAAsB;iEACnC,KAAa,EACb,QAAQ,GAAG,EAAE;;QAEb,MAAM,GAAG,GAAoB,EAAE,CAAC;QAEhC,IAAI,CAAC;YACH,IAAI,IAAI,GAAG,CAAC,CAAC;YACb,IAAI,UAAU,GAAG,CAAC,CAAC;YAEnB,OAAO,IAAI,IAAI,UAAU,EAAE,CAAC;gBAC1B,MAAM,GAAG,GAAG,MAAM,KAAK,CACrB,GAAG,kBAAkB,gCAAgC,IAAI,aAAa,QAAQ,EAAE,EAChF,EAAE,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,KAAK,EAAE,EAAE,EAAE,CAClD,CAAC;gBAEF,IAAI,CAAC,GAAG,CAAC,EAAE;oBAAE,MAAM;gBAEnB,MAAM,IAAI,GAA2B,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;gBACtD,MAAM,MAAM,GAAG,MAAA,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,0CAAE,MAAM,mCAAI,EAAE,CAAC;gBACxC,GAAG,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,CAAC;gBAEpB,IAAI,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,0CAAE,UAAU,EAAE,CAAC;oBAC3B,UAAU,GAAG,MAAA,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,UAAU,mCAAI,CAAC,CAAC;gBACpD,CAAC;qBAAM,CAAC;oBACN,MAAM;gBACR,CAAC;gBAED,IAAI,GAAG,CAAC,MAAM,IAAI,IAAI;oBAAE,MAAM;gBAE9B,IAAI,EAAE,CAAC;YACT,CAAC;QACH,CAAC;QAAC,WAAM,CAAC;YACP,0BAA0B;QAC5B,CAAC;QAED,OAAO,GAAG,CAAC;IACb,CAAC;CAAA;AAED;;GAEG;AACH,SAAS,mBAAmB,CAAC,MAAe;IAC1C,IAAI,MAAM,KAAK,IAAI,IAAI,MAAM,KAAK,SAAS;QAAE,OAAO,MAAM,CAAC;IAC3D,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAC1B,OAAO,MAAM;aACV,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,IAAI,IAAI,IAAI,KAAK,SAAS,CAAC;aACrD,GAAG,CAAC,mBAAmB,CAAC,CAAC;IAC9B,CAAC;IACD,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC/B,MAAM,MAAM,GAA4B,EAAE,CAAC;QAC3C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CACvC,MAAiC,CAClC,EAAE,CAAC;YACF,MAAM,CAAC,GAAG,CAAC,GAAG,mBAAmB,CAAC,KAAK,CAAC,CAAC;QAC3C,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,iFAAiF;AAEjF,MAAM,mBAAmB,GAA2B;IAClD,IAAI,EAAE,2BAAQ,CAAC,SAAS,CAAC,EAAE,WAAW,EAAE,WAAW,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;IACtE,QAAQ,EAAE,2BAAQ,CAAC,SAAS,CAAC,EAAE,WAAW,EAAE,WAAW,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;IAC1E,aAAa,EAAE,2BAAQ,CAAC,IAAI,CAAC;QAC3B,WAAW,EAAE,gBAAgB;QAC7B,QAAQ,EAAE,KAAK;KAChB,CAAC;IACF,cAAc,EAAE,2BAAQ,CAAC,SAAS,CAAC;QACjC,WAAW,EAAE,kBAAkB;QAC/B,QAAQ,EAAE,KAAK;KAChB,CAAC;IACF,SAAS,EAAE,2BAAQ,CAAC,SAAS,CAAC;QAC5B,WAAW,EAAE,gBAAgB;QAC7B,QAAQ,EAAE,KAAK;KAChB,CAAC;IACF,QAAQ,EAAE,2BAAQ,CAAC,SAAS,CAAC,EAAE,WAAW,EAAE,UAAU,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;IAC1E,IAAI,EAAE,2BAAQ,CAAC,IAAI,CAAC,EAAE,WAAW,EAAE,oBAAoB,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;CAC5E,CAAC;AAEF,iFAAiF;AAEpE,QAAA,QAAQ,GAAG,IAAA,+BAAY,EAAC;IACnC,IAAI,EAAE,kBAAkB;IACxB,WAAW,EAAE,WAAW;IACxB,WAAW,EACT,+HAA+H;IACjI,IAAI,EAAE,sBAAc;IACpB,KAAK,EAAE;QACL,CAAC,WAAW,CAAC,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YAC/B,WAAW,EAAE,YAAY;YACzB,QAAQ,EAAE,KAAK;YACf,WAAW,EACT,8FAA8F;SACjG,CAAC;QAEF,CAAC,wBAAe,CAAC,MAAM,CAAC,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YAC1C,WAAW,EAAE,QAAQ;YACrB,WAAW,EAAE,6CAA6C;YAC1D,QAAQ,EAAE,IAAI;SACf,CAAC;QAEF,CAAC,wBAAe,CAAC,iBAAiB,CAAC,EAAE,2BAAQ,CAAC,MAAM,CAAC;YACnD,WAAW,EAAE,UAAU;YACvB,QAAQ,EAAE,IAAI;SACf,CAAC;QAEF,yEAAyE;QACzE,oBAAoB,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YACtC,WAAW,EAAE,8BAA8B;YAC3C,WAAW,EACT,iGAAiG;YACnG,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,KAAK;SACpB,CAAC;QAEF,gBAAgB,EAAE,2BAAQ,CAAC,mBAAmB,CAAC;YAC7C,WAAW,EAAE,iBAAiB;YAC9B,WAAW,EACT,iGAAiG;YACnG,QAAQ,EAAE,KAAK;YACf,IAAI,EAAE,sBAAc;YACpB,UAAU,EAAE,CAAC,sBAAsB,CAAC;YACpC,OAAO,EAAE,CAAO,UAAU,EAAE,GAAG,EAAE,EAAE;gBACjC,MAAM,oBAAoB,GAAG,UAAU,CAAC,sBAAsB,CAEjD,CAAC;gBAEd,IAAI,CAAC,oBAAoB,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;oBACxD,OAAO;wBACL,QAAQ,EAAE,IAAI;wBACd,OAAO,EAAE,EAAE;wBACX,WAAW,EAAE,oCAAoC;qBAClD,CAAC;gBACJ,CAAC;gBAED,MAAM,MAAM,GAAG,MAAM,sBAAsB,CAAC,GAAG,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;gBAEvE,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;oBACnB,OAAO;wBACL,QAAQ,EAAE,IAAI;wBACd,OAAO,EAAE,EAAE;wBACX,WAAW,EAAE,0BAA0B;qBACxC,CAAC;gBACJ,CAAC;gBAED,OAAO;oBACL,QAAQ,EAAE,KAAK;oBACf,WAAW,EAAE,oCAAoC;oBACjD,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;wBAC1B,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE;wBAC/D,KAAK,EAAE,CAAC,CAAC,EAAE;qBACZ,CAAC,CAAC;iBACJ,CAAC;YACJ,CAAC,CAAA;SACF,CAAC;QAEF,uBAAuB,EAAE,2BAAQ,CAAC,cAAc,CAAC;YAC/C,WAAW,EAAE,aAAa;YAC1B,WAAW,EAAE,2CAA2C;YACxD,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,kBAAkB;YAChC,OAAO,EAAE;gBACP,OAAO,EAAE;oBACP;wBACE,KAAK,EAAE,gCAAgC;wBACvC,KAAK,EAAE,kBAAkB;qBAC1B;oBACD,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,UAAU,EAAE;oBACxC,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE;oBACpC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE;iBACvC;aACF;SACF,CAAC;QAEF,yEAAyE;QACzE,CAAC,wBAAe,CAAC,WAAW,CAAC,EAAE,2BAAQ,CAAC,KAAK,CAAC;YAC5C,WAAW,EAAE,aAAa;YAC1B,QAAQ,EAAE,KAAK;YACf,UAAU,EAAE,mBAAmB;SAChC,CAAC;QAEF,CAAC,wBAAe,CAAC,iBAAiB,CAAC,EAAE,2BAAQ,CAAC,KAAK,CAAC;YAClD,WAAW,EAAE,mBAAmB;YAChC,YAAY,EAAE,SAAS;YACvB,QAAQ,EAAE,KAAK;YACf,UAAU,EAAE;gBACV,WAAW,EAAE,2BAAQ,CAAC,SAAS,CAAC;oBAC9B,WAAW,EAAE,cAAc;oBAC3B,QAAQ,EAAE,IAAI;iBACf,CAAC;gBACF,WAAW,EAAE,2BAAQ,CAAC,SAAS,CAAC;oBAC9B,WAAW,EAAE,aAAa;oBAC1B,QAAQ,EAAE,KAAK;iBAChB,CAAC;gBACF,IAAI,EAAE,2BAAQ,CAAC,SAAS,CAAC,EAAE,WAAW,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;aAClE;SACF,CAAC;QAEF,CAAC,wBAAe,CAAC,SAAS,CAAC,EAAE,2BAAQ,CAAC,MAAM,CAAC;YAC3C,WAAW,EAAE,WAAW;YACxB,WAAW,EAAE,2CAA2C;YACxD,QAAQ,EAAE,IAAI;YACd,YAAY,EAAE,EAAE;SACjB,CAAC;KACH;IAED,6EAA6E;IACvE,GAAG,CAAC,OAAO;;;;YACf,MAAM,EACJ,MAAM,EACN,QAAQ,EACR,eAAe,EACf,oBAAoB,EACpB,gBAAgB,EAChB,uBAAuB,GACxB,GAAG,OAAO,CAAC,UAAU,CAAC;YAEvB,MAAM,kBAAkB,GAAG,eAAqC,CAAC;YAEjE,MAAM,KAAK,GAAG,MAAM,IAAA,sBAAa,EAAC;gBAChC,OAAO,EAAE,kBAAkB,CAAC,KAAK;gBACjC,QAAQ,EAAE,kBAAkB,CAAC,QAA0B;gBACvD,WAAW,EAAE,OAAO,CAAC,MAAM,CAAC,KAAK;gBACjC,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,MAAM;gBAC7B,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,EAAE;gBAC7B,MAAM,EAAE,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE;gBAChC,KAAK,EAAE,OAAO,CAAC,GAAG,CAAC,EAAE;aACtB,CAAC,CAAC;YAEH,MAAM,aAAa,GAAG,IAAA,yCAAkB,EAAC,MAAM,CAAC,CAAC;YAEjD,MAAM,mBAAmB,GACvB,CAAC,IAAA,cAAK,EAAC,OAAO,CAAC,UAAU,CAAC,gBAAgB,CAAC;gBAC3C,OAAO,CAAC,UAAU,CAAC,gBAAgB,CAAC,MAAM,GAAG,CAAC,CAAC;YACjD,MAAM,gBAAgB,GAAG,mBAAmB;gBAC1C,CAAC,CAAE,OAAO,CAAC,UAAU,CAAC,gBAAuC;gBAC7D,CAAC,CAAC,SAAS,CAAC;YAEd,MAAM,UAAU,GAAG,CAAC,MAAA,OAAO,CAAC,UAAU,CAAC,UAAU,mCAAI,EAAE,CAAgB,CAAC;YAExE,4EAA4E;YAC5E,MAAM,gBAAgB,GACpB,oBAAoB;gBACpB,KAAK,CAAC,OAAO,CAAC,gBAAgB,CAAC;gBAC/B,gBAAgB,CAAC,MAAM,GAAG,CAAC;gBACzB,CAAC,CAAC,gBAAgB;gBAClB,CAAC,CAAC,EAAE,CAAC;YAET,MAAM,oBAAoB,GACxB,gBAAgB,CAAC,MAAM,GAAG,CAAC;gBACzB,CAAC,CAAC;oBACE,GAAG,UAAU;oBACb;wBACE,IAAI,EAAE,gBAAgB;wBACtB,QAAQ,EAAE,uBAAuB;wBACjC,IAAI,EAAE;4BACJ,QAAQ,EAAE,gBAAgB;4BAC1B,UAAU,EAAE,uBAAuB,aAAvB,uBAAuB,cAAvB,uBAAuB,GAAI,kBAAkB;4BACzD,KAAK,EAAE,OAAO,CAAC,MAAM,CAAC,KAAK;yBAC5B;qBACsB;iBAC1B;gBACH,CAAC,CAAC,UAAU,CAAC;YAEjB,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,MAAM,IAAA,2BAAmB,EAAC;gBACtD,OAAO;gBACP,UAAU,EAAE,oBAAoB;gBAChC,KAAK;gBACL,aAAa;gBACb,gBAAgB;aACjB,CAAC,CAAC;YAEH,MAAM,MAAM,GAA2D,EAAE,CAAC;YAE1E,wEAAwE;YACxE,MAAM,UAAU,GAAG,kBAAU,CAAC,UAAU,CAAC,MAAgB,CAAC,CAAC,MAAM,CAAC;YAElE,MAAM,MAAM,GACV,gBAAgB,CAAC,MAAM,GAAG,CAAC;gBACzB,CAAC,CAAC,GAAG,UAAU;;;qBAGF,gBAAgB,CAAC,MAAM,kBAAkB,gBAAgB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;;;;8EAItB;gBACtE,CAAC,CAAC,UAAU,CAAC;YAEjB,MAAM,UAAU,GAAG,GAAS,EAAE;gBAC5B,MAAM,GAAG,GAAG,aAAa,CAAC,KAAK,EAAE,CAAC;gBAClC,MAAM,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC;oBAC1B,IAAI,EAAE,mBAAmB,CAAC,GAAG,CAAe;iBAC7C,CAAC,CAAC;YACL,CAAC,CAAA,CAAC;YAEF,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,IAAA,eAAU,EAAC;oBACxB,KAAK;oBACL,MAAM;oBACN,MAAM,EAAE,kBAAU,CAAC,UAAU,CAAC,MAAgB,CAAC,CAAC,MAAM;oBACtD,KAAK;oBACL,QAAQ,EAAE;wBACR,IAAA,gBAAW,EAAC,QAAQ,CAAC;wBACrB,IAAA,gBAAW,EAAC,kCAAyB,CAAC;qBACvC;oBACD,QAAQ,EAAE,GAAS,EAAE;wBACnB,MAAM,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;oBAChE,CAAC,CAAA;iBACF,CAAC,CAAC;;oBAEH,KAA0B,eAAA,KAAA,sBAAA,MAAM,CAAC,UAAU,CAAA,IAAA,sDAAE,CAAC;wBAApB,cAAiB;wBAAjB,WAAiB;wBAAhC,MAAM,KAAK,KAAA,CAAA;wBACpB,IAAI,CAAC;4BACH,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;gCACnB,KAAK,YAAY,CAAC,CAAC,CAAC;oCAClB,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;wCACnC,aAAa,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oCACxC,CAAC;oCACD,MAAM;gCACR,CAAC;gCAED,KAAK,WAAW,CAAC,CAAC,CAAC;oCACjB,IAAI,kBAAU,CAAC,wBAAwB,CAAC,KAAK,CAAC,QAAQ,CAAC;wCAAE,SAAS;oCAElE,IAAI,CAAC;wCACH,aAAa,CAAC,aAAa,CAAC;4CAC1B,QAAQ,EAAE,KAAK,CAAC,QAAQ;4CACxB,UAAU,EAAE,KAAK,CAAC,UAAU;4CAC5B,KAAK,EAAE,KAAK,CAAC,KAAgC;4CAC7C,UAAU,EAAE,oBAAoB;yCACjC,CAAC,CAAC;oCACL,CAAC;oCAAC,OAAO,CAAC,EAAE,CAAC;wCACX,OAAO,CAAC,IAAI,CACV,sCAAsC,EACtC,KAAK,CAAC,QAAQ,EACd,CAAC,CACF,CAAC;oCACJ,CAAC;oCACD,MAAM;gCACR,CAAC;gCAED,KAAK,aAAa,CAAC,CAAC,CAAC;oCACnB,IAAI,kBAAU,CAAC,wBAAwB,CAAC,KAAK,CAAC,QAAQ,CAAC;wCAAE,SAAS;oCAElE,IAAI,CAAC;wCACH,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,CAAC;wCAC/B,MAAM,UAAU,GACd,SAAS,KAAK,IAAI,IAAI,OAAO,SAAS,KAAK,QAAQ;4CACjD,CAAC,CAAE,SAAqC;4CACxC,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,SAAS,aAAT,SAAS,cAAT,SAAS,GAAI,EAAE,CAAC,EAAE,CAAC;wCAE1C,aAAa,CAAC,cAAc,CAAC;4CAC3B,UAAU,EAAE,KAAK,CAAC,UAAU;4CAC5B,MAAM,EAAE,UAAU;yCACnB,CAAC,CAAC;oCACL,CAAC;oCAAC,OAAO,CAAC,EAAE,CAAC;wCACX,OAAO,CAAC,IAAI,CACV,uCAAuC,EACvC,KAAK,CAAC,QAAQ,EACd,CAAC,CACF,CAAC;oCACJ,CAAC;oCACD,MAAM;gCACR,CAAC;gCAED,KAAK,YAAY,CAAC,CAAC,CAAC;oCAClB,MAAM,CAAC,IAAI,CAAC;wCACV,IAAI,EAAE,YAAY;wCAClB,OAAO,EAAE,QAAQ,KAAK,CAAC,QAAQ,SAAS;wCACxC,OAAO,EAAE,KAAK,CAAC,KAAK;qCACrB,CAAC,CAAC;oCACH,IAAI,CAAC;wCACH,aAAa,CAAC,YAAY,CAAC,EAAE,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC;oCAC/D,CAAC;oCAAC,OAAO,CAAC,EAAE,CAAC;wCACX,OAAO,CAAC,IAAI,CACV,qCAAqC,EACrC,KAAK,CAAC,QAAQ,EACd,CAAC,CACF,CAAC;oCACJ,CAAC;oCACD,MAAM;gCACR,CAAC;gCAED,KAAK,OAAO,CAAC,CAAC,CAAC;oCACb,MAAM,CAAC,IAAI,CAAC;wCACV,IAAI,EAAE,cAAc;wCACpB,OAAO,EAAE,wBAAwB;wCACjC,OAAO,EAAE,IAAA,cAAO,EAAC,KAAK,CAAC,KAAK,CAAC;qCAC9B,CAAC,CAAC;oCACH,MAAM;gCACR,CAAC;gCAED,KAAK,QAAQ;oCACX,MAAM;gCAER;oCACE,MAAM;4BACV,CAAC;wBACH,CAAC;wBAAC,OAAO,UAAU,EAAE,CAAC;4BACpB,OAAO,CAAC,KAAK,CACX,4CAA4C,EAC5C,KAAK,CAAC,IAAI,EACV,IAAA,cAAO,EAAC,UAAU,CAAC,CACpB,CAAC;4BACF,MAAM,CAAC,IAAI,CAAC;gCACV,IAAI,EAAE,wBAAwB;gCAC9B,OAAO,EAAE,wBAAwB,KAAK,CAAC,IAAI,MAAM,MAAM,CAAC,UAAU,CAAC,EAAE;gCACrE,OAAO,EAAE,IAAA,cAAO,EAAC,UAAU,CAAC;6BAC7B,CAAC,CAAC;wBACL,CAAC;wBAED,MAAM,UAAU,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE;4BAC7B,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,IAAA,cAAO,EAAC,CAAC,CAAC,CAAC,CAAC;wBAC7D,CAAC,CAAC,CAAC;oBACL,CAAC;;;;;;;;;gBAED,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACtB,MAAM,YAAY,GAAG,MAAM;yBACxB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,OAAO,EAAE,CAAC;yBACrC,IAAI,CAAC,IAAI,CAAC,CAAC;oBACd,aAAa,CAAC,WAAW,CACvB,gCAAgC,YAAY,EAAE,CAC/C,CAAC;oBACF,aAAa,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,6BAA6B,EAAE,CAAC,CAAC;oBAC/D,MAAM,UAAU,EAAE,CAAC;gBACrB,CAAC;qBAAM,CAAC;oBACN,aAAa,CAAC,SAAS,CAAC,wBAAe,CAAC,SAAS,CAAC,CAAC;gBACrD,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,YAAY,GAAG,8BAA8B,IAAA,cAAO,EAAC,KAAK,CAAC,EAAE,CAAC;gBACpE,aAAa,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;gBAC9C,MAAM,UAAU,EAAE,CAAC;gBACnB,MAAM,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;YAChE,CAAC;YAED,MAAM,WAAW,GAAG,aAAa,CAAC,KAAK,EAAE,CAAC;YAC1C,OAAO,mBAAmB,CAAC,WAAW,CAAuB,CAAC;QAChE,CAAC;KAAA;CACF,CAAC,CAAC"}

package/src/lib/actions/agents/search-document-store.d.ts

@@ -1,7 +1,7 @@
 export declare const searchDocumentStore: import("@activepieces/pieces-framework").IAction<import("@activepieces/pieces-framework").OAuth2Property<import("@activepieces/pieces-framework").OAuth2Props>, {
     adminFlow: import("@activepieces/pieces-framework").CheckboxProperty<false>;
-    documentStoreId: import("@activepieces/pieces-framework").DropdownProperty<string, true, undefined>;
-    datasetIds: import("@activepieces/pieces-framework").MultiSelectDropdownProperty<string, false, undefined>;
+    documentStoreId: import("@activepieces/pieces-framework").DropdownProperty<string, true, import("@activepieces/pieces-framework").OAuth2Property<import("@activepieces/pieces-framework").OAuth2Props>>;
+    datasetIds: import("@activepieces/pieces-framework").MultiSelectDropdownProperty<string, false, import("@activepieces/pieces-framework").OAuth2Property<import("@activepieces/pieces-framework").OAuth2Props>>;
     question: import("@activepieces/pieces-framework").LongTextProperty<true>;
     searchType: import("@activepieces/pieces-framework").StaticDropdownProperty<string, false>;
     enableCitation: import("@activepieces/pieces-framework").CheckboxProperty<false>;

package/src/lib/actions/agents/search-document-store.js

@@ -24,14 +24,6 @@ function fetchDocumentStores(token) {
         }
     });
 }
-function extractToken(auth, adminFlow, oauthUserToken) {
-    var _a;
-    const oauth = auth;
-    if (adminFlow) {
-        return oauth.access_token;
-    }
-    return (_a = oauthUserToken !== null && oauthUserToken !== void 0 ? oauthUserToken : oauthUserToken) !== null && _a !== void 0 ? _a : '';
-}
 // ─── Action ───────────────────────────────────────────────────────────────────
 exports.searchDocumentStore = (0, pieces_framework_1.createAction)({
     name: 'search_document_store',
@@ -51,7 +43,7 @@ exports.searchDocumentStore = (0, pieces_framework_1.createAction)({
             displayName: 'Document Store',
             description: 'Select the document store to search.',
             required: true,
-            auth: undefined,
+            auth: index_1.httpOauth2Auth,
             refreshers: [],
             options: (propsValue, context) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
                 console.log("From document store");
@@ -80,7 +72,7 @@ exports.searchDocumentStore = (0, pieces_framework_1.createAction)({
             displayName: 'Datasets (optional)',
             description: 'Restrict search to specific datasets. Leave empty to search the entire store.',
             required: false,
-            auth: undefined,
+            auth: index_1.httpOauth2Auth,
             refreshers: ['documentStoreId'],
             options: (propsValue, ctx) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
                 var _a;

package/src/lib/actions/agents/search-document-store.js.map

@@ -1 +1 @@
-{"version":3,"file":"search-document-store.js","sourceRoot":"","sources":["../../../../../../../../../packages/pieces/custom/ai-custom/src/lib/actions/agents/search-document-store.ts"],"names":[],"mappings":";;;;AAAA,qEAAwE;AACxE,0CAAgD;AAkChD,iFAAiF;AAEjF,MAAM,kBAAkB,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;AAE7D,iFAAiF;AAEjF,SAAe,mBAAmB,CAAC,KAAa;;;QAC9C,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,kBAAkB,yBAAyB,EAAE;gBACtE,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,KAAK,EAAE,EAAE;aAC9C,CAAC,CAAC;YACH,IAAI,CAAC,GAAG,CAAC,EAAE;gBAAE,OAAO,EAAE,CAAC;YACvB,MAAM,IAAI,GAA2B,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;YACtD,OAAO,MAAA,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,0CAAE,MAAM,mCAAI,EAAE,CAAC;QAClC,CAAC;QAAC,WAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;CAAA;AAED,SAAS,YAAY,CACnB,IAAa,EACb,SAAkB,EAClB,cAAyC;;IAEzC,MAAM,KAAK,GAAG,IAAoC,CAAC;IACnD,IAAI,SAAS,EAAE,CAAC;QACd,OAAO,KAAK,CAAC,YAAY,CAAC;IAC5B,CAAC;IACD,OAAO,MAAA,cAAc,aAAd,cAAc,cAAd,cAAc,GAAI,cAAc,mCAAI,EAAE,CAAC;AAChD,CAAC;AAED,iFAAiF;AAEpE,QAAA,mBAAmB,GAAG,IAAA,+BAAY,EAAC;IAC9C,IAAI,EAAE,uBAAuB;IAC7B,WAAW,EAAE,uBAAuB;IACpC,WAAW,EAAE,uFAAuF;IACpG,IAAI,EAAE,sBAAc;IACpB,KAAK,EAAE;QACL,yEAAyE;QACzE,SAAS,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YAC3B,WAAW,EAAE,YAAY;YACzB,WAAW,EAAE,sEAAsE;YACnF,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,KAAK;SACpB,CAAC;QAEF,yEAAyE;QACzE,eAAe,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YACjC,WAAW,EAAE,gBAAgB;YAC7B,WAAW,EAAE,sCAAsC;YACnD,QAAQ,EAAE,IAAI;YACd,IAAI,EAAE,SAAS;YACf,UAAU,EAAE,EAAE;YACd,OAAO,EAAE,CAAO,UAAU,EAAE,OAAO,EAAE,EAAE;gBACrC,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAA;gBAClC,OAAO,CAAC,GAAG,CAAC,EAAC,UAAU,EAAC,CAAC,CAAA;gBAEzB,4EAA4E;gBAC5E,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC,cAAc,CAAA;gBAE3C,IAAI,CAAC,KAAK,EAAE,CAAC;oBACX,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,WAAW,EAAE,4BAA4B,EAAE,CAAC;gBACpF,CAAC;gBAED,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,KAAK,CAAC,CAAC;gBAEhD,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;oBACnB,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,WAAW,EAAE,0BAA0B,EAAE,CAAC;gBAClF,CAAC;gBAED,OAAO;oBACL,QAAQ,EAAE,KAAK;oBACf,WAAW,EAAE,yBAAyB;oBACtC,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;wBAC1B,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE;wBAC/D,KAAK,EAAE,CAAC,CAAC,EAAE;qBACZ,CAAC,CAAC;iBACJ,CAAC;YACJ,CAAC,CAAA;SACF,CAAC;QAEF,yEAAyE;QACzE,UAAU,EAAE,2BAAQ,CAAC,mBAAmB,CAAC;YACvC,WAAW,EAAE,qBAAqB;YAClC,WAAW,EAAE,+EAA+E;YAC5F,QAAQ,EAAE,KAAK;YACf,IAAI,EAAE,SAAS;YACf,UAAU,EAAE,CAAC,iBAAiB,CAAC;YAC/B,OAAO,EAAE,CAAO,UAAU,EAAE,GAAG,EAAE,EAAE;;gBACjC,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAA;gBAC3C,OAAO,CAAC,GAAG,CAAC,EAAC,UAAU,EAAC,CAAC,CAAA;gBACzB,4EAA4E;gBAC5E,MAAM,eAAe,GAAG,UAAU,CAAC,iBAAiB,CAAuB,CAAC;gBAC5E,MAAM,KAAK,GAAG,GAAG,CAAC,MAAM,CAAC,cAAc,CAAA;gBAEvC,IAAI,CAAC,KAAK,IAAI,CAAC,eAAe,EAAE,CAAC;oBAC/B,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,WAAW,EAAE,+BAA+B,EAAE,CAAC;gBACvF,CAAC;gBAED,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,KAAK,CAAC,CAAC;gBAChD,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,eAAe,CAAC,CAAC;gBAE3D,IAAI,CAAC,CAAA,MAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,QAAQ,0CAAE,MAAM,CAAA,EAAE,CAAC;oBAC7B,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,WAAW,EAAE,iCAAiC,EAAE,CAAC;gBACzF,CAAC;gBAED,OAAO;oBACL,QAAQ,EAAE,KAAK;oBACf,WAAW,EAAE,kDAAkD;oBAC/D,OAAO,EAAE,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;wBAClC,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE;wBAC/D,KAAK,EAAE,CAAC,CAAC,EAAE;qBACZ,CAAC,CAAC;iBACJ,CAAC;YACJ,CAAC,CAAA;SACF,CAAC;QAEF,yEAAyE;QACzE,QAAQ,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YAC1B,WAAW,EAAE,kBAAkB;YAC/B,WAAW,EAAE,uDAAuD;YACpE,QAAQ,EAAE,IAAI;SACf,CAAC;QAEF,yEAAyE;QACzE,UAAU,EAAE,2BAAQ,CAAC,cAAc,CAAC;YAClC,WAAW,EAAE,aAAa;YAC1B,WAAW,EAAE,uDAAuD;YACpE,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,kBAAkB;YAChC,OAAO,EAAE;gBACP,OAAO,EAAE;oBACP,EAAE,KAAK,EAAE,gCAAgC,EAAE,KAAK,EAAE,kBAAkB,EAAE;oBACtE,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,UAAU,EAAE;oBACxC,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE;oBACpC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE;iBACvC;aACF;SACF,CAAC;QAEF,cAAc,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YAChC,WAAW,EAAE,mBAAmB;YAChC,WAAW,EAAE,qDAAqD;YAClE,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,IAAI;SACnB,CAAC;QAEF,iBAAiB,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YACnC,WAAW,EAAE,yBAAyB;YACtC,WAAW,EAAE,iDAAiD;YAC9D,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,KAAK;SACpB,CAAC;KACH;IAED,6EAA6E;IACvE,GAAG,CAAC,OAAO;;;YACf,MAAM,EACJ,SAAS,EACT,eAAe,EACf,UAAU,EACV,QAAQ,EACR,UAAU,EACV,cAAc,EACd,iBAAiB,GAClB,GAAG,OAAO,CAAC,UAAU,CAAC;YAGvB,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,CAAA;YAC5B,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC,cAAc,CAAC;YAE5C,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,MAAM,IAAI,KAAK,CAAC,iEAAiE,CAAC,CAAC;YACrF,CAAC;YAED,MAAM,IAAI,GAAkB;gBAC1B,SAAS,EAAE,CAAC,eAAyB,CAAC;gBACtC,QAAQ,EAAE,QAAkB;gBAC5B,eAAe,EAAE,cAAc,aAAd,cAAc,cAAd,cAAc,GAAI,IAAI;gBACvC,kBAAkB,EAAE,iBAAiB,aAAjB,iBAAiB,cAAjB,iBAAiB,GAAI,KAAK;gBAC9C,WAAW,EAAE,MAAC,UAAqB,mCAAI,kBAAkB;aAC1D,CAAC;YAEF,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACvD,IAAI,CAAC,WAAW,GAAG,UAAsB,CAAC;YAC5C,CAAC;YAED,OAAO,CAAC,GAAG,CAAC,EAAC,IAAI,EAAC,CAAC,CAAA;YACnB,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,kBAAkB,gCAAgC,EAAE;gBAC7E,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,MAAM,EAAE,kBAAkB;oBAC1B,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,UAAU,KAAK,EAAE;iBACjC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,CAAC,CAAC;YAEH,OAAO,CAAC,GAAG,CAAC,EAAC,GAAG,EAAC,CAAC,CAAA;YAElB,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;gBACZ,MAAM,SAAS,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;gBAC/D,MAAM,IAAI,KAAK,CAAC,sCAAsC,GAAG,CAAC,MAAM,MAAM,SAAS,EAAE,CAAC,CAAC;YACrF,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;YAE9B,mEAAmE;YACnE,6CAA6C;YAC7C,OAAO;gBACL,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,QAAQ;gBACR,OAAO,EAAE,eAAe;gBACxB,UAAU,EAAE,UAAU,aAAV,UAAU,cAAV,UAAU,GAAI,EAAE;gBAC5B,UAAU,EAAE,UAAU,aAAV,UAAU,cAAV,UAAU,GAAI,kBAAkB;gBAC5C,MAAM,EAAE,MAAA,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,0CAAE,MAAM,mCAAI,EAAE;gBAChC,SAAS,EAAE,MAAA,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,0CAAE,SAAS,mCAAI,EAAE;gBACtC,GAAG,EAAE,IAAI;aACV,CAAC;QACJ,CAAC;KAAA;CACF,CAAC,CAAC"}
+{"version":3,"file":"search-document-store.js","sourceRoot":"","sources":["../../../../../../../../../packages/pieces/custom/ai-custom/src/lib/actions/agents/search-document-store.ts"],"names":[],"mappings":";;;;AAAA,qEAAwE;AACxE,0CAAgD;AAiChD,iFAAiF;AAEjF,MAAM,kBAAkB,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;AAE7D,iFAAiF;AAEjF,SAAe,mBAAmB,CAAC,KAAa;;;QAC9C,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,kBAAkB,yBAAyB,EAAE;gBACtE,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,KAAK,EAAE,EAAE;aAC9C,CAAC,CAAC;YACH,IAAI,CAAC,GAAG,CAAC,EAAE;gBAAE,OAAO,EAAE,CAAC;YACvB,MAAM,IAAI,GAA2B,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;YACtD,OAAO,MAAA,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,0CAAE,MAAM,mCAAI,EAAE,CAAC;QAClC,CAAC;QAAC,WAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;CAAA;AAED,iFAAiF;AAEpE,QAAA,mBAAmB,GAAG,IAAA,+BAAY,EAAC;IAC9C,IAAI,EAAE,uBAAuB;IAC7B,WAAW,EAAE,uBAAuB;IACpC,WAAW,EAAE,uFAAuF;IACpG,IAAI,EAAE,sBAAc;IACpB,KAAK,EAAE;QACL,yEAAyE;QACzE,SAAS,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YAC3B,WAAW,EAAE,YAAY;YACzB,WAAW,EAAE,sEAAsE;YACnF,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,KAAK;SACpB,CAAC;QAEF,yEAAyE;QACzE,eAAe,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YACjC,WAAW,EAAE,gBAAgB;YAC7B,WAAW,EAAE,sCAAsC;YACnD,QAAQ,EAAE,IAAI;YACd,IAAI,EAAE,sBAAc;YACpB,UAAU,EAAE,EAAE;YACd,OAAO,EAAE,CAAO,UAAU,EAAE,OAAO,EAAE,EAAE;gBACrC,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAA;gBAClC,OAAO,CAAC,GAAG,CAAC,EAAC,UAAU,EAAC,CAAC,CAAA;gBAEzB,4EAA4E;gBAC5E,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC,cAAc,CAAA;gBAE3C,IAAI,CAAC,KAAK,EAAE,CAAC;oBACX,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,WAAW,EAAE,4BAA4B,EAAE,CAAC;gBACpF,CAAC;gBAED,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,KAAK,CAAC,CAAC;gBAEhD,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;oBACnB,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,WAAW,EAAE,0BAA0B,EAAE,CAAC;gBAClF,CAAC;gBAED,OAAO;oBACL,QAAQ,EAAE,KAAK;oBACf,WAAW,EAAE,yBAAyB;oBACtC,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;wBAC1B,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE;wBAC/D,KAAK,EAAE,CAAC,CAAC,EAAE;qBACZ,CAAC,CAAC;iBACJ,CAAC;YACJ,CAAC,CAAA;SACF,CAAC;QAEF,yEAAyE;QACzE,UAAU,EAAE,2BAAQ,CAAC,mBAAmB,CAAC;YACvC,WAAW,EAAE,qBAAqB;YAClC,WAAW,EAAE,+EAA+E;YAC5F,QAAQ,EAAE,KAAK;YACf,IAAI,EAAE,sBAAc;YACpB,UAAU,EAAE,CAAC,iBAAiB,CAAC;YAC/B,OAAO,EAAE,CAAO,UAAU,EAAE,GAAG,EAAE,EAAE;;gBACjC,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAA;gBAC3C,OAAO,CAAC,GAAG,CAAC,EAAC,UAAU,EAAC,CAAC,CAAA;gBACzB,4EAA4E;gBAC5E,MAAM,eAAe,GAAG,UAAU,CAAC,iBAAiB,CAAuB,CAAC;gBAC5E,MAAM,KAAK,GAAG,GAAG,CAAC,MAAM,CAAC,cAAc,CAAA;gBAEvC,IAAI,CAAC,KAAK,IAAI,CAAC,eAAe,EAAE,CAAC;oBAC/B,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,WAAW,EAAE,+BAA+B,EAAE,CAAC;gBACvF,CAAC;gBAED,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,KAAK,CAAC,CAAC;gBAChD,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,eAAe,CAAC,CAAC;gBAE3D,IAAI,CAAC,CAAA,MAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,QAAQ,0CAAE,MAAM,CAAA,EAAE,CAAC;oBAC7B,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,WAAW,EAAE,iCAAiC,EAAE,CAAC;gBACzF,CAAC;gBAED,OAAO;oBACL,QAAQ,EAAE,KAAK;oBACf,WAAW,EAAE,kDAAkD;oBAC/D,OAAO,EAAE,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;wBAClC,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE;wBAC/D,KAAK,EAAE,CAAC,CAAC,EAAE;qBACZ,CAAC,CAAC;iBACJ,CAAC;YACJ,CAAC,CAAA;SACF,CAAC;QAEF,yEAAyE;QACzE,QAAQ,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YAC1B,WAAW,EAAE,kBAAkB;YAC/B,WAAW,EAAE,uDAAuD;YACpE,QAAQ,EAAE,IAAI;SACf,CAAC;QAEF,yEAAyE;QACzE,UAAU,EAAE,2BAAQ,CAAC,cAAc,CAAC;YAClC,WAAW,EAAE,aAAa;YAC1B,WAAW,EAAE,uDAAuD;YACpE,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,kBAAkB;YAChC,OAAO,EAAE;gBACP,OAAO,EAAE;oBACP,EAAE,KAAK,EAAE,gCAAgC,EAAE,KAAK,EAAE,kBAAkB,EAAE;oBACtE,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,UAAU,EAAE;oBACxC,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE;oBACpC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE;iBACvC;aACF;SACF,CAAC;QAEF,cAAc,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YAChC,WAAW,EAAE,mBAAmB;YAChC,WAAW,EAAE,qDAAqD;YAClE,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,IAAI;SACnB,CAAC;QAEF,iBAAiB,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YACnC,WAAW,EAAE,yBAAyB;YACtC,WAAW,EAAE,iDAAiD;YAC9D,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,KAAK;SACpB,CAAC;KACH;IAED,6EAA6E;IACvE,GAAG,CAAC,OAAO;;;YACf,MAAM,EACJ,SAAS,EACT,eAAe,EACf,UAAU,EACV,QAAQ,EACR,UAAU,EACV,cAAc,EACd,iBAAiB,GAClB,GAAG,OAAO,CAAC,UAAU,CAAC;YAGvB,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,CAAA;YAC5B,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC,cAAc,CAAC;YAE5C,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,MAAM,IAAI,KAAK,CAAC,iEAAiE,CAAC,CAAC;YACrF,CAAC;YAED,MAAM,IAAI,GAAkB;gBAC1B,SAAS,EAAE,CAAC,eAAyB,CAAC;gBACtC,QAAQ,EAAE,QAAkB;gBAC5B,eAAe,EAAE,cAAc,aAAd,cAAc,cAAd,cAAc,GAAI,IAAI;gBACvC,kBAAkB,EAAE,iBAAiB,aAAjB,iBAAiB,cAAjB,iBAAiB,GAAI,KAAK;gBAC9C,WAAW,EAAE,MAAC,UAAqB,mCAAI,kBAAkB;aAC1D,CAAC;YAEF,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACvD,IAAI,CAAC,WAAW,GAAG,UAAsB,CAAC;YAC5C,CAAC;YAED,OAAO,CAAC,GAAG,CAAC,EAAC,IAAI,EAAC,CAAC,CAAA;YACnB,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,kBAAkB,gCAAgC,EAAE;gBAC7E,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,MAAM,EAAE,kBAAkB;oBAC1B,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,UAAU,KAAK,EAAE;iBACjC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,CAAC,CAAC;YAEH,OAAO,CAAC,GAAG,CAAC,EAAC,GAAG,EAAC,CAAC,CAAA;YAElB,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;gBACZ,MAAM,SAAS,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;gBAC/D,MAAM,IAAI,KAAK,CAAC,sCAAsC,GAAG,CAAC,MAAM,MAAM,SAAS,EAAE,CAAC,CAAC;YACrF,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;YAE9B,mEAAmE;YACnE,6CAA6C;YAC7C,OAAO;gBACL,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,QAAQ;gBACR,OAAO,EAAE,eAAe;gBACxB,UAAU,EAAE,UAAU,aAAV,UAAU,cAAV,UAAU,GAAI,EAAE;gBAC5B,UAAU,EAAE,UAAU,aAAV,UAAU,cAAV,UAAU,GAAI,kBAAkB;gBAC5C,MAAM,EAAE,MAAA,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,0CAAE,MAAM,mCAAI,EAAE;gBAChC,SAAS,EAAE,MAAA,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,0CAAE,SAAS,mCAAI,EAAE;gBACtC,GAAG,EAAE,IAAI;aACV,CAAC;QACJ,CAAC;KAAA;CACF,CAAC,CAAC"}

package/src/lib/actions/agents/tools.js

@@ -8,8 +8,8 @@ const utils_1 = require("./utils");
 const shared_1 = require("@activepieces/shared");
 const streamableHttp_js_1 = require("@modelcontextprotocol/sdk/client/streamableHttp.js");
 const mcp_1 = require("@ai-sdk/mcp");
-const document_store_tool_1 = require("./document-store-tool"); // 👈 ADD THIS
-// ─── (all existing helpers unchanged below) ───────────────────────────────────
+const document_store_tool_1 = require("./document-store-tool");
+// ─── Transport ────────────────────────────────────────────────────────────────
 function createTransportConfig(protocol, serverUrl, headers = {}) {
     const url = new URL(serverUrl);
     switch (protocol) {
@@ -107,7 +107,12 @@ function constructMcpServersTools(tools, token) {
         return collected;
     });
 }
-// ─── 👇 NEW: build document store tools from the injected DOCUMENT_STORE entries ─
+// ─── Document store tools ─────────────────────────────────────────────────────
+/**
+ * Finds all DOCUMENT_STORE entries in agentTools and builds a single
+ * AI SDK tool for each one. The config's `storeIds` array is passed
+ * directly to the search API so multiple stores are queried in one call.
+ */
 function buildDocumentStoreTools(agentTools, token) {
     var _a;
     const tools = {};
@@ -115,11 +120,12 @@ function buildDocumentStoreTools(agentTools, token) {
     for (const dsTool of dsTools) {
         const config = dsTool.auth;
         const toolName = (_a = dsTool.toolName) !== null && _a !== void 0 ? _a : 'search_document_store';
+        // config.storeIds is already the full array set in run-agent.ts
         tools[toolName] = (0, document_store_tool_1.buildDocumentStoreTool)(config, token);
     }
     return tools;
 }
-// ─────────────────────────────────────────────────────────────────────────────
+// ─── Main export ──────────────────────────────────────────────────────────────
 function constructAgentTools(params) {
     return tslib_1.__awaiter(this, void 0, void 0, function* () {
         var _a, _b;
@@ -156,7 +162,6 @@ function constructAgentTools(params) {
         }
         const mcpServerTools = yield constructMcpServersTools(agentTools.filter(tool => tool.type === shared_1.AgentToolType.MCP), token ? token : '');
         const { mcpClients, tools: mcpTools } = flattenMcpServers(mcpServerTools);
-        // 👇 ADD THIS: build document store tools and merge them in
         const documentStoreTools = buildDocumentStoreTools(agentTools, token);
         const combinedTools = Object.assign(Object.assign(Object.assign(Object.assign({}, agentPieceTools), flowsTools), mcpTools), documentStoreTools);
         return {

package/src/lib/actions/agents/tools.js.map

@@ -1 +1 @@
-{"version":3,"file":"tools.js","sourceRoot":"","sources":["../../../../../../../../../packages/pieces/custom/ai-custom/src/lib/actions/agents/tools.ts"],"names":[],"mappings":";;AA+JA,kDA0GC;;AAzQD,2BAAsD;AACtD,sDAAoB;AACpB,mCAAqC;AAErC,iDAA0N;AAC1N,0FAAkG;AAElG,qCAAuG;AACvG,+DAAwF,CAAC,cAAc;AAEvG,iFAAiF;AAEjF,SAAS,qBAAqB,CAC1B,QAAqB,EACrB,SAAiB,EACjB,UAAkC,EAAE;IAEpC,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,SAAS,CAAC,CAAA;IAE9B,QAAQ,QAAQ,EAAE,CAAC;QACf,KAAK,oBAAW,CAAC,WAAW,CAAC,CAAC,CAAC;YAC3B,OAAO;gBACH,IAAI,EAAE,MAAM;gBACZ,GAAG,EAAE,SAAS;gBACd,OAAO;aACV,CAAA;QACL,CAAC;QACD,KAAK,oBAAW,CAAC,eAAe,CAAC,CAAC,CAAC;YAC/B,OAAO,IAAI,iDAA6B,CAAC,GAAG,EAAE;gBAC1C,WAAW,EAAE;oBACT,OAAO;iBACV;aACJ,CAAC,CAAA;QACN,CAAC;QACD,KAAK,oBAAW,CAAC,GAAG,CAAC,CAAC,CAAC;YACnB,OAAO;gBACH,IAAI,EAAE,KAAK;gBACX,GAAG,EAAE,SAAS;gBACd,OAAO;aACV,CAAA;QACL,CAAC;QACD;YACI,MAAM,IAAI,KAAK,CAAC,kCAAkC,QAAQ,EAAE,CAAC,CAAA;IACrE,CAAC;AACL,CAAC;AAOD,SAAS,iBAAiB,CACtB,OAAyB;IAEzB,MAAM,UAAU,GAAgB,EAAE,CAAC;IACnC,MAAM,KAAK,GAAyB,EAAE,CAAC;IAEvC,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAElC,KAAK,MAAM,CAAC,QAAQ,EAAE,EAAE,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1D,MAAM,gBAAgB,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;YAC7D,KAAK,CAAC,GAAG,QAAQ,IAAI,gBAAgB,EAAE,CAAC,GAAG,EAAE,CAAC;QAChD,CAAC;IACH,CAAC;IAED,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC;AACjC,CAAC;AAED,SAAS,gBAAgB,CAAC,UAAyB,EAAE,KAAoB;IACrE,IAAI,OAAO,GAA2B,EAAE,CAAA;IACxC,OAAO,CAAC,GAAG,CAAC,uBAAuB,EAAE,UAAU,CAAC,CAAA;IAChD,OAAO,CAAC,GAAG,CAAC,EAAE,UAAU,EAAE,CAAC,CAAA;IAC3B,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAA;IACtB,OAAO,CAAC,GAAG,CAAC,EAAE,KAAK,EAAE,CAAC,CAAA;IACtB,QAAQ,UAAU,CAAC,IAAI,EAAE,CAAC;QACtB,KAAK,oBAAW,CAAC,IAAI;YACjB,MAAK;QACT,KAAK,oBAAW,CAAC,OAAO,CAAC,CAAC,CAAC;YACvB,OAAO,GAAG,UAAU,CAAC,OAAO,CAAA;YAC5B,MAAK;QACT,CAAC;QACD,KAAK,oBAAW,CAAC,YAAY,CAAC,CAAC,CAAC;YAC5B,IAAI,KAAK,IAAI,IAAI,EAAI,CAAC;gBAClB,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAA;gBAClC,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,KAAK,EAAE,CAAA;YAChD,CAAC;iBAAM,CAAC;gBACJ,OAAO,CAAC,GAAG,CAAC,wCAAwC,CAAC,CAAA;gBACrD,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,UAAU,CAAC,WAAW,EAAE,CAAA;YACjE,CAAC;YACD,MAAK;QACT,CAAC;QACD,KAAK,oBAAW,CAAC,OAAO,CAAC,CAAC,CAAC;YACvB,MAAM,UAAU,GAAG,UAAU,CAAC,YAAY,CAAA;YAC1C,OAAO,CAAC,UAAU,CAAC,GAAG,UAAU,CAAC,MAAM,CAAA;YACvC,MAAK;QACT,CAAC;IACL,CAAC;IAED,OAAO,OAAO,CAAA;AAClB,CAAC;AAED,SAAe,wBAAwB,CACnC,KAAqB,EACrB,KAAoB;;QAEpB,MAAM,SAAS,GAAqB,EAAE,CAAC;QACvC,OAAO,CAAC,GAAG,CAAC,+BAA+B,CAAC,CAAA;QAC5C,OAAO,CAAC,GAAG,CAAC,EAAC,KAAK,EAAC,CAAC,CAAA;QACpB,OAAO,CAAC,GAAG,CAAC,EAAC,KAAK,EAAC,CAAC,CAAA;QAEpB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC;gBACH,MAAM,SAAS,GAAG,MAAM,IAAA,kCAAe,EAAC;oBACtC,SAAS,EAAE,qBAAqB,CAC9B,IAAI,CAAC,QAAQ,EACb,IAAI,CAAC,SAAS,EACd,gBAAgB,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,CACnB;iBAClB,CAAC,CAAC;gBAEH,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,KAAK,EAAE,CAAC;gBAEzC,SAAS,CAAC,IAAI,CAAC;oBACb,OAAO,EAAE,IAAI,CAAC,QAAQ;oBACtB,SAAS;oBACT,KAAK,EAAE,QAAgC;iBACxC,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CACX,mCAAmC,IAAI,CAAC,SAAS,GAAG,EACpD,KAAK,CACN,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;CAAA;AAEH,oFAAoF;AAEpF,SAAS,uBAAuB,CAAC,UAAuB,EAAE,KAAa;;IACrE,MAAM,KAAK,GAAyB,EAAE,CAAC;IAEvC,MAAM,OAAO,GAAG,UAAU,CAAC,MAAM,CAC/B,CAAC,CAAC,EAAE,EAAE,CAAE,CAAS,CAAC,IAAI,KAAK,gBAAgB,CAC5C,CAAC;IAEF,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAI,MAAc,CAAC,IAA+B,CAAC;QAC/D,MAAM,QAAQ,GAAG,MAAC,MAAc,CAAC,QAAQ,mCAAI,uBAAuB,CAAC;QACrE,KAAK,CAAC,QAAQ,CAAC,GAAG,IAAA,4CAAsB,EAAC,MAAM,EAAE,KAAK,CAAoB,CAAC;IAC7E,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,gFAAgF;AAEhF,SAAsB,mBAAmB,CACvC,MAAgC;;;QAG9B,MAAM,EAAE,aAAa,EAAE,gBAAgB,EAAE,UAAU,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC;QAC/E,MAAM,eAAe,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC;YAChD,KAAK,EAAE,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,sBAAa,CAAC,KAAK,CAAC;YACnE,KAAK,EAAE,KAAK;SACb,CAAC,CAAC;QACH,MAAM,UAAU,GAAG,MAAM,kBAAU,CAAC,mBAAmB,CAAC;YACtD,KAAK,EAAE,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,sBAAa,CAAC,IAAI,CAAC;YAClE,UAAU,EAAE,OAAO,CAAC,KAAK,CAAC,IAAI;YAC9B,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,SAAS;YACnC,KAAK,EAAE,OAAO,CAAC,MAAM,CAAC,KAAK;SAC5B,CAAC,CAAA;QACF,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,cAAc,CAAC;QAEjD,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAA;QAC9B,OAAO,CAAC,GAAG,CAAC,EAAC,UAAU,EAAC,CAAC,CAAA;QACzB,MAAM,KAAK,GAAG,OAAO,CAAC,IAAoC,CAAC;QAC3D,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAA;QACnC,OAAO,CAAC,GAAG,CAAC,EAAC,KAAK,EAAC,CAAC,CAAA;QACpB,MAAM,WAAW,GAAG,CAAC,IAAA,cAAK,EAAC,OAAO,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,WAAW,CAAC,KAAK,IAAI,CAAA;QAEvG,IAAI,KAAK,CAAC;QACV,IAAG,WAAW,EAAC,CAAC;YACb,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAA;YACrC,KAAK,GAAG,KAAK,CAAC,YAAY,CAAC;QAC9B,CAAC;aAAI,CAAC;YACJ,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAA;YACjC,KAAK,GAAG,UAAU,CAAC;QACtB,CAAC;QAED,IAAG,CAAC,KAAK,EAAC,CAAC;YACT,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC;YAC/B,KAAK,GAAG,EAAE,CAAC;QACb,CAAC;QAED,MAAM,cAAc,GAAG,MAAM,wBAAwB,CACnD,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,sBAAa,CAAC,GAAG,CAAC,EAC1D,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CACnB,CAAA;QACD,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,iBAAiB,CAAC,cAAc,CAAC,CAAA;QAEzE,4DAA4D;QAC5D,MAAM,kBAAkB,GAAG,uBAAuB,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAEtE,MAAM,aAAa,+DACd,eAAe,GACf,UAAU,GACV,QAAQ,GACR,kBAAkB,CACtB,CAAC;QAEF,OAAO;YACH,UAAU;YACV,KAAK,kCACE,aAAa,KAChB,CAAC,kCAAyB,CAAC,EAAE,IAAA,gBAAW,EAAC;oBAC3C,WAAW,EACT,iQAAiQ;oBACnQ,WAAW,EAAE,aAAC,CAAC,MAAM,iBACnB,OAAO,EAAE,aAAC;6BACP,OAAO,EAAE;6BACT,QAAQ,CACP,8FAA8F,CAC/F,IACA,CAAC,CAAC,IAAA,cAAK,EAAC,gBAAgB,CAAC;wBAC1B,CAAC,CAAC;4BACE,MAAM,EAAE,aAAC;iCACN,MAAM,CACL,MAAA,MAAA,kBAAU,CAAC,sBAAsB,CAAC,gBAAgB,CAAC,0CAAE,KAAK,mCAAI,EAAE,CACjE;iCACA,QAAQ,EAAE;iCACV,QAAQ,CACP,4GAA4G,CAC7G;yBACJ;wBACH,CAAC,CAAC;4BACE,MAAM,EAAE,aAAC;iCACN,MAAM,EAAE;iCACR,QAAQ,EAAE;iCACV,QAAQ,CACP,8HAA8H,CAC/H;yBACJ,CAAC,EACN;oBACF,OAAO,EAAE,CAAO,MAAM,EAAE,EAAE;wBACxB,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,MAG3B,CAAC;wBACF,aAAa,CAAC,SAAS,CACrB,OAAO,CAAC,CAAC,CAAC,wBAAe,CAAC,SAAS,CAAC,CAAC,CAAC,wBAAe,CAAC,MAAM,CAC7D,CAAC;wBACF,IAAI,CAAC,IAAA,cAAK,EAAC,gBAAgB,CAAC,IAAI,CAAC,IAAA,cAAK,EAAC,MAAM,CAAC,EAAE,CAAC;4BAC/C,aAAa,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;wBAC5C,CAAC;wBACD,IAAI,CAAC,IAAA,cAAK,EAAC,gBAAgB,CAAC,IAAI,CAAC,IAAA,cAAK,EAAC,MAAM,CAAC,EAAE,CAAC;4BAC/C,aAAa,CAAC,WAAW,CAAC,MAA2B,CAAC,CAAC;wBACzD,CAAC;wBACD,OAAO,EAAE,CAAC;oBACZ,CAAC,CAAA;iBACF,CAAC,GACL;SACA,CAAA;IACL,CAAC;CAAA"}
+{"version":3,"file":"tools.js","sourceRoot":"","sources":["../../../../../../../../../packages/pieces/custom/ai-custom/src/lib/actions/agents/tools.ts"],"names":[],"mappings":";;AA4KA,kDAyGC;;AArRD,2BAAsD;AACtD,sDAAoB;AACpB,mCAAqC;AAErC,iDAA0N;AAC1N,0FAAkG;AAElG,qCAAuG;AACvG,+DAAwF;AAExF,iFAAiF;AAEjF,SAAS,qBAAqB,CAC1B,QAAqB,EACrB,SAAiB,EACjB,UAAkC,EAAE;IAEpC,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,SAAS,CAAC,CAAA;IAE9B,QAAQ,QAAQ,EAAE,CAAC;QACf,KAAK,oBAAW,CAAC,WAAW,CAAC,CAAC,CAAC;YAC3B,OAAO;gBACH,IAAI,EAAE,MAAM;gBACZ,GAAG,EAAE,SAAS;gBACd,OAAO;aACV,CAAA;QACL,CAAC;QACD,KAAK,oBAAW,CAAC,eAAe,CAAC,CAAC,CAAC;YAC/B,OAAO,IAAI,iDAA6B,CAAC,GAAG,EAAE;gBAC1C,WAAW,EAAE;oBACT,OAAO;iBACV;aACJ,CAAC,CAAA;QACN,CAAC;QACD,KAAK,oBAAW,CAAC,GAAG,CAAC,CAAC,CAAC;YACnB,OAAO;gBACH,IAAI,EAAE,KAAK;gBACX,GAAG,EAAE,SAAS;gBACd,OAAO;aACV,CAAA;QACL,CAAC;QACD;YACI,MAAM,IAAI,KAAK,CAAC,kCAAkC,QAAQ,EAAE,CAAC,CAAA;IACrE,CAAC;AACL,CAAC;AASD,SAAS,iBAAiB,CACtB,OAAyB;IAEzB,MAAM,UAAU,GAAgB,EAAE,CAAC;IACnC,MAAM,KAAK,GAAyB,EAAE,CAAC;IAEvC,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAElC,KAAK,MAAM,CAAC,QAAQ,EAAE,EAAE,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1D,MAAM,gBAAgB,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;YAC7D,KAAK,CAAC,GAAG,QAAQ,IAAI,gBAAgB,EAAE,CAAC,GAAG,EAAE,CAAC;QAChD,CAAC;IACH,CAAC;IAED,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC;AACjC,CAAC;AAED,SAAS,gBAAgB,CAAC,UAAyB,EAAE,KAAoB;IACrE,IAAI,OAAO,GAA2B,EAAE,CAAA;IACxC,OAAO,CAAC,GAAG,CAAC,uBAAuB,EAAE,UAAU,CAAC,CAAA;IAChD,OAAO,CAAC,GAAG,CAAC,EAAE,UAAU,EAAE,CAAC,CAAA;IAC3B,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAA;IACtB,OAAO,CAAC,GAAG,CAAC,EAAE,KAAK,EAAE,CAAC,CAAA;IACtB,QAAQ,UAAU,CAAC,IAAI,EAAE,CAAC;QACtB,KAAK,oBAAW,CAAC,IAAI;YACjB,MAAK;QACT,KAAK,oBAAW,CAAC,OAAO,CAAC,CAAC,CAAC;YACvB,OAAO,GAAG,UAAU,CAAC,OAAO,CAAA;YAC5B,MAAK;QACT,CAAC;QACD,KAAK,oBAAW,CAAC,YAAY,CAAC,CAAC,CAAC;YAC5B,IAAI,KAAK,IAAI,IAAI,EAAI,CAAC;gBAClB,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAA;gBAClC,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,KAAK,EAAE,CAAA;YAChD,CAAC;iBAAM,CAAC;gBACJ,OAAO,CAAC,GAAG,CAAC,wCAAwC,CAAC,CAAA;gBACrD,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,UAAU,CAAC,WAAW,EAAE,CAAA;YACjE,CAAC;YACD,MAAK;QACT,CAAC;QACD,KAAK,oBAAW,CAAC,OAAO,CAAC,CAAC,CAAC;YACvB,MAAM,UAAU,GAAG,UAAU,CAAC,YAAY,CAAA;YAC1C,OAAO,CAAC,UAAU,CAAC,GAAG,UAAU,CAAC,MAAM,CAAA;YACvC,MAAK;QACT,CAAC;IACL,CAAC;IAED,OAAO,OAAO,CAAA;AAClB,CAAC;AAED,SAAe,wBAAwB,CACnC,KAAqB,EACrB,KAAoB;;QAEpB,MAAM,SAAS,GAAqB,EAAE,CAAC;QACvC,OAAO,CAAC,GAAG,CAAC,+BAA+B,CAAC,CAAA;QAC5C,OAAO,CAAC,GAAG,CAAC,EAAC,KAAK,EAAC,CAAC,CAAA;QACpB,OAAO,CAAC,GAAG,CAAC,EAAC,KAAK,EAAC,CAAC,CAAA;QAEpB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC;gBACH,MAAM,SAAS,GAAG,MAAM,IAAA,kCAAe,EAAC;oBACtC,SAAS,EAAE,qBAAqB,CAC9B,IAAI,CAAC,QAAQ,EACb,IAAI,CAAC,SAAS,EACd,gBAAgB,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,CACnB;iBAClB,CAAC,CAAC;gBAEH,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,KAAK,EAAE,CAAC;gBAEzC,SAAS,CAAC,IAAI,CAAC;oBACb,OAAO,EAAE,IAAI,CAAC,QAAQ;oBACtB,SAAS;oBACT,KAAK,EAAE,QAAgC;iBACxC,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CACX,mCAAmC,IAAI,CAAC,SAAS,GAAG,EACpD,KAAK,CACN,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;CAAA;AAEH,iFAAiF;AAEjF;;;;GAIG;AACH,SAAS,uBAAuB,CAC9B,UAAuB,EACvB,KAAa;;IAEb,MAAM,KAAK,GAAyB,EAAE,CAAC;IAEvC,MAAM,OAAO,GAAG,UAAU,CAAC,MAAM,CAC/B,CAAC,CAAC,EAAE,EAAE,CAAE,CAAS,CAAC,IAAI,KAAK,gBAAgB,CAC5C,CAAC;IAEF,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAI,MAAc,CAAC,IAA+B,CAAC;QAC/D,MAAM,QAAQ,GACZ,MAAC,MAAc,CAAC,QAAQ,mCAAI,uBAAuB,CAAC;QAEtD,gEAAgE;QAChE,KAAK,CAAC,QAAQ,CAAC,GAAG,IAAA,4CAAsB,EAAC,MAAM,EAAE,KAAK,CAAoB,CAAC;IAC7E,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,iFAAiF;AAEjF,SAAsB,mBAAmB,CACvC,MAAgC;;;QAG9B,MAAM,EAAE,aAAa,EAAE,gBAAgB,EAAE,UAAU,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC;QAC/E,MAAM,eAAe,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC;YAChD,KAAK,EAAE,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,sBAAa,CAAC,KAAK,CAAC;YACnE,KAAK,EAAE,KAAK;SACb,CAAC,CAAC;QACH,MAAM,UAAU,GAAG,MAAM,kBAAU,CAAC,mBAAmB,CAAC;YACtD,KAAK,EAAE,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,sBAAa,CAAC,IAAI,CAAC;YAClE,UAAU,EAAE,OAAO,CAAC,KAAK,CAAC,IAAI;YAC9B,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,SAAS;YACnC,KAAK,EAAE,OAAO,CAAC,MAAM,CAAC,KAAK;SAC5B,CAAC,CAAA;QACF,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,cAAc,CAAC;QAEjD,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAA;QAC9B,OAAO,CAAC,GAAG,CAAC,EAAC,UAAU,EAAC,CAAC,CAAA;QACzB,MAAM,KAAK,GAAG,OAAO,CAAC,IAAoC,CAAC;QAC3D,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAA;QACnC,OAAO,CAAC,GAAG,CAAC,EAAC,KAAK,EAAC,CAAC,CAAA;QACpB,MAAM,WAAW,GAAG,CAAC,IAAA,cAAK,EAAC,OAAO,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,WAAW,CAAC,KAAK,IAAI,CAAA;QAEvG,IAAI,KAAK,CAAC;QACV,IAAG,WAAW,EAAC,CAAC;YACb,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAA;YACrC,KAAK,GAAG,KAAK,CAAC,YAAY,CAAC;QAC9B,CAAC;aAAI,CAAC;YACJ,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAA;YACjC,KAAK,GAAG,UAAU,CAAC;QACtB,CAAC;QAED,IAAG,CAAC,KAAK,EAAC,CAAC;YACT,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC;YAC/B,KAAK,GAAG,EAAE,CAAC;QACb,CAAC;QAED,MAAM,cAAc,GAAG,MAAM,wBAAwB,CACnD,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,sBAAa,CAAC,GAAG,CAAC,EAC1D,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CACnB,CAAA;QACD,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,iBAAiB,CAAC,cAAc,CAAC,CAAA;QAEzE,MAAM,kBAAkB,GAAG,uBAAuB,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAEtE,MAAM,aAAa,+DACd,eAAe,GACf,UAAU,GACV,QAAQ,GACR,kBAAkB,CACtB,CAAC;QAEF,OAAO;YACH,UAAU;YACV,KAAK,kCACE,aAAa,KAChB,CAAC,kCAAyB,CAAC,EAAE,IAAA,gBAAW,EAAC;oBAC3C,WAAW,EACT,iQAAiQ;oBACnQ,WAAW,EAAE,aAAC,CAAC,MAAM,iBACnB,OAAO,EAAE,aAAC;6BACP,OAAO,EAAE;6BACT,QAAQ,CACP,8FAA8F,CAC/F,IACA,CAAC,CAAC,IAAA,cAAK,EAAC,gBAAgB,CAAC;wBAC1B,CAAC,CAAC;4BACE,MAAM,EAAE,aAAC;iCACN,MAAM,CACL,MAAA,MAAA,kBAAU,CAAC,sBAAsB,CAAC,gBAAgB,CAAC,0CAAE,KAAK,mCAAI,EAAE,CACjE;iCACA,QAAQ,EAAE;iCACV,QAAQ,CACP,4GAA4G,CAC7G;yBACJ;wBACH,CAAC,CAAC;4BACE,MAAM,EAAE,aAAC;iCACN,MAAM,EAAE;iCACR,QAAQ,EAAE;iCACV,QAAQ,CACP,8HAA8H,CAC/H;yBACJ,CAAC,EACN;oBACF,OAAO,EAAE,CAAO,MAAM,EAAE,EAAE;wBACxB,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,MAG3B,CAAC;wBACF,aAAa,CAAC,SAAS,CACrB,OAAO,CAAC,CAAC,CAAC,wBAAe,CAAC,SAAS,CAAC,CAAC,CAAC,wBAAe,CAAC,MAAM,CAC7D,CAAC;wBACF,IAAI,CAAC,IAAA,cAAK,EAAC,gBAAgB,CAAC,IAAI,CAAC,IAAA,cAAK,EAAC,MAAM,CAAC,EAAE,CAAC;4BAC/C,aAAa,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;wBAC5C,CAAC;wBACD,IAAI,CAAC,IAAA,cAAK,EAAC,gBAAgB,CAAC,IAAI,CAAC,IAAA,cAAK,EAAC,MAAM,CAAC,EAAE,CAAC;4BAC/C,aAAa,CAAC,WAAW,CAAC,MAA2B,CAAC,CAAC;wBACzD,CAAC;wBACD,OAAO,EAAE,CAAC;oBACZ,CAAC,CAAA;iBACF,CAAC,GACL;SACA,CAAA;IACL,CAAC;CAAA"}