@hashgraphonline/conversational-agent 0.1.217 → 0.1.219

package/dist/esm/index15.js.map

@@ -1 +1 @@
-{"version":3,"file":"index15.js","sources":["../../src/memory/SmartMemoryManager.ts"],"sourcesContent":["import type { BaseMessage } from '@langchain/core/messages';\nimport { Logger } from '@hashgraphonline/standards-sdk';\nimport { MemoryWindow } from './MemoryWindow';\nimport { ContentStorage } from './ContentStorage';\nimport { TokenCounter } from './TokenCounter';\n\n/**\n * Entity association for storing blockchain entity contexts\n */\nexport interface EntityAssociation {\n  /** The blockchain entity ID (e.g., tokenId, accountId, topicId) */\n  entityId: string;\n  /** User-provided or derived friendly name */\n  entityName: string;\n  /** Type of entity (token, account, topic, schedule, etc.) */\n  entityType: string;\n  /** When the entity was created/associated */\n  createdAt: Date;\n  /** Transaction ID that created this entity */\n  transactionId?: string;\n}\n\n/**\n * Options for resolving entity references\n */\nexport interface EntityResolutionOptions {\n  /** Filter by specific entity type */\n  entityType?: string;\n  /** Maximum number of results to return */\n  limit?: number;\n  /** Whether to use fuzzy matching for natural language queries */\n  fuzzyMatch?: boolean;\n}\n\n/**\n * Configuration for SmartMemoryManager\n */\nexport interface SmartMemoryConfig {\n  /** Maximum tokens for active memory window */\n  maxTokens?: number;\n  /** Reserve tokens for response generation */\n  reserveTokens?: number;\n  /** Model name for token counting */\n  modelName?: string;\n  /** Maximum messages to store in content storage */\n  storageLimit?: number;\n}\n\n/**\n * Search options for history search\n */\nexport interface SearchOptions {\n  /** Whether to perform case-sensitive search */\n  caseSensitive?: boolean;\n  /** Maximum number of results to return */\n  limit?: number;\n  /** Whether to use regex pattern matching */\n  useRegex?: boolean;\n}\n\n/**\n * Memory statistics for active memory window\n */\nexport interface MemoryStats {\n  /** Total active messages in memory window */\n  totalActiveMessages: number;\n  /** Current token count including system prompt */\n  currentTokenCount: number;\n  /** Maximum token capacity */\n  maxTokens: number;\n  /** Remaining token capacity */\n  remainingCapacity: number;\n  /** System prompt token count */\n  systemPromptTokens: number;\n  /** Memory usage percentage */\n  usagePercentage: number;\n}\n\nconst IS_ENTITY_ASSOCIATION_FLAG = '\"isEntityAssociation\":true';\n\n\n/**\n * TODO: investigate using chroma / rag for long term memory\n * Smart memory manager that combines active memory window with long-term storage\n * Provides context-aware memory management with automatic pruning and searchable history\n */\nexport class SmartMemoryManager {\n  private memoryWindow: MemoryWindow;\n  private _contentStorage: ContentStorage;\n  private tokenCounter: TokenCounter;\n  private config: Required<SmartMemoryConfig>;\n  private logger: Logger;\n\n  private static readonly DEFAULT_CONFIG: Required<SmartMemoryConfig> = {\n    maxTokens: 8000,\n    reserveTokens: 1000,\n    modelName: 'gpt-4o',\n    storageLimit: 1000,\n  };\n\n  constructor(config: SmartMemoryConfig = {}) {\n    this.config = { ...SmartMemoryManager.DEFAULT_CONFIG, ...config };\n    this.logger = new Logger({ module: 'SmartMemoryManager' });\n\n    this.tokenCounter = new TokenCounter(this.config.modelName);\n    this._contentStorage = new ContentStorage(this.config.storageLimit);\n    this.memoryWindow = new MemoryWindow(\n      this.config.maxTokens,\n      this.config.reserveTokens,\n      this.tokenCounter\n    );\n  }\n\n  /**\n   * Get the content storage instance for file/content reference operations\n   * @returns ContentStorage instance\n   */\n  get contentStorage(): ContentStorage {\n    return this._contentStorage;\n  }\n\n  /**\n   * Add a message to the active memory window\n   * Automatically handles pruning and storage of displaced messages\n   * @param message - Message to add\n   */\n  addMessage(message: BaseMessage): void {\n    const result = this.memoryWindow.addMessage(message);\n\n    if (result.prunedMessages.length > 0) {\n      this._contentStorage.storeMessages(result.prunedMessages);\n    }\n  }\n\n  /**\n   * Get all active messages from the memory window\n   * @returns Array of active messages in chronological order\n   */\n  getMessages(): BaseMessage[] {\n    return this.memoryWindow.getMessages();\n  }\n\n  /**\n   * Clear active memory window\n   * @param clearStorage - Whether to also clear the content storage (default: false)\n   */\n  clear(clearStorage: boolean = false): void {\n    this.memoryWindow.clear();\n\n    if (clearStorage) {\n      this._contentStorage.clear();\n    }\n  }\n\n  /**\n   * Set the system prompt for the memory window\n   * @param systemPrompt - System prompt text\n   */\n  setSystemPrompt(systemPrompt: string): void {\n    this.memoryWindow.setSystemPrompt(systemPrompt);\n  }\n\n  /**\n   * Get the current system prompt\n   * @returns Current system prompt text\n   */\n  getSystemPrompt(): string {\n    return this.memoryWindow.getSystemPrompt();\n  }\n\n  /**\n   * Search through stored message history\n   * @param query - Search term or pattern\n   * @param options - Search configuration\n   * @returns Array of matching messages from history\n   */\n  searchHistory(query: string, options: SearchOptions = {}): BaseMessage[] {\n    return this._contentStorage.searchMessages(query, options);\n  }\n\n  /**\n   * Get recent messages from storage history\n   * @param count - Number of recent messages to retrieve\n   * @returns Array of recent messages from storage\n   */\n  getRecentHistory(count: number): BaseMessage[] {\n    return this._contentStorage.getRecentMessages(count);\n  }\n\n  /**\n   * Check if a message can be added without exceeding limits\n   * @param message - Message to test\n   * @returns True if message can be added\n   */\n  canAddMessage(message: BaseMessage): boolean {\n    return this.memoryWindow.canAddMessage(message);\n  }\n\n  /**\n   * Get statistics about the active memory window\n   * @returns Memory usage statistics\n   */\n  getMemoryStats(): MemoryStats {\n    const windowStats = this.memoryWindow.getStats();\n\n    return {\n      totalActiveMessages: windowStats.totalMessages,\n      currentTokenCount: windowStats.currentTokens,\n      maxTokens: windowStats.maxTokens,\n      remainingCapacity: windowStats.remainingCapacity,\n      systemPromptTokens: windowStats.systemPromptTokens,\n      usagePercentage: windowStats.usagePercentage,\n    };\n  }\n\n  /**\n   * Get statistics about the content storage\n   * @returns Storage usage statistics\n   */\n  getStorageStats(): ReturnType<ContentStorage['getStorageStats']> {\n    return this._contentStorage.getStorageStats();\n  }\n\n  /**\n   * Get combined statistics for both active memory and storage\n   * @returns Combined memory and storage statistics\n   */\n  getOverallStats(): {\n    activeMemory: MemoryStats;\n    storage: ReturnType<ContentStorage['getStorageStats']>;\n    totalMessagesManaged: number;\n    activeMemoryUtilization: number;\n    storageUtilization: number;\n  } {\n    const memoryStats = this.getMemoryStats();\n    const storageStats = this.getStorageStats();\n\n    return {\n      activeMemory: memoryStats,\n      storage: storageStats,\n      totalMessagesManaged:\n        memoryStats.totalActiveMessages + storageStats.totalMessages,\n      activeMemoryUtilization: memoryStats.usagePercentage,\n      storageUtilization: storageStats.usagePercentage,\n    };\n  }\n\n  /**\n   * Update the configuration and apply changes\n   * @param newConfig - New configuration options\n   */\n  updateConfig(newConfig: Partial<SmartMemoryConfig>): void {\n    this.config = { ...this.config, ...newConfig };\n\n    if (\n      newConfig.maxTokens !== undefined ||\n      newConfig.reserveTokens !== undefined\n    ) {\n      this.memoryWindow.updateLimits(\n        this.config.maxTokens,\n        this.config.reserveTokens\n      );\n    }\n\n    if (newConfig.storageLimit !== undefined) {\n      this._contentStorage.updateStorageLimit(this.config.storageLimit);\n    }\n  }\n\n  /**\n   * Get current configuration\n   * @returns Current configuration settings\n   */\n  getConfig(): Required<SmartMemoryConfig> {\n    return { ...this.config };\n  }\n\n  /**\n   * Get messages from storage within a time range\n   * @param startTime - Start of time range\n   * @param endTime - End of time range\n   * @returns Messages within the specified time range\n   */\n  getHistoryFromTimeRange(startTime: Date, endTime: Date): BaseMessage[] {\n    return this._contentStorage.getMessagesFromTimeRange(startTime, endTime);\n  }\n\n  /**\n   * Get messages from storage by message type\n   * @param messageType - Type of messages to retrieve ('human', 'ai', 'system', etc.)\n   * @param limit - Maximum number of messages to return\n   * @returns Messages of the specified type\n   */\n  getHistoryByType(messageType: string, limit?: number): BaseMessage[] {\n    return this._contentStorage.getMessagesByType(messageType, limit);\n  }\n\n  /**\n   * Get recent messages from storage within the last N minutes\n   * @param minutes - Number of minutes to look back\n   * @returns Messages from the last N minutes\n   */\n  getRecentHistoryByTime(minutes: number): BaseMessage[] {\n    return this._contentStorage.getRecentMessagesByTime(minutes);\n  }\n\n  /**\n   * Export the current state for persistence or analysis\n   * @returns Serializable representation of memory state\n   */\n  exportState(): {\n    config: Required<SmartMemoryConfig>;\n    activeMessages: Array<{ content: unknown; type: string }>;\n    systemPrompt: string;\n    memoryStats: MemoryStats;\n    storageStats: ReturnType<ContentStorage['getStorageStats']>;\n    storedMessages: ReturnType<ContentStorage['exportMessages']>;\n  } {\n    return {\n      config: this.config,\n      activeMessages: this.memoryWindow.getMessages().map((msg) => ({\n        content: msg.content,\n        type: msg._getType(),\n      })),\n      systemPrompt: this.memoryWindow.getSystemPrompt(),\n      memoryStats: this.getMemoryStats(),\n      storageStats: this.getStorageStats(),\n      storedMessages: this._contentStorage.exportMessages(),\n    };\n  }\n\n  /**\n   * Get a summary of conversation context for external use\n   * Useful for providing context to other systems or for logging\n   * @param includeStoredContext - Whether to include recent stored messages\n   * @returns Context summary object\n   */\n  getContextSummary(includeStoredContext: boolean = false): {\n    activeMessageCount: number;\n    systemPrompt: string;\n    recentMessages: BaseMessage[];\n    memoryUtilization: number;\n    hasStoredHistory: boolean;\n    recentStoredMessages?: BaseMessage[];\n    storageStats?: ReturnType<ContentStorage['getStorageStats']>;\n  } {\n    const activeMessages = this.getMessages();\n    const summary = {\n      activeMessageCount: activeMessages.length,\n      systemPrompt: this.getSystemPrompt(),\n      recentMessages: activeMessages.slice(-5),\n      memoryUtilization: this.getMemoryStats().usagePercentage,\n      hasStoredHistory: this.getStorageStats().totalMessages > 0,\n    };\n\n    if (includeStoredContext) {\n      return {\n        ...summary,\n        recentStoredMessages: this.getRecentHistory(10),\n        storageStats: this.getStorageStats(),\n      };\n    }\n\n    return summary;\n  }\n\n  /**\n   * Perform maintenance operations\n   * Optimizes storage and cleans up resources\n   */\n  performMaintenance(): void {}\n\n  /**\n   * Store an entity association for later resolution\n   * @param entityId - The blockchain entity ID\n   * @param entityName - User-provided or derived friendly name\n   * @param entityType - Type of entity (token, account, topic, etc.)\n   * @param transactionId - Optional transaction ID that created this entity\n   */\n  storeEntityAssociation(\n    entityId: string,\n    entityName: string,\n    entityType: string,\n    transactionId?: string\n  ): void {\n    try {\n      if (\n        !entityId ||\n        typeof entityId !== 'string' ||\n        entityId.trim().length === 0\n      ) {\n        return;\n      }\n\n      if (\n        !entityName ||\n        typeof entityName !== 'string' ||\n        entityName.trim().length === 0\n      ) {\n        return;\n      }\n\n      if (\n        !entityType ||\n        typeof entityType !== 'string' ||\n        entityType.trim().length === 0\n      ) {\n        return;\n      }\n\n      const sanitizedEntityId = entityId.trim();\n      const sanitizedEntityName = entityName.trim().substring(0, 100);\n      const sanitizedEntityType = entityType.trim().toLowerCase();\n\n      const association: EntityAssociation & { isEntityAssociation: boolean } =\n        {\n          entityId: sanitizedEntityId,\n          entityName: sanitizedEntityName,\n          entityType: sanitizedEntityType,\n          createdAt: new Date(),\n          isEntityAssociation: true,\n          ...(transactionId !== undefined &&\n          transactionId !== null &&\n          transactionId.trim() !== ''\n            ? { transactionId: transactionId.trim() }\n            : {}),\n        };\n\n      const content = JSON.stringify(association);\n      type LangChainLikeMessage = {\n        _getType: () => string;\n        content: unknown;\n        id: string;\n        name?: string;\n        additional_kwargs?: Record<string, unknown>;\n      };\n\n      const entityMessage: LangChainLikeMessage = {\n        _getType: () => 'system',\n        content: content,\n        id: `entity_${sanitizedEntityId}_${Date.now()}`,\n        name: 'entity_association',\n        additional_kwargs: {\n          entityId: sanitizedEntityId,\n          entityName: sanitizedEntityName,\n          entityType: sanitizedEntityType,\n          isEntityAssociation: true,\n        },\n      };\n\n      this._contentStorage.storeMessages([entityMessage as BaseMessage]);\n    } catch (error) {\n      this.logger.error('Failed to store entity association', {\n        entityId,\n        entityName,\n        entityType,\n        error: error instanceof Error ? error.message : String(error)\n      });\n    }\n  }\n\n  /**\n   * Resolve entity references from natural language queries\n   * @param query - Search query (entity name or natural language reference)\n   * @param options - Resolution options for filtering and fuzzy matching\n   * @returns Array of matching entity associations\n   */\n  resolveEntityReference(\n    query: string,\n    options: EntityResolutionOptions = {}\n  ): EntityAssociation[] {\n    try {\n      if (!query || typeof query !== 'string') {\n        return [];\n      }\n\n      const sanitizedQuery = query.trim();\n      if (sanitizedQuery.length === 0) {\n        return [];\n      }\n\n      if (sanitizedQuery.length > 200) {\n      }\n\n      const { entityType, limit = 10, fuzzyMatch = true } = options;\n\n      const safeLimit = Math.max(1, Math.min(limit || 10, 100));\n\n      const isEntityIdQuery = /^0\\.0\\.\\d+$/.test(sanitizedQuery);\n\n      const searchResults = this._contentStorage.searchMessages(\n        sanitizedQuery.substring(0, 200),\n        {\n          caseSensitive: false,\n          limit: safeLimit * 2,\n        }\n      );\n\n      const associations: EntityAssociation[] = [];\n\n      for (const message of searchResults) {\n        try {\n          const content = message.content as string;\n          if (\n            content.includes(IS_ENTITY_ASSOCIATION_FLAG) ||\n            content.includes('entityId')\n          ) {\n            const parsed = JSON.parse(content);\n            if (parsed.entityId && parsed.entityName && parsed.entityType) {\n              if (entityType && parsed.entityType !== entityType) {\n                continue;\n              }\n\n              if (isEntityIdQuery) {\n                if (parsed.entityId !== sanitizedQuery) {\n                  continue;\n                }\n              }\n\n              associations.push(parsed as EntityAssociation);\n            }\n          }\n        } catch {\n          continue;\n        }\n      }\n\n      if (fuzzyMatch && associations.length === 0 && !isEntityIdQuery) {\n        const fuzzyQueries = [\n          query.toLowerCase(),\n          `token`,\n          `account`,\n          entityType || '',\n        ].filter(Boolean);\n\n        for (const fuzzyQuery of fuzzyQueries) {\n          if (fuzzyQuery === query.toLowerCase()) continue;\n\n          const fuzzyResults = this._contentStorage.searchMessages(fuzzyQuery, {\n            caseSensitive: false,\n            limit: limit,\n          });\n\n          for (const message of fuzzyResults) {\n            try {\n              const content = message.content as string;\n              if (content.includes(IS_ENTITY_ASSOCIATION_FLAG)) {\n                const parsed = JSON.parse(content);\n                if (parsed.entityId && parsed.entityName && parsed.entityType) {\n                  if (entityType && parsed.entityType !== entityType) {\n                    continue;\n                  }\n                  associations.push(parsed as EntityAssociation);\n                }\n              }\n            } catch {\n              continue;\n            }\n          }\n        }\n      }\n\n      const uniqueAssociations = associations\n        .filter(\n          (assoc, index, arr) =>\n            arr.findIndex((a) => a.entityId === assoc.entityId) === index\n        )\n        .sort(\n          (a, b) =>\n            new Date(b.createdAt).getTime() - new Date(a.createdAt).getTime()\n        );\n\n      const results = uniqueAssociations.slice(0, safeLimit);\n\n      return results;\n    } catch (error) {\n      this.logger.error('Failed to resolve entity reference', {\n        query,\n        options,\n        error: error instanceof Error ? error.message : String(error)\n      });\n      return [];\n    }\n  }\n\n  /**\n   * Get all entity associations, optionally filtered by type\n   * @param entityType - Optional filter by entity type\n   * @returns Array of entity associations\n   */\n  getEntityAssociations(entityType?: string): EntityAssociation[] {\n    try {\n      const sanitizedEntityType = entityType\n        ? entityType.trim().toLowerCase()\n        : undefined;\n\n      if (\n        entityType &&\n        (!sanitizedEntityType || sanitizedEntityType.length === 0)\n      ) {\n        return [];\n      }\n\n      const SEARCH_ANY_ENTITY = 'entityId';\n      const searchQuery = sanitizedEntityType || SEARCH_ANY_ENTITY;\n      const searchResults = this._contentStorage.searchMessages(searchQuery, {\n        caseSensitive: false,\n        limit: 100,\n      });\n\n      const associations: EntityAssociation[] = [];\n\n      for (const message of searchResults) {\n        try {\n          const content = message.content as string;\n          if (content.includes(IS_ENTITY_ASSOCIATION_FLAG)) {\n            const parsed = JSON.parse(content);\n\n            if (parsed.entityId && parsed.entityName && parsed.entityType) {\n              if (\n                sanitizedEntityType &&\n                parsed.entityType !== sanitizedEntityType\n              ) {\n                continue;\n              }\n\n              if (parsed.createdAt && typeof parsed.createdAt === 'string') {\n                parsed.createdAt = new Date(parsed.createdAt);\n              }\n\n              associations.push(parsed as EntityAssociation);\n            }\n          }\n        } catch (parseError) {\n          this.logger.warn('Failed to parse entity association from message', {\n            messageContent: typeof message.content === 'string' ? message.content.substring(0, 100) : 'non-string',\n            error: parseError instanceof Error ? parseError.message : String(parseError)\n          });\n          continue;\n        }\n      }\n\n      const results = associations\n        .filter(\n          (assoc, index, arr) =>\n            arr.findIndex((a) => a.entityId === assoc.entityId) === index\n        )\n        .sort((a, b): number => {\n          const getTime = (d: Date | string): number =>\n            d instanceof Date ? d.getTime() : new Date(d).getTime();\n          const aTime = getTime(a.createdAt);\n          const bTime = getTime(b.createdAt);\n          return bTime - aTime;\n        });\n\n      return results;\n    } catch (error) {\n      this.logger.error('Failed to get entity associations', {\n        entityType,\n        error: error instanceof Error ? error.message : String(error)\n      });\n      return [];\n    }\n  }\n\n  /**\n   * Clean up resources and dispose of components\n   */\n  dispose(): void {\n    this.memoryWindow.dispose();\n    this._contentStorage.dispose();\n    this.tokenCounter.dispose();\n  }\n}\n"],"names":[],"mappings":";;;;AA8EA,MAAM,6BAA6B;AAQ5B,MAAM,sBAAN,MAAM,oBAAmB;AAAA,EAc9B,YAAY,SAA4B,IAAI;AAC1C,SAAK,SAAS,EAAE,GAAG,oBAAmB,gBAAgB,GAAG,OAAA;AACzD,SAAK,SAAS,IAAI,OAAO,EAAE,QAAQ,sBAAsB;AAEzD,SAAK,eAAe,IAAI,aAAa,KAAK,OAAO,SAAS;AAC1D,SAAK,kBAAkB,IAAI,eAAe,KAAK,OAAO,YAAY;AAClE,SAAK,eAAe,IAAI;AAAA,MACtB,KAAK,OAAO;AAAA,MACZ,KAAK,OAAO;AAAA,MACZ,KAAK;AAAA,IAAA;AAAA,EAET;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,IAAI,iBAAiC;AACnC,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,WAAW,SAA4B;AACrC,UAAM,SAAS,KAAK,aAAa,WAAW,OAAO;AAEnD,QAAI,OAAO,eAAe,SAAS,GAAG;AACpC,WAAK,gBAAgB,cAAc,OAAO,cAAc;AAAA,IAC1D;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,cAA6B;AAC3B,WAAO,KAAK,aAAa,YAAA;AAAA,EAC3B;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,eAAwB,OAAa;AACzC,SAAK,aAAa,MAAA;AAElB,QAAI,cAAc;AAChB,WAAK,gBAAgB,MAAA;AAAA,IACvB;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,gBAAgB,cAA4B;AAC1C,SAAK,aAAa,gBAAgB,YAAY;AAAA,EAChD;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,kBAA0B;AACxB,WAAO,KAAK,aAAa,gBAAA;AAAA,EAC3B;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,cAAc,OAAe,UAAyB,IAAmB;AACvE,WAAO,KAAK,gBAAgB,eAAe,OAAO,OAAO;AAAA,EAC3D;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,iBAAiB,OAA8B;AAC7C,WAAO,KAAK,gBAAgB,kBAAkB,KAAK;AAAA,EACrD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,cAAc,SAA+B;AAC3C,WAAO,KAAK,aAAa,cAAc,OAAO;AAAA,EAChD;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,iBAA8B;AAC5B,UAAM,cAAc,KAAK,aAAa,SAAA;AAEtC,WAAO;AAAA,MACL,qBAAqB,YAAY;AAAA,MACjC,mBAAmB,YAAY;AAAA,MAC/B,WAAW,YAAY;AAAA,MACvB,mBAAmB,YAAY;AAAA,MAC/B,oBAAoB,YAAY;AAAA,MAChC,iBAAiB,YAAY;AAAA,IAAA;AAAA,EAEjC;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,kBAAiE;AAC/D,WAAO,KAAK,gBAAgB,gBAAA;AAAA,EAC9B;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,kBAME;AACA,UAAM,cAAc,KAAK,eAAA;AACzB,UAAM,eAAe,KAAK,gBAAA;AAE1B,WAAO;AAAA,MACL,cAAc;AAAA,MACd,SAAS;AAAA,MACT,sBACE,YAAY,sBAAsB,aAAa;AAAA,MACjD,yBAAyB,YAAY;AAAA,MACrC,oBAAoB,aAAa;AAAA,IAAA;AAAA,EAErC;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,aAAa,WAA6C;AACxD,SAAK,SAAS,EAAE,GAAG,KAAK,QAAQ,GAAG,UAAA;AAEnC,QACE,UAAU,cAAc,UACxB,UAAU,kBAAkB,QAC5B;AACA,WAAK,aAAa;AAAA,QAChB,KAAK,OAAO;AAAA,QACZ,KAAK,OAAO;AAAA,MAAA;AAAA,IAEhB;AAEA,QAAI,UAAU,iBAAiB,QAAW;AACxC,WAAK,gBAAgB,mBAAmB,KAAK,OAAO,YAAY;AAAA,IAClE;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,YAAyC;AACvC,WAAO,EAAE,GAAG,KAAK,OAAA;AAAA,EACnB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,wBAAwB,WAAiB,SAA8B;AACrE,WAAO,KAAK,gBAAgB,yBAAyB,WAAW,OAAO;AAAA,EACzE;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,iBAAiB,aAAqB,OAA+B;AACnE,WAAO,KAAK,gBAAgB,kBAAkB,aAAa,KAAK;AAAA,EAClE;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,uBAAuB,SAAgC;AACrD,WAAO,KAAK,gBAAgB,wBAAwB,OAAO;AAAA,EAC7D;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,cAOE;AACA,WAAO;AAAA,MACL,QAAQ,KAAK;AAAA,MACb,gBAAgB,KAAK,aAAa,cAAc,IAAI,CAAC,SAAS;AAAA,QAC5D,SAAS,IAAI;AAAA,QACb,MAAM,IAAI,SAAA;AAAA,MAAS,EACnB;AAAA,MACF,cAAc,KAAK,aAAa,gBAAA;AAAA,MAChC,aAAa,KAAK,eAAA;AAAA,MAClB,cAAc,KAAK,gBAAA;AAAA,MACnB,gBAAgB,KAAK,gBAAgB,eAAA;AAAA,IAAe;AAAA,EAExD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,kBAAkB,uBAAgC,OAQhD;AACA,UAAM,iBAAiB,KAAK,YAAA;AAC5B,UAAM,UAAU;AAAA,MACd,oBAAoB,eAAe;AAAA,MACnC,cAAc,KAAK,gBAAA;AAAA,MACnB,gBAAgB,eAAe,MAAM,EAAE;AAAA,MACvC,mBAAmB,KAAK,eAAA,EAAiB;AAAA,MACzC,kBAAkB,KAAK,gBAAA,EAAkB,gBAAgB;AAAA,IAAA;AAG3D,QAAI,sBAAsB;AACxB,aAAO;AAAA,QACL,GAAG;AAAA,QACH,sBAAsB,KAAK,iBAAiB,EAAE;AAAA,QAC9C,cAAc,KAAK,gBAAA;AAAA,MAAgB;AAAA,IAEvC;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,qBAA2B;AAAA,EAAC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAS5B,uBACE,UACA,YACA,YACA,eACM;AACN,QAAI;AACF,UACE,CAAC,YACD,OAAO,aAAa,YACpB,SAAS,KAAA,EAAO,WAAW,GAC3B;AACA;AAAA,MACF;AAEA,UACE,CAAC,cACD,OAAO,eAAe,YACtB,WAAW,KAAA,EAAO,WAAW,GAC7B;AACA;AAAA,MACF;AAEA,UACE,CAAC,cACD,OAAO,eAAe,YACtB,WAAW,KAAA,EAAO,WAAW,GAC7B;AACA;AAAA,MACF;AAEA,YAAM,oBAAoB,SAAS,KAAA;AACnC,YAAM,sBAAsB,WAAW,KAAA,EAAO,UAAU,GAAG,GAAG;AAC9D,YAAM,sBAAsB,WAAW,KAAA,EAAO,YAAA;AAE9C,YAAM,cACJ;AAAA,QACE,UAAU;AAAA,QACV,YAAY;AAAA,QACZ,YAAY;AAAA,QACZ,+BAAe,KAAA;AAAA,QACf,qBAAqB;AAAA,QACrB,GAAI,kBAAkB,UACtB,kBAAkB,QAClB,cAAc,KAAA,MAAW,KACrB,EAAE,eAAe,cAAc,KAAA,EAAK,IACpC,CAAA;AAAA,MAAC;AAGT,YAAM,UAAU,KAAK,UAAU,WAAW;AAS1C,YAAM,gBAAsC;AAAA,QAC1C,UAAU,MAAM;AAAA,QAChB;AAAA,QACA,IAAI,UAAU,iBAAiB,IAAI,KAAK,KAAK;AAAA,QAC7C,MAAM;AAAA,QACN,mBAAmB;AAAA,UACjB,UAAU;AAAA,UACV,YAAY;AAAA,UACZ,YAAY;AAAA,UACZ,qBAAqB;AAAA,QAAA;AAAA,MACvB;AAGF,WAAK,gBAAgB,cAAc,CAAC,aAA4B,CAAC;AAAA,IACnE,SAAS,OAAO;AACd,WAAK,OAAO,MAAM,sCAAsC;AAAA,QACtD;AAAA,QACA;AAAA,QACA;AAAA,QACA,OAAO,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;AAAA,MAAA,CAC7D;AAAA,IACH;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,uBACE,OACA,UAAmC,IACd;AACrB,QAAI;AACF,UAAI,CAAC,SAAS,OAAO,UAAU,UAAU;AACvC,eAAO,CAAA;AAAA,MACT;AAEA,YAAM,iBAAiB,MAAM,KAAA;AAC7B,UAAI,eAAe,WAAW,GAAG;AAC/B,eAAO,CAAA;AAAA,MACT;AAEA,UAAI,eAAe,SAAS,KAAK;AAAA,MACjC;AAEA,YAAM,EAAE,YAAY,QAAQ,IAAI,aAAa,SAAS;AAEtD,YAAM,YAAY,KAAK,IAAI,GAAG,KAAK,IAAI,SAAS,IAAI,GAAG,CAAC;AAExD,YAAM,kBAAkB,cAAc,KAAK,cAAc;AAEzD,YAAM,gBAAgB,KAAK,gBAAgB;AAAA,QACzC,eAAe,UAAU,GAAG,GAAG;AAAA,QAC/B;AAAA,UACE,eAAe;AAAA,UACf,OAAO,YAAY;AAAA,QAAA;AAAA,MACrB;AAGF,YAAM,eAAoC,CAAA;AAE1C,iBAAW,WAAW,eAAe;AACnC,YAAI;AACF,gBAAM,UAAU,QAAQ;AACxB,cACE,QAAQ,SAAS,0BAA0B,KAC3C,QAAQ,SAAS,UAAU,GAC3B;AACA,kBAAM,SAAS,KAAK,MAAM,OAAO;AACjC,gBAAI,OAAO,YAAY,OAAO,cAAc,OAAO,YAAY;AAC7D,kBAAI,cAAc,OAAO,eAAe,YAAY;AAClD;AAAA,cACF;AAEA,kBAAI,iBAAiB;AACnB,oBAAI,OAAO,aAAa,gBAAgB;AACtC;AAAA,gBACF;AAAA,cACF;AAEA,2BAAa,KAAK,MAA2B;AAAA,YAC/C;AAAA,UACF;AAAA,QACF,QAAQ;AACN;AAAA,QACF;AAAA,MACF;AAEA,UAAI,cAAc,aAAa,WAAW,KAAK,CAAC,iBAAiB;AAC/D,cAAM,eAAe;AAAA,UACnB,MAAM,YAAA;AAAA,UACN;AAAA,UACA;AAAA,UACA,cAAc;AAAA,QAAA,EACd,OAAO,OAAO;AAEhB,mBAAW,cAAc,cAAc;AACrC,cAAI,eAAe,MAAM,cAAe;AAExC,gBAAM,eAAe,KAAK,gBAAgB,eAAe,YAAY;AAAA,YACnE,eAAe;AAAA,YACf;AAAA,UAAA,CACD;AAED,qBAAW,WAAW,cAAc;AAClC,gBAAI;AACF,oBAAM,UAAU,QAAQ;AACxB,kBAAI,QAAQ,SAAS,0BAA0B,GAAG;AAChD,sBAAM,SAAS,KAAK,MAAM,OAAO;AACjC,oBAAI,OAAO,YAAY,OAAO,cAAc,OAAO,YAAY;AAC7D,sBAAI,cAAc,OAAO,eAAe,YAAY;AAClD;AAAA,kBACF;AACA,+BAAa,KAAK,MAA2B;AAAA,gBAC/C;AAAA,cACF;AAAA,YACF,QAAQ;AACN;AAAA,YACF;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAEA,YAAM,qBAAqB,aACxB;AAAA,QACC,CAAC,OAAO,OAAO,QACb,IAAI,UAAU,CAAC,MAAM,EAAE,aAAa,MAAM,QAAQ,MAAM;AAAA,MAAA,EAE3D;AAAA,QACC,CAAC,GAAG,MACF,IAAI,KAAK,EAAE,SAAS,EAAE,QAAA,IAAY,IAAI,KAAK,EAAE,SAAS,EAAE,QAAA;AAAA,MAAQ;AAGtE,YAAM,UAAU,mBAAmB,MAAM,GAAG,SAAS;AAErD,aAAO;AAAA,IACT,SAAS,OAAO;AACd,WAAK,OAAO,MAAM,sCAAsC;AAAA,QACtD;AAAA,QACA;AAAA,QACA,OAAO,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;AAAA,MAAA,CAC7D;AACD,aAAO,CAAA;AAAA,IACT;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,sBAAsB,YAA0C;AAC9D,QAAI;AACF,YAAM,sBAAsB,aACxB,WAAW,KAAA,EAAO,gBAClB;AAEJ,UACE,eACC,CAAC,uBAAuB,oBAAoB,WAAW,IACxD;AACA,eAAO,CAAA;AAAA,MACT;AAEA,YAAM,oBAAoB;AAC1B,YAAM,cAAc,uBAAuB;AAC3C,YAAM,gBAAgB,KAAK,gBAAgB,eAAe,aAAa;AAAA,QACrE,eAAe;AAAA,QACf,OAAO;AAAA,MAAA,CACR;AAED,YAAM,eAAoC,CAAA;AAE1C,iBAAW,WAAW,eAAe;AACnC,YAAI;AACF,gBAAM,UAAU,QAAQ;AACxB,cAAI,QAAQ,SAAS,0BAA0B,GAAG;AAChD,kBAAM,SAAS,KAAK,MAAM,OAAO;AAEjC,gBAAI,OAAO,YAAY,OAAO,cAAc,OAAO,YAAY;AAC7D,kBACE,uBACA,OAAO,eAAe,qBACtB;AACA;AAAA,cACF;AAEA,kBAAI,OAAO,aAAa,OAAO,OAAO,cAAc,UAAU;AAC5D,uBAAO,YAAY,IAAI,KAAK,OAAO,SAAS;AAAA,cAC9C;AAEA,2BAAa,KAAK,MAA2B;AAAA,YAC/C;AAAA,UACF;AAAA,QACF,SAAS,YAAY;AACnB,eAAK,OAAO,KAAK,mDAAmD;AAAA,YAClE,gBAAgB,OAAO,QAAQ,YAAY,WAAW,QAAQ,QAAQ,UAAU,GAAG,GAAG,IAAI;AAAA,YAC1F,OAAO,sBAAsB,QAAQ,WAAW,UAAU,OAAO,UAAU;AAAA,UAAA,CAC5E;AACD;AAAA,QACF;AAAA,MACF;AAEA,YAAM,UAAU,aACb;AAAA,QACC,CAAC,OAAO,OAAO,QACb,IAAI,UAAU,CAAC,MAAM,EAAE,aAAa,MAAM,QAAQ,MAAM;AAAA,MAAA,EAE3D,KAAK,CAAC,GAAG,MAAc;AACtB,cAAM,UAAU,CAAC,MACf,aAAa,OAAO,EAAE,QAAA,IAAY,IAAI,KAAK,CAAC,EAAE,QAAA;AAChD,cAAM,QAAQ,QAAQ,EAAE,SAAS;AACjC,cAAM,QAAQ,QAAQ,EAAE,SAAS;AACjC,eAAO,QAAQ;AAAA,MACjB,CAAC;AAEH,aAAO;AAAA,IACT,SAAS,OAAO;AACd,WAAK,OAAO,MAAM,qCAAqC;AAAA,QACrD;AAAA,QACA,OAAO,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;AAAA,MAAA,CAC7D;AACD,aAAO,CAAA;AAAA,IACT;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,UAAgB;AACd,SAAK,aAAa,QAAA;AAClB,SAAK,gBAAgB,QAAA;AACrB,SAAK,aAAa,QAAA;AAAA,EACpB;AACF;AApkBE,oBAAwB,iBAA8C;AAAA,EACpE,WAAW;AAAA,EACX,eAAe;AAAA,EACf,WAAW;AAAA,EACX,cAAc;AAAA;AAXX,IAAM,qBAAN;"}
+{"version":3,"file":"index15.js","sources":["../../src/mcp/mcp-client-manager.ts"],"sourcesContent":["import { Client } from '@modelcontextprotocol/sdk/client/index.js';\nimport { StdioClientTransport } from '@modelcontextprotocol/sdk/client/stdio.js';\nimport type { MCPServerConfig, MCPToolInfo, MCPConnectionStatus } from './types';\nimport { Logger } from '@hashgraphonline/standards-sdk';\nimport type { ContentStorage } from '../memory/content-storage';\nimport { MCPContentProcessor } from './content-processor';\n\n/**\n * Manages connections to MCP servers and tool discovery\n */\nexport class MCPClientManager {\n  private clients: Map<string, Client> = new Map();\n  private tools: Map<string, MCPToolInfo[]> = new Map();\n  private logger: Logger;\n  private contentProcessor?: MCPContentProcessor;\n\n  constructor(logger: Logger, contentStorage?: ContentStorage) {\n    this.logger = logger;\n    if (contentStorage) {\n      this.contentProcessor = new MCPContentProcessor(contentStorage, logger);\n    }\n  }\n\n  /**\n   * Connect to an MCP server and discover its tools\n   */\n  async connectServer(config: MCPServerConfig): Promise<MCPConnectionStatus> {\n    try {\n      if (this.isServerConnected(config.name)) {\n        return {\n          serverName: config.name,\n          connected: false,\n          error: `Server ${config.name} is already connected`,\n          tools: [],\n        };\n      }\n\n      if (config.transport && config.transport !== 'stdio') {\n        throw new Error(`Transport ${config.transport} not yet supported`);\n      }\n\n      const transport = new StdioClientTransport({\n        command: config.command,\n        args: config.args,\n        ...(config.env && { env: config.env }),\n      });\n\n      const client = new Client({\n        name: `conversational-agent-${config.name}`,\n        version: '1.0.0',\n      }, {\n        capabilities: {},\n      });\n\n      await client.connect(transport);\n      this.clients.set(config.name, client);\n\n      const toolsResponse = await client.listTools();\n      const toolsWithServer: MCPToolInfo[] = toolsResponse.tools.map(tool => ({\n        ...tool,\n        serverName: config.name,\n      }));\n\n      this.tools.set(config.name, toolsWithServer);\n      this.logger.info(`Connected to MCP server ${config.name} with ${toolsWithServer.length} tools`);\n\n      return {\n        serverName: config.name,\n        connected: true,\n        tools: toolsWithServer,\n      };\n    } catch (error) {\n      this.logger.error(`Failed to connect to MCP server ${config.name}:`, error);\n      return {\n        serverName: config.name,\n        connected: false,\n        error: error instanceof Error ? error.message : 'Unknown error',\n        tools: [],\n      };\n    }\n  }\n\n  /**\n   * Execute a tool on a specific MCP server\n   */\n  async executeTool(serverName: string, toolName: string, args: Record<string, unknown>): Promise<unknown> {\n    const client = this.clients.get(serverName);\n    if (!client) {\n      throw new Error(`MCP server ${serverName} not connected`);\n    }\n\n    this.logger.debug(`Executing MCP tool ${toolName} on server ${serverName}`, args);\n\n    try {\n      const result = await client.callTool({\n        name: toolName,\n        arguments: args,\n      });\n\n      if (this.contentProcessor) {\n        const processed = await this.contentProcessor.processResponse(result, serverName, toolName);\n        \n        if (processed.wasProcessed) {\n          this.logger.debug(\n            `Processed MCP response from ${serverName}::${toolName}`,\n            {\n              referenceCreated: processed.referenceCreated,\n              originalSize: processed.originalSize,\n              errors: processed.errors\n            }\n          );\n\n          if (processed.errors && processed.errors.length > 0) {\n            this.logger.warn(`Content processing warnings for ${serverName}::${toolName}:`, processed.errors);\n          }\n        }\n\n        return processed.content;\n      }\n\n      return result;\n    } catch (error) {\n      this.logger.error(`Error executing MCP tool ${toolName}:`, error);\n      throw error;\n    }\n  }\n\n  /**\n   * Disconnect all MCP servers\n   */\n  async disconnectAll(): Promise<void> {\n    for (const [name, client] of this.clients) {\n      try {\n        await client.close();\n        this.logger.info(`Disconnected from MCP server ${name}`);\n      } catch (error) {\n        this.logger.error(`Error disconnecting MCP server ${name}:`, error);\n      }\n    }\n    this.clients.clear();\n    this.tools.clear();\n  }\n\n  /**\n   * Get all discovered tools from all connected servers\n   */\n  getAllTools(): MCPToolInfo[] {\n    const allTools: MCPToolInfo[] = [];\n    for (const tools of this.tools.values()) {\n      allTools.push(...tools);\n    }\n    return allTools;\n  }\n\n  /**\n   * Get tools from a specific server\n   */\n  getServerTools(serverName: string): MCPToolInfo[] {\n    return this.tools.get(serverName) || [];\n  }\n\n  /**\n   * Check if a server is connected\n   */\n  isServerConnected(serverName: string): boolean {\n    return this.clients.has(serverName);\n  }\n\n  /**\n   * Get list of connected server names\n   */\n  getConnectedServers(): string[] {\n    return Array.from(this.clients.keys());\n  }\n\n  /**\n   * Enable content processing with content storage\n   */\n  enableContentProcessing(contentStorage: ContentStorage): void {\n    this.contentProcessor = new MCPContentProcessor(contentStorage, this.logger);\n    this.logger.info('Content processing enabled for MCP responses');\n  }\n\n  /**\n   * Disable content processing\n   */\n  disableContentProcessing(): void {\n    delete this.contentProcessor;\n    this.logger.info('Content processing disabled for MCP responses');\n  }\n\n  /**\n   * Check if content processing is enabled\n   */\n  isContentProcessingEnabled(): boolean {\n    return this.contentProcessor !== undefined;\n  }\n\n  /**\n   * Analyze a response without processing it (for testing/debugging)\n   */\n  analyzeResponseContent(response: unknown): unknown {\n    if (!this.contentProcessor) {\n      throw new Error('Content processing is not enabled');\n    }\n    return this.contentProcessor.analyzeResponse(response);\n  }\n}"],"names":[],"mappings":";;;AAUO,MAAM,iBAAiB;AAAA,EAM5B,YAAY,QAAgB,gBAAiC;AAL7D,SAAQ,8BAAmC,IAAA;AAC3C,SAAQ,4BAAwC,IAAA;AAK9C,SAAK,SAAS;AACd,QAAI,gBAAgB;AAClB,WAAK,mBAAmB,IAAI,oBAAoB,gBAAgB,MAAM;AAAA,IACxE;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,cAAc,QAAuD;AACzE,QAAI;AACF,UAAI,KAAK,kBAAkB,OAAO,IAAI,GAAG;AACvC,eAAO;AAAA,UACL,YAAY,OAAO;AAAA,UACnB,WAAW;AAAA,UACX,OAAO,UAAU,OAAO,IAAI;AAAA,UAC5B,OAAO,CAAA;AAAA,QAAC;AAAA,MAEZ;AAEA,UAAI,OAAO,aAAa,OAAO,cAAc,SAAS;AACpD,cAAM,IAAI,MAAM,aAAa,OAAO,SAAS,oBAAoB;AAAA,MACnE;AAEA,YAAM,YAAY,IAAI,qBAAqB;AAAA,QACzC,SAAS,OAAO;AAAA,QAChB,MAAM,OAAO;AAAA,QACb,GAAI,OAAO,OAAO,EAAE,KAAK,OAAO,IAAA;AAAA,MAAI,CACrC;AAED,YAAM,SAAS,IAAI,OAAO;AAAA,QACxB,MAAM,wBAAwB,OAAO,IAAI;AAAA,QACzC,SAAS;AAAA,MAAA,GACR;AAAA,QACD,cAAc,CAAA;AAAA,MAAC,CAChB;AAED,YAAM,OAAO,QAAQ,SAAS;AAC9B,WAAK,QAAQ,IAAI,OAAO,MAAM,MAAM;AAEpC,YAAM,gBAAgB,MAAM,OAAO,UAAA;AACnC,YAAM,kBAAiC,cAAc,MAAM,IAAI,CAAA,UAAS;AAAA,QACtE,GAAG;AAAA,QACH,YAAY,OAAO;AAAA,MAAA,EACnB;AAEF,WAAK,MAAM,IAAI,OAAO,MAAM,eAAe;AAC3C,WAAK,OAAO,KAAK,2BAA2B,OAAO,IAAI,SAAS,gBAAgB,MAAM,QAAQ;AAE9F,aAAO;AAAA,QACL,YAAY,OAAO;AAAA,QACnB,WAAW;AAAA,QACX,OAAO;AAAA,MAAA;AAAA,IAEX,SAAS,OAAO;AACd,WAAK,OAAO,MAAM,mCAAmC,OAAO,IAAI,KAAK,KAAK;AAC1E,aAAO;AAAA,QACL,YAAY,OAAO;AAAA,QACnB,WAAW;AAAA,QACX,OAAO,iBAAiB,QAAQ,MAAM,UAAU;AAAA,QAChD,OAAO,CAAA;AAAA,MAAC;AAAA,IAEZ;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,YAAY,YAAoB,UAAkB,MAAiD;AACvG,UAAM,SAAS,KAAK,QAAQ,IAAI,UAAU;AAC1C,QAAI,CAAC,QAAQ;AACX,YAAM,IAAI,MAAM,cAAc,UAAU,gBAAgB;AAAA,IAC1D;AAEA,SAAK,OAAO,MAAM,sBAAsB,QAAQ,cAAc,UAAU,IAAI,IAAI;AAEhF,QAAI;AACF,YAAM,SAAS,MAAM,OAAO,SAAS;AAAA,QACnC,MAAM;AAAA,QACN,WAAW;AAAA,MAAA,CACZ;AAED,UAAI,KAAK,kBAAkB;AACzB,cAAM,YAAY,MAAM,KAAK,iBAAiB,gBAAgB,QAAQ,YAAY,QAAQ;AAE1F,YAAI,UAAU,cAAc;AAC1B,eAAK,OAAO;AAAA,YACV,+BAA+B,UAAU,KAAK,QAAQ;AAAA,YACtD;AAAA,cACE,kBAAkB,UAAU;AAAA,cAC5B,cAAc,UAAU;AAAA,cACxB,QAAQ,UAAU;AAAA,YAAA;AAAA,UACpB;AAGF,cAAI,UAAU,UAAU,UAAU,OAAO,SAAS,GAAG;AACnD,iBAAK,OAAO,KAAK,mCAAmC,UAAU,KAAK,QAAQ,KAAK,UAAU,MAAM;AAAA,UAClG;AAAA,QACF;AAEA,eAAO,UAAU;AAAA,MACnB;AAEA,aAAO;AAAA,IACT,SAAS,OAAO;AACd,WAAK,OAAO,MAAM,4BAA4B,QAAQ,KAAK,KAAK;AAChE,YAAM;AAAA,IACR;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,gBAA+B;AACnC,eAAW,CAAC,MAAM,MAAM,KAAK,KAAK,SAAS;AACzC,UAAI;AACF,cAAM,OAAO,MAAA;AACb,aAAK,OAAO,KAAK,gCAAgC,IAAI,EAAE;AAAA,MACzD,SAAS,OAAO;AACd,aAAK,OAAO,MAAM,kCAAkC,IAAI,KAAK,KAAK;AAAA,MACpE;AAAA,IACF;AACA,SAAK,QAAQ,MAAA;AACb,SAAK,MAAM,MAAA;AAAA,EACb;AAAA;AAAA;AAAA;AAAA,EAKA,cAA6B;AAC3B,UAAM,WAA0B,CAAA;AAChC,eAAW,SAAS,KAAK,MAAM,OAAA,GAAU;AACvC,eAAS,KAAK,GAAG,KAAK;AAAA,IACxB;AACA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,eAAe,YAAmC;AAChD,WAAO,KAAK,MAAM,IAAI,UAAU,KAAK,CAAA;AAAA,EACvC;AAAA;AAAA;AAAA;AAAA,EAKA,kBAAkB,YAA6B;AAC7C,WAAO,KAAK,QAAQ,IAAI,UAAU;AAAA,EACpC;AAAA;AAAA;AAAA;AAAA,EAKA,sBAAgC;AAC9B,WAAO,MAAM,KAAK,KAAK,QAAQ,MAAM;AAAA,EACvC;AAAA;AAAA;AAAA;AAAA,EAKA,wBAAwB,gBAAsC;AAC5D,SAAK,mBAAmB,IAAI,oBAAoB,gBAAgB,KAAK,MAAM;AAC3E,SAAK,OAAO,KAAK,8CAA8C;AAAA,EACjE;AAAA;AAAA;AAAA;AAAA,EAKA,2BAAiC;AAC/B,WAAO,KAAK;AACZ,SAAK,OAAO,KAAK,+CAA+C;AAAA,EAClE;AAAA;AAAA;AAAA;AAAA,EAKA,6BAAsC;AACpC,WAAO,KAAK,qBAAqB;AAAA,EACnC;AAAA;AAAA;AAAA;AAAA,EAKA,uBAAuB,UAA4B;AACjD,QAAI,CAAC,KAAK,kBAAkB;AAC1B,YAAM,IAAI,MAAM,mCAAmC;AAAA,IACrD;AACA,WAAO,KAAK,iBAAiB,gBAAgB,QAAQ;AAAA,EACvD;AACF;"}

package/dist/esm/index16.js

@@ -1,138 +1,256 @@
-import { StructuredTool } from "@langchain/core/tools";
-import { z } from "zod";
-import { ChatOpenAI } from "@langchain/openai";
-import { Logger } from "@hashgraphonline/standards-sdk";
-const logger = new Logger({ module: "EntityResolverTool" });
-const ResolveEntitiesSchema = z.object({
-  message: z.string().describe("The message containing entity references to resolve"),
-  entities: z.array(z.object({
-    entityId: z.string(),
-    entityName: z.string(),
-    entityType: z.string()
-  })).describe("Available entities in memory")
-});
-const ExtractEntitiesSchema = z.object({
-  response: z.string().describe("Agent response text to extract entities from"),
-  userMessage: z.string().describe("Original user message for context")
-});
-class ResolveEntitiesTool extends StructuredTool {
-  constructor(apiKey, modelName = "gpt-4o-mini") {
-    super();
-    this.name = "resolve_entities";
-    this.description = 'Resolves entity references like "the topic", "it", "that" to actual entity IDs';
-    this.schema = ResolveEntitiesSchema;
-    this.llm = new ChatOpenAI({
-      apiKey,
-      modelName,
-      temperature: 0
-    });
+class MCPContentProcessor {
+  constructor(contentStorage, logger) {
+    this.contentStorage = contentStorage;
+    this.logger = logger;
   }
-  async _call(input) {
-    const { message, entities } = input;
-    if (!entities || entities.length === 0) {
-      return message;
-    }
-    const byType = this.groupEntitiesByType(entities);
-    const context = this.buildEntityContext(byType);
-    const prompt = `Task: Replace entity references with IDs.
-
-${context}
-
-Message: "${message}"
-
-Rules:
-- "the topic" or "that topic" → replace with most recent topic ID
-- "the token" or "that token" → replace with most recent token ID
-- "it" or "that" after action verb → replace with most recent entity ID
-- "airdrop X" without token ID → add most recent token ID as first parameter
-- Token operations without explicit token → use most recent token ID
-
-Examples:
-- "submit on the topic" → "submit on 0.0.6543472"
-- "airdrop the token" → "airdrop 0.0.123456"
-- "airdrop 10 to 0.0.5842697" → "airdrop 0.0.123456 10 to 0.0.5842697"
-- "mint 100" → "mint 0.0.123456 100"
-
-Return ONLY the resolved message:`;
+  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 response = await this.llm.invoke(prompt);
-      return response.content.trim();
+      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) {
-      logger.error("ResolveEntitiesTool failed:", error);
-      return message;
+      this.logger.error("Error processing MCP response:", error);
+      return {
+        content: response,
+        wasProcessed: false,
+        errors: [
+          error instanceof Error ? error.message : "Unknown processing error"
+        ]
+      };
     }
   }
-  groupEntitiesByType(entities) {
-    return entities.reduce((acc, entity) => {
-      if (!acc[entity.entityType]) {
-        acc[entity.entityType] = [];
+  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)
+        });
       }
-      acc[entity.entityType].push(entity);
-      return acc;
-    }, {});
+    }
   }
-  buildEntityContext(groupedEntities) {
-    let context = "Available entities:\n";
-    for (const [type, list] of Object.entries(groupedEntities)) {
-      const recent = list[0];
-      context += `- Most recent ${type}: "${recent.entityName}" = ${recent.entityId}
-`;
-    }
-    return context;
+  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;
   }
-}
-class ExtractEntitiesTool extends StructuredTool {
-  constructor(apiKey, modelName = "gpt-4o-mini") {
-    super();
-    this.name = "extract_entities";
-    this.description = "Extracts newly created entities from agent responses";
-    this.schema = ExtractEntitiesSchema;
-    this.llm = new ChatOpenAI({
-      apiKey,
-      modelName,
-      temperature: 0
-    });
+  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
+    };
   }
-  async _call(input) {
-    const { response, userMessage } = input;
-    const prompt = `Extract ONLY newly created entities from this response.
-
-User asked: "${userMessage.substring(0, 200)}"
-Response: ${response.substring(0, 3e3)}
-
-Look for:
-- Success messages with new entity IDs
-- Words like "created", "new", "successfully" followed by entity IDs
-
-Return JSON array of created entities:
-[{"id": "0.0.XXX", "name": "name", "type": "topic|token|account"}]
-
-If none created, return: []
-
-JSON:`;
-    try {
-      const llmResponse = await this.llm.invoke(prompt);
-      const content = llmResponse.content;
-      const match = content.match(/\[[\s\S]*?\]/);
-      if (match) {
-        return match[0];
+  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];
+        }
+        if (typeof newContent === "object" && newContent !== null) {
+          const newContentRecord = newContent;
+          for (const key of Object.keys(newContentRecord)) {
+            record[key] = newContentRecord[key];
+          }
+        }
+        return;
+      }
+      for (const key in record) {
+        if (record[key] === oldContent) {
+          record[key] = newContent;
+        } else {
+          this.replaceContentInResponse(record[key], oldContent, newContent);
+        }
       }
-      return "[]";
-    } catch (error) {
-      logger.error("ExtractEntitiesTool failed:", error);
-      return "[]";
     }
   }
-}
-function createEntityTools(apiKey, modelName = "gpt-4o-mini") {
-  return {
-    resolveEntities: new ResolveEntitiesTool(apiKey, modelName),
-    extractEntities: new ExtractEntitiesTool(apiKey, modelName)
-  };
+  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 {
-  ExtractEntitiesTool,
-  ResolveEntitiesTool,
-  createEntityTools
+  MCPContentProcessor
 };
 //# sourceMappingURL=index16.js.map

package/dist/esm/index16.js.map

@@ -1 +1 @@
-{"version":3,"file":"index16.js","sources":["../../src/tools/EntityResolverTool.ts"],"sourcesContent":["import { StructuredTool } from '@langchain/core/tools';\nimport { z } from 'zod';\nimport { ChatOpenAI } from '@langchain/openai';\nimport { Logger } from '@hashgraphonline/standards-sdk';\n\nconst logger = new Logger({ module: 'EntityResolverTool' });\n\nconst ResolveEntitiesSchema = z.object({\n  message: z.string().describe('The message containing entity references to resolve'),\n  entities: z.array(z.object({\n    entityId: z.string(),\n    entityName: z.string(),\n    entityType: z.string(),\n  })).describe('Available entities in memory'),\n});\n\nconst ExtractEntitiesSchema = z.object({\n  response: z.string().describe('Agent response text to extract entities from'),\n  userMessage: z.string().describe('Original user message for context'),\n});\n\nexport class ResolveEntitiesTool extends StructuredTool {\n  name = 'resolve_entities';\n  description = 'Resolves entity references like \"the topic\", \"it\", \"that\" to actual entity IDs';\n  schema = ResolveEntitiesSchema;\n  \n  private llm: ChatOpenAI;\n  \n  constructor(apiKey: string, modelName = 'gpt-4o-mini') {\n    super();\n    this.llm = new ChatOpenAI({\n      apiKey,\n      modelName,\n      temperature: 0,\n    });\n  }\n  \n  async _call(input: z.infer<typeof ResolveEntitiesSchema>): Promise<string> {\n    const { message, entities } = input;\n    \n    if (!entities || entities.length === 0) {\n      return message;\n    }\n    \n    const byType = this.groupEntitiesByType(entities);\n    const context = this.buildEntityContext(byType);\n    \n    const prompt = `Task: Replace entity references with IDs.\n\n${context}\n\nMessage: \"${message}\"\n\nRules:\n- \"the topic\" or \"that topic\" → replace with most recent topic ID\n- \"the token\" or \"that token\" → replace with most recent token ID\n- \"it\" or \"that\" after action verb → replace with most recent entity ID\n- \"airdrop X\" without token ID → add most recent token ID as first parameter\n- Token operations without explicit token → use most recent token ID\n\nExamples:\n- \"submit on the topic\" → \"submit on 0.0.6543472\"\n- \"airdrop the token\" → \"airdrop 0.0.123456\"\n- \"airdrop 10 to 0.0.5842697\" → \"airdrop 0.0.123456 10 to 0.0.5842697\"\n- \"mint 100\" → \"mint 0.0.123456 100\"\n\nReturn ONLY the resolved message:`;\n    \n    try {\n      const response = await this.llm.invoke(prompt);\n      return (response.content as string).trim();\n    } catch (error) {\n      logger.error('ResolveEntitiesTool failed:', error);\n      return message;\n    }\n  }\n  \n  private groupEntitiesByType(entities: EntityGroup): GroupedEntities {\n    return entities.reduce((acc, entity) => {\n      if (!acc[entity.entityType]) {\n        acc[entity.entityType] = [];\n      }\n      acc[entity.entityType].push(entity);\n      return acc;\n    }, {} as GroupedEntities);\n  }\n  \n  private buildEntityContext(groupedEntities: GroupedEntities): string {\n    let context = 'Available entities:\\n';\n    for (const [type, list] of Object.entries(groupedEntities)) {\n      const recent = list[0];\n      context += `- Most recent ${type}: \"${recent.entityName}\" = ${recent.entityId}\\n`;\n    }\n    return context;\n  }\n}\n\nexport class ExtractEntitiesTool extends StructuredTool {\n  name = 'extract_entities';\n  description = 'Extracts newly created entities from agent responses';\n  schema = ExtractEntitiesSchema;\n  \n  private llm: ChatOpenAI;\n  \n  constructor(apiKey: string, modelName = 'gpt-4o-mini') {\n    super();\n    this.llm = new ChatOpenAI({\n      apiKey,\n      modelName,\n      temperature: 0,\n    });\n  }\n  \n  async _call(input: z.infer<typeof ExtractEntitiesSchema>): Promise<string> {\n    const { response, userMessage } = input;\n    \n    const prompt = `Extract ONLY newly created entities from this response.\n\nUser asked: \"${userMessage.substring(0, 200)}\"\nResponse: ${response.substring(0, 3000)}\n\nLook for:\n- Success messages with new entity IDs\n- Words like \"created\", \"new\", \"successfully\" followed by entity IDs\n\nReturn JSON array of created entities:\n[{\"id\": \"0.0.XXX\", \"name\": \"name\", \"type\": \"topic|token|account\"}]\n\nIf none created, return: []\n\nJSON:`;\n    \n    try {\n      const llmResponse = await this.llm.invoke(prompt);\n      const content = llmResponse.content as string;\n      const match = content.match(/\\[[\\s\\S]*?\\]/);\n      if (match) {\n        return match[0];\n      }\n      return '[]';\n    } catch (error) {\n      logger.error('ExtractEntitiesTool failed:', error);\n      return '[]';\n    }\n  }\n}\n\nexport function createEntityTools(apiKey: string, modelName = 'gpt-4o-mini'): {\n  resolveEntities: ResolveEntitiesTool;\n  extractEntities: ExtractEntitiesTool;\n} {\n  return {\n    resolveEntities: new ResolveEntitiesTool(apiKey, modelName),\n    extractEntities: new ExtractEntitiesTool(apiKey, modelName),\n  };\n}\n\ninterface EntityReference {\n  entityId: string;\n  entityName: string;\n  entityType: string;\n}\n\ntype EntityGroup = EntityReference[];\n\ntype GroupedEntities = Record<string, EntityGroup>;\n\n"],"names":[],"mappings":";;;;AAKA,MAAM,SAAS,IAAI,OAAO,EAAE,QAAQ,sBAAsB;AAE1D,MAAM,wBAAwB,EAAE,OAAO;AAAA,EACrC,SAAS,EAAE,SAAS,SAAS,qDAAqD;AAAA,EAClF,UAAU,EAAE,MAAM,EAAE,OAAO;AAAA,IACzB,UAAU,EAAE,OAAA;AAAA,IACZ,YAAY,EAAE,OAAA;AAAA,IACd,YAAY,EAAE,OAAA;AAAA,EAAO,CACtB,CAAC,EAAE,SAAS,8BAA8B;AAC7C,CAAC;AAED,MAAM,wBAAwB,EAAE,OAAO;AAAA,EACrC,UAAU,EAAE,SAAS,SAAS,8CAA8C;AAAA,EAC5E,aAAa,EAAE,OAAA,EAAS,SAAS,mCAAmC;AACtE,CAAC;AAEM,MAAM,4BAA4B,eAAe;AAAA,EAOtD,YAAY,QAAgB,YAAY,eAAe;AACrD,UAAA;AAPF,SAAA,OAAO;AACP,SAAA,cAAc;AACd,SAAA,SAAS;AAMP,SAAK,MAAM,IAAI,WAAW;AAAA,MACxB;AAAA,MACA;AAAA,MACA,aAAa;AAAA,IAAA,CACd;AAAA,EACH;AAAA,EAEA,MAAM,MAAM,OAA+D;AACzE,UAAM,EAAE,SAAS,SAAA,IAAa;AAE9B,QAAI,CAAC,YAAY,SAAS,WAAW,GAAG;AACtC,aAAO;AAAA,IACT;AAEA,UAAM,SAAS,KAAK,oBAAoB,QAAQ;AAChD,UAAM,UAAU,KAAK,mBAAmB,MAAM;AAE9C,UAAM,SAAS;AAAA;AAAA,EAEjB,OAAO;AAAA;AAAA,YAEG,OAAO;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAiBf,QAAI;AACF,YAAM,WAAW,MAAM,KAAK,IAAI,OAAO,MAAM;AAC7C,aAAQ,SAAS,QAAmB,KAAA;AAAA,IACtC,SAAS,OAAO;AACd,aAAO,MAAM,+BAA+B,KAAK;AACjD,aAAO;AAAA,IACT;AAAA,EACF;AAAA,EAEQ,oBAAoB,UAAwC;AAClE,WAAO,SAAS,OAAO,CAAC,KAAK,WAAW;AACtC,UAAI,CAAC,IAAI,OAAO,UAAU,GAAG;AAC3B,YAAI,OAAO,UAAU,IAAI,CAAA;AAAA,MAC3B;AACA,UAAI,OAAO,UAAU,EAAE,KAAK,MAAM;AAClC,aAAO;AAAA,IACT,GAAG,CAAA,CAAqB;AAAA,EAC1B;AAAA,EAEQ,mBAAmB,iBAA0C;AACnE,QAAI,UAAU;AACd,eAAW,CAAC,MAAM,IAAI,KAAK,OAAO,QAAQ,eAAe,GAAG;AAC1D,YAAM,SAAS,KAAK,CAAC;AACrB,iBAAW,iBAAiB,IAAI,MAAM,OAAO,UAAU,OAAO,OAAO,QAAQ;AAAA;AAAA,IAC/E;AACA,WAAO;AAAA,EACT;AACF;AAEO,MAAM,4BAA4B,eAAe;AAAA,EAOtD,YAAY,QAAgB,YAAY,eAAe;AACrD,UAAA;AAPF,SAAA,OAAO;AACP,SAAA,cAAc;AACd,SAAA,SAAS;AAMP,SAAK,MAAM,IAAI,WAAW;AAAA,MACxB;AAAA,MACA;AAAA,MACA,aAAa;AAAA,IAAA,CACd;AAAA,EACH;AAAA,EAEA,MAAM,MAAM,OAA+D;AACzE,UAAM,EAAE,UAAU,YAAA,IAAgB;AAElC,UAAM,SAAS;AAAA;AAAA,eAEJ,YAAY,UAAU,GAAG,GAAG,CAAC;AAAA,YAChC,SAAS,UAAU,GAAG,GAAI,CAAC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAanC,QAAI;AACF,YAAM,cAAc,MAAM,KAAK,IAAI,OAAO,MAAM;AAChD,YAAM,UAAU,YAAY;AAC5B,YAAM,QAAQ,QAAQ,MAAM,cAAc;AAC1C,UAAI,OAAO;AACT,eAAO,MAAM,CAAC;AAAA,MAChB;AACA,aAAO;AAAA,IACT,SAAS,OAAO;AACd,aAAO,MAAM,+BAA+B,KAAK;AACjD,aAAO;AAAA,IACT;AAAA,EACF;AACF;AAEO,SAAS,kBAAkB,QAAgB,YAAY,eAG5D;AACA,SAAO;AAAA,IACL,iBAAiB,IAAI,oBAAoB,QAAQ,SAAS;AAAA,IAC1D,iBAAiB,IAAI,oBAAoB,QAAQ,SAAS;AAAA,EAAA;AAE9D;"}
+{"version":3,"file":"index16.js","sources":["../../src/mcp/content-processor.ts"],"sourcesContent":["import type { ContentType, ContentSource } from '../types/content-reference';\nimport type { ContentStorage } from '../memory/content-storage';\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\n/**\n * Content reference interface\n */\ninterface ContentReference {\n  referenceId: string;\n  preview?: string;\n  metadata: {\n    sizeBytes: number;\n    contentType?: string;\n    [key: string]: unknown;\n  };\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(\n      (max, content) => Math.max(max, content.sizeBytes),\n      0\n    );\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: [\n          error instanceof Error ? error.message : 'Unknown processing error',\n        ],\n      };\n    }\n  }\n\n  private extractContentFromResponse(\n    obj: unknown,\n    contents: MCPResponseContent[]\n  ): 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      );\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 (\n        this.contentStorage.shouldUseReference(\n          typeof contentInfo.content === 'string'\n            ? contentInfo.content\n            : JSON.stringify(contentInfo.content)\n        )\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(\n            contentInfo.mimeType\n          );\n\n          const metadata: Parameters<\n            typeof this.contentStorage.storeContentIfLarge\n          >[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(\n            `Failed to create reference: ${\n              error instanceof Error ? error.message : 'Unknown error'\n            }`\n          );\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(\n    reference: ContentReference\n  ): 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(\n    obj: unknown,\n    oldContent: unknown,\n    newContent: unknown\n  ): 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        if (typeof newContent === 'object' && newContent !== null) {\n          const newContentRecord = newContent as Record<string, unknown>;\n          for (const key of Object.keys(newContentRecord)) {\n            record[key] = newContentRecord[key];\n          }\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}\n"],"names":[],"mappings":"AAwCO,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;AAAA,MAClC,CAAC,KAAK,YAAY,KAAK,IAAI,KAAK,QAAQ,SAAS;AAAA,MACjD;AAAA,IAAA;AAGF,UAAM,gBAAgB,SAAS;AAAA,MAAK,CAAC,YACnC,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;AAAA,UACN,iBAAiB,QAAQ,MAAM,UAAU;AAAA,QAAA;AAAA,MAC3C;AAAA,IAEJ;AAAA,EACF;AAAA,EAEQ,2BACN,KACA,UACM;AACN,QAAI,QAAQ,QAAQ,QAAQ,QAAW;AACrC;AAAA,IACF;AAEA,QAAI,MAAM,QAAQ,GAAG,GAAG;AACtB,UAAI,QAAQ,CAAC,SAAS,KAAK,2BAA2B,MAAM,QAAQ,CAAC;AACrE;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,UAAW,OAAO,YAAuB;AAAA,QAAA,CAC1C;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;AAAA,QAAQ,CAAC,UAC7B,KAAK,2BAA2B,OAAO,QAAQ;AAAA,MAAA;AAEjD;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,UACE,KAAK,eAAe;AAAA,QAClB,OAAO,YAAY,YAAY,WAC3B,YAAY,UACZ,KAAK,UAAU,YAAY,OAAO;AAAA,MAAA,GAExC;AACA,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;AAAA,YACvB,YAAY;AAAA,UAAA;AAGd,gBAAM,WAEC;AAAA,YACL;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;AAAA,YACL,+BACE,iBAAiB,QAAQ,MAAM,UAAU,eAC3C;AAAA,UAAA;AAAA,QAEJ;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,2BACN,WACyB;AACzB,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,yBACN,KACA,YACA,YACM;AACN,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,YAAI,OAAO,eAAe,YAAY,eAAe,MAAM;AACzD,gBAAM,mBAAmB;AACzB,qBAAW,OAAO,OAAO,KAAK,gBAAgB,GAAG;AAC/C,mBAAO,GAAG,IAAI,iBAAiB,GAAG;AAAA,UACpC;AAAA,QACF;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,CAAC,SAAS,KAAK,UAAU,IAAI,CAAC;AAAA,IAC/C;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;"}