@hashgraphonline/hedera-agent-kit/src/langchain/tools/scs/get-contracts-tool.ts

Build LLM-powered applications that interact with the Hedera Network. Create conversational agents that can understand user requests in natural language and execute Hedera transactions, or build backend systems that leverage AI for on-chain operations.

import { z } from 'zod';
import {
  BaseHederaQueryTool,
  BaseHederaQueryToolParams,
} from '../common/base-hedera-query-tool';

const GetContractsZodSchema = z.object({
  contractId: z.string().optional().describe('Filter by specific contract ID'),
  limit: z
    .number()
    .optional()
    .describe('Maximum number of contracts to return'),
  order: z.enum(['asc', 'desc']).optional().describe('Order of results'),
});

/**
 * Tool for retrieving contract entities from the network.
 */
export class HederaGetContractsTool extends BaseHederaQueryTool<
  typeof GetContractsZodSchema
> {
  name = 'hedera-get-contracts';
  description =
    'Retrieves contract entities from the Hedera network with optional filtering.';
  specificInputSchema = GetContractsZodSchema;
  namespace = 'scs';

  constructor(params: BaseHederaQueryToolParams) {
    super(params);
  }

  protected async executeQuery(
    args: z.infer<typeof GetContractsZodSchema>
  ): Promise<unknown> {
    this.logger.info('Getting contracts from the network');

    const contracts = await this.hederaKit.query().getContracts(args);

    if (contracts === null) {
      return {
        success: false,
        error: 'Could not retrieve contracts from the network',
      };
    }

    return {
      success: true,
      contracts,
      count: contracts.length,
    };
  }
}