prediction-market-agent-tooling 0.69.4.dev1072__py3-none-any.whl → 0.69.5__py3-none-any.whl

prediction_market_agent_tooling/markets/seer/data_models.py

@@ -163,13 +163,6 @@ class SeerMarketWithQuestions(SeerMarket):
     questions: list[SeerMarketQuestions]
 
 
-class RedeemParams(BaseModel):
-    model_config = ConfigDict(populate_by_name=True)
-    market: ChecksumAddress
-    outcome_indices: list[int] = Field(alias="outcomeIndexes")
-    amounts: list[OutcomeWei]
-
-
 class ExactInputSingleParams(BaseModel):
     # from https://gnosisscan.io/address/0xffb643e73f280b97809a8b41f7232ab401a04ee1#code
     model_config = ConfigDict(populate_by_name=True)

prediction_market_agent_tooling/markets/seer/seer.py

@@ -49,7 +49,6 @@ from prediction_market_agent_tooling.markets.omen.omen_contracts import (
     SeerAgentResultMappingContract,
 )
 from prediction_market_agent_tooling.markets.seer.data_models import (
-    RedeemParams,
     SeerMarket,
     SeerMarketWithQuestions,
 )
@@ -83,6 +82,7 @@ from prediction_market_agent_tooling.tools.cow.cow_order import (
     OrderStatusError,
     get_orders_by_owner,
     get_trades_by_order_uid,
+    handle_allowance,
     swap_tokens_waiting,
     wait_for_order_completion,
 )
@@ -279,6 +279,7 @@ class SeerAgentMarket(AgentMarket):
         """
         We filter the markets using previous trades by the user so that we don't have to process all Seer markets.
         """
+
         trades_by_user = get_seer_transactions(
             api_keys.bet_from_address, RPCConfig().CHAIN_ID
         )
@@ -299,10 +300,12 @@ class SeerAgentMarket(AgentMarket):
         return filtered_markets
 
     @staticmethod
-    def redeem_winnings(api_keys: APIKeys) -> None:
-        web3 = RPCConfig().get_web3()
+    def redeem_winnings(api_keys: APIKeys, web3: Web3 | None = None) -> None:
+        web3 = web3 or RPCConfig().get_web3()
         subgraph = SeerSubgraphHandler()
 
+        # ToDo - Find open positions by user directly
+
         closed_markets = subgraph.get_markets(
             filter_by=FilterBy.RESOLVED, sort_by=SortBy.NEWEST
         )
@@ -311,8 +314,8 @@ class SeerAgentMarket(AgentMarket):
         )
 
         market_balances = {
-            market.id: market.get_outcome_token_balances(
-                api_keys.bet_from_address, web3
+            market.id: list(
+                market.get_outcome_token_balances(api_keys.bet_from_address, web3)
             )
             for market in filtered_markets
         }
@@ -329,17 +332,15 @@ class SeerAgentMarket(AgentMarket):
             try:
                 # GnosisRouter needs approval to use our outcome tokens
                 for i, token in enumerate(market.wrapped_tokens):
-                    ContractERC20OnGnosisChain(
-                        address=Web3.to_checksum_address(token)
-                    ).approve(
-                        api_keys,
+                    handle_allowance(
+                        api_keys=api_keys,
+                        sell_token=Web3.to_checksum_address(token),
+                        amount_to_check_wei=market_balances[market.id][i].as_wei,
                         for_address=gnosis_router.address,
-                        amount_wei=market_balances[market.id][i].as_wei,
                         web3=web3,
                     )
 
                 # We can only ask for redeem of outcome tokens on correct outcomes
-                # TODO: Implement more complex use-cases: https://github.com/gnosis/prediction-market-agent-tooling/issues/850
                 amounts_to_redeem = [
                     (amount if numerator > 0 else OutcomeWei(0))
                     for amount, numerator in zip(
@@ -347,13 +348,13 @@ class SeerAgentMarket(AgentMarket):
                     )
                 ]
 
-                # Redeem!
-                params = RedeemParams(
+                gnosis_router.redeem_to_base(
+                    api_keys,
                     market=Web3.to_checksum_address(market.id),
-                    outcome_indices=list(range(len(market.payout_numerators))),
+                    outcome_indexes=list(range(len(market.payout_numerators))),
                     amounts=amounts_to_redeem,
+                    web3=web3,
                 )
-                gnosis_router.redeem_to_base(api_keys, params=params, web3=web3)
                 logger.info(f"Redeemed market {market.url}.")
             except Exception:
                 logger.exception(

prediction_market_agent_tooling/markets/seer/seer_contracts.py

@@ -8,13 +8,13 @@ from prediction_market_agent_tooling.gtypes import (
     ABI,
     ChecksumAddress,
     OutcomeStr,
+    OutcomeWei,
     TxReceipt,
     Wei,
     xDai,
 )
 from prediction_market_agent_tooling.markets.seer.data_models import (
     ExactInputSingleParams,
-    RedeemParams,
 )
 from prediction_market_agent_tooling.markets.seer.subgraph_data_models import (
     CreateCategoricalMarketsParams,
@@ -102,12 +102,18 @@ class GnosisRouter(ContractOnGnosisChain):
     def redeem_to_base(
         self,
         api_keys: APIKeys,
-        params: RedeemParams,
+        market: ChecksumAddress,
+        outcome_indexes: list[int],
+        amounts: list[OutcomeWei],
         web3: Web3 | None = None,
     ) -> TxReceipt:
-        params_dict = params.model_dump(by_alias=True)
         # We explicity set amounts since OutcomeWei gets serialized as dict
-        params_dict["amounts"] = [amount.value for amount in params.amounts]
+        params_dict = {
+            "market": market,
+            "outcomeIndexes": outcome_indexes,
+            "amounts": [amount.value for amount in amounts],
+        }
+
         receipt_tx = self.send(
             api_keys=api_keys,
             function_name="redeemToBase",
@@ -116,6 +122,22 @@ class GnosisRouter(ContractOnGnosisChain):
         )
         return receipt_tx
 
+    def split_from_base(
+        self,
+        api_keys: APIKeys,
+        market_id: ChecksumAddress,
+        amount_wei: Wei,
+        web3: Web3 | None = None,
+    ) -> TxReceipt:
+        """Splits using xDAI and receives outcome tokens"""
+        return self.send_with_value(
+            api_keys=api_keys,
+            function_name="splitFromBase",
+            amount_wei=amount_wei,
+            function_params=[market_id],
+            web3=web3,
+        )
+
     def split_position(
         self,
         api_keys: APIKeys,

prediction_market_agent_tooling/tools/contract.py

@@ -434,6 +434,25 @@ class ContractWrapped1155BaseClass(ContractERC20BaseClass):
         )
     )
 
+    def factory(self, web3: Web3 | None = None) -> ChecksumAddress:
+        return Web3.to_checksum_address(self.call("factory", web3=web3))
+
+    def mint(
+        self,
+        api_keys: APIKeys,
+        to_address: ChecksumAddress,
+        amount: Wei,
+        tx_params: t.Optional[TxParams] = None,
+        web3: Web3 | None = None,
+    ) -> TxReceipt:
+        return self.send(
+            api_keys=api_keys,
+            function_name="mint",
+            function_params=[to_address, amount],
+            tx_params=tx_params,
+            web3=web3,
+        )
+
 
 class OwnableContract(ContractBaseClass):
     abi: ABI = abi_field_validator(

{prediction_market_agent_tooling-0.69.4.dev1072.dist-info → prediction_market_agent_tooling-0.69.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: prediction-market-agent-tooling
-Version: 0.69.4.dev1072
+Version: 0.69.5
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.13

{prediction_market_agent_tooling-0.69.4.dev1072.dist-info → prediction_market_agent_tooling-0.69.5.dist-info}/RECORD

@@ -72,12 +72,12 @@ prediction_market_agent_tooling/markets/polymarket/polymarket.py,sha256=-zAx64Mc
 prediction_market_agent_tooling/markets/polymarket/polymarket_contracts.py,sha256=x8yvAUPl55nbJALsQAAHJtstahhpH8WcEOj8or_QloQ,1165
 prediction_market_agent_tooling/markets/polymarket/polymarket_subgraph_handler.py,sha256=2b1hVAp3tCPkmp_FFAZwy6baW7CrPhpUSGk7TlrvgtU,1631
 prediction_market_agent_tooling/markets/polymarket/utils.py,sha256=A_diygWKYyp4WHbxAlAVAuC0S0ZqbEKE80wxL6mxHKQ,1275
-prediction_market_agent_tooling/markets/seer/data_models.py,sha256=Y8tIyOyTHfcIYu0b6YfMDxXcDnW87o3Fam4NEHBITmw,7833
+prediction_market_agent_tooling/markets/seer/data_models.py,sha256=nnuclpajQCkb_s52cCf_rwGWdgjMutaIPyfUq7CqX4M,7626
 prediction_market_agent_tooling/markets/seer/exceptions.py,sha256=cEObdjluivD94tgOLzmimR7wgQEOt6SRakrYdhsRQtk,112
 prediction_market_agent_tooling/markets/seer/price_manager.py,sha256=Sz8NkbDd5CYruKe6KZ9GU--Jf8FDSkniE-PVrbqGiPM,9952
-prediction_market_agent_tooling/markets/seer/seer.py,sha256=OBQWj8IXU-6NNztvQqH0STNP_56f9G7DLc8jrgDogjY,31238
+prediction_market_agent_tooling/markets/seer/seer.py,sha256=1coDkaHX66s_lWHjzhLEgoaNGeOo87Sxd83e0YGgNDA,31153
 prediction_market_agent_tooling/markets/seer/seer_api.py,sha256=4iiTWskxWm__oGgUUd1GhV_ItPSrAr0OfnYQ7lKndnQ,1143
-prediction_market_agent_tooling/markets/seer/seer_contracts.py,sha256=p2cW1Tk7Zj5Rq1oJ1Cil42MNJjKg2EPIxCOm9621PKY,5595
+prediction_market_agent_tooling/markets/seer/seer_contracts.py,sha256=LEmZg6PO5GnxI2izxkUYtfU7W58MQJrp8CRoMnF-HSc,6170
 prediction_market_agent_tooling/markets/seer/seer_subgraph_handler.py,sha256=nmeSofxwtdB6osrP0ASmDWhL684U52Qp4bqJRbRFkCE,18794
 prediction_market_agent_tooling/markets/seer/subgraph_data_models.py,sha256=96v41jdNNxBqTCp5g8SLMSrIgeeITVx5IsBjVgm-qdo,2920
 prediction_market_agent_tooling/markets/seer/swap_pool_handler.py,sha256=k_sCEJZLroVDjOVkZ084VKJGNODLGjBGezhsWEZvlH4,3528
@@ -90,7 +90,7 @@ prediction_market_agent_tooling/tools/betting_strategies/utils.py,sha256=MpS3FOM
 prediction_market_agent_tooling/tools/caches/db_cache.py,sha256=rZIGhgijquwwPtp_qncSAPR1SDF2XxIVZL1ir0fgzWw,12127
 prediction_market_agent_tooling/tools/caches/inmemory_cache.py,sha256=ZW5iI5rmjqeAebu5T7ftRnlkxiL02IC-MxCfDB80x7w,1506
 prediction_market_agent_tooling/tools/caches/serializers.py,sha256=vFDx4fsPxclXp2q0sv27j4al_M_Tj9aR2JJP-xNHQXA,2151
-prediction_market_agent_tooling/tools/contract.py,sha256=8k4wF4p19JI3YjVob3o6t4J6sdyPjoSP8jCiAiEVEtM,32305
+prediction_market_agent_tooling/tools/contract.py,sha256=ZuypmdAB82LYpVZE_qvlax13ikwg7D6OV-YafiQYxoE,32870
 prediction_market_agent_tooling/tools/costs.py,sha256=EaAJ7v9laD4VEV3d8B44M4u3_oEO_H16jRVCdoZ93Uw,954
 prediction_market_agent_tooling/tools/cow/cow_order.py,sha256=kdvhWVTb31oHVspQFowDNEIoDtx8hwGTlKsWYGRh3oQ,14050
 prediction_market_agent_tooling/tools/cow/models.py,sha256=2hTW-Ye6SamBBAO32nVo4f3vjAOOySw8ehECReRcA3o,756
@@ -132,8 +132,8 @@ prediction_market_agent_tooling/tools/tokens/usd.py,sha256=DPO-4HBTy1-TZHKL_9CnH
 prediction_market_agent_tooling/tools/transaction_cache.py,sha256=K5YKNL2_tR10Iw2TD9fuP-CTGpBbZtNdgbd0B_R7pjg,1814
 prediction_market_agent_tooling/tools/utils.py,sha256=ruq6P5TFs8CBHxeBLj1Plpx7kuNFPpDgMsJGQgDiRNs,8785
 prediction_market_agent_tooling/tools/web3_utils.py,sha256=CDbaidlLeQ4VHzSg150L7QNfHfGveljSePGuDVFEYqc,13963
-prediction_market_agent_tooling-0.69.4.dev1072.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.69.4.dev1072.dist-info/METADATA,sha256=WzwC0O9_CJf22VVwHY20fP_QA26TFt1ICzmevaVYLUw,8813
-prediction_market_agent_tooling-0.69.4.dev1072.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-prediction_market_agent_tooling-0.69.4.dev1072.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.69.4.dev1072.dist-info/RECORD,,
+prediction_market_agent_tooling-0.69.5.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.69.5.dist-info/METADATA,sha256=Uaa9YSg_wOcxC5vKVW3qxMF6rRj5NJTDXC6buuKqJuU,8805
+prediction_market_agent_tooling-0.69.5.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+prediction_market_agent_tooling-0.69.5.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.69.5.dist-info/RECORD,,