sei-agent-kit/dist/src/langchain/sei-erc20/balance.js

A package for building AI agents on the SEI blockchain

import { StructuredTool } from "langchain/tools";
import { getTokenAddressFromTicker } from "../../tools";
import { z } from "zod";
const SeiERC20BalanceInputSchema = z.object({
    contract_address: z.string().optional(),
    ticker: z.string().optional(),
});
export class SeiERC20BalanceTool extends StructuredTool {
    seiKit;
    name = "sei_erc20_balance";
    description = `Get the balance of ERC20 tokens in a Sei wallet.

  This tool retrieves token balances without requiring a wallet address (uses connected wallet).

  If neither parameter is provided, returns the native SEI token balance.

  Parameters:
  - contract_address: Optional. The contract address of the token.
  - ticker: Optional. The token symbol/ticker (e.g., "USDC").

  One of these parameters can be used to specify a non-SEI token.`;
    schema = SeiERC20BalanceInputSchema;
    constructor(seiKit) {
        super();
        this.seiKit = seiKit;
    }
    async _call(input) {
        try {
            var balance;
            if (input) {
                let contract_address;
                if (input.ticker) {
                    contract_address = await getTokenAddressFromTicker(input.ticker);
                }
                else if (input.contract_address) {
                    contract_address = input.contract_address;
                }
                balance = await this.seiKit.getERC20Balance(contract_address);
            }
            else {
                balance = await this.seiKit.getERC20Balance();
            }
            return JSON.stringify({
                status: "success",
                balance,
                token: input || "SEI",
            });
        }
        catch (error) {
            return JSON.stringify({
                status: "error",
                message: error.message,
                code: error.code || "UNKNOWN_ERROR",
            });
        }
    }
}
//# sourceMappingURL=balance.js.map