@hashgraphonline/conversational-agent 0.1.217 → 0.1.218

package/src/memory/index.ts

@@ -1,8 +1,5 @@
-export { TokenCounter } from './TokenCounter';
-export { MemoryWindow } from './MemoryWindow';
-export { ContentStorage } from './ContentStorage';
-export { SmartMemoryManager } from './SmartMemoryManager';
-
-export type { SmartMemoryConfig, SearchOptions, MemoryStats, EntityAssociation, EntityResolutionOptions } from './SmartMemoryManager';
-export type { AddMessageResult } from './MemoryWindow';
-export type { StorageStats } from './ContentStorage';
+export * from './smart-memory-manager';
+export * from './token-counter';
+export * from './memory-window';
+export * from './content-storage';
+export * from './reference-id-generator';

package/src/memory/{MemoryWindow.ts → memory-window.ts}

@@ -1,5 +1,5 @@
 import type { BaseMessage } from '@langchain/core/messages';
-import { TokenCounter } from './TokenCounter';
+import { TokenCounter } from './token-counter';
 
 /**
  * Result of adding a message to the memory window
@@ -52,19 +52,19 @@ export class MemoryWindow {
    */
   addMessage(message: BaseMessage): AddMessageResult {
     this.tokenCounter.countMessageTokens(message);
-    
+
     this.messages.push(message);
-    
+
     const currentTokens = this.getCurrentTokenCount();
     const availableTokens = this.maxTokens - this.reserveTokens;
-    
+
     let prunedMessages: BaseMessage[] = [];
-    
+
     if (currentTokens > availableTokens) {
       this.messages.pop();
-      
+
       prunedMessages = this.pruneToFit();
-      
+
       this.messages.push(message);
     }
 
@@ -72,7 +72,7 @@ export class MemoryWindow {
       added: true,
       prunedMessages,
       currentTokenCount: this.getCurrentTokenCount(),
-      remainingCapacity: this.getRemainingTokenCapacity()
+      remainingCapacity: this.getRemainingTokenCapacity(),
     };
   }
 
@@ -84,17 +84,23 @@ export class MemoryWindow {
   pruneToFit(): BaseMessage[] {
     const prunedMessages: BaseMessage[] = [];
     const targetTokens = this.maxTokens - this.reserveTokens;
-    
-    while (this.getCurrentTokenCount() > targetTokens && this.messages.length > 0) {
-      const batchSize = Math.min(MemoryWindow.PRUNING_BATCH_SIZE, this.messages.length);
-      
+
+    while (
+      this.getCurrentTokenCount() > targetTokens &&
+      this.messages.length > 0
+    ) {
+      const batchSize = Math.min(
+        MemoryWindow.PRUNING_BATCH_SIZE,
+        this.messages.length
+      );
+
       for (let i = 0; i < batchSize; i++) {
         const prunedMessage = this.messages.shift();
         if (prunedMessage) {
           prunedMessages.push(prunedMessage);
         }
       }
-      
+
       if (prunedMessages.length > 1000) {
         break;
       }
@@ -128,12 +134,13 @@ export class MemoryWindow {
   canAddMessage(message: BaseMessage): boolean {
     const messageTokens = this.tokenCounter.countMessageTokens(message);
     const currentTokens = this.getCurrentTokenCount();
-    const wouldExceedReserve = (currentTokens + messageTokens) > (this.maxTokens - this.reserveTokens);
-    
+    const wouldExceedReserve =
+      currentTokens + messageTokens > this.maxTokens - this.reserveTokens;
+
     if (messageTokens > this.maxTokens) {
       return false;
     }
-    
+
     return !wouldExceedReserve || this.messages.length > 0;
   }
 
@@ -158,7 +165,8 @@ export class MemoryWindow {
    */
   setSystemPrompt(systemPrompt: string): void {
     this.systemPrompt = systemPrompt;
-    this.systemPromptTokens = this.tokenCounter.estimateSystemPromptTokens(systemPrompt);
+    this.systemPromptTokens =
+      this.tokenCounter.estimateSystemPromptTokens(systemPrompt);
   }
 
   /**
@@ -173,13 +181,19 @@ export class MemoryWindow {
    * Get current configuration
    * @returns Memory window configuration
    */
-  getConfig() {
+  getConfig(): {
+    maxTokens: number;
+    reserveTokens: number;
+    currentTokens: number;
+    messageCount: number;
+    systemPromptTokens: number;
+  } {
     return {
       maxTokens: this.maxTokens,
       reserveTokens: this.reserveTokens,
       currentTokens: this.getCurrentTokenCount(),
       messageCount: this.messages.length,
-      systemPromptTokens: this.systemPromptTokens
+      systemPromptTokens: this.systemPromptTokens,
     };
   }
 
@@ -198,7 +212,7 @@ export class MemoryWindow {
       this.reserveTokens = reserveTokens;
     }
 
-    if (this.getCurrentTokenCount() > (this.maxTokens - this.reserveTokens)) {
+    if (this.getCurrentTokenCount() > this.maxTokens - this.reserveTokens) {
       this.pruneToFit();
     }
   }
@@ -207,11 +221,20 @@ export class MemoryWindow {
    * Get statistics about the memory window
    * @returns Memory usage statistics
    */
-  getStats() {
+  getStats(): {
+    totalMessages: number;
+    currentTokens: number;
+    maxTokens: number;
+    reserveTokens: number;
+    systemPromptTokens: number;
+    usagePercentage: number;
+    remainingCapacity: number;
+    canAcceptMore: boolean;
+  } {
     const currentTokens = this.getCurrentTokenCount();
     const capacity = this.maxTokens;
     const usagePercentage = (currentTokens / capacity) * 100;
-    
+
     return {
       totalMessages: this.messages.length,
       currentTokens,
@@ -220,7 +243,7 @@ export class MemoryWindow {
       systemPromptTokens: this.systemPromptTokens,
       usagePercentage: Math.round(usagePercentage * 100) / 100,
       remainingCapacity: this.getRemainingTokenCapacity(),
-      canAcceptMore: this.getRemainingTokenCapacity() > this.reserveTokens
+      canAcceptMore: this.getRemainingTokenCapacity() > this.reserveTokens,
     };
   }
 
@@ -231,4 +254,4 @@ export class MemoryWindow {
     this.clear();
     this.tokenCounter.dispose();
   }
-}
+}

package/src/memory/{SmartMemoryManager.ts → smart-memory-manager.ts}

@@ -1,8 +1,9 @@
 import type { BaseMessage } from '@langchain/core/messages';
+import { SystemMessage } from '@langchain/core/messages';
 import { Logger } from '@hashgraphonline/standards-sdk';
-import { MemoryWindow } from './MemoryWindow';
-import { ContentStorage } from './ContentStorage';
-import { TokenCounter } from './TokenCounter';
+import { MemoryWindow } from './memory-window';
+import { ContentStorage } from './content-storage';
+import { TokenCounter } from './token-counter';
 
 /**
  * Entity association for storing blockchain entity contexts
@@ -78,7 +79,6 @@ export interface MemoryStats {
 
 const IS_ENTITY_ASSOCIATION_FLAG = '"isEntityAssociation":true';
 
-
 /**
  * TODO: investigate using chroma / rag for long term memory
  * Smart memory manager that combines active memory window with long-term storage
@@ -412,19 +412,36 @@ export class SmartMemoryManager {
       const sanitizedEntityName = entityName.trim().substring(0, 100);
       const sanitizedEntityType = entityType.trim().toLowerCase();
 
-      const association: EntityAssociation & { isEntityAssociation: boolean } =
-        {
-          entityId: sanitizedEntityId,
-          entityName: sanitizedEntityName,
-          entityType: sanitizedEntityType,
-          createdAt: new Date(),
-          isEntityAssociation: true,
-          ...(transactionId !== undefined &&
-          transactionId !== null &&
-          transactionId.trim() !== ''
-            ? { transactionId: transactionId.trim() }
-            : {}),
-        };
+      let usageHint = '';
+      if (sanitizedEntityType === 'tokenid') {
+        usageHint = 'Use this as tokenId for HTS operations';
+      } else if (sanitizedEntityType === 'topicid') {
+        usageHint =
+          'Can be used for HCS operations, HRLs for minting with the format hcs://1/<topicId>, etc.';
+      } else if (sanitizedEntityType === 'accountid') {
+        usageHint = 'Can be used for account based operations';
+      }
+
+      const association: EntityAssociation & {
+        isEntityAssociation: boolean;
+        usage?: string;
+        hrl?: string;
+      } = {
+        entityId: sanitizedEntityId,
+        entityName: sanitizedEntityName,
+        entityType: sanitizedEntityType,
+        createdAt: new Date(),
+        isEntityAssociation: true,
+        ...(usageHint ? { usage: usageHint } : {}),
+        ...(sanitizedEntityType === 'topicid'
+          ? { hrl: `hcs://1/${sanitizedEntityId}` }
+          : {}),
+        ...(transactionId !== undefined &&
+        transactionId !== null &&
+        transactionId.trim() !== ''
+          ? { transactionId: transactionId.trim() }
+          : {}),
+      };
 
       const content = JSON.stringify(association);
       type LangChainLikeMessage = {
@@ -448,13 +465,17 @@ export class SmartMemoryManager {
         },
       };
 
+      try {
+        this.memoryWindow.addMessage(new SystemMessage(content));
+      } catch {}
+
       this._contentStorage.storeMessages([entityMessage as BaseMessage]);
     } catch (error) {
       this.logger.error('Failed to store entity association', {
         entityId,
         entityName,
         entityType,
-        error: error instanceof Error ? error.message : String(error)
+        error: error instanceof Error ? error.message : String(error),
       });
     }
   }
@@ -577,7 +598,7 @@ export class SmartMemoryManager {
       this.logger.error('Failed to resolve entity reference', {
         query,
         options,
-        error: error instanceof Error ? error.message : String(error)
+        error: error instanceof Error ? error.message : String(error),
       });
       return [];
     }
@@ -633,8 +654,14 @@ export class SmartMemoryManager {
           }
         } catch (parseError) {
           this.logger.warn('Failed to parse entity association from message', {
-            messageContent: typeof message.content === 'string' ? message.content.substring(0, 100) : 'non-string',
-            error: parseError instanceof Error ? parseError.message : String(parseError)
+            messageContent:
+              typeof message.content === 'string'
+                ? message.content.substring(0, 100)
+                : 'non-string',
+            error:
+              parseError instanceof Error
+                ? parseError.message
+                : String(parseError),
           });
           continue;
         }
@@ -657,7 +684,7 @@ export class SmartMemoryManager {
     } catch (error) {
       this.logger.error('Failed to get entity associations', {
         entityType,
-        error: error instanceof Error ? error.message : String(error)
+        error: error instanceof Error ? error.message : String(error),
       });
       return [];
     }

package/src/plugins/hbar/HbarPlugin.ts

@@ -76,7 +76,7 @@ export class HbarPlugin extends BasePlugin {
   }
 
   override getTools(): HederaTool[] {
-    return this.tools;
+    return this.tools as unknown as HederaTool[];
   }
 
   async shutdown(): Promise<void> {

package/src/plugins/hcs-10/HCS10Plugin.ts

@@ -22,6 +22,41 @@ import {
 } from '@hashgraphonline/standards-agent-kit';
 import { HCS10Client } from '@hashgraphonline/standards-sdk';
 
+interface HCS10ClientManager {
+  initializeConnectionsManager(client: HCS10Client): void;
+}
+
+/**
+ * Extracts private key string from operator key
+ */
+function extractPrivateKey(opKey: unknown): string {
+  const key = opKey as {
+    toString?: () => string;
+    toStringRaw?: () => string;
+  };
+  
+  if (typeof key?.toStringRaw === 'function') {
+    return key.toStringRaw();
+  }
+  
+  if (typeof key?.toString === 'function') {
+    return key.toString();
+  }
+  
+  return String(key);
+}
+
+function hasInitializeConnectionsManager(
+  stateManager: IStateManager
+): stateManager is IStateManager & HCS10ClientManager {
+  return (
+    typeof stateManager === 'object' &&
+    stateManager !== null &&
+    'initializeConnectionsManager' in stateManager &&
+    typeof stateManager.initializeConnectionsManager === 'function'
+  );
+}
+
 export class HCS10Plugin extends BasePlugin {
   id = 'hcs-10';
   name = 'HCS-10 Plugin';
@@ -58,15 +93,8 @@ export class HCS10Plugin extends BasePlugin {
       let outboundTopicId = '';
 
       try {
-        const opKey = hederaKit.signer.getOperatorPrivateKey() as {
-          toString?: () => string;
-          toStringRaw?: () => string;
-        };
-        const privateKey = typeof opKey?.toStringRaw === 'function'
-          ? opKey.toStringRaw()
-          : typeof opKey?.toString === 'function'
-            ? opKey.toString()
-            : String(opKey);
+        const opKey = hederaKit.signer.getOperatorPrivateKey();
+        const privateKey = extractPrivateKey(opKey);
 
         const hcs10Client = new HCS10Client({
           network: hederaKit.network as 'mainnet' | 'testnet',
@@ -93,15 +121,8 @@ export class HCS10Plugin extends BasePlugin {
         inboundTopicId,
         outboundTopicId,
         privateKey: ((): string => {
-          const opKey = hederaKit.signer.getOperatorPrivateKey() as {
-            toString?: () => string;
-            toStringRaw?: () => string;
-          };
-          return typeof opKey?.toStringRaw === 'function'
-            ? opKey.toStringRaw()
-            : typeof opKey?.toString === 'function'
-              ? opKey.toString()
-              : String(opKey);
+          const opKey = hederaKit.signer.getOperatorPrivateKey();
+          return extractPrivateKey(opKey);
         })(),
       });
 
@@ -111,15 +132,8 @@ export class HCS10Plugin extends BasePlugin {
 
       if (this.stateManager && !this.stateManager.getConnectionsManager()) {
         try {
-          const opKey = hederaKit.signer.getOperatorPrivateKey() as {
-            toString?: () => string;
-            toStringRaw?: () => string;
-          };
-          const privateKey = typeof opKey?.toStringRaw === 'function'
-            ? opKey.toStringRaw()
-            : typeof opKey?.toString === 'function'
-              ? opKey.toString()
-              : String(opKey);
+          const opKey = hederaKit.signer.getOperatorPrivateKey();
+          const privateKey = extractPrivateKey(opKey);
           const hcs10Client = new HCS10Client({
             network: hederaKit.network as 'mainnet' | 'testnet',
             operatorId: accountId,
@@ -127,7 +141,11 @@ export class HCS10Plugin extends BasePlugin {
             logLevel: 'error',
           });
 
-          this.stateManager.initializeConnectionsManager(hcs10Client as any);
+          if (hasInitializeConnectionsManager(this.stateManager)) {
+            this.stateManager.initializeConnectionsManager(hcs10Client);
+          } else {
+            this.context.logger.warn('StateManager does not support connection manager initialization');
+          }
           this.context.logger.info(
             'ConnectionsManager initialized in HCS10Plugin'
           );

package/src/scripts/test-external-tool-wrapper.ts

@@ -5,7 +5,7 @@
 
 import { examples } from '../examples/external-tool-wrapper-example';
 
-async function testExternalToolWrapper() {
+async function testExternalToolWrapper(): Promise<void> {
   console.log('🧪 Testing External Tool Wrapper\n');
 
   try {
@@ -23,7 +23,7 @@ async function testExternalToolWrapper() {
       memo: 'Test transfer'
     };
     
-    const result = await (basicTool as any)._call(testInput);
+    const result = await (basicTool as { _call: (input: unknown) => Promise<string> })._call(testInput);
     console.log(`   Execution result: ${result}\n`);
 
     console.log('📝 Test 2: Preset configurations');
@@ -47,10 +47,10 @@ async function testExternalToolWrapper() {
     const incompleteInput = {
       fromAccountId: '0.0.123'
 
-    } as any;
+    } as { fromAccountId: string };
     
     try {
-      const formResult = await (formTool as any)._call(incompleteInput);
+      const formResult = await (formTool as { _call: (input: unknown) => Promise<string> })._call(incompleteInput);
       const parsedResult = JSON.parse(formResult);
       if (parsedResult.requiresForm) {
         console.log(`   ✅ Form generation triggered for missing fields`);
@@ -68,11 +68,11 @@ async function testExternalToolWrapper() {
     console.log(`✅ Tool "${customTool.name}" wrapped with custom configs`);
     console.log(`   Description: ${customTool.description}`);
     
-    const schema = customTool.schema as any;
+    const schema = customTool.schema as { _def: { shape: () => Record<string, { _renderConfig?: unknown }> } };
     const shape = schema._def.shape();
     let fieldConfigCount = 0;
     for (const [, fieldSchema] of Object.entries(shape)) {
-      if ((fieldSchema as any)._renderConfig) {
+      if (fieldSchema._renderConfig) {
         fieldConfigCount++;
       }
     }

package/src/scripts/test-inscribe-form-generation.ts

@@ -17,6 +17,7 @@ import dotenv from 'dotenv';
 import { ConversationalAgent } from '../conversational-agent';
 import type { ChatResponse } from '../base-agent';
 import type { FormMessage } from '../forms/types';
+import { NetworkType } from '@hashgraphonline/standards-sdk';
 
 dotenv.config();
 
@@ -65,7 +66,7 @@ async function createTestAgent(): Promise<ConversationalAgent> {
   const options = {
     accountId: process.env.HEDERA_OPERATOR_ID!,
     privateKey: process.env.HEDERA_OPERATOR_KEY!,
-    network: (process.env.HEDERA_NETWORK as any) || 'testnet',
+    network: (process.env.HEDERA_NETWORK as NetworkType) || NetworkType.TESTNET,
     openAIApiKey: process.env.OPENAI_API_KEY!,
     openAIModelName: 'gpt-4o-mini',
     verbose: true, // Enable verbose for debugging
@@ -87,11 +88,11 @@ async function testDirectToolAccess(agent: ConversationalAgent): Promise<FormGen
   
   try {
     const underlyingAgent = agent.getAgent();
-    const tools = (underlyingAgent as any).tools;
+    const tools = (underlyingAgent as unknown as { tools: Array<{ name: string; constructor: { name: string }; _call: (input: unknown) => Promise<string> }> }).tools;
     
-    console.log('Available tools:', tools.map((t: any) => t.name));
+    console.log('Available tools:', tools.map(t => t.name));
     
-    const inscribeHashinalTool = tools.find((t: any) => t.name === 'inscribeHashinal');
+    const inscribeHashinalTool = tools.find(t => t.name === 'inscribeHashinal');
     
     if (!inscribeHashinalTool) {
       console.log('❌ InscribeHashinal tool not found in tools array');
@@ -114,7 +115,7 @@ async function testDirectToolAccess(agent: ConversationalAgent): Promise<FormGen
     };
     
     console.log('🧪 Calling tool directly with incomplete input...');
-    const result = await (inscribeHashinalTool as any)._call(incompleteInput);
+    const result = await inscribeHashinalTool._call(incompleteInput);
     
     console.log('📋 Direct tool call result:', {
       resultType: typeof result,
@@ -128,20 +129,20 @@ async function testDirectToolAccess(agent: ConversationalAgent): Promise<FormGen
         console.log('🎉 Form generated by direct tool call!');
         const formMessage = parsed.formMessage;
         
-        const formConfig = (formMessage as any).formConfig;
+        const formConfig = (formMessage as { formConfig?: { fields?: Array<{ name: string; type: string }> } }).formConfig;
         const attributesField = formConfig?.fields?.find(
-          (field: any) => field.name === 'attributes'
+          field => field.name === 'attributes'
         );
         
         const hasAttributesField = !!attributesField;
         
         console.log('📝 Form analysis (direct call):', {
-          formId: (formMessage as any).id,
-          toolName: (formMessage as any).toolName,
+          formId: (formMessage as { id?: string }).id,
+          toolName: (formMessage as { toolName?: string }).toolName,
           fieldCount: formConfig?.fields?.length || 0,
           hasAttributesField,
           attributesFieldType: attributesField?.type,
-          fieldsPresent: formConfig?.fields?.map((f: any) => f.name) || []
+          fieldsPresent: formConfig?.fields?.map(f => f.name) || []
         });
 
         return {
@@ -214,20 +215,20 @@ async function testInscribeFormGeneration(agent: ConversationalAgent): Promise<F
               console.log('📝 Found form in tool call output!');
               const formMessage = parsed.formMessage;
               
-              const formConfig = (formMessage as any).formConfig;
+              const formConfig = (formMessage as { formConfig?: { fields?: Array<{ name: string; type: string }> } }).formConfig;
               const attributesField = formConfig?.fields?.find(
-                (field: any) => field.name === 'attributes'
+                field => field.name === 'attributes'
               );
               
               const hasAttributesField = !!attributesField;
               
               console.log('📝 Form analysis:', {
-                formId: (formMessage as any).id,
-                toolName: (formMessage as any).toolName,
+                formId: (formMessage as { id?: string }).id,
+                toolName: (formMessage as { toolName?: string }).toolName,
                 fieldCount: formConfig?.fields?.length || 0,
                 hasAttributesField,
                 attributesFieldType: attributesField?.type,
-                fieldsPresent: formConfig?.fields?.map((f: any) => f.name) || []
+                fieldsPresent: formConfig?.fields?.map(f => f.name) || []
               });
 
               return {
@@ -237,7 +238,7 @@ async function testInscribeFormGeneration(agent: ConversationalAgent): Promise<F
                 originalResponse: response
               };
             }
-          } catch {
+          } catch (parseError) {
             console.log('Could not parse tool output as JSON:', parseError);
           }
         }
@@ -247,20 +248,20 @@ async function testInscribeFormGeneration(agent: ConversationalAgent): Promise<F
     if (response.requiresForm && response.formMessage) {
       const formMessage = response.formMessage;
       
-      const formConfig = (formMessage as any).formConfig;
+      const formConfig = (formMessage as { formConfig?: { fields?: Array<{ name: string; type: string }> } }).formConfig;
       const attributesField = formConfig?.fields?.find(
-        (field: any) => field.name === 'attributes'
+        field => field.name === 'attributes'
       );
       
       const hasAttributesField = !!attributesField;
       
       console.log('📝 Form analysis (top-level):', {
-        formId: (formMessage as any).id,
-        toolName: (formMessage as any).toolName,
+        formId: (formMessage as { id?: string }).id,
+        toolName: (formMessage as { toolName?: string }).toolName,
         fieldCount: formConfig?.fields?.length || 0,
         hasAttributesField,
         attributesFieldType: attributesField?.type,
-        fieldsPresent: formConfig?.fields?.map((f: any) => f.name) || []
+        fieldsPresent: formConfig?.fields?.map(f => f.name) || []
       });
 
       return {

package/src/scripts/test-inscribe-wrapper-verification.ts

@@ -13,6 +13,7 @@
 
 import dotenv from 'dotenv';
 import { ConversationalAgent } from '../conversational-agent';
+import { NetworkType } from '@hashgraphonline/standards-sdk';
 
 dotenv.config();
 
@@ -60,7 +61,7 @@ async function createTestAgent(): Promise<ConversationalAgent> {
   const options = {
     accountId: process.env.HEDERA_OPERATOR_ID!,
     privateKey: process.env.HEDERA_OPERATOR_KEY!,
-    network: (process.env.HEDERA_NETWORK as any) || 'testnet',
+    network: (process.env.HEDERA_NETWORK as NetworkType) || NetworkType.TESTNET,
     openAIApiKey: process.env.OPENAI_API_KEY!,
     openAIModelName: 'gpt-4o-mini',
     verbose: false, // Reduce verbosity
@@ -82,12 +83,12 @@ async function testInscribeHashinalWrapper(agent: ConversationalAgent): Promise<
   
   try {
     const underlyingAgent = agent.getAgent();
-    const tools = (underlyingAgent as any).tools;
+    const tools = (underlyingAgent as unknown as { tools: Array<{ name: string; constructor: { name: string }; schema?: { _renderConfig?: unknown } }> }).tools;
     
     const toolsCount = tools.length;
     console.log(`📊 Total tools loaded: ${toolsCount}`);
     
-    const inscribeHashinalTool = tools.find((t: any) => t.name === 'inscribeHashinal');
+    const inscribeHashinalTool = tools.find(t => t.name === 'inscribeHashinal');
     
     if (!inscribeHashinalTool) {
       return {
@@ -103,7 +104,7 @@ async function testInscribeHashinalWrapper(agent: ConversationalAgent): Promise<
     
     const toolType = inscribeHashinalTool.constructor.name;
     const isFormValidatingWrapper = toolType === 'FormValidatingToolWrapper';
-    const hasRenderConfig = !!(inscribeHashinalTool.schema as any)?._renderConfig;
+    const hasRenderConfig = !!inscribeHashinalTool.schema?._renderConfig;
     
     console.log('🔧 InscribeHashinal tool analysis:', {
       name: inscribeHashinalTool.name,