@hashgraphonline/hedera-agent-kit/src/langchain/tools/account/get-account-nfts-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 GetAccountNftsZodSchema = z.object({
  accountId: z
    .string()
    .describe('The account ID to get NFTs for (e.g., "0.0.12345")'),
  tokenId: z
    .string()
    .optional()
    .describe('Optional token ID to filter NFTs by (e.g., "0.0.67890")'),
  limit: z
    .number()
    .int()
    .positive()
    .optional()
    .default(100)
    .describe('Maximum number of NFTs to return (default: 100)'),
});

/**
 * Tool for retrieving NFTs owned by a Hedera account.
 * This is a read-only operation that queries the mirror node.
 */
export class HederaGetAccountNftsTool extends BaseHederaQueryTool<
  typeof GetAccountNftsZodSchema
> {
  name = 'hedera-get-account-nfts';
  description =
    'Retrieves all NFTs owned by a Hedera account. Optionally filter by token ID.';
  specificInputSchema = GetAccountNftsZodSchema;
  namespace = 'account';

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

  protected async executeQuery(
    args: z.infer<typeof GetAccountNftsZodSchema>
  ): Promise<unknown> {
    this.logger.info(`Getting NFTs for account ID: ${args.accountId}`);
    
    const nfts = await this.hederaKit.query().getAccountNfts(
      args.accountId,
      args.tokenId,
      args.limit
    );
    
    if (!nfts) {
      return {
        success: false,
        error: `Could not retrieve NFTs for account ${args.accountId}`,
      };
    }

    return {
      success: true,
      accountId: args.accountId,
      tokenId: args.tokenId,
      nftCount: nfts.length,
      nfts,
    };
  }
}