thinyai 0.1.11 → 0.1.13

package/dist/bin.js

@@ -1535,18 +1535,13 @@ function explorerTxUrl(network, digest) {
 function suiPlugin(opts) {
   const sig = opts.signer;
   const resolve2 = typeof sig === "function" ? sig : () => sig;
-  const signerOrThrow = () => {
-    const s = resolve2();
-    if (!s) {
-      throw new Error(
-        "No Sui wallet configured yet. Call sui_setup to create or import one (or tell the user to run `thiny sui init`)."
-      );
-    }
-    return s;
+  const SETUP_NEEDED = {
+    ok: false,
+    setupNeeded: true,
+    message: "Sui isn't set up yet. Ask the user which network (testnet or mainnet) and which wallet (generate a new key, import a private key, or use a Rill agent wallet), then call sui_setup. Do NOT retry this tool until setup is complete."
   };
   const requireSimSuccess = opts.policy?.requireSimSuccess ?? true;
-  const executeTx = async (tx, toolName, approvalArgs, reason) => {
-    const signer = signerOrThrow();
+  const executeTx = async (signer, tx, toolName, approvalArgs, reason) => {
     const sim = await signer.devInspect(tx);
     const status = sim.effects.status.status;
     if (requireSimSuccess && status !== "success") {
@@ -1576,10 +1571,11 @@ function suiPlugin(opts) {
       coinType: z4.string().optional().describe("Coin type, e.g. 0x2::sui::SUI (default: SUI).")
     }),
     execute: async ({ address, coinType }) => {
-      const signer = signerOrThrow();
+      const signer = resolve2();
+      if (!signer) return SETUP_NEEDED;
       const owner = address ?? signer.address;
       if (owner === void 0) {
-        throw new Error("sui_balance: no address given and the signer has no key/address.");
+        return { ok: false, message: "No address given and no wallet is set up. Run sui_setup first." };
       }
       const bal = await signer.client.getBalance({ owner, ...coinType ? { coinType } : {} });
       return {
@@ -1595,7 +1591,8 @@ function suiPlugin(opts) {
     description: "Read a Sui object's type and fields by id.",
     parameters: z4.object({ objectId: z4.string().min(1).describe("The object id (0x\u2026).") }),
     execute: async ({ objectId }) => {
-      const signer = signerOrThrow();
+      const signer = resolve2();
+      if (!signer) return SETUP_NEEDED;
       const res = await signer.client.getObject({
         id: objectId,
         options: { showContent: true, showType: true, showOwner: true }
@@ -1617,8 +1614,10 @@ function suiPlugin(opts) {
       unsignedTx: z4.string().min(1).describe("Unsigned PTB \u2014 the JSON string from Transaction.toJSON() (no sender, no gas).")
     }),
     execute: async ({ unsignedTx }) => {
+      const signer = resolve2();
+      if (!signer) return SETUP_NEEDED;
       const tx = Transaction2.from(unsignedTx);
-      return await executeTx(tx, "sui_execute_ptb", { unsignedTx }, "sign and submit a Sui PTB");
+      return await executeTx(signer, tx, "sui_execute_ptb", { unsignedTx }, "sign and submit a Sui PTB");
     }
   });
   const transfer = defineTool({
@@ -1631,7 +1630,8 @@ function suiPlugin(opts) {
       coinType: z4.string().optional().describe("Coin type (default 0x2::sui::SUI).")
     }),
     execute: async ({ recipient, amountMist, coinType }) => {
-      const signer = signerOrThrow();
+      const signer = resolve2();
+      if (!signer) return SETUP_NEEDED;
       const amount = BigInt(amountMist);
       const type = coinType ?? "0x2::sui::SUI";
       const tx = new Transaction2();
@@ -1640,7 +1640,7 @@ function suiPlugin(opts) {
         tx.transferObjects([coin], tx.pure.address(recipient));
       } else {
         const owner = signer.address;
-        if (owner === void 0) throw new Error("sui_transfer: signer has no address.");
+        if (owner === void 0) return { ok: false, message: "Wallet has no address. Run sui_setup." };
         const { data } = await signer.client.getCoins({ owner, coinType: type });
         const [first, ...rest] = data;
         if (!first) throw new Error(`sui_transfer: no ${type} coins owned by ${owner}.`);
@@ -1652,6 +1652,7 @@ function suiPlugin(opts) {
         tx.transferObjects([coin], tx.pure.address(recipient));
       }
       return executeTx(
+        signer,
         tx,
         "sui_transfer",
         { recipient, amountMist, coinType: type },
@@ -1675,6 +1676,8 @@ function suiPlugin(opts) {
       ).optional().describe("Ordered arguments to the function.")
     }),
     execute: async ({ target, typeArguments, args }) => {
+      const signer = resolve2();
+      if (!signer) return SETUP_NEEDED;
       const tx = new Transaction2();
       const built = (args ?? []).map((a) => {
         if (a.kind === "gas") return tx.gas;
@@ -1705,7 +1708,7 @@ function suiPlugin(opts) {
         }
       });
       tx.moveCall({ target, typeArguments: typeArguments ?? [], arguments: built });
-      return await executeTx(tx, "sui_move_call", { target, typeArguments, args }, `Move call ${target}`);
+      return await executeTx(signer, tx, "sui_move_call", { target, typeArguments, args }, `Move call ${target}`);
     }
   });
   return { name: "sui", tools: [balance, object, executePtb, transfer, moveCall] };
@@ -2796,12 +2799,55 @@ async function runCli() {
       return { activeAddress: address, note: `Now signing as ${address}.` };
     }
   });
+  const suiBalancesTool = defineTool({
+    name: "sui_balances",
+    description: "Fetch ALL coin balances across ALL of the user's Sui addresses on a network. Use for 'what's my balance / what coins do I have'. Returns each address with its coins. SUI amounts are also given in whole SUI (1 SUI = 1e9 MIST).",
+    parameters: z7.object({
+      network: z7.enum(["testnet", "mainnet"]).optional().describe("Network (default: the active one)."),
+      address: z7.string().optional().describe("Limit to one address (default: all the user's wallets).")
+    }),
+    execute: async ({ network: network2, address }) => {
+      const net = network2 ?? suiNetwork;
+      const client = net === suiNetwork && suiSignerRef ? suiSignerRef.client : suiSigner({ network: net }).client;
+      const walletAddrs = suiWalletsOf(loadConfig()).map((w) => w.address);
+      if (suiSignerRef?.address && !walletAddrs.includes(suiSignerRef.address)) {
+        walletAddrs.push(suiSignerRef.address);
+      }
+      const addrs = address ? [address] : walletAddrs;
+      if (addrs.length === 0) {
+        return {
+          ok: false,
+          setupNeeded: true,
+          message: "Sui isn't set up. Ask the user which network and wallet (generate / import / Rill), then call sui_setup. Don't retry until then."
+        };
+      }
+      const addresses = [];
+      for (const addr of addrs) {
+        const balances = await client.getAllBalances({ owner: addr });
+        addresses.push({
+          address: addr,
+          coins: balances.map((b) => {
+            const symbol = b.coinType.split("::").pop() ?? b.coinType;
+            const isSui = b.coinType.endsWith("::sui::SUI");
+            return {
+              symbol,
+              coinType: b.coinType,
+              balanceMist: b.totalBalance,
+              ...isSui ? { sui: Number(b.totalBalance) / 1e9 } : {}
+            };
+          })
+        });
+      }
+      return { network: net, addresses };
+    }
+  });
   const walletTools = [
     suiWalletsTool,
     suiCreateWalletTool,
     suiImportWalletTool,
     suiExportWalletTool,
-    suiUseWalletTool
+    suiUseWalletTool,
+    suiBalancesTool
   ];
   const fetchUrlTool = defineTool({
     name: "fetch_url",
@@ -2874,7 +2920,7 @@ HOW TO ACT: When a request maps to one of your tools, CALL THE TOOL automaticall
 YOUR TOOLS:
 \u2022 Memory \u2014 remember_fact: durable memory across sessions. Whenever the user shares anything durable about themselves (name, role, preferences, projects, goals), call remember_fact ONCE to save it. Your known facts are AUTO-INJECTED at the top of every conversation under \u201C[User Memory \u2026]\u201D, so answer \u201Cwhat do you remember / what's my name\u201D directly from that context \u2014 do NOT call recall_memory unless the injected memory is empty and you truly need to re-check. You DO remember across sessions; never say otherwise.
 \u2022 Links \u2014 fetch_url: read ANY URL the user shares (a skill.md, docs, JSON, an API/MCP endpoint). Always fetch shared links instead of saying you can't open URLs.
-` + (webSearchOn ? "\u2022 Web search \u2014 web_search: search the web for anything you don't know (news, prices, docs). web_search FINDS pages by query; fetch_url READS a specific URL \u2014 use them together.\n" : "") + "\u2022 Planning \u2014 update_plan (track multi-step work), delegate_task (hand a focused subtask to a sub-agent).\n\u2022 Sui blockchain \u2014 you transact yourself; NEVER tell the user to install a browser wallet. " + (suiSignerRef ? `The active wallet is on ${suiNetwork} at ${suiSignerRef.address ?? "?"}. ` : "No wallet yet \u2014 call sui_create_wallet (or sui_import_wallet) when the user wants Sui, then have them fund the address. ") + "Wallets: sui_wallets (list ALL the user's wallets + addresses \u2014 use this to answer 'what's my address / what wallets do I have'), sui_create_wallet (new key pair), sui_import_wallet (restore from a suiprivkey), sui_export_wallet (reveal a private key \u2014 only when asked), sui_use_wallet (switch the active wallet). On-chain: sui_balance & sui_object (read), sui_transfer (send SUI/any coin \u2014 amounts in MIST, 1 SUI = 1e9), sui_move_call (call ANY Move function), sui_execute_ptb (sign a builder/Rill PTB). Prefer sui_transfer for sends and sui_move_call for contract calls; confirm details before signing.",
+` + (webSearchOn ? "\u2022 Web search \u2014 web_search: search the web for anything you don't know (news, prices, docs). web_search FINDS pages by query; fetch_url READS a specific URL \u2014 use them together.\n" : "") + "\u2022 Planning \u2014 update_plan (track multi-step work), delegate_task (hand a focused subtask to a sub-agent).\n\u2022 Sui blockchain \u2014 you transact yourself; NEVER tell the user to install a browser wallet. " + (suiSignerRef ? `Sui IS set up. The user's primary (active) wallet is on ${suiNetwork} at ${suiSignerRef.address ?? "?"}. ` : "Sui is NOT set up yet. When the user wants anything Sui-related, FIRST ask if they'd like you to set it up, and ask which network (testnet/mainnet) and which wallet option (generate a new key, import a suiprivkey, or use a Rill agent wallet). Then call sui_setup with their choices. Do not attempt other Sui tools until setup succeeds. ") + "Wallet tools: sui_wallets (list ALL wallets + which is primary/active \u2014 answer 'what's my address' from this), sui_balances (ALL coins across ALL addresses on a network \u2014 answer 'what's my balance' with this), sui_create_wallet, sui_import_wallet, sui_export_wallet (reveal a key only when asked), sui_use_wallet (switch primary). Never overwrite or replace an existing wallet \u2014 adding a wallet keeps the others. If it's unclear which wallet to use, ask the user which is their primary. On-chain: sui_balance & sui_object (read), sui_transfer (send SUI/any coin \u2014 amounts in MIST, 1 SUI = 1e9), sui_move_call (call ANY Move function), sui_execute_ptb (sign a builder/Rill PTB). Prefer sui_transfer for sends and sui_move_call for contract calls; confirm details before signing.\nWhen ANY Sui tool fails, do NOT paste raw JSON or stack traces \u2014 explain what went wrong in one plain sentence and what to do next. Present balances/results readably (SUI amounts in whole SUI, short addresses), and keep every answer brief and to the point.",
     tools: [echoTool, suiSetupTool, ...walletTools, fetchUrlTool, ...webTools],
     plugins: [
       {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "thinyai",
-  "version": "0.1.11",
+  "version": "0.1.13",
   "description": "Thiny AI — a beautiful terminal agent: interactive chat, tools, Walrus memory, and Sui execution.",
   "type": "module",
   "license": "MIT",
@@ -38,16 +38,16 @@
     "tsup": "^8.5.1",
     "typescript": "^5.5.0",
     "@thiny/core": "0.1.0",
-    "@thiny/logger-pino": "0.1.0",
     "@thiny/memory-memwal": "0.1.0",
-    "@thiny/model-aisdk": "0.1.0",
+    "@thiny/logger-pino": "0.1.0",
     "@thiny/mcp": "0.1.0",
-    "@thiny/plugin-sui": "0.1.0",
+    "@thiny/walrus": "0.1.0",
     "@thiny/plugin-agents": "0.1.0",
+    "@thiny/plugin-sui": "0.1.0",
     "@thiny/plugin-web-search": "0.1.0",
-    "@thiny/walrus": "0.1.0",
-    "@thiny/signer-sui": "0.1.0",
-    "@thiny/skills": "0.1.0"
+    "@thiny/model-aisdk": "0.1.0",
+    "@thiny/skills": "0.1.0",
+    "@thiny/signer-sui": "0.1.0"
   },
   "author": "Thiny AI",
   "engines": {