UNPKG

@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
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