22 lines
700 B
JavaScript
22 lines
700 B
JavaScript
|
import { LAMPORTS_PER_SOL } from "@solana/web3.js"
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Get the balance of SOL or an SPL token for the agent's wallet
|
||
|
|
* @param agent - SolanaAgentKit instance
|
||
|
|
* @param token_address - Optional SPL token mint address. If not provided, returns SOL balance
|
||
|
|
* @returns Promise resolving to the balance as a number (in UI units) or null if account doesn't exist
|
||
|
|
*/
|
||
|
|
export async function get_balance(agent, token_address) {
|
||
|
|
if (!token_address) {
|
||
|
|
return (
|
||
|
|
(await agent.connection.getBalance(agent.wallet_address)) /
|
||
|
|
LAMPORTS_PER_SOL
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
const token_account = await agent.connection.getTokenAccountBalance(
|
||
|
|
token_address
|
||
|
|
)
|
||
|
|
return token_account.value.uiAmount || 0
|
||
|
|
}
|