@hashgraphonline/conversational-agent 0.1.208 → 0.1.210

package/dist/esm/index27.js

@@ -0,0 +1,242 @@
+class MCPContentProcessor {
+  constructor(contentStorage, logger) {
+    this.contentStorage = contentStorage;
+    this.logger = logger;
+  }
+  analyzeResponse(response) {
+    const contents = [];
+    let totalSize = 0;
+    this.extractContentFromResponse(response, contents);
+    totalSize = contents.reduce((sum, content) => sum + content.sizeBytes, 0);
+    const largestContentSize = contents.reduce((max, content) => Math.max(max, content.sizeBytes), 0);
+    const shouldProcess = contents.some(
+      (content) => this.contentStorage.shouldUseReference(
+        typeof content.content === "string" ? content.content : JSON.stringify(content.content)
+      )
+    );
+    return {
+      shouldProcess,
+      contents,
+      totalSize,
+      largestContentSize
+    };
+  }
+  async processResponse(response, serverName, toolName) {
+    try {
+      const analysis = this.analyzeResponse(response);
+      if (!analysis.shouldProcess) {
+        return {
+          content: response,
+          wasProcessed: false
+        };
+      }
+      const processedResponse = await this.createReferencedResponse(
+        response,
+        analysis,
+        serverName,
+        toolName
+      );
+      return processedResponse;
+    } catch (error) {
+      this.logger.error("Error processing MCP response:", error);
+      return {
+        content: response,
+        wasProcessed: false,
+        errors: [error instanceof Error ? error.message : "Unknown processing error"]
+      };
+    }
+  }
+  extractContentFromResponse(obj, contents) {
+    if (obj === null || obj === void 0) {
+      return;
+    }
+    if (Array.isArray(obj)) {
+      obj.forEach((item) => this.extractContentFromResponse(item, contents));
+      return;
+    }
+    if (typeof obj === "object") {
+      const record = obj;
+      if (record.type === "text" && typeof record.text === "string") {
+        contents.push({
+          content: record.text,
+          type: "text",
+          sizeBytes: Buffer.byteLength(record.text, "utf8"),
+          mimeType: this.detectMimeType(record.text)
+        });
+        return;
+      }
+      if (record.type === "image" && typeof record.data === "string") {
+        contents.push({
+          content: record.data,
+          type: "image",
+          sizeBytes: Math.ceil(record.data.length * 0.75),
+          mimeType: record.mimeType || "image/jpeg"
+        });
+        return;
+      }
+      if (record.type === "resource" && record.resource) {
+        const resourceStr = JSON.stringify(record.resource);
+        contents.push({
+          content: resourceStr,
+          type: "resource",
+          sizeBytes: Buffer.byteLength(resourceStr, "utf8"),
+          mimeType: "application/json"
+        });
+        return;
+      }
+      Object.values(record).forEach((value) => this.extractContentFromResponse(value, contents));
+      return;
+    }
+    if (typeof obj === "string") {
+      if (obj.length > 1e3) {
+        contents.push({
+          content: obj,
+          type: "text",
+          sizeBytes: Buffer.byteLength(obj, "utf8"),
+          mimeType: this.detectMimeType(obj)
+        });
+      }
+    }
+  }
+  async createReferencedResponse(originalResponse, analysis, serverName, toolName) {
+    const processedResponse = this.deepClone(originalResponse);
+    const errors = [];
+    let referenceCreated = false;
+    let totalReferenceSize = 0;
+    for (const contentInfo of analysis.contents) {
+      if (this.contentStorage.shouldUseReference(
+        typeof contentInfo.content === "string" ? contentInfo.content : JSON.stringify(contentInfo.content)
+      )) {
+        try {
+          const contentBuffer = Buffer.from(
+            typeof contentInfo.content === "string" ? contentInfo.content : JSON.stringify(contentInfo.content),
+            "utf8"
+          );
+          const contentType = this.mapMimeTypeToContentType(contentInfo.mimeType);
+          const metadata = {
+            contentType,
+            source: "mcp_tool",
+            mcpToolName: `${serverName}::${toolName}`,
+            tags: ["mcp_response", serverName, toolName]
+          };
+          if (contentInfo.mimeType !== void 0) {
+            metadata.mimeType = contentInfo.mimeType;
+          }
+          const reference = await this.contentStorage.storeContentIfLarge(
+            contentBuffer,
+            metadata
+          );
+          if (reference) {
+            this.replaceContentInResponse(
+              processedResponse,
+              contentInfo.content,
+              this.createLightweightReference(reference)
+            );
+            referenceCreated = true;
+            totalReferenceSize += contentBuffer.length;
+          }
+        } catch (error) {
+          errors.push(`Failed to create reference: ${error instanceof Error ? error.message : "Unknown error"}`);
+        }
+      }
+    }
+    const result = {
+      content: processedResponse,
+      wasProcessed: true,
+      referenceCreated,
+      originalSize: totalReferenceSize
+    };
+    if (errors.length > 0) {
+      result.errors = errors;
+    }
+    return result;
+  }
+  createLightweightReference(reference) {
+    return {
+      type: "content_reference",
+      referenceId: reference.referenceId,
+      preview: reference.preview,
+      size: reference.metadata.sizeBytes,
+      contentType: reference.metadata.contentType,
+      format: "ref://{id}",
+      _isReference: true
+    };
+  }
+  replaceContentInResponse(obj, oldContent, newContent) {
+    if (obj === null || obj === void 0) {
+      return;
+    }
+    if (Array.isArray(obj)) {
+      for (let i = 0; i < obj.length; i++) {
+        if (obj[i] === oldContent) {
+          obj[i] = newContent;
+        } else {
+          this.replaceContentInResponse(obj[i], oldContent, newContent);
+        }
+      }
+      return;
+    }
+    if (typeof obj === "object") {
+      const record = obj;
+      if (record.type === "text" && record.text === oldContent) {
+        for (const key of Object.keys(record)) {
+          delete record[key];
+        }
+        for (const key of Object.keys(newContent)) {
+          record[key] = newContent[key];
+        }
+        return;
+      }
+      for (const key in record) {
+        if (record[key] === oldContent) {
+          record[key] = newContent;
+        } else {
+          this.replaceContentInResponse(record[key], oldContent, newContent);
+        }
+      }
+    }
+  }
+  detectMimeType(content) {
+    if (content.trim().startsWith("{") || content.trim().startsWith("[")) {
+      return "application/json";
+    }
+    if (content.includes("<html>") || content.includes("<!DOCTYPE")) {
+      return "text/html";
+    }
+    if (content.includes("# ") || content.includes("## ")) {
+      return "text/markdown";
+    }
+    return "text/plain";
+  }
+  mapMimeTypeToContentType(mimeType) {
+    if (!mimeType) return "text";
+    if (mimeType.startsWith("text/plain")) return "text";
+    if (mimeType === "application/json") return "json";
+    if (mimeType === "text/html") return "html";
+    if (mimeType === "text/markdown") return "markdown";
+    if (mimeType.startsWith("text/")) return "text";
+    return "binary";
+  }
+  deepClone(obj) {
+    if (obj === null || typeof obj !== "object") {
+      return obj;
+    }
+    if (obj instanceof Date) {
+      return new Date(obj.getTime());
+    }
+    if (Array.isArray(obj)) {
+      return obj.map((item) => this.deepClone(item));
+    }
+    const cloned = {};
+    for (const key in obj) {
+      if (obj.hasOwnProperty(key)) {
+        cloned[key] = this.deepClone(obj[key]);
+      }
+    }
+    return cloned;
+  }
+}
+export {
+  MCPContentProcessor
+};
+//# sourceMappingURL=index27.js.map

package/dist/esm/index27.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index27.js","sources":["../../src/mcp/ContentProcessor.ts"],"sourcesContent":["import type { ContentType, ContentSource } from '../types/content-reference';\nimport type { ContentStorage } from '../memory/ContentStorage';\nimport { Logger } from '@hashgraphonline/standards-sdk';\n\nexport interface MCPResponseContent {\n  content: unknown;\n  type: 'text' | 'image' | 'resource' | 'text[]' | 'image[]';\n  sizeBytes: number;\n  mimeType?: string;\n}\n\nexport interface ProcessedResponse {\n  content: unknown;\n  wasProcessed: boolean;\n  referenceCreated?: boolean;\n  referenceId?: string;\n  originalSize?: number;\n  errors?: string[];\n}\n\nexport interface ContentAnalysis {\n  shouldProcess: boolean;\n  contents: MCPResponseContent[];\n  totalSize: number;\n  largestContentSize: number;\n}\n\nexport class MCPContentProcessor {\n  private contentStorage: ContentStorage;\n  private logger: Logger;\n\n  constructor(contentStorage: ContentStorage, logger: Logger) {\n    this.contentStorage = contentStorage;\n    this.logger = logger;\n  }\n\n  analyzeResponse(response: unknown): ContentAnalysis {\n    const contents: MCPResponseContent[] = [];\n    let totalSize = 0;\n\n    this.extractContentFromResponse(response, contents);\n\n    totalSize = contents.reduce((sum, content) => sum + content.sizeBytes, 0);\n    const largestContentSize = contents.reduce((max, content) => \n      Math.max(max, content.sizeBytes), 0);\n\n    const shouldProcess = contents.some(content => \n      this.contentStorage.shouldUseReference(\n        typeof content.content === 'string' \n          ? content.content \n          : JSON.stringify(content.content)\n      )\n    );\n\n    return {\n      shouldProcess,\n      contents,\n      totalSize,\n      largestContentSize\n    };\n  }\n\n  async processResponse(\n    response: unknown,\n    serverName: string,\n    toolName: string\n  ): Promise<ProcessedResponse> {\n    try {\n      const analysis = this.analyzeResponse(response);\n      \n      if (!analysis.shouldProcess) {\n        return {\n          content: response,\n          wasProcessed: false\n        };\n      }\n\n      const processedResponse = await this.createReferencedResponse(\n        response,\n        analysis,\n        serverName,\n        toolName\n      );\n\n      return processedResponse;\n    } catch (error) {\n      this.logger.error('Error processing MCP response:', error);\n      return {\n        content: response,\n        wasProcessed: false,\n        errors: [error instanceof Error ? error.message : 'Unknown processing error']\n      };\n    }\n  }\n\n  private extractContentFromResponse(obj: unknown, contents: MCPResponseContent[]): void {\n    if (obj === null || obj === undefined) {\n      return;\n    }\n\n    if (Array.isArray(obj)) {\n      obj.forEach(item => this.extractContentFromResponse(item, contents));\n      return;\n    }\n\n    if (typeof obj === 'object') {\n      const record = obj as Record<string, unknown>;\n      \n      if (record.type === 'text' && typeof record.text === 'string') {\n        contents.push({\n          content: record.text,\n          type: 'text',\n          sizeBytes: Buffer.byteLength(record.text, 'utf8'),\n          mimeType: this.detectMimeType(record.text as string)\n        });\n        return;\n      }\n\n      if (record.type === 'image' && typeof record.data === 'string') {\n        contents.push({\n          content: record.data,\n          type: 'image',\n          sizeBytes: Math.ceil(record.data.length * 0.75),\n          mimeType: record.mimeType as string || 'image/jpeg'\n        });\n        return;\n      }\n\n      if (record.type === 'resource' && record.resource) {\n        const resourceStr = JSON.stringify(record.resource);\n        contents.push({\n          content: resourceStr,\n          type: 'resource',\n          sizeBytes: Buffer.byteLength(resourceStr, 'utf8'),\n          mimeType: 'application/json'\n        });\n        return;\n      }\n\n      Object.values(record).forEach(value => \n        this.extractContentFromResponse(value, contents));\n      return;\n    }\n\n    if (typeof obj === 'string') {\n      if (obj.length > 1000) {\n        contents.push({\n          content: obj,\n          type: 'text',\n          sizeBytes: Buffer.byteLength(obj, 'utf8'),\n          mimeType: this.detectMimeType(obj)\n        });\n      }\n    }\n  }\n\n  private async createReferencedResponse(\n    originalResponse: unknown,\n    analysis: ContentAnalysis,\n    serverName: string,\n    toolName: string\n  ): Promise<ProcessedResponse> {\n    const processedResponse = this.deepClone(originalResponse);\n    const errors: string[] = [];\n    let referenceCreated = false;\n    let totalReferenceSize = 0;\n\n    for (const contentInfo of analysis.contents) {\n      if (this.contentStorage.shouldUseReference(\n        typeof contentInfo.content === 'string' \n          ? contentInfo.content \n          : JSON.stringify(contentInfo.content)\n      )) {\n        try {\n          const contentBuffer = Buffer.from(\n            typeof contentInfo.content === 'string' \n              ? contentInfo.content \n              : JSON.stringify(contentInfo.content),\n            'utf8'\n          );\n\n          const contentType = this.mapMimeTypeToContentType(contentInfo.mimeType);\n          \n          const metadata: Parameters<typeof this.contentStorage.storeContentIfLarge>[1] = {\n            contentType,\n            source: 'mcp_tool' as ContentSource,\n            mcpToolName: `${serverName}::${toolName}`,\n            tags: ['mcp_response', serverName, toolName]\n          };\n          \n          if (contentInfo.mimeType !== undefined) {\n            metadata.mimeType = contentInfo.mimeType;\n          }\n          \n          const reference = await this.contentStorage.storeContentIfLarge(\n            contentBuffer,\n            metadata\n          );\n\n          if (reference) {\n            this.replaceContentInResponse(\n              processedResponse,\n              contentInfo.content,\n              this.createLightweightReference(reference)\n            );\n            referenceCreated = true;\n            totalReferenceSize += contentBuffer.length;\n          }\n        } catch (error) {\n          errors.push(`Failed to create reference: ${error instanceof Error ? error.message : 'Unknown error'}`);\n        }\n      }\n    }\n\n    const result: ProcessedResponse = {\n      content: processedResponse,\n      wasProcessed: true,\n      referenceCreated,\n      originalSize: totalReferenceSize\n    };\n    \n    if (errors.length > 0) {\n      result.errors = errors;\n    }\n    \n    return result;\n  }\n\n  private createLightweightReference(reference: any): Record<string, unknown> {\n    return {\n      type: 'content_reference',\n      referenceId: reference.referenceId,\n      preview: reference.preview,\n      size: reference.metadata.sizeBytes,\n      contentType: reference.metadata.contentType,\n      format: 'ref://{id}',\n      _isReference: true\n    };\n  }\n\n  private replaceContentInResponse(obj: unknown, oldContent: unknown, newContent: any): void {\n    if (obj === null || obj === undefined) {\n      return;\n    }\n\n    if (Array.isArray(obj)) {\n      for (let i = 0; i < obj.length; i++) {\n        if (obj[i] === oldContent) {\n          obj[i] = newContent;\n        } else {\n          this.replaceContentInResponse(obj[i], oldContent, newContent);\n        }\n      }\n      return;\n    }\n\n    if (typeof obj === 'object') {\n      const record = obj as Record<string, unknown>;\n      if (record.type === 'text' && record.text === oldContent) {\n        for (const key of Object.keys(record)) {\n          delete record[key];\n        }\n        for (const key of Object.keys(newContent)) {\n          record[key] = newContent[key];\n        }\n        return;\n      }\n      for (const key in record) {\n        if (record[key] === oldContent) {\n          record[key] = newContent;\n        } else {\n          this.replaceContentInResponse(record[key], oldContent, newContent);\n        }\n      }\n    }\n  }\n\n  private detectMimeType(content: string): string {\n    if (content.trim().startsWith('{') || content.trim().startsWith('[')) {\n      return 'application/json';\n    }\n    if (content.includes('<html>') || content.includes('<!DOCTYPE')) {\n      return 'text/html';\n    }\n    if (content.includes('# ') || content.includes('## ')) {\n      return 'text/markdown';\n    }\n    return 'text/plain';\n  }\n\n  private mapMimeTypeToContentType(mimeType?: string): ContentType {\n    if (!mimeType) return 'text';\n    \n    if (mimeType.startsWith('text/plain')) return 'text';\n    if (mimeType === 'application/json') return 'json';\n    if (mimeType === 'text/html') return 'html';\n    if (mimeType === 'text/markdown') return 'markdown';\n    if (mimeType.startsWith('text/')) return 'text';\n    \n    return 'binary';\n  }\n\n  private deepClone<T>(obj: T): T {\n    if (obj === null || typeof obj !== 'object') {\n      return obj;\n    }\n    \n    if (obj instanceof Date) {\n      return new Date(obj.getTime()) as T;\n    }\n    \n    if (Array.isArray(obj)) {\n      return obj.map(item => this.deepClone(item)) as T;\n    }\n    \n    const cloned = {} as T;\n    for (const key in obj) {\n      if (obj.hasOwnProperty(key)) {\n        cloned[key] = this.deepClone(obj[key]);\n      }\n    }\n    \n    return cloned;\n  }\n}"],"names":[],"mappings":"AA2BO,MAAM,oBAAoB;AAAA,EAI/B,YAAY,gBAAgC,QAAgB;AAC1D,SAAK,iBAAiB;AACtB,SAAK,SAAS;AAAA,EAChB;AAAA,EAEA,gBAAgB,UAAoC;AAClD,UAAM,WAAiC,CAAA;AACvC,QAAI,YAAY;AAEhB,SAAK,2BAA2B,UAAU,QAAQ;AAElD,gBAAY,SAAS,OAAO,CAAC,KAAK,YAAY,MAAM,QAAQ,WAAW,CAAC;AACxE,UAAM,qBAAqB,SAAS,OAAO,CAAC,KAAK,YAC/C,KAAK,IAAI,KAAK,QAAQ,SAAS,GAAG,CAAC;AAErC,UAAM,gBAAgB,SAAS;AAAA,MAAK,CAAA,YAClC,KAAK,eAAe;AAAA,QAClB,OAAO,QAAQ,YAAY,WACvB,QAAQ,UACR,KAAK,UAAU,QAAQ,OAAO;AAAA,MAAA;AAAA,IACpC;AAGF,WAAO;AAAA,MACL;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IAAA;AAAA,EAEJ;AAAA,EAEA,MAAM,gBACJ,UACA,YACA,UAC4B;AAC5B,QAAI;AACF,YAAM,WAAW,KAAK,gBAAgB,QAAQ;AAE9C,UAAI,CAAC,SAAS,eAAe;AAC3B,eAAO;AAAA,UACL,SAAS;AAAA,UACT,cAAc;AAAA,QAAA;AAAA,MAElB;AAEA,YAAM,oBAAoB,MAAM,KAAK;AAAA,QACnC;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MAAA;AAGF,aAAO;AAAA,IACT,SAAS,OAAO;AACd,WAAK,OAAO,MAAM,kCAAkC,KAAK;AACzD,aAAO;AAAA,QACL,SAAS;AAAA,QACT,cAAc;AAAA,QACd,QAAQ,CAAC,iBAAiB,QAAQ,MAAM,UAAU,0BAA0B;AAAA,MAAA;AAAA,IAEhF;AAAA,EACF;AAAA,EAEQ,2BAA2B,KAAc,UAAsC;AACrF,QAAI,QAAQ,QAAQ,QAAQ,QAAW;AACrC;AAAA,IACF;AAEA,QAAI,MAAM,QAAQ,GAAG,GAAG;AACtB,UAAI,QAAQ,CAAA,SAAQ,KAAK,2BAA2B,MAAM,QAAQ,CAAC;AACnE;AAAA,IACF;AAEA,QAAI,OAAO,QAAQ,UAAU;AAC3B,YAAM,SAAS;AAEf,UAAI,OAAO,SAAS,UAAU,OAAO,OAAO,SAAS,UAAU;AAC7D,iBAAS,KAAK;AAAA,UACZ,SAAS,OAAO;AAAA,UAChB,MAAM;AAAA,UACN,WAAW,OAAO,WAAW,OAAO,MAAM,MAAM;AAAA,UAChD,UAAU,KAAK,eAAe,OAAO,IAAc;AAAA,QAAA,CACpD;AACD;AAAA,MACF;AAEA,UAAI,OAAO,SAAS,WAAW,OAAO,OAAO,SAAS,UAAU;AAC9D,iBAAS,KAAK;AAAA,UACZ,SAAS,OAAO;AAAA,UAChB,MAAM;AAAA,UACN,WAAW,KAAK,KAAK,OAAO,KAAK,SAAS,IAAI;AAAA,UAC9C,UAAU,OAAO,YAAsB;AAAA,QAAA,CACxC;AACD;AAAA,MACF;AAEA,UAAI,OAAO,SAAS,cAAc,OAAO,UAAU;AACjD,cAAM,cAAc,KAAK,UAAU,OAAO,QAAQ;AAClD,iBAAS,KAAK;AAAA,UACZ,SAAS;AAAA,UACT,MAAM;AAAA,UACN,WAAW,OAAO,WAAW,aAAa,MAAM;AAAA,UAChD,UAAU;AAAA,QAAA,CACX;AACD;AAAA,MACF;AAEA,aAAO,OAAO,MAAM,EAAE,QAAQ,WAC5B,KAAK,2BAA2B,OAAO,QAAQ,CAAC;AAClD;AAAA,IACF;AAEA,QAAI,OAAO,QAAQ,UAAU;AAC3B,UAAI,IAAI,SAAS,KAAM;AACrB,iBAAS,KAAK;AAAA,UACZ,SAAS;AAAA,UACT,MAAM;AAAA,UACN,WAAW,OAAO,WAAW,KAAK,MAAM;AAAA,UACxC,UAAU,KAAK,eAAe,GAAG;AAAA,QAAA,CAClC;AAAA,MACH;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAc,yBACZ,kBACA,UACA,YACA,UAC4B;AAC5B,UAAM,oBAAoB,KAAK,UAAU,gBAAgB;AACzD,UAAM,SAAmB,CAAA;AACzB,QAAI,mBAAmB;AACvB,QAAI,qBAAqB;AAEzB,eAAW,eAAe,SAAS,UAAU;AAC3C,UAAI,KAAK,eAAe;AAAA,QACtB,OAAO,YAAY,YAAY,WAC3B,YAAY,UACZ,KAAK,UAAU,YAAY,OAAO;AAAA,MAAA,GACrC;AACD,YAAI;AACF,gBAAM,gBAAgB,OAAO;AAAA,YAC3B,OAAO,YAAY,YAAY,WAC3B,YAAY,UACZ,KAAK,UAAU,YAAY,OAAO;AAAA,YACtC;AAAA,UAAA;AAGF,gBAAM,cAAc,KAAK,yBAAyB,YAAY,QAAQ;AAEtE,gBAAM,WAA0E;AAAA,YAC9E;AAAA,YACA,QAAQ;AAAA,YACR,aAAa,GAAG,UAAU,KAAK,QAAQ;AAAA,YACvC,MAAM,CAAC,gBAAgB,YAAY,QAAQ;AAAA,UAAA;AAG7C,cAAI,YAAY,aAAa,QAAW;AACtC,qBAAS,WAAW,YAAY;AAAA,UAClC;AAEA,gBAAM,YAAY,MAAM,KAAK,eAAe;AAAA,YAC1C;AAAA,YACA;AAAA,UAAA;AAGF,cAAI,WAAW;AACb,iBAAK;AAAA,cACH;AAAA,cACA,YAAY;AAAA,cACZ,KAAK,2BAA2B,SAAS;AAAA,YAAA;AAE3C,+BAAmB;AACnB,kCAAsB,cAAc;AAAA,UACtC;AAAA,QACF,SAAS,OAAO;AACd,iBAAO,KAAK,+BAA+B,iBAAiB,QAAQ,MAAM,UAAU,eAAe,EAAE;AAAA,QACvG;AAAA,MACF;AAAA,IACF;AAEA,UAAM,SAA4B;AAAA,MAChC,SAAS;AAAA,MACT,cAAc;AAAA,MACd;AAAA,MACA,cAAc;AAAA,IAAA;AAGhB,QAAI,OAAO,SAAS,GAAG;AACrB,aAAO,SAAS;AAAA,IAClB;AAEA,WAAO;AAAA,EACT;AAAA,EAEQ,2BAA2B,WAAyC;AAC1E,WAAO;AAAA,MACL,MAAM;AAAA,MACN,aAAa,UAAU;AAAA,MACvB,SAAS,UAAU;AAAA,MACnB,MAAM,UAAU,SAAS;AAAA,MACzB,aAAa,UAAU,SAAS;AAAA,MAChC,QAAQ;AAAA,MACR,cAAc;AAAA,IAAA;AAAA,EAElB;AAAA,EAEQ,yBAAyB,KAAc,YAAqB,YAAuB;AACzF,QAAI,QAAQ,QAAQ,QAAQ,QAAW;AACrC;AAAA,IACF;AAEA,QAAI,MAAM,QAAQ,GAAG,GAAG;AACtB,eAAS,IAAI,GAAG,IAAI,IAAI,QAAQ,KAAK;AACnC,YAAI,IAAI,CAAC,MAAM,YAAY;AACzB,cAAI,CAAC,IAAI;AAAA,QACX,OAAO;AACL,eAAK,yBAAyB,IAAI,CAAC,GAAG,YAAY,UAAU;AAAA,QAC9D;AAAA,MACF;AACA;AAAA,IACF;AAEA,QAAI,OAAO,QAAQ,UAAU;AAC3B,YAAM,SAAS;AACf,UAAI,OAAO,SAAS,UAAU,OAAO,SAAS,YAAY;AACxD,mBAAW,OAAO,OAAO,KAAK,MAAM,GAAG;AACrC,iBAAO,OAAO,GAAG;AAAA,QACnB;AACA,mBAAW,OAAO,OAAO,KAAK,UAAU,GAAG;AACzC,iBAAO,GAAG,IAAI,WAAW,GAAG;AAAA,QAC9B;AACA;AAAA,MACF;AACA,iBAAW,OAAO,QAAQ;AACxB,YAAI,OAAO,GAAG,MAAM,YAAY;AAC9B,iBAAO,GAAG,IAAI;AAAA,QAChB,OAAO;AACL,eAAK,yBAAyB,OAAO,GAAG,GAAG,YAAY,UAAU;AAAA,QACnE;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,eAAe,SAAyB;AAC9C,QAAI,QAAQ,OAAO,WAAW,GAAG,KAAK,QAAQ,KAAA,EAAO,WAAW,GAAG,GAAG;AACpE,aAAO;AAAA,IACT;AACA,QAAI,QAAQ,SAAS,QAAQ,KAAK,QAAQ,SAAS,WAAW,GAAG;AAC/D,aAAO;AAAA,IACT;AACA,QAAI,QAAQ,SAAS,IAAI,KAAK,QAAQ,SAAS,KAAK,GAAG;AACrD,aAAO;AAAA,IACT;AACA,WAAO;AAAA,EACT;AAAA,EAEQ,yBAAyB,UAAgC;AAC/D,QAAI,CAAC,SAAU,QAAO;AAEtB,QAAI,SAAS,WAAW,YAAY,EAAG,QAAO;AAC9C,QAAI,aAAa,mBAAoB,QAAO;AAC5C,QAAI,aAAa,YAAa,QAAO;AACrC,QAAI,aAAa,gBAAiB,QAAO;AACzC,QAAI,SAAS,WAAW,OAAO,EAAG,QAAO;AAEzC,WAAO;AAAA,EACT;AAAA,EAEQ,UAAa,KAAW;AAC9B,QAAI,QAAQ,QAAQ,OAAO,QAAQ,UAAU;AAC3C,aAAO;AAAA,IACT;AAEA,QAAI,eAAe,MAAM;AACvB,aAAO,IAAI,KAAK,IAAI,SAAS;AAAA,IAC/B;AAEA,QAAI,MAAM,QAAQ,GAAG,GAAG;AACtB,aAAO,IAAI,IAAI,CAAA,SAAQ,KAAK,UAAU,IAAI,CAAC;AAAA,IAC7C;AAEA,UAAM,SAAS,CAAA;AACf,eAAW,OAAO,KAAK;AACrB,UAAI,IAAI,eAAe,GAAG,GAAG;AAC3B,eAAO,GAAG,IAAI,KAAK,UAAU,IAAI,GAAG,CAAC;AAAA,MACvC;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AACF;"}

package/dist/esm/index5.js

@@ -1,35 +1,32 @@
-import { BasePlugin } from "hedera-agent-kit";
-import { TransferHbarTool } from "./index12.js";
-class HbarTransferPlugin extends BasePlugin {
+import { BasePlugin, HederaAirdropTokenTool } from "hedera-agent-kit";
+import { TransferHbarTool } from "./index18.js";
+import { AirdropToolWrapper } from "./index19.js";
+class HbarPlugin extends BasePlugin {
   constructor() {
     super(...arguments);
-    this.id = "hbar-transfer";
-    this.name = "HBAR Transfer Plugin";
-    this.description = "HBAR transfer tool with proper decimal handling for multi-signature transactions";
+    this.id = "hbar";
+    this.name = "HBAR Plugin";
+    this.description = "HBAR operations: transfer tool with robust decimal handling and compatibility with airdrop improvements";
     this.version = "1.0.0";
     this.author = "Hashgraph Online";
     this.namespace = "account";
     this.tools = [];
+    this.originalAirdropTool = null;
   }
   async initialize(context) {
     await super.initialize(context);
     const hederaKit = context.config.hederaKit;
     if (!hederaKit) {
       this.context.logger.warn(
-        "HederaKit not found in context. HBAR transfer tools will not be available."
+        "HederaKit not found in context. HBAR tools will not be available."
       );
       return;
     }
     try {
       this.initializeTools();
-      this.context.logger.info(
-        "HBAR Transfer Plugin initialized successfully"
-      );
+      this.context.logger.info("HBAR Plugin initialized successfully");
     } catch (error) {
-      this.context.logger.error(
-        "Failed to initialize HBAR Transfer plugin:",
-        error
-      );
+      this.context.logger.error("Failed to initialize HBAR plugin:", error);
     }
   }
   initializeTools() {
@@ -37,12 +34,28 @@ class HbarTransferPlugin extends BasePlugin {
     if (!hederaKit) {
       throw new Error("HederaKit not found in context config");
     }
-    this.tools = [
-      new TransferHbarTool({
+    const transfer = new TransferHbarTool({
+      hederaKit,
+      logger: this.context.logger
+    });
+    this.tools = [transfer];
+    try {
+      this.context.logger.info(
+        "Creating wrapper for passed original airdrop tool"
+      );
+      const airdropTool = new HederaAirdropTokenTool({
         hederaKit,
         logger: this.context.logger
-      })
-    ];
+      });
+      const wrappedAirdropTool = new AirdropToolWrapper(airdropTool, hederaKit);
+      this.tools.push(wrappedAirdropTool);
+      this.context.logger.info("Added wrapped airdrop tool to HBAR Plugin");
+    } catch (error) {
+      this.context.logger.error("Error creating airdrop tool wrapper:", error);
+    }
+    this.context.logger.info(
+      `HBAR Plugin tools initialized with ${this.tools.length} tools`
+    );
   }
   getTools() {
     return this.tools;
@@ -52,6 +65,6 @@ class HbarTransferPlugin extends BasePlugin {
   }
 }
 export {
-  HbarTransferPlugin
+  HbarPlugin
 };
 //# sourceMappingURL=index5.js.map

package/dist/esm/index5.js.map

@@ -1 +1 @@
-{"version":3,"file":"index5.js","sources":["../../src/plugins/hbar-transfer/HbarTransferPlugin.ts"],"sourcesContent":["import {\n  GenericPluginContext,\n  HederaTool,\n  BasePlugin,\n  HederaAgentKit,\n} from 'hedera-agent-kit';\nimport { TransferHbarTool } from './TransferHbarTool';\n\nexport class HbarTransferPlugin extends BasePlugin {\n  id = 'hbar-transfer';\n  name = 'HBAR Transfer Plugin';\n  description =\n    'HBAR transfer tool with proper decimal handling for multi-signature transactions';\n  version = '1.0.0';\n  author = 'Hashgraph Online';\n  namespace = 'account';\n\n  private tools: HederaTool[] = [];\n\n  override async initialize(context: GenericPluginContext): Promise<void> {\n    await super.initialize(context);\n\n    const hederaKit = context.config.hederaKit as HederaAgentKit;\n    if (!hederaKit) {\n      this.context.logger.warn(\n        'HederaKit not found in context. HBAR transfer tools will not be available.'\n      );\n      return;\n    }\n\n    try {\n      this.initializeTools();\n\n      this.context.logger.info(\n        'HBAR Transfer Plugin initialized successfully'\n      );\n    } catch (error) {\n      this.context.logger.error(\n        'Failed to initialize HBAR Transfer plugin:',\n        error\n      );\n    }\n  }\n\n  private initializeTools(): void {\n    const hederaKit = this.context.config.hederaKit as HederaAgentKit;\n    if (!hederaKit) {\n      throw new Error('HederaKit not found in context config');\n    }\n\n    this.tools = [\n      new TransferHbarTool({\n        hederaKit: hederaKit,\n        logger: this.context.logger,\n      }),\n    ];\n  }\n\n  override getTools(): HederaTool[] {\n    return this.tools;\n  }\n\n  async shutdown(): Promise<void> {\n    this.tools = [];\n  }\n}"],"names":[],"mappings":";;AAQO,MAAM,2BAA2B,WAAW;AAAA,EAA5C,cAAA;AAAA,UAAA,GAAA,SAAA;AACL,SAAA,KAAK;AACL,SAAA,OAAO;AACP,SAAA,cACE;AACF,SAAA,UAAU;AACV,SAAA,SAAS;AACT,SAAA,YAAY;AAEZ,SAAQ,QAAsB,CAAA;AAAA,EAAC;AAAA,EAE/B,MAAe,WAAW,SAA8C;AACtE,UAAM,MAAM,WAAW,OAAO;AAE9B,UAAM,YAAY,QAAQ,OAAO;AACjC,QAAI,CAAC,WAAW;AACd,WAAK,QAAQ,OAAO;AAAA,QAClB;AAAA,MAAA;AAEF;AAAA,IACF;AAEA,QAAI;AACF,WAAK,gBAAA;AAEL,WAAK,QAAQ,OAAO;AAAA,QAClB;AAAA,MAAA;AAAA,IAEJ,SAAS,OAAO;AACd,WAAK,QAAQ,OAAO;AAAA,QAClB;AAAA,QACA;AAAA,MAAA;AAAA,IAEJ;AAAA,EACF;AAAA,EAEQ,kBAAwB;AAC9B,UAAM,YAAY,KAAK,QAAQ,OAAO;AACtC,QAAI,CAAC,WAAW;AACd,YAAM,IAAI,MAAM,uCAAuC;AAAA,IACzD;AAEA,SAAK,QAAQ;AAAA,MACX,IAAI,iBAAiB;AAAA,QACnB;AAAA,QACA,QAAQ,KAAK,QAAQ;AAAA,MAAA,CACtB;AAAA,IAAA;AAAA,EAEL;AAAA,EAES,WAAyB;AAChC,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,MAAM,WAA0B;AAC9B,SAAK,QAAQ,CAAA;AAAA,EACf;AACF;"}
+{"version":3,"file":"index5.js","sources":["../../src/plugins/hbar/HbarPlugin.ts"],"sourcesContent":["import {\n  GenericPluginContext,\n  HederaTool,\n  BasePlugin,\n  HederaAgentKit,\n  HederaAirdropTokenTool,\n} from 'hedera-agent-kit';\nimport { TransferHbarTool } from './TransferHbarTool';\nimport { AirdropToolWrapper } from './AirdropToolWrapper';\nimport { StructuredTool } from '@langchain/core/tools';\n\nexport class HbarPlugin extends BasePlugin {\n  id = 'hbar';\n  name = 'HBAR Plugin';\n  description =\n    'HBAR operations: transfer tool with robust decimal handling and compatibility with airdrop improvements';\n  version = '1.0.0';\n  author = 'Hashgraph Online';\n  namespace = 'account';\n\n  private tools: (HederaTool | AirdropToolWrapper)[] = [];\n  private originalAirdropTool: StructuredTool | null = null;\n\n  override async initialize(context: GenericPluginContext): Promise<void> {\n    await super.initialize(context);\n\n    const hederaKit = context.config.hederaKit as HederaAgentKit;\n    if (!hederaKit) {\n      this.context.logger.warn(\n        'HederaKit not found in context. HBAR tools will not be available.'\n      );\n      return;\n    }\n\n    try {\n      this.initializeTools();\n\n      this.context.logger.info('HBAR Plugin initialized successfully');\n    } catch (error) {\n      this.context.logger.error('Failed to initialize HBAR plugin:', error);\n    }\n  }\n\n  private initializeTools(): void {\n    const hederaKit = this.context.config.hederaKit as HederaAgentKit;\n    if (!hederaKit) {\n      throw new Error('HederaKit not found in context config');\n    }\n\n    const transfer = new TransferHbarTool({\n      hederaKit: hederaKit,\n      logger: this.context.logger,\n    });\n\n    this.tools = [transfer];\n\n    try {\n      this.context.logger.info(\n        'Creating wrapper for passed original airdrop tool'\n      );\n\n      const airdropTool = new HederaAirdropTokenTool({\n        hederaKit: hederaKit,\n        logger: this.context.logger,\n      });\n      const wrappedAirdropTool = new AirdropToolWrapper(airdropTool, hederaKit);\n      this.tools.push(wrappedAirdropTool);\n      this.context.logger.info('Added wrapped airdrop tool to HBAR Plugin');\n    } catch (error) {\n      this.context.logger.error('Error creating airdrop tool wrapper:', error);\n    }\n\n    this.context.logger.info(\n      `HBAR Plugin tools initialized with ${this.tools.length} tools`\n    );\n  }\n\n  override getTools(): HederaTool[] {\n    // @ts-ignore\n    return this.tools;\n  }\n\n  async shutdown(): Promise<void> {\n    this.tools = [];\n  }\n}\n"],"names":[],"mappings":";;;AAWO,MAAM,mBAAmB,WAAW;AAAA,EAApC,cAAA;AAAA,UAAA,GAAA,SAAA;AACL,SAAA,KAAK;AACL,SAAA,OAAO;AACP,SAAA,cACE;AACF,SAAA,UAAU;AACV,SAAA,SAAS;AACT,SAAA,YAAY;AAEZ,SAAQ,QAA6C,CAAA;AACrD,SAAQ,sBAA6C;AAAA,EAAA;AAAA,EAErD,MAAe,WAAW,SAA8C;AACtE,UAAM,MAAM,WAAW,OAAO;AAE9B,UAAM,YAAY,QAAQ,OAAO;AACjC,QAAI,CAAC,WAAW;AACd,WAAK,QAAQ,OAAO;AAAA,QAClB;AAAA,MAAA;AAEF;AAAA,IACF;AAEA,QAAI;AACF,WAAK,gBAAA;AAEL,WAAK,QAAQ,OAAO,KAAK,sCAAsC;AAAA,IACjE,SAAS,OAAO;AACd,WAAK,QAAQ,OAAO,MAAM,qCAAqC,KAAK;AAAA,IACtE;AAAA,EACF;AAAA,EAEQ,kBAAwB;AAC9B,UAAM,YAAY,KAAK,QAAQ,OAAO;AACtC,QAAI,CAAC,WAAW;AACd,YAAM,IAAI,MAAM,uCAAuC;AAAA,IACzD;AAEA,UAAM,WAAW,IAAI,iBAAiB;AAAA,MACpC;AAAA,MACA,QAAQ,KAAK,QAAQ;AAAA,IAAA,CACtB;AAED,SAAK,QAAQ,CAAC,QAAQ;AAEtB,QAAI;AACF,WAAK,QAAQ,OAAO;AAAA,QAClB;AAAA,MAAA;AAGF,YAAM,cAAc,IAAI,uBAAuB;AAAA,QAC7C;AAAA,QACA,QAAQ,KAAK,QAAQ;AAAA,MAAA,CACtB;AACD,YAAM,qBAAqB,IAAI,mBAAmB,aAAa,SAAS;AACxE,WAAK,MAAM,KAAK,kBAAkB;AAClC,WAAK,QAAQ,OAAO,KAAK,2CAA2C;AAAA,IACtE,SAAS,OAAO;AACd,WAAK,QAAQ,OAAO,MAAM,wCAAwC,KAAK;AAAA,IACzE;AAEA,SAAK,QAAQ,OAAO;AAAA,MAClB,sCAAsC,KAAK,MAAM,MAAM;AAAA,IAAA;AAAA,EAE3D;AAAA,EAES,WAAyB;AAEhC,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,MAAM,WAA0B;AAC9B,SAAK,QAAQ,CAAA;AAAA,EACf;AACF;"}