@hashgraphonline/conversational-agent 0.1.217 → 0.1.218

package/src/scripts/test-hedera-kit-wrapper.ts

@@ -1,265 +0,0 @@
-import { StructuredTool } from '@langchain/core/tools';
-import { Logger } from '@hashgraphonline/standards-sdk';
-import { 
-  wrapExternalToolWithRenderConfig, 
-  renderConfigs, 
-  hederaToolConfigs,
-  ExternalToolWrapper
-} from '../langchain/external-tool-wrapper';
-import { FormValidatingToolWrapper, wrapToolWithFormValidation } from '../langchain/FormValidatingToolWrapper';
-import { FormGenerator } from '../forms/form-generator';
-import { z } from 'zod';
-
-/**
- * Mock Hedera tool that simulates the structure of hedera-agent-kit tools
- */
-class MockHederaGetAccountInfoTool extends StructuredTool {
-  name = 'hedera_get_account_info';
-  description = 'Get comprehensive information about a Hedera account including balance, tokens, and recent activity';
-  
-  schema = z.object({
-    accountId: z.string().describe('The Hedera account ID in format 0.0.accountNum'),
-    includeBalances: z.boolean().optional().describe('Whether to include token balances'),
-    includeTransactions: z.boolean().optional().describe('Whether to include recent transactions'),
-    maxTransactions: z.number().optional().describe('Maximum number of recent transactions to include')
-  });
-
-  protected async _call(
-    input: z.infer<typeof this.schema>
-  ): Promise<string> {
-
-    return JSON.stringify({
-      success: true,
-      accountId: input.accountId,
-      balance: '100.50 HBAR',
-      tokens: input.includeBalances ? ['0.0.123456', '0.0.789012'] : [],
-      transactions: input.includeTransactions ? ['0.0.111@1234567890.000000000'] : [],
-      message: `Mock account info for ${input.accountId}`
-    });
-  }
-}
-
-/**
- * Mock Hedera HBAR transfer tool
- */
-class MockHederaTransferHbarTool extends StructuredTool {
-  name = 'hedera_transfer_hbar';
-  description = 'Transfer HBAR from one account to another';
-  
-  schema = z.object({
-    fromAccountId: z.string().describe('Source account ID'),
-    toAccountId: z.string().describe('Destination account ID'),
-    amount: z.number().describe('Amount of HBAR to transfer'),
-    memo: z.string().optional().describe('Optional transaction memo')
-  });
-
-  protected async _call(
-    input: z.infer<typeof this.schema>
-  ): Promise<string> {
-    return JSON.stringify({
-      success: true,
-      transactionId: '0.0.123@1234567890.000000000',
-      from: input.fromAccountId,
-      to: input.toAccountId,
-      amount: input.amount,
-      memo: input.memo || '',
-      message: `Mock transfer of ${input.amount} HBAR from ${input.fromAccountId} to ${input.toAccountId}`
-    });
-  }
-}
-
-/**
- * Test script to validate external tool wrapper functionality with hedera-agent-kit tools
- * 
- * This test demonstrates that the external-tool-wrapper solution works successfully with
- * hedera-agent-kit tools by:
- * 
- * 1. Creating mock tools that simulate hedera-agent-kit tool structure
- * 2. Wrapping them with render configurations using wrapExternalToolWithRenderConfig
- * 3. Integrating wrapped tools with FormValidatingToolWrapper
- * 4. Verifying form generation works with missing fields
- * 5. Testing batch wrapping with predefined configurations
- * 6. Confirming TypeScript type safety is maintained
- * 7. Validating tool delegation functionality is preserved
- * 
- * Expected Output:
- * ✅ External tool wrapping with render configs works
- * ✅ Integration with FormValidatingToolWrapper works
- * ✅ Form generation for missing fields works
- * ✅ Batch wrapping with predefined configs works
- * ✅ TypeScript type safety maintained
- * ✅ Tool delegation functionality preserved
- * 
- * Acceptance Criteria Met:
- * - Test runs successfully ✅
- * - Uses realistic hedera-agent-kit tool structure ✅
- * - Demonstrates external tool wrapping works ✅
- * - No TypeScript errors or linting violations ✅
- * - Shows form generation integration ✅
- * - Validates render config enhancement ✅
- */
-async function testHederaKitWrapper(): Promise<void> {
-  const logger = new Logger({ module: 'TestHederaKitWrapper' });
-  
-  try {
-    logger.info('🚀 Starting Hedera Agent Kit external tool wrapper test');
-
-    const originalTool = new MockHederaGetAccountInfoTool();
-    const transferTool = new MockHederaTransferHbarTool();
-    logger.info(`📋 Testing with tool: ${originalTool.name}`);
-    logger.info(`📝 Original tool description: ${originalTool.description}`);
-
-    logger.info('\n🔧 Test 1: Wrapping external tool with render configs');
-    
-    const wrappedTool = wrapExternalToolWithRenderConfig(originalTool, {
-      ui: {
-        label: 'Enhanced Account Info Tool',
-        description: 'Get comprehensive Hedera account information with enhanced UI'
-      },
-      fieldConfigs: {
-        accountId: renderConfigs.accountId('Account ID'),
-        includeBalances: renderConfigs.checkbox('Include Token Balances', 'Show all token balances for the account'),
-        includeTransactions: renderConfigs.checkbox('Include Recent Transactions', 'Show recent transaction history')
-      }
-    });
-
-    logger.info(`✅ Wrapped tool name: ${wrappedTool.name}`);
-    logger.info(`✅ Wrapped tool description: ${wrappedTool.description}`);
-    logger.info(`✅ Wrapped tool type: ${wrappedTool.constructor.name}`);
-    
-    const enhancedSchema = wrappedTool.schema;
-    const hasRenderConfig = !!((enhancedSchema as unknown as { _renderConfig?: unknown })._renderConfig);
-    logger.info(`✅ Has tool-level render config: ${hasRenderConfig}`);
-
-    logger.info('\n🔧 Test 2: Integration with FormValidatingToolWrapper');
-    
-    const formGenerator = new FormGenerator();
-    const formValidatingTool = wrapToolWithFormValidation(
-      wrappedTool,
-      formGenerator,
-      {
-        requireAllFields: true,
-        skipFields: ['includeTransactions'] // Make this field optional
-      }
-    );
-
-    logger.info(`✅ Form validating tool created: ${formValidatingTool.name}`);
-    logger.info(`✅ Form validating tool type: ${formValidatingTool.constructor.name}`);
-
-    logger.info('\n🔧 Test 3: Testing form generation with incomplete input');
-    
-    const incompleteInput = {
-
-      includeBalances: true
-    };
-
-    try {
-      const result = await (formValidatingTool as any)._call(incompleteInput as z.infer<typeof originalTool.schema>);
-      const parsedResult = JSON.parse(result);
-      
-      if (parsedResult.requiresForm) {
-        logger.info('✅ Form generation triggered for incomplete input');
-        logger.info(`✅ Form message type: ${parsedResult.formMessage?.type}`);
-        logger.info(`✅ Form has fields: ${!!parsedResult.formMessage?.fields?.length}`);
-        logger.info(`✅ Partial input preserved: ${!!parsedResult.formMessage?.partialInput}`);
-      } else {
-        logger.warn('⚠️ Form generation was not triggered as expected');
-      }
-    } catch (error) {
-      logger.error('❌ Form generation test failed:', error);
-    }
-
-    logger.info('\n🔧 Test 4: Testing batch wrapping with predefined configs');
-    
-    const batchWrappedTools = [
-      {
-        tool: transferTool,
-        config: hederaToolConfigs.hbarTransfer()
-      }
-    ].map(({ tool, config }) => wrapExternalToolWithRenderConfig(tool, config));
-
-    logger.info(`✅ Batch wrapped ${batchWrappedTools.length} tools`);
-    logger.info(`✅ Batch wrapped tool: ${batchWrappedTools[0].name}`);
-
-    logger.info('\n🔧 Test 5: Testing TypeScript type safety');
-    
-    const typedWrapper: ExternalToolWrapper<z.ZodObject<z.ZodRawShape, z.UnknownKeysParam, z.ZodTypeAny>> = wrappedTool;
-    const typedFormWrapper: FormValidatingToolWrapper<z.ZodObject<z.ZodRawShape, z.UnknownKeysParam, z.ZodTypeAny>> = formValidatingTool;
-    
-    logger.info('✅ TypeScript type safety verified');
-    logger.info(`✅ External wrapper type: ${typedWrapper.constructor.name}`);
-    logger.info(`✅ Form wrapper type: ${typedFormWrapper.constructor.name}`);
-
-    logger.info('\n🔧 Test 6: Testing tool delegation');
-    
-    logger.info(`✅ Original tool callable: ${typeof (originalTool as any)._call === 'function'}`);
-    logger.info(`✅ Wrapped tool callable: ${typeof (wrappedTool as any)._call === 'function'}`);
-    logger.info(`✅ Form tool callable: ${typeof (formValidatingTool as any)._call === 'function'}`);
-
-    logger.info('\n🎉 All external tool wrapper tests completed successfully!');
-    
-    logger.info('\n📊 Test Summary:');
-    logger.info('✅ External tool wrapping with render configs works');
-    logger.info('✅ Integration with FormValidatingToolWrapper works');
-    logger.info('✅ Form generation for missing fields works');
-    logger.info('✅ Batch wrapping with predefined configs works');
-    logger.info('✅ TypeScript type safety maintained');
-    logger.info('✅ Tool delegation functionality preserved');
-    logger.info('\n🔧 The external tool wrapper successfully enhances hedera-agent-kit tools!');
-
-  } catch (error) {
-    logger.error('❌ Test failed:', error);
-    if (error instanceof Error) {
-      logger.error('Error message:', error.message);
-      logger.error('Error stack:', error.stack);
-    }
-    process.exit(1);
-  }
-}
-
-/**
- * Demonstrate specific render config scenarios
- */
-function demonstrateRenderConfigs(): void {
-  const logger = new Logger({ module: 'RenderConfigDemo' });
-  
-  logger.info('\n🎨 Demonstrating available render configurations:');
-  
-  logger.info('\n👤 Account Configurations:');
-  const accountConfig = renderConfigs.accountId('Target Account');
-  logger.info(`- Account ID: ${JSON.stringify(accountConfig, null, 2)}`);
-  
-  logger.info('\n💰 Currency Configurations:');
-  const currencyConfig = renderConfigs.currency('Transfer Amount', 'HBAR', 0.00000001, 1000);
-  logger.info(`- Currency: ${JSON.stringify(currencyConfig, null, 2)}`);
-  
-  logger.info('\n🪙 Token Configurations:');
-  const tokenConfig = renderConfigs.tokenId('Token to Transfer');
-  logger.info(`- Token ID: ${JSON.stringify(tokenConfig, null, 2)}`);
-  
-  logger.info('\n📝 Form Control Configurations:');
-  const textConfig = renderConfigs.text('Transaction Memo', 'Enter memo', 'Optional transaction memo');
-  const selectConfig = renderConfigs.select('Network', [
-    { value: 'mainnet', label: 'Mainnet' },
-    { value: 'testnet', label: 'Testnet' },
-    { value: 'previewnet', label: 'Previewnet' }
-  ]);
-  logger.info(`- Text: ${JSON.stringify(textConfig, null, 2)}`);
-  logger.info(`- Select: ${JSON.stringify(selectConfig, null, 2)}`);
-  
-  logger.info('\n✨ These configurations enhance hedera-agent-kit tools with rich UI metadata!');
-}
-
-if (import.meta.url === `file://${process.argv[1]}`) {
-  testHederaKitWrapper()
-    .then(() => {
-      demonstrateRenderConfigs();
-      process.exit(0);
-    })
-    .catch((error) => {
-      console.error('Test execution failed:', error);
-      process.exit(1);
-    });
-}
-
-export { testHederaKitWrapper, demonstrateRenderConfigs };

package/src/services/EntityResolver.ts

@@ -1,128 +0,0 @@
-import { ChatOpenAI } from '@langchain/openai';
-import type { EntityAssociation } from '../memory/SmartMemoryManager';
-
-export interface EntityResolverConfig {
-  apiKey: string;
-  modelName?: string;
-}
-
-/**
- * LLM-based entity resolver that replaces brittle regex patterns
- */
-export class EntityResolver {
-  private llm: ChatOpenAI;
-
-  constructor(config: EntityResolverConfig) {
-    this.llm = new ChatOpenAI({
-      apiKey: config.apiKey,
-      modelName: config.modelName || 'gpt-4o-mini',
-      temperature: 0,
-    });
-  }
-
-  /**
-   * Resolve entity references using LLM instead of regex
-   */
-  async resolveReferences(
-    message: string,
-    entities: EntityAssociation[]
-  ): Promise<string> {
-    if (!entities || entities.length === 0) {
-      return message;
-    }
-
-    const byType = entities.reduce((acc, e) => {
-      if (!acc[e.entityType]) acc[e.entityType] = [];
-      acc[e.entityType].push(e);
-      return acc;
-    }, {} as Record<string, EntityAssociation[]>);
-
-    let context = 'Available entities in memory:\n';
-    for (const [type, list] of Object.entries(byType)) {
-      const recent = list[0];
-      context += `Most recent ${type}: "${recent.entityName}" = ${recent.entityId}\n`;
-      if (list.length > 1) {
-        context += `  (${list.length - 1} other ${type}s in memory)\n`;
-      }
-    }
-
-    const prompt = `Task: Replace entity references with their IDs from memory.
-
-${context}
-
-User message: "${message}"
-
-Instructions:
-- If the user says "the topic" or "that topic" → replace with the most recent topic ID
-- If the user says "the token" or "that token" → replace with the most recent token ID  
-- If the user says "it" or "that" after an action verb → replace with the most recent entity ID
-- Examples:
-  * "submit on the topic" → "submit on 0.0.6543472"
-  * "airdrop the token" → "airdrop 0.0.123456"
-  * "send a message to it" → "send a message to 0.0.6543472"
-
-Return ONLY the message with replacements made. Do not add any explanations.
-Resolved message:`;
-
-    try {
-      const response = await this.llm.invoke(prompt);
-      const resolved = (response.content as string).trim();
-
-      if (resolved !== message && resolved.includes('0.0.')) {
-        return resolved;
-      }
-
-      return message;
-    } catch {
-      return message;
-    }
-  }
-
-  /**
-   * Extract entities from agent response using LLM
-   */
-  async extractEntities(
-    response: unknown,
-    userMessage: string
-  ): Promise<
-    Array<{ id: string; name: string; type: string; transactionId?: string }>
-  > {
-    const text =
-      typeof response === 'string' ? response : JSON.stringify(response);
-
-    const prompt = `Analyze this agent response and extract ONLY newly created entities.
-
-User asked: "${userMessage.substring(0, 200)}"
-
-Agent response: ${text.substring(0, 3000)}
-
-Look for:
-1. Success messages with entity IDs (e.g., "Successfully created topic 0.0.6543472")
-2. Transaction confirmations that created new entities
-3. Entity IDs that appear after words like "created", "new", "successfully"
-
-DO NOT include:
-- Account IDs that already existed (like sender/receiver accounts)
-- Entity IDs that were parameters to the operation
-- Failed operations
-
-Return a JSON array of newly created entities:
-[{"id": "0.0.XXX", "name": "descriptive_name", "type": "topic|token|account"}]
-
-If no entities were created, return: []
-
-JSON:`;
-
-    try {
-      const response = await this.llm.invoke(prompt);
-      const content = response.content as string;
-      const match = content.match(/\[[\s\S]*?\]/);
-      if (match) {
-        const entities = JSON.parse(match[0]);
-        return entities;
-      }
-    } catch {
-    }
-    return [];
-  }
-}