iwa 0.0.2__py3-none-any.whl → 0.0.11__py3-none-any.whl

iwa/plugins/olas/contracts/service.py

@@ -64,10 +64,10 @@ class ServiceRegistryContract(ContractInstance):
         """Get the token address for a service."""
         return self.call("token", service_id)
 
-    def get_agent_params(self, service_id: int, agent_id: int) -> Dict:
-        """Get agent params (slots, bond) for a service."""
-        (slots, bond) = self.call("getAgentParams", service_id, agent_id)
-        return {"slots": slots, "bond": bond}
+    def get_agent_params(self, service_id: int) -> list:
+        """Get agent params (slots, bond) for all agents in a service."""
+        num_ids, params = self.call("getAgentParams", service_id)
+        return [{"slots": p[0], "bond": p[1]} for p in params]
 
     def prepare_approve_tx(
         self,
@@ -213,3 +213,53 @@ class ServiceManagerContract(ContractInstance):
             tx_params={"from": from_address},
         )
         return tx
+
+
+class ServiceRegistryTokenUtilityContract(ContractInstance):
+    """Class to interact with the service registry token utility contract.
+
+    This contract manages token-bonded services, tracking agent bonds and
+    security deposits for services that use ERC20 tokens (like OLAS) instead
+    of native currency.
+    """
+
+    name = "service_registry_token_utility"
+    abi_path = OLAS_ABI_PATH / "service_registry_token_utility.json"
+
+    def get_agent_bond(self, service_id: int, agent_id: int) -> int:
+        """Get the agent bond for a specific agent in a service.
+
+        Args:
+            service_id: The service ID.
+            agent_id: The agent ID within the service.
+
+        Returns:
+            The bond amount in wei.
+
+        """
+        return self.call("getAgentBond", service_id, agent_id)
+
+    def get_operator_balance(self, operator: str, service_id: int) -> int:
+        """Get the operator balance for a service.
+
+        Args:
+            operator: The operator address.
+            service_id: The service ID.
+
+        Returns:
+            The balance amount in wei.
+
+        """
+        return self.call("getOperatorBalance", operator, service_id)
+
+    def get_service_token_deposit(self, service_id: int) -> tuple:
+        """Get the token deposit info for a service.
+
+        Args:
+            service_id: The service ID.
+
+        Returns:
+            Tuple of (token_address, security_deposit).
+
+        """
+        return self.call("mapServiceIdTokenDeposit", service_id)

iwa/plugins/olas/contracts/staking.py

@@ -30,20 +30,19 @@ The staking contract checks that:
 4. Agent bond was deposited during service registration
 """
 
-import logging
 import math
 import time
 from datetime import datetime, timezone
 from enum import Enum
 from typing import Dict, List, Optional, Union
 
+from loguru import logger
+
 from iwa.core.contracts.contract import ContractInstance
 from iwa.core.types import EthereumAddress
 from iwa.plugins.olas.contracts.activity_checker import ActivityCheckerContract
 from iwa.plugins.olas.contracts.base import OLAS_ABI_PATH
 
-logger = logging.getLogger(__name__)
-
 
 class StakingState(Enum):
     """Enum representing the staking state of a service."""

iwa/plugins/olas/plugin.py

@@ -48,8 +48,12 @@ class OlasPlugin(Plugin):
         """Create a new Olas service"""
         wallet = Wallet()
         manager = ServiceManager(wallet)
-        # Note: Manager logic currently depends on internal config state which might need setup
-        manager.create(chain_name, owner, token, bond)
+        manager.create(
+            chain_name=chain_name,
+            service_owner_address_or_tag=owner,
+            token_address_or_tag=token,
+            bond_amount_wei=bond,
+        )
 
     def _get_safe_signers(self, safe_address: str, chain_name: str) -> tuple:
         """Query Safe signers on-chain.
@@ -65,13 +69,16 @@ class OlasPlugin(Plugin):
             from safe_eth.eth import EthereumClient
             from safe_eth.safe import Safe
 
-            from iwa.core.secrets import secrets
+            from iwa.core.chain import ChainInterfaces
 
-            rpc_secret = getattr(secrets, f"{chain_name}_rpc", None)
-            if not rpc_secret:
-                return None, None  # Can't verify, skip
+            try:
+                chain_interface = ChainInterfaces().get(chain_name)
+                if not chain_interface.chain.rpcs:
+                    return None, None
+            except ValueError:
+                return None, None  # Chain not supported/configured
 
-            ethereum_client = EthereumClient(rpc_secret.get_secret_value())
+            ethereum_client = EthereumClient(chain_interface.chain.rpc)
             safe = Safe(safe_address, ethereum_client)
             owners = safe.retrieve_owners()
             return owners, True