Improve SPL error handling

Author: 0xaguspunk

typescript/.changeset/chilled-bugs-develop.md

@@ -0,0 +1,5 @@
+---
+"@goat-sdk/plugin-spl-token": patch
+---
+
+Improve error handling

typescript/packages/plugins/spl-token/src/parameters.ts

@@ -18,7 +18,7 @@ export class TransferTokenByMintAddressParameters extends createToolParameters(
     z.object({
         mintAddress: z.string().describe("The mint address of the token to transfer"),
         to: z.string().describe("The address to transfer the token to"),
-        amount: z.string().describe("The amount of tokens to transfer"),
+        amount: z.string().describe("The amount of tokens to transfer in base unit"),
     }),
 ) {}
 

typescript/packages/plugins/spl-token/src/spl-token.service.ts

@@ -49,9 +49,24 @@ export class SplTokenService {
         parameters: GetTokenBalanceByMintAddressParameters,
     ) {
         const { walletAddress, mintAddress } = parameters;
-        const tokenAccount = getAssociatedTokenAddressSync(new PublicKey(mintAddress), new PublicKey(walletAddress));
-        const balance = walletClient.getConnection().getTokenAccountBalance(tokenAccount);
-        return balance;
+        try {
+            const tokenAccount = getAssociatedTokenAddressSync(
+                new PublicKey(mintAddress),
+                new PublicKey(walletAddress),
+            );
+
+            const accountExists = await doesAccountExist(walletClient.getConnection(), tokenAccount);
+
+            if (!accountExists) {
+                return 0;
+            }
+
+            const balance = await walletClient.getConnection().getTokenAccountBalance(tokenAccount);
+
+            return balance;
+        } catch (error) {
+            throw new Error(`Failed to get token balance: ${error}`);
+        }
     }
 
     @Tool({