fix: use get_balance instead of get_credit_balance who does not exist anymore

Author: 1yam

src/aleph_client/commands/account.py

@@ -335,11 +335,11 @@ async def balance(
             try:
                 # Fetch user Credits
                 async with AlephHttpClient() as client:
-                    credits_balance = await client.get_credit_balance(address)
+                    credits_balance = await client.get_balance(address).credit_balance
                     infos += [
                         Text("\nCredits:"),
                         Text.from_markup(
-                            f"[bright_cyan] {displayable_amount(credits_balance.credits, decimals=2)}[/bright_cyan]"
+                            f"[bright_cyan] {displayable_amount(credits_balance, decimals=2)}[/bright_cyan]"
                         ),
                     ]
             except Exception as e:

src/aleph_client/commands/credit.py

@@ -48,14 +48,14 @@ async def show(
 
     if address:
         async with AlephHttpClient(api_server=settings.API_HOST) as client:
-            credit = await client.get_credit_balance(address=address)
+            credit = await client.get_balance(address=address)
             if json:
                 typer.echo(credit.model_dump_json(indent=4))
             else:
                 infos = [
                     Text.from_markup(f"Address: {address}\n"),
                     Text("Credits:"),
-                    Text.from_markup(f" {displayable_amount(credit.credits, decimals=2)}"),
+                    Text.from_markup(f" {displayable_amount(credit.credit_balance, decimals=2)}"),
                 ]
                 console.print(
                     Panel(

src/aleph_client/commands/instance/__init__.py

@@ -451,12 +451,14 @@ async def create(
     if payment_type == PaymentType.credit:
         async with AlephHttpClient(api_server=settings.API_HOST) as client:
             try:
-                credit_info = await client.get_credit_balance(address=address)
+                credit_info = await client.get_balance(address=address)
                 if isinstance(compute_unit_price, Price) and compute_unit_price.credit:
                     credit_price = Decimal(str(compute_unit_price.credit)) * tier.compute_units
-                    if credit_info.credits < credit_price:
-                        raise InsufficientFundsError(TokenType.CREDIT, float(credit_price), float(credit_info.credits))
-                    available_funds = credit_info.credits
+                    if credit_info.credit_balance < credit_price:
+                        raise InsufficientFundsError(
+                            TokenType.CREDIT, float(credit_price), float(credit_info.credit_balance)
+                        )
+                    available_funds = credit_info.credit_balance
                 else:
                     echo("No credits price available for this tier.")
                     raise typer.Exit(code=1)

tests/unit/test_instance.py

@@ -290,9 +290,9 @@ def create_mock_client(mock_crn_list_obj, mock_pricing_info_response, mock_setti
                 payment_type=payment_type,
             )
         ),
-        get_credit_balance=AsyncMock(
+        get_balance=AsyncMock(
             return_value=MagicMock(
-                credits=5000,  # Enough credits for testing
+                credit_balance=5000,  # Enough credits for testing
             )
         ),
     )
@@ -339,9 +339,9 @@ def response_get_program_price(ptype):
         create_instance=AsyncMock(return_value=[mock_response_create_instance, 200]),
         get_program_price=None,
         forget=AsyncMock(return_value=(MagicMock(), 200)),
-        get_credit_balance=AsyncMock(
+        get_balance=AsyncMock(
             return_value=MagicMock(
-                credits=5000,  # Enough credits for testing
+                credit_balance=5000,  # Enough credits for testing
             )
         ),
     )