olas-operate-middleware 0.10.9__py3-none-any.whl → 0.10.11__py3-none-any.whl

{olas_operate_middleware-0.10.9.dist-info → olas_operate_middleware-0.10.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: olas-operate-middleware
-Version: 0.10.9
+Version: 0.10.11
 Summary: 
 Author: David Vilela
 Author-email: dvilelaf@gmail.com

{olas_operate_middleware-0.10.9.dist-info → olas_operate_middleware-0.10.11.dist-info}/RECORD

@@ -79,7 +79,7 @@ operate/services/__init__.py,sha256=isrThS-Ccu5Sc15JZgkN4uTAVaSg-NwUUSDeTyJEqLk,
 operate/services/agent_runner.py,sha256=6tJePUJmlRxlIugT2fDaCJHSrQlDnl1t9pbg3-7EmCQ,7560
 operate/services/deployment_runner.py,sha256=Su73o7cdH6fkQfj468K77J04a_TWiokJwbMyVQ25xko,27067
 operate/services/health_checker.py,sha256=2KSEDxG3YmGolUDU--648ny0UJpTAAKvxkcr_VZQv-I,9654
-operate/services/manage.py,sha256=cmZ2pb3P1pHZrOdLAYkM0Azxv81dZdNQwnanM5ag7Zo,118090
+operate/services/manage.py,sha256=5yUZPOFZYHVBV8KSZfGQf9JGsoTFZgfaCUsYbQdPlSw,118137
 operate/services/protocol.py,sha256=FoJmHz5nukI01-VYtReMsIl3JT5RgZZSLX-mf9EXQ90,62903
 operate/services/service.py,sha256=ZbkTryuK3kBkdvhxd9QDyXRxRFsGGG9WZqbLGOUPloc,39301
 operate/services/utils/__init__.py,sha256=TvioaZ1mfTRUSCtrQoLNAp4WMVXyqEJqFJM4PxSQCRU,24
@@ -91,8 +91,8 @@ operate/utils/ssl.py,sha256=O5DrDoZD4T4qQuHP8GLwWUVxQ-1qXeefGp6uDJiF2lM,4308
 operate/wallet/__init__.py,sha256=NGiozD3XhvkBi7_FaOWQ8x1thZPK4uGpokJaeDY_o2w,813
 operate/wallet/master.py,sha256=jLFLcRgO6ADok2DigpK59PFJ4mnc_0H9a4RRioYzlvw,30811
 operate/wallet/wallet_recovery_manager.py,sha256=sXEZyvFMePxQKf9NJg4HT90mPg4-7ZcTbvggMnKKzhA,7795
-olas_operate_middleware-0.10.9.dist-info/LICENSE,sha256=mdBDB-mWKV5Cz4ejBzBiKqan6Z8zVLAh9xwM64O2FW4,11339
-olas_operate_middleware-0.10.9.dist-info/METADATA,sha256=ljzrFEYIuZlVQMFuMZtspZeau_-1Yp0SvIeyPJiONP8,2035
-olas_operate_middleware-0.10.9.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-olas_operate_middleware-0.10.9.dist-info/entry_points.txt,sha256=dM1g2I7ODApKQFcgl5J4NGA7pfBTo6qsUTXM-j2OLlw,44
-olas_operate_middleware-0.10.9.dist-info/RECORD,,
+olas_operate_middleware-0.10.11.dist-info/LICENSE,sha256=mdBDB-mWKV5Cz4ejBzBiKqan6Z8zVLAh9xwM64O2FW4,11339
+olas_operate_middleware-0.10.11.dist-info/METADATA,sha256=xwCWaxrKY3mwm1EYluGPODMlj_fdPGyPdoU7xt6m7_U,2036
+olas_operate_middleware-0.10.11.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+olas_operate_middleware-0.10.11.dist-info/entry_points.txt,sha256=dM1g2I7ODApKQFcgl5J4NGA7pfBTo6qsUTXM-j2OLlw,44
+olas_operate_middleware-0.10.11.dist-info/RECORD,,

operate/services/manage.py

@@ -1895,21 +1895,23 @@ class ServiceManager:
         service = self.load(service_config_id=service_config_id)
         chain_config = service.chain_configs[chain]
         ledger_config = chain_config.ledger_config
-        staking_program_id = chain_config.chain_data.user_params.staking_program_id
         wallet = self.wallet_manager.load(ledger_config.chain.ledger_type)
         ledger_api = wallet.ledger_api(chain=ledger_config.chain, rpc=ledger_config.rpc)
         self.logger.info(
             f"OLAS Balance on service Safe {chain_config.chain_data.multisig}: "
             f"{get_asset_balance(ledger_api, OLAS[Chain(chain)], chain_config.chain_data.multisig)}"
         )
+        current_staking_program = self._get_current_staking_program(
+            service=service, chain=chain
+        )
         staking_contract = get_staking_contract(
             chain=ledger_config.chain,
-            staking_program_id=staking_program_id,
+            staking_program_id=current_staking_program,
         )
         if staking_contract is None:
             raise RuntimeError(
-                "No staking contract found for the current staking_program_id: "
-                f"{staking_program_id}. Not claiming the rewards."
+                "No staking contract found for the "
+                f"{current_staking_program=}. Not claiming the rewards."
             )
 
         sftxb = self.get_eth_safe_tx_builder(ledger_config=ledger_config)
@@ -1925,18 +1927,21 @@ class ServiceManager:
             .add(
                 sftxb.get_claiming_data(
                     service_id=chain_config.chain_data.token,
-                    staking_contract=get_staking_contract(
-                        chain=ledger_config.chain,
-                        staking_program_id=staking_program_id,
-                    ),
+                    staking_contract=staking_contract,
                 )
             )
             .settle()
         )
 
+        if receipt.status != 1:
+            self.logger.error(
+                f"Failed to claim staking rewards. Tx hash: {receipt.tx_hash}"
+            )
+            return 0
+
         # transfer claimed amount from agents safe to master safe
         # TODO: remove after staking contract directly starts sending the rewards to master safe
-        amount_claimed = receipt["logs"][0]["data"]
+        amount_claimed = int(receipt["logs"][0]["data"].hex(), 16)
         self.logger.info(f"Claimed amount: {amount_claimed}")
         ethereum_crypto = KeysManager().get_crypto_instance(service.agent_addresses[0])
         transfer_erc20_from_safe(
@@ -2323,8 +2328,8 @@ class ServiceManager:
                         await loop.run_in_executor(
                             executor,
                             self.claim_on_chain_from_safe,
-                            service_config_id=service_config_id,
-                            chain=service.home_chain,
+                            service_config_id,
+                            service.home_chain,
                         )
                     except Exception:  # pylint: disable=broad-except
                         logging.info(