@suiware/ai-tools
Version:
Pluggable tools for Vercel AI SDK which allow AI assistants to interact with Sui Network and perform various actions.
28 lines (26 loc) • 1.07 kB
JavaScript
import { NaviService } from './chunk-OZX7KUX2.mjs';
import { SuiStakingService } from './chunk-EIQJRNDC.mjs';
import { __async, disableConsoleLog, enableConsoleLog } from './chunk-HFORSGSH.mjs';
import { tool } from 'ai';
import z from 'zod';
var suiWalletBalanceTool = tool({
description: "Get non-zero wallet balances. Note that the nUSDC balance should be displayed as USDC.",
parameters: z.object({}),
execute: () => __async(null, null, function* () {
const originalConsoleLog = disableConsoleLog();
const naviService = new NaviService();
const balances = yield naviService.getWalletNonZeroCoins();
enableConsoleLog(originalConsoleLog);
const suiStakingService = new SuiStakingService();
const stakedSuiBalance = yield suiStakingService.getTotalStakedBalance();
if (Number(stakedSuiBalance) > 0) {
balances["Natively Staked SUI"] = stakedSuiBalance;
}
return {
balances
};
})
});
export { suiWalletBalanceTool };
//# sourceMappingURL=chunk-ICA6CUNG.mjs.map
//# sourceMappingURL=chunk-ICA6CUNG.mjs.map