@dainprotocol/service-sdk 2.0.77 → 2.0.79

package/dist/plugins/citations-plugin.js

@@ -3,10 +3,8 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.CitationsPlugin = void 0;
 const base_1 = require("./base");
 /**
- * CitationsPlugin for extracting and managing citations from tool results
- *
- * This plugin automatically extracts citations from tool responses that contain
- * citation data, making them easily accessible for clients to display or process.
+ * CitationsPlugin extracts and manages citations from tool results,
+ * making them easily accessible for clients to display or process.
  */
 class CitationsPlugin extends base_1.BaseDainPlugin {
     id = 'citations-plugin';
@@ -15,176 +13,83 @@ class CitationsPlugin extends base_1.BaseDainPlugin {
     description = 'Extracts and manages citations from tool results';
     type = 'both';
     config;
-    /**
-     * Create a new CitationsPlugin instance
-     * @param config Plugin configuration (optional)
-     */
     constructor(config = {}) {
         super();
         this.config = config;
     }
-    /**
-     * Get plugin input data from a tool context with proper typing
-     */
     getInputData(context) {
         return super.getInputData(context);
     }
-    /**
-     * Get plugin output data from a response with proper typing
-     */
     getResponseData(response) {
         return super.getResponseData(response);
     }
-    /**
-     * Create a standardized response for this plugin with proper typing
-     */
     createResponse(data) {
         return super.createResponse(data);
     }
-    /**
-     * Extract citations from various data structures
-     * @param data Any data structure that might contain citations
-     * @returns Array of citations found
-     */
-    extractCitations(data) {
-        const citations = [];
-        if (!data)
-            return citations;
-        // Helper function to recursively search for citations
-        const searchForCitations = (obj, visited = new Set()) => {
-            // Prevent infinite recursion
-            if (!obj || typeof obj !== 'object' || visited.has(obj)) {
-                return;
-            }
-            visited.add(obj);
-            // Check if this object is a citation
-            if (this.isCitation(obj)) {
-                citations.push(obj);
-                return;
-            }
-            // Check if this object has a citation property
-            if (obj.citation && this.isCitation(obj.citation)) {
-                citations.push(obj.citation);
-            }
-            // Recursively search arrays and objects
-            if (Array.isArray(obj)) {
-                obj.forEach(item => searchForCitations(item, visited));
-            }
-            else {
-                Object.values(obj).forEach(value => searchForCitations(value, visited));
-            }
-        };
-        searchForCitations(data);
-        // Remove duplicates based on citation ID
-        const uniqueCitations = citations.filter((citation, index, self) => index === self.findIndex(c => c.id === citation.id));
-        return uniqueCitations;
+    isCitation(obj) {
+        return obj && typeof obj === 'object' && typeof obj.id === 'string' && obj.ui !== undefined;
     }
-    /**
-     * Extract content with citations from data structures
-     * @param data Any data structure that might contain content with citations
-     * @returns Array of content objects with their citations
-     */
-    extractContentWithCitations(data) {
-        const contentWithCitations = [];
+    isContentWithCitation(obj) {
+        return (obj &&
+            typeof obj === 'object' &&
+            typeof obj.content === 'string' &&
+            typeof obj.id === 'string' &&
+            this.isCitation(obj.citation));
+    }
+    recursiveSearch(data, predicate, checkNested) {
+        const results = [];
         if (!data)
-            return contentWithCitations;
-        // Helper function to recursively search for content with citations
-        const searchForContent = (obj, visited = new Set()) => {
-            // Prevent infinite recursion
-            if (!obj || typeof obj !== 'object' || visited.has(obj)) {
+            return results;
+        const search = (obj, visited = new Set()) => {
+            if (!obj || typeof obj !== 'object' || visited.has(obj))
                 return;
-            }
             visited.add(obj);
-            // Check if this object is content with citation
-            if (this.isContentWithCitation(obj)) {
-                contentWithCitations.push(obj);
+            if (predicate(obj)) {
+                results.push(obj);
                 return;
             }
-            // Recursively search arrays and objects
-            if (Array.isArray(obj)) {
-                obj.forEach(item => searchForContent(item, visited));
-            }
-            else {
-                Object.values(obj).forEach(value => searchForContent(value, visited));
+            if (checkNested) {
+                const nested = checkNested(obj);
+                if (nested)
+                    results.push(nested);
             }
+            const values = Array.isArray(obj) ? obj : Object.values(obj);
+            values.forEach(value => search(value, visited));
         };
-        searchForContent(data);
-        // Remove duplicates based on content ID
-        const uniqueContent = contentWithCitations.filter((content, index, self) => index === self.findIndex(c => c.id === content.id));
-        return uniqueContent;
+        search(data);
+        const seen = new Set();
+        return results.filter(item => {
+            const id = item.id;
+            if (seen.has(id))
+                return false;
+            seen.add(id);
+            return true;
+        });
     }
-    /**
-     * Check if an object is a valid citation
-     * @param obj Object to check
-     * @returns True if the object is a valid citation
-     */
-    isCitation(obj) {
-        return obj &&
-            typeof obj === 'object' &&
-            typeof obj.id === 'string' &&
-            obj.ui !== undefined;
+    extractCitations(data) {
+        return this.recursiveSearch(data, (obj) => this.isCitation(obj), (obj) => this.isCitation(obj.citation) ? obj.citation : undefined);
     }
-    /**
-     * Check if an object is content with citation
-     * @param obj Object to check
-     * @returns True if the object is content with citation
-     */
-    isContentWithCitation(obj) {
-        return obj &&
-            typeof obj === 'object' &&
-            typeof obj.content === 'string' &&
-            typeof obj.id === 'string' &&
-            this.isCitation(obj.citation);
+    extractContentWithCitations(data) {
+        return this.recursiveSearch(data, (obj) => this.isContentWithCitation(obj));
     }
-    /**
-     * Get citations from a tool response
-     * @param response Tool response
-     * @returns Array of citations found in the response
-     */
     getCitationsFromResponse(response) {
-        // First try to get from plugin data
         const pluginData = this.getResponseData(response);
-        if (pluginData?.citations) {
+        if (pluginData?.citations)
             return pluginData.citations;
-        }
-        // If not found in plugin data, extract from the entire response
         return this.extractCitations(response);
     }
-    /**
-     * Get content with citations from a tool response
-     * @param response Tool response
-     * @returns Array of content with citations found in the response
-     */
     getContentWithCitationsFromResponse(response) {
-        // First try to get from plugin data
         const pluginData = this.getResponseData(response);
-        if (pluginData?.contentWithCitations) {
+        if (pluginData?.contentWithCitations)
             return pluginData.contentWithCitations;
-        }
-        // If not found in plugin data, extract from the entire response
         return this.extractContentWithCitations(response);
     }
-    /**
-     * Get a specific citation by ID from a response
-     * @param response Tool response
-     * @param id Citation ID to find
-     * @returns Citation or undefined if not found
-     */
     getCitationById(response, id) {
-        const citations = this.getCitationsFromResponse(response);
-        return citations.find(citation => citation.id === id);
+        return this.getCitationsFromResponse(response).find(c => c.id === id);
     }
-    /**
-     * Process client input before sending to service
-     * No input processing needed for citations plugin
-     */
-    async processInputClient(input) {
+    async processInputClient(_input) {
         return {};
     }
-    /**
-     * Process service output before sending to client
-     * Extract citations from the response data
-     */
     async processOutputService(output) {
         const citations = this.extractCitations(output);
         const contentWithCitations = this.extractContentWithCitations(output);
@@ -194,32 +99,18 @@ class CitationsPlugin extends base_1.BaseDainPlugin {
             contentWithCitations: contentWithCitations.length > 0 ? contentWithCitations : undefined
         };
     }
-    /**
-     * Process client output after receiving service response
-     * Ensure citations are properly formatted
-     */
     async processOutputClient(output) {
-        // Validate and clean up citations in the plugin data
         if (output?.plugins?.[this.id]?.citations) {
-            output.plugins[this.id].citations = output.plugins[this.id].citations.filter((citation) => this.isCitation(citation));
+            output.plugins[this.id].citations = output.plugins[this.id].citations.filter((c) => this.isCitation(c));
         }
         if (output?.plugins?.[this.id]?.contentWithCitations) {
-            output.plugins[this.id].contentWithCitations = output.plugins[this.id].contentWithCitations.filter((content) => this.isContentWithCitation(content));
+            output.plugins[this.id].contentWithCitations = output.plugins[this.id].contentWithCitations.filter((c) => this.isContentWithCitation(c));
         }
         return output;
     }
-    /**
-     * Process service input before handling the request
-     */
     async processInputService(input) {
-        // No modifications needed for service input processing
         return input;
     }
-    /**
-     * Helper method to format citations for display
-     * @param citations Array of citations to format
-     * @returns Formatted citation strings
-     */
     formatCitations(citations) {
         return citations.map(citation => {
             const parts = [];
@@ -236,20 +127,14 @@ class CitationsPlugin extends base_1.BaseDainPlugin {
             return parts.join(' ');
         });
     }
-    /**
-     * Helper method to group citations by type
-     * @param citations Array of citations to group
-     * @returns Object with citations grouped by type
-     */
     groupCitationsByType(citations) {
         const grouped = {};
-        citations.forEach(citation => {
+        for (const citation of citations) {
             const type = citation.type || 'other';
-            if (!grouped[type]) {
+            if (!grouped[type])
                 grouped[type] = [];
-            }
             grouped[type].push(citation);
-        });
+        }
         return grouped;
     }
 }

package/dist/plugins/citations-plugin.js.map

@@ -1 +1 @@
-{"version":3,"file":"citations-plugin.js","sourceRoot":"","sources":["../../src/plugins/citations-plugin.ts"],"names":[],"mappings":";;;AAAA,iCAAwC;AA+BxC;;;;;GAKG;AACH,MAAa,eAAgB,SAAQ,qBAAkF;IACrH,EAAE,GAAG,kBAAkB,CAAC;IACxB,IAAI,GAAG,kBAAkB,CAAC;IAC1B,OAAO,GAAG,OAAO,CAAC;IAClB,WAAW,GAAG,kDAAkD,CAAC;IACjE,IAAI,GAAG,MAAe,CAAC;IACvB,MAAM,CAAwB;IAE9B;;;OAGG;IACH,YAAY,SAAgC,EAAE;QAC5C,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED;;OAEG;IACM,YAAY,CAAC,OAAY;QAChC,OAAO,KAAK,CAAC,YAAY,CAAC,OAAO,CAAqC,CAAC;IACzE,CAAC;IAED;;OAEG;IACM,eAAe,CAAC,QAAa;QACpC,OAAO,KAAK,CAAC,eAAe,CAAC,QAAQ,CAAsC,CAAC;IAC9E,CAAC;IAED;;OAEG;IACM,cAAc,CAAC,IAAoC;QAC1D,OAAO,KAAK,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC;IAED;;;;OAIG;IACH,gBAAgB,CAAC,IAAS;QACxB,MAAM,SAAS,GAAe,EAAE,CAAC;QAEjC,IAAI,CAAC,IAAI;YAAE,OAAO,SAAS,CAAC;QAE5B,sDAAsD;QACtD,MAAM,kBAAkB,GAAG,CAAC,GAAQ,EAAE,OAAO,GAAG,IAAI,GAAG,EAAE,EAAQ,EAAE;YACjE,6BAA6B;YAC7B,IAAI,CAAC,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gBACxD,OAAO;YACT,CAAC;YACD,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAEjB,qCAAqC;YACrC,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;gBACzB,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACpB,OAAO;YACT,CAAC;YAED,+CAA+C;YAC/C,IAAI,GAAG,CAAC,QAAQ,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAClD,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC/B,CAAC;YAED,wCAAwC;YACxC,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;gBACvB,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,kBAAkB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC;YACzD,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,kBAAkB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC;YAC1E,CAAC;QACH,CAAC,CAAC;QAEF,kBAAkB,CAAC,IAAI,CAAC,CAAC;QAEzB,yCAAyC;QACzC,MAAM,eAAe,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,CACjE,KAAK,KAAK,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,QAAQ,CAAC,EAAE,CAAC,CACpD,CAAC;QAEF,OAAO,eAAe,CAAC;IACzB,CAAC;IAED;;;;OAIG;IACH,2BAA2B,CAAC,IAAS;QACnC,MAAM,oBAAoB,GAA0B,EAAE,CAAC;QAEvD,IAAI,CAAC,IAAI;YAAE,OAAO,oBAAoB,CAAC;QAEvC,mEAAmE;QACnE,MAAM,gBAAgB,GAAG,CAAC,GAAQ,EAAE,OAAO,GAAG,IAAI,GAAG,EAAE,EAAQ,EAAE;YAC/D,6BAA6B;YAC7B,IAAI,CAAC,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gBACxD,OAAO;YACT,CAAC;YACD,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAEjB,gDAAgD;YAChD,IAAI,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,EAAE,CAAC;gBACpC,oBAAoB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAC/B,OAAO;YACT,CAAC;YAED,wCAAwC;YACxC,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;gBACvB,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,gBAAgB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC;YACvD,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,gBAAgB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC;YACxE,CAAC;QACH,CAAC,CAAC;QAEF,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAEvB,wCAAwC;QACxC,MAAM,aAAa,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,CACzE,KAAK,KAAK,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,OAAO,CAAC,EAAE,CAAC,CACnD,CAAC;QAEF,OAAO,aAAa,CAAC;IACvB,CAAC;IAED;;;;OAIG;IACK,UAAU,CAAC,GAAQ;QACzB,OAAO,GAAG;YACH,OAAO,GAAG,KAAK,QAAQ;YACvB,OAAO,GAAG,CAAC,EAAE,KAAK,QAAQ;YAC1B,GAAG,CAAC,EAAE,KAAK,SAAS,CAAC;IAC9B,CAAC;IAED;;;;OAIG;IACK,qBAAqB,CAAC,GAAQ;QACpC,OAAO,GAAG;YACH,OAAO,GAAG,KAAK,QAAQ;YACvB,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ;YAC/B,OAAO,GAAG,CAAC,EAAE,KAAK,QAAQ;YAC1B,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACvC,CAAC;IAED;;;;OAIG;IACH,wBAAwB,CAAC,QAAa;QACpC,oCAAoC;QACpC,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;QAClD,IAAI,UAAU,EAAE,SAAS,EAAE,CAAC;YAC1B,OAAO,UAAU,CAAC,SAAS,CAAC;QAC9B,CAAC;QAED,gEAAgE;QAChE,OAAO,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IACzC,CAAC;IAED;;;;OAIG;IACH,mCAAmC,CAAC,QAAa;QAC/C,oCAAoC;QACpC,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;QAClD,IAAI,UAAU,EAAE,oBAAoB,EAAE,CAAC;YACrC,OAAO,UAAU,CAAC,oBAAoB,CAAC;QACzC,CAAC;QAED,gEAAgE;QAChE,OAAO,IAAI,CAAC,2BAA2B,CAAC,QAAQ,CAAC,CAAC;IACpD,CAAC;IAED;;;;;OAKG;IACH,eAAe,CAAC,QAAa,EAAE,EAAU;QACvC,MAAM,SAAS,GAAG,IAAI,CAAC,wBAAwB,CAAC,QAAQ,CAAC,CAAC;QAC1D,OAAO,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IACxD,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,kBAAkB,CAAC,KAAU;QACjC,OAAO,EAAE,CAAC;IACZ,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,oBAAoB,CAAC,MAAW;QACpC,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;QAChD,MAAM,oBAAoB,GAAG,IAAI,CAAC,2BAA2B,CAAC,MAAM,CAAC,CAAC;QAEtE,OAAO;YACL,SAAS,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;YACvD,aAAa,EAAE,SAAS,CAAC,MAAM;YAC/B,oBAAoB,EAAE,oBAAoB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,SAAS;SACzF,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,mBAAmB,CAAC,MAAW;QACnC,qDAAqD;QACrD,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC;YAC1C,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,SAAS,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,MAAM,CAC1E,CAAC,QAAa,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAC7C,CAAC;QACJ,CAAC;QAED,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,oBAAoB,EAAE,CAAC;YACrD,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,oBAAoB,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAChG,CAAC,OAAY,EAAE,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,CACtD,CAAC;QACJ,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,mBAAmB,CAAC,KAAU;QAClC,uDAAuD;QACvD,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;;;OAIG;IACH,eAAe,CAAC,SAAqB;QACnC,OAAO,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE;YAC9B,MAAM,KAAK,GAAa,EAAE,CAAC;YAE3B,IAAI,QAAQ,CAAC,KAAK;gBAAE,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;YAC/C,IAAI,QAAQ,CAAC,MAAM;gBAAE,KAAK,CAAC,IAAI,CAAC,MAAM,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;YACzD,IAAI,QAAQ,CAAC,MAAM;gBAAE,KAAK,CAAC,IAAI,CAAC,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;YACxD,IAAI,QAAQ,CAAC,IAAI;gBAAE,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YAC7C,IAAI,QAAQ,CAAC,GAAG;gBAAE,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;YAE3C,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACzB,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;OAIG;IACH,oBAAoB,CAAC,SAAqB;QACxC,MAAM,OAAO,GAA+B,EAAE,CAAC;QAE/C,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;YAC3B,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,IAAI,OAAO,CAAC;YACtC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;gBACnB,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;YACrB,CAAC;YACD,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,OAAO,OAAO,CAAC;IACjB,CAAC;CACF;AA5RD,0CA4RC"}
+{"version":3,"file":"citations-plugin.js","sourceRoot":"","sources":["../../src/plugins/citations-plugin.ts"],"names":[],"mappings":";;;AAAA,iCAAwC;AAsBxC;;;GAGG;AACH,MAAa,eAAgB,SAAQ,qBAAkF;IACrH,EAAE,GAAG,kBAAkB,CAAC;IACxB,IAAI,GAAG,kBAAkB,CAAC;IAC1B,OAAO,GAAG,OAAO,CAAC;IAClB,WAAW,GAAG,kDAAkD,CAAC;IACjE,IAAI,GAAG,MAAe,CAAC;IACvB,MAAM,CAAwB;IAE9B,YAAY,SAAgC,EAAE;QAC5C,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAEQ,YAAY,CAAC,OAAY;QAChC,OAAO,KAAK,CAAC,YAAY,CAAC,OAAO,CAAqC,CAAC;IACzE,CAAC;IAEQ,eAAe,CAAC,QAAa;QACpC,OAAO,KAAK,CAAC,eAAe,CAAC,QAAQ,CAAsC,CAAC;IAC9E,CAAC;IAEQ,cAAc,CAAC,IAAoC;QAC1D,OAAO,KAAK,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC;IAEO,UAAU,CAAC,GAAQ;QACzB,OAAO,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,OAAO,GAAG,CAAC,EAAE,KAAK,QAAQ,IAAI,GAAG,CAAC,EAAE,KAAK,SAAS,CAAC;IAC9F,CAAC;IAEO,qBAAqB,CAAC,GAAQ;QACpC,OAAO,CACL,GAAG;YACH,OAAO,GAAG,KAAK,QAAQ;YACvB,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ;YAC/B,OAAO,GAAG,CAAC,EAAE,KAAK,QAAQ;YAC1B,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAC9B,CAAC;IACJ,CAAC;IAEO,eAAe,CACrB,IAAS,EACT,SAAiC,EACjC,WAAyC;QAEzC,MAAM,OAAO,GAAQ,EAAE,CAAC;QACxB,IAAI,CAAC,IAAI;YAAE,OAAO,OAAO,CAAC;QAE1B,MAAM,MAAM,GAAG,CAAC,GAAQ,EAAE,OAAO,GAAG,IAAI,GAAG,EAAE,EAAQ,EAAE;YACrD,IAAI,CAAC,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC;gBAAE,OAAO;YAChE,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAEjB,IAAI,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC;gBACnB,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAClB,OAAO;YACT,CAAC;YAED,IAAI,WAAW,EAAE,CAAC;gBAChB,MAAM,MAAM,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC;gBAChC,IAAI,MAAM;oBAAE,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACnC,CAAC;YAED,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC7D,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC;QAClD,CAAC,CAAC;QAEF,MAAM,CAAC,IAAI,CAAC,CAAC;QAEb,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;QAC/B,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE;YAC3B,MAAM,EAAE,GAAI,IAAuB,CAAC,EAAE,CAAC;YACvC,IAAI,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;gBAAE,OAAO,KAAK,CAAC;YAC/B,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACb,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;IACL,CAAC;IAED,gBAAgB,CAAC,IAAS;QACxB,OAAO,IAAI,CAAC,eAAe,CACzB,IAAI,EACJ,CAAC,GAAG,EAAmB,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAC9C,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAClE,CAAC;IACJ,CAAC;IAED,2BAA2B,CAAC,IAAS;QACnC,OAAO,IAAI,CAAC,eAAe,CACzB,IAAI,EACJ,CAAC,GAAG,EAA8B,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,CACrE,CAAC;IACJ,CAAC;IAED,wBAAwB,CAAC,QAAa;QACpC,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;QAClD,IAAI,UAAU,EAAE,SAAS;YAAE,OAAO,UAAU,CAAC,SAAS,CAAC;QACvD,OAAO,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IACzC,CAAC;IAED,mCAAmC,CAAC,QAAa;QAC/C,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;QAClD,IAAI,UAAU,EAAE,oBAAoB;YAAE,OAAO,UAAU,CAAC,oBAAoB,CAAC;QAC7E,OAAO,IAAI,CAAC,2BAA2B,CAAC,QAAQ,CAAC,CAAC;IACpD,CAAC;IAED,eAAe,CAAC,QAAa,EAAE,EAAU;QACvC,OAAO,IAAI,CAAC,wBAAwB,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IACxE,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,MAAW;QAClC,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,MAAW;QACpC,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;QAChD,MAAM,oBAAoB,GAAG,IAAI,CAAC,2BAA2B,CAAC,MAAM,CAAC,CAAC;QAEtE,OAAO;YACL,SAAS,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;YACvD,aAAa,EAAE,SAAS,CAAC,MAAM;YAC/B,oBAAoB,EAAE,oBAAoB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,SAAS;SACzF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,MAAW;QACnC,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC;YAC1C,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,SAAS,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,MAAM,CAC1E,CAAC,CAAM,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAC/B,CAAC;QACJ,CAAC;QAED,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,oBAAoB,EAAE,CAAC;YACrD,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,oBAAoB,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAChG,CAAC,CAAM,EAAE,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAC1C,CAAC;QACJ,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,KAAU;QAClC,OAAO,KAAK,CAAC;IACf,CAAC;IAED,eAAe,CAAC,SAAqB;QACnC,OAAO,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE;YAC9B,MAAM,KAAK,GAAa,EAAE,CAAC;YAC3B,IAAI,QAAQ,CAAC,KAAK;gBAAE,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;YAC/C,IAAI,QAAQ,CAAC,MAAM;gBAAE,KAAK,CAAC,IAAI,CAAC,MAAM,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;YACzD,IAAI,QAAQ,CAAC,MAAM;gBAAE,KAAK,CAAC,IAAI,CAAC,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;YACxD,IAAI,QAAQ,CAAC,IAAI;gBAAE,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YAC7C,IAAI,QAAQ,CAAC,GAAG;gBAAE,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;YAC3C,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACzB,CAAC,CAAC,CAAC;IACL,CAAC;IAED,oBAAoB,CAAC,SAAqB;QACxC,MAAM,OAAO,GAA+B,EAAE,CAAC;QAE/C,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YACjC,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,IAAI,OAAO,CAAC;YACtC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;gBAAE,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;YACvC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/B,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;CACF;AArKD,0CAqKC"}

package/dist/plugins/crypto-plugin.d.ts

@@ -1,5 +1,11 @@
 import { BaseDainPlugin } from './base';
-export type BlockchainNetwork = 'eth' | 'sol' | 'arb' | 'base' | 'hyperEvm' | 'bsc' | 'btc';
+/**
+ * Supported blockchain networks:
+ * - Layer 1: sol (Solana), eth (Ethereum), btc (Bitcoin)
+ * - EVM L2/Sidechains: arb (Arbitrum), base (Base), bsc (BSC), polygon (Polygon POS)
+ * - App-specific: hyperEvm (HyperLiquid EVM)
+ */
+export type BlockchainNetwork = 'eth' | 'sol' | 'arb' | 'base' | 'hyperEvm' | 'bsc' | 'btc' | 'polygon';
 export interface WalletInfo {
     chain: BlockchainNetwork;
     address: string;
@@ -9,14 +15,14 @@ export interface UnsignedTransaction {
     encodedTx: string;
     chain: BlockchainNetwork;
     signer: string;
-    metadata?: Record<string, any>;
-    ui?: any;
+    metadata?: Record<string, unknown>;
+    ui?: unknown;
 }
 export interface TransactionSignatureRequest {
     txs: UnsignedTransaction[];
     executeSimultaneously?: boolean;
     toolsShouldWaitForConfirmation?: boolean;
-    ui?: any;
+    ui?: unknown;
 }
 export interface CryptoPluginConfig {
     supportedChains?: BlockchainNetwork[];
@@ -29,11 +35,8 @@ export interface CryptoPluginOutput {
     signatureRequests?: TransactionSignatureRequest[];
 }
 /**
- * CryptoPlugin for handling cryptocurrency transaction workflows
- *
- * This plugin allows services to request transaction signatures from clients
- * by sending back unsigned transactions. Clients can provide their wallet
- * addresses which tools can access and use to create appropriate transactions.
+ * CryptoPlugin handles cryptocurrency wallet information and transaction signing requests.
+ * Services can request transaction signatures by sending unsigned transactions to clients.
  */
 export declare class CryptoPlugin extends BaseDainPlugin<CryptoPluginConfig, CryptoPluginInput, CryptoPluginOutput> {
     id: string;
@@ -43,142 +46,34 @@ export declare class CryptoPlugin extends BaseDainPlugin<CryptoPluginConfig, Cry
     type: "both";
     config: CryptoPluginConfig;
     private wallets;
-    /**
-     * Create a new CryptoPlugin instance
-     * @param config Plugin configuration or array of wallet information
-     *
-     * @example
-     * // Initialize with wallet information directly
-     * const cryptoPlugin = new CryptoPlugin([
-     *   { chain: 'eth', address: '0x1234...' },
-     *   { chain: 'sol', address: 'Sol456...' }
-     * ]);
-     *
-     * @example
-     * // Initialize with full configuration
-     * const cryptoPlugin = new CryptoPlugin({
-     *   supportedChains: ['eth', 'sol'],
-     *   wallets: [
-     *     { chain: 'eth', address: '0x1234...' },
-     *     { chain: 'sol', address: 'Sol456...' }
-     *   ]
-     * });
-     */
     constructor(configOrWallets?: CryptoPluginConfig | WalletInfo[]);
-    /**
-     * Set or update wallet information
-     * @param wallets Array of wallet information
-     */
     setWallets(wallets: WalletInfo[]): void;
-    /**
-     * Add a single wallet
-     * @param wallet Wallet information to add
-     */
     addWallet(wallet: WalletInfo): void;
-    /**
-     * Get all configured wallets
-     * @returns Array of wallet information
-     */
     getWallets(): WalletInfo[];
-    /**
-     * Get plugin input data from a tool context with proper typing
-     */
     getInputData(context: any): CryptoPluginInput | undefined;
-    /**
-     * Get plugin output data from a response with proper typing
-     */
     getResponseData(response: any): CryptoPluginOutput | undefined;
-    /**
-     * Create a standardized response for this plugin with proper typing
-     */
     createResponse(data: Partial<CryptoPluginOutput>): CryptoPluginOutput;
     /**
-     * Get user wallets from various sources
-     * @param input Tool context or context parameter from context/datasource handlers
-     * @returns Array of wallet information or empty array if none found
-     *
-     * @example
-     * // In a tool handler
-     * const wallets = cryptoPlugin.getUserWallets(context);
-     *
-     * @example
-     * // In a context handler
-     * const wallets = cryptoPlugin.getUserWallets(context);
-     *
-     * @example
-     * // In a datasource handler
-     * const wallets = cryptoPlugin.getUserWallets(context);
+     * Get user wallets from tool context, context/datasource context, or direct input.
+     * Checks multiple paths: extraData.plugins, plugins, and getInputData result.
      */
     getUserWallets(input: any): WalletInfo[];
-    /**
-     * Get wallet address for a specific chain
-     * @param input Tool context or context parameter from context/datasource handlers
-     * @param chain Blockchain network
-     * @returns Wallet address or undefined if not found
-     *
-     * @example
-     * // In a tool handler
-     * const ethAddress = cryptoPlugin.getUserWalletForChain(context, "eth");
-     *
-     * @example
-     * // In a context handler
-     * const solAddress = cryptoPlugin.getUserWalletForChain(context, "sol");
-     *
-     * @example
-     * // In a datasource handler
-     * const arbAddress = cryptoPlugin.getUserWalletForChain(context, "arb");
-     */
     getUserWalletForChain(input: any, chain: BlockchainNetwork): string | undefined;
-    /**
-     * Helper to create a transaction signature request
-     * @param transactions Array of unsigned transactions
-     * @param options Optional configuration for the transaction request
-     * @param options.executeSimultaneously Whether to execute all transactions simultaneously
-     * @param options.toolsShouldWaitForConfirmation Whether to wait for transaction confirmation before completing
-     * @param options.ui Optional UI component for the entire transaction request
-     * @returns Properly formatted transaction signature request
-     */
+    private ensureTransactionIds;
+    private normalizeSignatureRequest;
     createTransactionRequest(transactions: UnsignedTransaction[], options?: {
         executeSimultaneously?: boolean;
         toolsShouldWaitForConfirmation?: boolean;
-        ui?: any;
+        ui?: unknown;
     }): TransactionSignatureRequest;
-    /**
-     * Create a single transaction with a unique ID
-     * @param transaction Transaction data, ID will be generated if not provided
-     * @returns Transaction with a guaranteed unique ID
-     */
     createTransaction(transaction: Omit<UnsignedTransaction, 'id'> & {
         id?: string;
     }): UnsignedTransaction;
-    /**
-     * Get requested transaction signatures from a response
-     * @param response Tool response
-     * @returns Array of transaction signature requests or empty array if none
-     */
     getRequestedCryptoSignatures(response: any): TransactionSignatureRequest[];
-    /**
-     * Get a specific transaction by ID from all signature requests
-     * @param response Tool response
-     * @param id Transaction ID to find
-     * @returns Transaction or undefined if not found
-     */
     getTransactionById(response: any, id: string): UnsignedTransaction | undefined;
-    /**
-     * Process client input before sending to service
-     * Automatically adds wallet information to the request
-     */
     processInputClient(input: any): Promise<CryptoPluginInput>;
-    /**
-     * Process service output before sending to client
-     */
+    private extractSignatureRequests;
     processOutputService(output: any): Promise<CryptoPluginOutput>;
-    /**
-     * Process client output after receiving service response
-     */
     processOutputClient(output: any): Promise<any>;
-    /**
-     * Process service input before handling the request
-     */
     processInputService(input: any): Promise<any>;
 }