diff --git a/pyproject.toml b/pyproject.toml index 890b1262b..f213d2bc9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ build-backend = "flit_core.buildapi" [project] name = "vanta-cli" -version = "2.0.2" +version = "2.0.3" description = "Vanta Network CLI" readme = "README.md" authors = [ diff --git a/vanta_cli/src/commands/collateral/deposit.py b/vanta_cli/src/commands/collateral/deposit.py index 6c5e5ced9..4a5730953 100644 --- a/vanta_cli/src/commands/collateral/deposit.py +++ b/vanta_cli/src/commands/collateral/deposit.py @@ -114,10 +114,12 @@ async def deposit( # Check if source_stake is empty to avoid index error if not source_stake: console.print("[red]❌ No source stake found for this coldkey[/red]") + manager.subtensor_api.close() return None if not matching_stake: console.print(f"[red]❌ No stake found for hotkey {hotkey} on netuid {netuid}[/red]") + manager.subtensor_api.close() return None if verbose: @@ -143,6 +145,9 @@ async def deposit( encoded = manager.encode_extrinsic(extrinsic) decoded = manager.decode_extrinsic(encoded) + # Close the subtensor connection to prevent CLI from hanging + manager.subtensor_api.close() + if verbose: console.print("[cyan]Encoded extrinsic:[/cyan]") console.print(json.dumps(str(encoded), indent=2)) diff --git a/vanta_cli/vanta.py b/vanta_cli/vanta.py index baab26d04..a6755107f 100644 --- a/vanta_cli/vanta.py +++ b/vanta_cli/vanta.py @@ -216,7 +216,7 @@ def asset_select( validate=WV.WALLET_AND_HOTKEY, ) - assets = ["crypto", "forex"] + assets = ["crypto", "forex", "equities"] for idx, asset in enumerate(assets, start=1): console.print(f"{idx}. {asset}")