iwa 0.0.32__py3-none-any.whl → 0.0.58__py3-none-any.whl

iwa/plugins/olas/models.py

@@ -134,8 +134,5 @@ class OlasConfig(BaseModel):
         target = multisig_address.lower()
         for service in self.services.values():
             if service.multisig_address and str(service.multisig_address).lower() == target:
-                # The following line is from the Code Edit, but it does not fit syntactically here.
-                # It appears to be from a different file (decoder.py) as indicated by the instruction.
-                # args_str = ", ".join(f"{n}={v}" for n, v in zip(d["arg_names"], decoded, strict=False))
                 return service
         return None

iwa/plugins/olas/service_manager/drain.py

@@ -107,12 +107,18 @@ class DrainManagerMixin:
             logger.error("Service has no multisig address")
             return False, 0
 
-        if not self.olas_config.withdrawal_address:
-            logger.error("No withdrawal address configured in OlasConfig")
-            return False, 0
-
+        withdrawal_address = (
+            str(self.olas_config.withdrawal_address)
+            if self.olas_config.withdrawal_address
+            else str(self.wallet.master_account.address)
+        )
         multisig_address = str(self.service.multisig_address)
-        withdrawal_address = str(self.olas_config.withdrawal_address)
+
+        if not self.olas_config.withdrawal_address:
+            logger.warning(
+                "No withdrawal address configured. Defaulting to master account: "
+                f"{withdrawal_address}"
+            )
 
         # Get OLAS balance of the Safe
         olas_token = ERC20Contract(
@@ -126,8 +132,11 @@ class DrainManagerMixin:
             return False, 0
 
         olas_amount = olas_balance / 1e18
+        withdrawal_tag = self.wallet.get_tag_by_address(withdrawal_address) or withdrawal_address
+        multisig_tag = self.wallet.get_tag_by_address(multisig_address) or multisig_address
+
         logger.info(
-            f"Withdrawing {olas_amount:.4f} OLAS from {multisig_address} to {withdrawal_address}"
+            f"Withdrawing {olas_amount:.4f} OLAS from {multisig_tag} to {withdrawal_tag}"
         )
 
         # Transfer from Safe to withdrawal address
@@ -143,7 +152,7 @@ class DrainManagerMixin:
             logger.error("Failed to transfer OLAS")
             return False, 0
 
-        logger.info(f"Withdrew {olas_amount:.4f} OLAS to {withdrawal_address}")
+        logger.info(f"Withdrew {olas_amount:.4f} OLAS to {withdrawal_tag}")
         return True, olas_amount
 
     def drain_service(

iwa/plugins/olas/service_manager/lifecycle.py

@@ -632,7 +632,7 @@ class LifecycleManagerMixin:
         # Use service_name for consistency with Safe naming
         agent_tag = f"{self.service.service_name}_agent"
         try:
-            agent_account = self.wallet.key_storage.create_account(agent_tag)
+            agent_account = self.wallet.key_storage.generate_new_account(agent_tag)
             agent_account_address = agent_account.address
             logger.info(f"Created new agent account: {agent_account_address}")
 
@@ -855,16 +855,27 @@ class LifecycleManagerMixin:
             _, agent_instances = self.registry.call("getAgentInstances", self.service.service_id)
             service_info = self.registry.get_service(self.service.service_id)
             threshold = service_info["threshold"]
+           # Store the multisig in the wallet with tag
+            multisig_tag = f"{self.service.service_name}_multisig"
+
+            # ARCHIVING LOGIC: If tag is already taken by a different address, rename the old one
+            existing = self.wallet.key_storage.find_stored_account(multisig_tag)
+            if existing and existing.address != multisig_address:
+                archive_tag = f"{multisig_tag}_old_{existing.address[:6]}"
+                logger.info(f"[DEPLOY] Archiving old multisig: {multisig_tag} -> {archive_tag}")
+                try:
+                    self.wallet.key_storage.rename_account(existing.address, archive_tag)
+                except Exception as ex:
+                    logger.warning(f"[DEPLOY] Failed to rename old multisig (collision?): {ex}")
 
             safe_account = StoredSafeAccount(
-                tag=f"{self.service.service_name}_multisig",
+                tag=multisig_tag,
                 address=multisig_address,
                 chains=[self.chain_name],
                 threshold=threshold,
                 signers=agent_instances,
             )
-            self.wallet.key_storage.accounts[multisig_address] = safe_account
-            self.wallet.key_storage.save()
+            self.wallet.key_storage.register_account(safe_account)
             logger.debug("[DEPLOY] Registered multisig in wallet")
         except Exception as e:
             logger.warning(f"[DEPLOY] Failed to register multisig in wallet: {e}")

iwa/plugins/olas/service_manager/staking.py

@@ -211,14 +211,14 @@ class StakingManagerMixin:
         # Helper to safely get min_staking_duration
         try:
             min_duration = staking.min_staking_duration
-            logger.info(f"min_staking_duration: {min_duration}")
+            logger.debug(f"min_staking_duration: {min_duration}")
         except Exception as e:
             logger.error(f"Failed to get min_staking_duration: {e}")
             min_duration = 0
 
         unstake_at = None
         ts_start = info.get("ts_start", 0)
-        logger.info(f"ts_start: {ts_start}")
+        logger.debug(f"ts_start: {ts_start}")
 
         if ts_start > 0:
             try:
@@ -227,12 +227,12 @@ class StakingManagerMixin:
                     unstake_ts,
                     tz=timezone.utc,
                 ).isoformat()
-                logger.info(f"unstake_available_at: {unstake_at} (ts={unstake_ts})")
+                logger.debug(f"unstake_available_at: {unstake_at} (ts={unstake_ts})")
             except Exception as e:
                 logger.error(f"calc error: {e}")
                 pass
         else:
-            logger.warning("ts_start is 0, cannot calculate unstake time")
+            logger.debug("ts_start is 0, cannot calculate unstake time")
 
         return unstake_at, ts_start, min_duration
 

iwa/plugins/olas/tests/test_importer_error_handling.py

@@ -296,6 +296,12 @@ def test_import_key_success(importer):
     importer.key_storage.find_stored_account.return_value = None
     importer.key_storage.accounts = {}
 
+    # Define side effect to update accounts dict when register_account is called
+    def mock_register(acc):
+        importer.key_storage.accounts[acc.address] = acc
+
+    importer.key_storage.register_account.side_effect = mock_register
+
     with patch("iwa.core.keys.EncryptedAccount.encrypt_private_key") as mock_enc:
         mock_enc.return_value = MagicMock(address=addr)
         success, msg = importer._import_key(key, "my_service")
@@ -310,10 +316,17 @@ def test_import_safe_success(importer):
     importer.key_storage.find_stored_account.return_value = None
     importer.key_storage.accounts = {}
 
+    # Define side effect to update accounts dict when register_account is called
+    def mock_register(acc):
+        importer.key_storage.accounts[acc.address] = acc
+
+    importer.key_storage.register_account.side_effect = mock_register
+
     success, msg = importer._import_safe(safe_address, service_name="my_service")
     assert success is True
     assert msg == "ok"
     assert safe_address in importer.key_storage.accounts
+    assert importer.key_storage.accounts[safe_address].tag == "my_service_multisig"
 
 
 def test_import_service_config_success(importer):

iwa/plugins/olas/tests/test_olas_archiving.py

@@ -0,0 +1,73 @@
+"""Tests for Olas multisig archiving logic."""
+
+import unittest
+from unittest.mock import MagicMock, patch
+
+from iwa.core.models import StoredAccount
+from iwa.plugins.olas.contracts.service import ServiceState
+from iwa.plugins.olas.models import Service
+from iwa.plugins.olas.service_manager import ServiceManager
+
+
+class TestOlasArchiving(unittest.TestCase):
+    """Test multisig archiving logic during deployment."""
+
+    def setUp(self):
+        """Set up test fixtures."""
+        self.mock_wallet = MagicMock()
+        # Mock KeyStorage
+        self.mock_key_storage = MagicMock()
+        self.mock_wallet.key_storage = self.mock_key_storage
+
+        # Initialize ServiceManager
+        with patch("iwa.core.models.Config"), \
+             patch("iwa.plugins.olas.service_manager.base.ChainInterfaces"), \
+             patch("iwa.plugins.olas.service_manager.base.ContractCache"), \
+             patch("iwa.plugins.olas.service_manager.base.ServiceRegistryContract"):
+            self.sm = ServiceManager(self.mock_wallet)
+            self.sm.service = Service(service_name="trader_psi", chain_name="gnosis", service_id=1)
+            self.sm.chain_name = "gnosis"
+
+    def test_multisig_archiving_logic(self):
+        """Test that old multisig is archived if tag is taken by different address."""
+        multisig_tag = "trader_psi_multisig"
+        old_address = "0x1111111111111111111111111111111111111111"
+        new_address = "0x2222222222222222222222222222222222222222"
+
+        # 1. Simulate existing multisig in wallet with same tag
+        existing_acc = StoredAccount(address=old_address, tag=multisig_tag)
+        self.mock_key_storage.find_stored_account.return_value = existing_acc
+
+        # 2. Mock required contract calls for deploy()
+        self.sm.registry.get_service.return_value = {
+            "state": ServiceState.FINISHED_REGISTRATION, # DEPLOYED
+            "security_deposit": 0,
+            "multisig": new_address,
+            "threshold": 1,
+            "configHash": b"\x00" * 32
+        }
+        self.sm.registry.call.return_value = (None, []) # getAgentInstances
+
+        # 3. Trigger deploy (archiving happens here)
+        with patch.object(self.mock_wallet, "sign_and_send_transaction", return_value=(True, {"status": 1})), \
+             patch.object(self.sm.registry, "extract_events", return_value=[
+                 {"name": "DeployService", "args": {}},
+                 {"name": "CreateMultisigWithAgents", "args": {"multisig": new_address}}
+             ]), \
+             patch("iwa.plugins.olas.service_manager.lifecycle.get_tx_hash", return_value="0xhash"), \
+             patch("iwa.core.models.StoredSafeAccount") as mock_safe_cls:
+
+            self.sm.deploy(fund_multisig=False)
+
+            # 4. Verify rename_account was called for the old address
+            archive_tag = f"{multisig_tag}_old_{old_address[:6]}"
+            self.mock_key_storage.rename_account.assert_called_with(old_address, archive_tag)
+
+            # 5. Verify new account constructor was called with correct params
+            mock_safe_cls.assert_called()
+            _, kwargs = mock_safe_cls.call_args
+            self.assertEqual(kwargs["address"], new_address)
+            self.assertEqual(kwargs["tag"], multisig_tag)
+
+if __name__ == "__main__":
+    unittest.main()

iwa/plugins/olas/tests/test_olas_view.py

@@ -147,21 +147,25 @@ async def test_olas_view_actions(mock_wallet, mock_olas_config):
 
             app = OlasTestApp(mock_wallet)
             async with app.run_test() as pilot:
-                await pilot.pause()
+                # Wait for initial load to finish
+                await pilot.pause(0.5)
 
                 # 1. Test Claim
                 with patch.object(OlasView, "claim_rewards") as mock_claim:
                     await pilot.click("#claim-gnosis_1")
+                    await pilot.pause()
                     mock_claim.assert_called_with("gnosis:1")
 
                 # 2. Test Unstake
                 with patch.object(OlasView, "unstake_service") as mock_unstake:
                     await pilot.click("#unstake-gnosis_1")
+                    await pilot.pause()
                     mock_unstake.assert_called_with("gnosis:1")
 
                 # 3. Test Checkpoint
                 with patch.object(OlasView, "checkpoint_service") as mock_checkpoint:
                     await pilot.click("#checkpoint-gnosis_1")
+                    await pilot.pause()
                     mock_checkpoint.assert_called_with("gnosis:1")
 
 

iwa/plugins/olas/tests/test_service_manager.py

@@ -65,11 +65,11 @@ def mock_wallet():
     wallet.master_account.address = "0x1234567890123456789012345678901234567890"
     wallet.key_storage = MagicMock()
     wallet.key_storage.get_account.return_value = None  # Default
-    # Mock create_account which returns a StoredAccount or similar
+    # Mock generate_new_account which returns a StoredAccount or similar
     new_acc = MagicMock()
     new_acc.address = "0x0987654321098765432109876543210987654321"
-    wallet.key_storage.create_account.return_value = new_acc
-    wallet.key_storage.create_account.return_value = new_acc
+    wallet.key_storage.generate_new_account.return_value = new_acc
+    wallet.key_storage.generate_new_account.return_value = new_acc
     # Mock account_service
     wallet.account_service = MagicMock()
     wallet.account_service.get_tag_by_address.return_value = "mock_tag"
@@ -219,7 +219,7 @@ def test_register_agent_success(service_manager, mock_wallet):
         "security_deposit": 50000000000000000000,
     }
 
-    # create_account is already mocked
+    # generate_new_account is already mocked
     mock_wallet.send.return_value = "0xMockTxHash"  # wallet.send returns tx_hash
     mock_wallet.sign_and_send_transaction.return_value = (True, {})
     service_manager.registry.extract_events.return_value = [{"name": "RegisterInstance"}]
@@ -331,7 +331,7 @@ def test_register_agent_with_existing_address(service_manager, mock_wallet):
     assert service_manager.register_agent(agent_address=existing_agent) is True
     assert service_manager.service.agent_address == TEST_EXISTING_AGENT_ADDR
     # Should NOT create a new account
-    mock_wallet.key_storage.create_account.assert_not_called()
+    mock_wallet.key_storage.generate_new_account.assert_not_called()
     # Should NOT fund the agent (only for new accounts)
     mock_wallet.send.assert_not_called()
 
@@ -349,7 +349,7 @@ def test_register_agent_creates_new_if_none(service_manager, mock_wallet):
 
     assert service_manager.register_agent() is True
     # Should create a new account
-    mock_wallet.key_storage.create_account.assert_called()
+    mock_wallet.key_storage.generate_new_account.assert_called()
     # Should fund the new agent
     mock_wallet.send.assert_called()
 
@@ -661,7 +661,7 @@ def test_spin_up_with_existing_agent(service_manager, mock_wallet):
     existing_agent = TEST_EXISTING_AGENT_ADDR
     assert service_manager.spin_up(agent_address=existing_agent) is True
     # Verify agent address was not newly created
-    mock_wallet.key_storage.create_account.assert_not_called()
+    mock_wallet.key_storage.generate_new_account.assert_not_called()
 
 
 # --- Tests for wind_down ---

iwa/plugins/olas/tests/test_service_manager_errors.py

@@ -18,7 +18,7 @@ def mock_wallet():
     wallet.master_account.address = VALID_ADDR
     wallet.chain_interface.chain_name = "gnosis"
     wallet.sign_and_send_transaction.return_value = (True, {"transactionHash": b"\x00" * 32})
-    wallet.key_storage.create_account.return_value.address = VALID_ADDR
+    wallet.key_storage.generate_new_account.return_value.address = VALID_ADDR
     wallet.key_storage.get_account.return_value.address = VALID_ADDR
     return wallet
 

iwa/plugins/olas/tests/test_service_manager_flows.py

@@ -17,7 +17,7 @@ def mock_wallet():
     wallet.key_storage.get_account.return_value.address = (
         "0xAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA"
     )
-    wallet.key_storage.create_account.return_value.address = (
+    wallet.key_storage.generate_new_account.return_value.address = (
         "0xBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBB"
     )
     wallet.sign_and_send_transaction.return_value = (True, {"status": 1})

iwa/plugins/olas/tests/test_service_manager_rewards.py

@@ -197,7 +197,11 @@ def test_withdraw_rewards_no_withdrawal_address(mock_wallet):
     )
     manager.olas_config.withdrawal_address = None
 
-    success, amount = manager.withdraw_rewards()
+    with patch("iwa.plugins.olas.service_manager.drain.ERC20Contract") as mock_erc20_cls:
+        mock_erc20 = mock_erc20_cls.return_value
+        mock_erc20.balance_of_wei.return_value = 0
+
+        success, amount = manager.withdraw_rewards()
 
     assert success is False
     assert amount == 0

iwa/tools/drain_accounts.py

@@ -0,0 +1,60 @@
+
+"""Tool to drain specific accounts to a master address."""
+import argparse
+import sys
+from pathlib import Path
+
+from dotenv import load_dotenv
+
+# Ensure src is in pythonpath
+sys.path.append(str(Path(__file__).resolve().parents[2]))
+
+from loguru import logger
+
+from iwa.core.constants import SECRETS_PATH
+from iwa.core.wallet import Wallet
+
+
+def main() -> None:
+    """Run the account draining tool."""
+    parser = argparse.ArgumentParser(description="Drain specific accounts to master.")
+    parser.add_argument(
+        "tags",
+        nargs="+",
+        help="List of account tags to drain (or 'all' for all configured accounts)",
+    )
+    parser.add_argument(
+        "--chain",
+        "-c",
+        default="gnosis",
+        help="Target chain (default: gnosis)",
+    )
+    args = parser.parse_args()
+
+    # Load secrets
+    if SECRETS_PATH.exists():
+        load_dotenv(SECRETS_PATH, override=True)
+
+    wallet = Wallet()
+
+    tags = args.tags
+    if "all" in tags:
+        # Get all configured accounts except master
+        all_accounts = wallet.account_service.get_account_data()
+        tags = [acc.tag for acc in all_accounts.values() if acc.tag != "master"]
+        logger.info(f"Draining ALL accounts: {tags}")
+
+    for tag in tags:
+        logger.info(f"Processing drain for account: {tag}")
+        try:
+            tx_hash = wallet.drain(from_address_or_tag=tag, chain_name=args.chain)
+            if tx_hash:
+                logger.success(f"Drain tx sent for {tag}: {tx_hash}")
+                # Optional: link to explorer
+            else:
+                logger.warning(f"Drain failed or nothing to drain for {tag}")
+        except Exception:
+            logger.exception(f"Error draining {tag}")
+
+if __name__ == "__main__":
+    main()

iwa/tools/list_contracts.py

@@ -101,6 +101,7 @@ def print_table(console, contract_data, chain_name, sort_criterion):
     table.add_column("Available Rewards", justify="right", style="yellow")
     table.add_column("Contract Balance", justify="right", style="blue")
     table.add_column("Epoch End (UTC)", justify="right", style="white")
+    table.add_column("Epoch End (Local)", justify="right", style="white")
 
     for item in contract_data:
         if item.get("error"):
@@ -113,6 +114,7 @@ def print_table(console, contract_data, chain_name, sort_criterion):
                 f"{item['rewards_olas']:,.2f} OLAS",
                 f"{item['balance_olas']:,.2f} OLAS",
                 item["epoch_end"].strftime("%Y-%m-%d %H:%M:%S"),
+                item["epoch_end"].astimezone().strftime("%Y-%m-%d %H:%M:%S"),
             )
     console.print(table)
 

iwa/tui/screens/wallets.py

@@ -330,7 +330,7 @@ class WalletsScreen(VerticalScroll):
 
     def _fetch_single_token_balance(self, address: str, token: str, chain_name: str) -> str:
         """Fetch a single token balance using BalanceService."""
-        val_token = self.wallet.balance_service.get_erc20_balance_with_retry(
+        val_token = self.wallet.balance_service.get_erc20_balance_eth(
             address, token, chain_name
         )
         val_token_str = f"{val_token:.4f}" if val_token is not None else "-"
@@ -442,7 +442,7 @@ class WalletsScreen(VerticalScroll):
             def handler(tag):
                 if tag is not None:
                     tag = tag or f"Account {len(self.wallet.key_storage.accounts) + 1}"
-                    self.wallet.key_storage.create_account(tag)
+                    self.wallet.key_storage.generate_new_account(tag)
                     self.notify(f"Created new EOA: {escape(tag)}")
                     self.refresh_accounts()
 

iwa/web/routers/accounts.py

@@ -68,7 +68,7 @@ def get_accounts(
 def create_eoa(request: Request, req: AccountCreateRequest, auth: bool = Depends(verify_auth)):
     """Create a new EOA account with the given tag."""
     try:
-        wallet.key_storage.create_account(req.tag)
+        wallet.key_storage.generate_new_account(req.tag)
         return {"status": "success"}
     except Exception as e:
         raise HTTPException(status_code=400, detail=str(e)) from None

iwa/web/static/app.js

@@ -2096,13 +2096,24 @@ document.addEventListener("DOMContentLoaded", () => {
                         <span class="value ${
                           isLoading
                             ? ""
-                            : isStaked
-                              ? "staked"
-                              : service.state === "DEPLOYED"
-                                ? "deployed"
-                                : "not-staked"
+                            : staking.staking_state === "EVICTED"
+                              ? "evicted"
+                              : isStaked
+                                ? "staked"
+                                : service.state === "DEPLOYED"
+                                  ? "deployed"
+                                  : "not-staked"
                         }">
-                            ${isLoading ? '<span class="cell-spinner"></span>' : service.state ? service.state : isStaked ? "✓ STAKED" : "○ NOT STAKED"}
+                            ${
+                              isLoading
+                                ? '<span class="cell-spinner"></span>'
+                                : (service.state || "UNKNOWN") +
+                                  (staking.staking_state
+                                    ? `, ${staking.staking_state.replace(/_/g, " ")}`
+                                    : isStaked
+                                      ? ", STAKED"
+                                      : ", NOT STAKED")
+                            }
                         </span>
                     </div>
                     <div class="staking-row">
@@ -2638,11 +2649,11 @@ document.addEventListener("DOMContentLoaded", () => {
 
       // Show select, hide spinner
       select.style.display = "";
-      spinnerDiv.style.display = "none";
+      spinnerDiv.classList.add("hidden");
     } catch (err) {
       select.innerHTML = '<option value="">Error loading contracts</option>';
       select.style.display = "";
-      spinnerDiv.style.display = "none";
+      spinnerDiv.classList.add("hidden");
       confirmBtn.disabled = false;
     }
   };
@@ -2699,7 +2710,8 @@ document.addEventListener("DOMContentLoaded", () => {
 
     // Load staking contracts
     contractSelect.style.display = "none";
-    spinnerDiv.style.display = "block";
+    spinnerDiv.classList.remove("hidden");
+    spinnerDiv.style.display = "block"; // Ensure display block for visibility
     spinnerDiv.innerHTML =
       '<span class="loading-spinner"></span> Loading contracts...';
     submitBtn.disabled = true;

iwa/web/static/style.css

@@ -1101,6 +1101,10 @@ textarea:focus {
   color: var(--warning);
 }
 
+.staking-row .value.evicted {
+  color: var(--error);
+}
+
 .staking-row .value.countdown {
   color: var(--accent-color);
 }

iwa/web/tests/test_web_endpoints.py

@@ -284,7 +284,7 @@ def test_safe_create_request_validation(client):
 
 def test_create_eoa_success(client):
     """Cover create_eoa success (lines 308-315)."""
-    wallet.key_storage.create_account = MagicMock()
+    wallet.key_storage.generate_new_account = MagicMock()
 
     response = client.post("/api/accounts/eoa", json={"tag": "my_wallet"})
     assert response.status_code == 200
@@ -293,7 +293,7 @@ def test_create_eoa_success(client):
 
 def test_create_eoa_error(client):
     """Cover create_eoa error (lines 314-315)."""
-    wallet.key_storage.create_account = MagicMock(side_effect=Exception("Tag exists"))
+    wallet.key_storage.generate_new_account = MagicMock(side_effect=Exception("Tag exists"))
 
     response = client.post("/api/accounts/eoa", json={"tag": "existing"})
     assert response.status_code == 400

{iwa-0.0.32.dist-info → iwa-0.0.58.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iwa
-Version: 0.0.32
+Version: 0.0.58
 Summary: A secure, modular, and plugin-based framework for crypto agents and ops
 Requires-Python: <4.0,>=3.12
 Description-Content-Type: text/markdown
@@ -17,6 +17,7 @@ Requires-Dist: tomli-w>=1.2.0
 Requires-Dist: typer>=0.19.2
 Requires-Dist: web3>=7.13.0
 Requires-Dist: pyyaml<7.0.0,>=6.0.3
+Requires-Dist: ruamel.yaml>=0.18.0
 Requires-Dist: safe-eth-py>=7.14.0
 Requires-Dist: twine>=6.2.0
 Requires-Dist: cowdao-cowpy>=1.0.1
@@ -82,7 +83,7 @@ iwa/
 ├── core/               # Core wallet functionality
 │   ├── keys.py         # KeyStorage - Encrypted key management
 │   ├── wallet.py       # Wallet - High-level interface
-│   ├── chain.py        # ChainInterface - Blockchain interaction with rate limiting
+│   ├── chain/          # Blockchain interface with rate limiting
 │   ├── services/       # Service layer (accounts, balances, transactions)
 │   └── contracts/      # Contract abstractions (ERC20, Safe)
 ├── plugins/            # Protocol integrations
@@ -134,6 +135,9 @@ GNOSIS_RPC=https://rpc.gnosis.io,https://gnosis.drpc.org
 ETHEREUM_RPC=https://mainnet.infura.io/v3/YOUR_KEY
 BASE_RPC=https://mainnet.base.org
 
+# Testing mode (default: true uses Tenderly test RPCs)
+TESTING=false
+
 # Optional
 GNOSISSCAN_API_KEY=your_api_key
 COINGECKO_API_KEY=your_api_key
@@ -150,7 +154,6 @@ just web
 
 # Use CLI
 iwa wallet list --chain gnosis
-iwa wallet balance <address> --chain gnosis
 ```
 
 ### Running Tests