d0zingcat/solana-agent-kit
1
0
Fork 0
mirror of https://github.com/d0zingcat/solana-agent-kit.git synced 2026-05-13 23:16:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: type for get_balance

Browse Source
This commit is contained in:
aryan
2024-12-23 11:29:11 +05:30
parent f84a618c6c
commit c82f59533b
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/agent/index.ts
View File

@@ -84,7 +84,7 @@ export class SolanaAgentKit {
return deploy_collection(this, options);
}
async getBalance(token_address?: PublicKey): Promise<number | null> {
async getBalance(token_address?: PublicKey): Promise<number> {
return get_balance(this, token_address);
}

4
src/tools/get_balance.ts
View File

@@ -10,7 +10,7 @@ import { SolanaAgentKit } from "../index";
export async function get_balance(
agent: SolanaAgentKit,
token_address?: PublicKey,
): Promise<number | null> {
): Promise<number> {
if (!token_address)
return (
(await agent.connection.getBalance(agent.wallet_address)) /
@@ -19,5 +19,5 @@ export async function get_balance(
const token_account =
await agent.connection.getTokenAccountBalance(token_address);
return token_account.value.uiAmount;
return token_account.value.uiAmount || 0;
}