Chain Wallet Authoring

import { AgentWallet, type AgentTool } from '@agentic-wallet/core';
import { z } from 'zod';

export class NewChainWallet extends AgentWallet {
  constructor(readonly rpcUrl: string, readonly address: string) {
    super();
  }

  async balance(input?: { address?: string }) {
    return { address: input?.address ?? this.address, amount: '0' };
  }

  override getTools(): AgentTool[] {
    return [
      ...super.getTools(),
      {
        name: 'newchain_get_balance',
        description: 'Fetch native balance',
        inputSchema: z.object({ address: z.string().optional() }),
        execute: (input) => this.balance(input),
      },
    ];
  }
}