prediction-market-agent-tooling 0.67.2__py3-none-any.whl → 0.67.3__py3-none-any.whl

prediction_market_agent_tooling/tools/tokens/auto_deposit.py

@@ -3,14 +3,22 @@ from web3 import Web3
 from prediction_market_agent_tooling.config import APIKeys
 from prediction_market_agent_tooling.gtypes import USD, Wei
 from prediction_market_agent_tooling.loggers import logger
+from prediction_market_agent_tooling.markets.seer.seer_contracts import GnosisRouter
+from prediction_market_agent_tooling.markets.seer.seer_subgraph_handler import (
+    SeerSubgraphHandler,
+)
 from prediction_market_agent_tooling.tools.contract import (
     ContractDepositableWrapperERC20BaseClass,
     ContractERC20BaseClass,
     ContractERC20OnGnosisChain,
     ContractERC4626BaseClass,
+    ContractWrapped1155BaseClass,
+    init_collateral_token_contract,
+    to_gnosis_chain_contract,
 )
 from prediction_market_agent_tooling.tools.cow.cow_order import (
     get_sell_token_amount,
+    handle_allowance,
     swap_tokens_waiting,
 )
 from prediction_market_agent_tooling.tools.tokens.main_token import KEEPING_ERC20_TOKEN
@@ -49,6 +57,9 @@ def auto_deposit_collateral_token(
             collateral_token_contract, collateral_amount_wei, api_keys, web3
         )
 
+    elif isinstance(collateral_token_contract, ContractWrapped1155BaseClass):
+        mint_full_set(collateral_token_contract, collateral_amount_wei, api_keys, web3)
+
     elif isinstance(collateral_token_contract, ContractERC20BaseClass):
         auto_deposit_erc20(
             collateral_token_contract, collateral_amount_wei, api_keys, web3
@@ -170,3 +181,49 @@ def auto_deposit_erc20(
         web3=web3,
         slippage_tolerance=slippage_tolerance,
     )
+
+
+def mint_full_set(
+    collateral_token_contract: ContractERC20BaseClass,
+    collateral_amount_wei: Wei,
+    api_keys: APIKeys,
+    web3: Web3 | None,
+) -> None:
+    router = GnosisRouter()
+    # We need to fetch the parent's market collateral token, to split it and get the collateral token
+    # of the child market.
+    seer_subgraph_handler = SeerSubgraphHandler()
+    market = seer_subgraph_handler.get_market_by_wrapped_token(
+        token=collateral_token_contract.address
+    )
+    market_collateral_token = Web3.to_checksum_address(market.collateral_token)
+
+    balance_market_collateral = ContractERC20OnGnosisChain(
+        address=market_collateral_token
+    ).balanceOf(for_address=api_keys.bet_from_address, web3=web3)
+    if balance_market_collateral < collateral_amount_wei:
+        logger.debug(
+            f"Not enough collateral token in the market. Expected {collateral_amount_wei} but got {balance_market_collateral}. Auto-depositing market collateral."
+        )
+        market_collateral_token_contract = to_gnosis_chain_contract(
+            init_collateral_token_contract(market_collateral_token, web3=web3)
+        )
+        auto_deposit_collateral_token(
+            market_collateral_token_contract, collateral_amount_wei, api_keys, web3
+        )
+
+    handle_allowance(
+        api_keys=api_keys,
+        sell_token=market_collateral_token,
+        amount_wei=collateral_amount_wei,
+        for_address=router.address,
+        web3=web3,
+    )
+
+    router.split_position(
+        api_keys=api_keys,
+        collateral_token=market_collateral_token,
+        market_id=Web3.to_checksum_address(market.id),
+        amount=collateral_amount_wei,
+        web3=web3,
+    )

{prediction_market_agent_tooling-0.67.2.dist-info → prediction_market_agent_tooling-0.67.3.dist-info}/METADATA

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

{prediction_market_agent_tooling-0.67.2.dist-info → prediction_market_agent_tooling-0.67.3.dist-info}/RECORD

@@ -1,6 +1,7 @@
 prediction_market_agent_tooling/abis/agentresultmapping.abi.json,sha256=YWyJBpZFlpLoSDmYZYNavTETAh3Hj7SRpI_iNEebMOg,4753
 prediction_market_agent_tooling/abis/debuggingcontract.abi.json,sha256=KdkCWmif_WO421RFKhc03kGJiccFqsxsZNrgCd8EH_4,571
 prediction_market_agent_tooling/abis/depositablewrapper_erc20.abi.json,sha256=m0Wk3uQyLM8apWRRvX4d3u1d77bWVuXfV38D-aV48t0,4612
+prediction_market_agent_tooling/abis/erc1155.abi.json,sha256=AdgD245luodQ69SYxS7QmySPtiFR-pc366RctM9YGfs,6449
 prediction_market_agent_tooling/abis/erc20.abi.json,sha256=b8t6tPUhjBo5dHFq8ipsK-ih_g29dOTbWI0GBZiEWL8,3685
 prediction_market_agent_tooling/abis/erc4626.abi.json,sha256=qvmonhj3tJU5flR4pJjXCSSwB8tgEiJcx4Sc5ilQ5VY,17801
 prediction_market_agent_tooling/abis/erc721.abi.json,sha256=sslrlTfLAkHA7cAJAaAHmGBSsY8_3K6jNK-2OoPUTPs,5396
@@ -26,9 +27,9 @@ prediction_market_agent_tooling/benchmark/utils.py,sha256=vmVTQLER8I7MM_bHFiavrN
 prediction_market_agent_tooling/chains.py,sha256=1qQstoqXMwqwM7k-KH7MjMz8Ei-D83KZByvDbCZpAxs,116
 prediction_market_agent_tooling/config.py,sha256=-kJfdDr-m0R-tGZ1KRI-hJJk0mXDt142CAlvwaJ2N2I,11778
 prediction_market_agent_tooling/data_download/langfuse_data_downloader.py,sha256=VY23h324VKIVkevj1B1O-zL1eEp9AElmcfn6SwYDUSc,14246
-prediction_market_agent_tooling/deploy/agent.py,sha256=-kLE5US4F3tFOhpztO4PHgopz5xTTGU-iHdgrRBIOLI,30623
+prediction_market_agent_tooling/deploy/agent.py,sha256=eFiOaBgaEcUuxUNnqRdXrWen3lEeyhzM9EzLY4NY07Q,30380
 prediction_market_agent_tooling/deploy/agent_example.py,sha256=yS1fWkHynr9MYGNOM2WsCnRWLPaffY4bOc6bIudrdd4,1377
-prediction_market_agent_tooling/deploy/betting_strategy.py,sha256=wiUV_zyB9ASLH-VUETIme0H_IMih8jZwiwEmcRa3W5Q,21125
+prediction_market_agent_tooling/deploy/betting_strategy.py,sha256=iSipsauuAI4Ij1BVF_0w9NJT5n83vouMG7aPnrrTfWY,21149
 prediction_market_agent_tooling/deploy/constants.py,sha256=iobTlZpQD6_2UL9TfoElAnBEbqzIIAKZSsAoMCGhwmA,331
 prediction_market_agent_tooling/deploy/gcp/deploy.py,sha256=CYUgnfy-9XVk04kkxA_5yp0GE9Mw5caYqlFUZQ2j3ks,3739
 prediction_market_agent_tooling/deploy/gcp/kubernetes_models.py,sha256=OsPboCFGiZKsvGyntGZHwdqPlLTthITkNF5rJFvGgU8,2582
@@ -40,7 +41,7 @@ prediction_market_agent_tooling/jobs/jobs_models.py,sha256=DoZ9dlvVhpNrnINiR1uy6
 prediction_market_agent_tooling/jobs/omen/omen_jobs.py,sha256=lCymxn0iH4xDmqouTP2LMORoGCiTzlK1_yqYtx1Njj4,5132
 prediction_market_agent_tooling/loggers.py,sha256=o1HyvwtK1DbuC0YWQwJNqzXLLbSC41gNBkEUxiAziEg,5796
 prediction_market_agent_tooling/logprobs_parser.py,sha256=DBlBQtWX8_URXhzTU3YWIPa76Zx3QDHlx1ARqbgJsVI,5008
-prediction_market_agent_tooling/markets/agent_market.py,sha256=4ktJolOR4z4M7o3qnj2HBnBIA6WgXtYNe_Mbzgp5gTc,20411
+prediction_market_agent_tooling/markets/agent_market.py,sha256=tlNiVhP5m0Wtz4MEjTCCGY-nsJUVGF4JZKuo94y8w1Y,20564
 prediction_market_agent_tooling/markets/base_subgraph_handler.py,sha256=7RaYO_4qAmQ6ZGM8oPK2-CkiJfKmV9MxM-rJlduaecU,1971
 prediction_market_agent_tooling/markets/blockchain_utils.py,sha256=6REOt70v3vnzmtCbuRcUTdwt6htXy9nAfNkLOH3Bv1U,2987
 prediction_market_agent_tooling/markets/categorize.py,sha256=orLZlPaHgeREU66m1amxfWikeV77idV4sZDPB8NgSD0,1300
@@ -48,17 +49,17 @@ prediction_market_agent_tooling/markets/data_models.py,sha256=x4dBbdPb7dd2JiaD15
 prediction_market_agent_tooling/markets/manifold/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prediction_market_agent_tooling/markets/manifold/api.py,sha256=tWnjuqvU8pcCuja2B_ynHeds1iiEFc6QWHjeSO_GSxY,7676
 prediction_market_agent_tooling/markets/manifold/data_models.py,sha256=3z1gFbPMEgCDGqeH-IK8wcvqmIHgLdZX8C2M1UQ7iDw,6740
-prediction_market_agent_tooling/markets/manifold/manifold.py,sha256=wbv2dMWFinxCiNJoGJQuzC9o9XiwmWPJ4fI2dJAx_IQ,5377
+prediction_market_agent_tooling/markets/manifold/manifold.py,sha256=hHloNKYQRyd4OWTCUh5kCk-Kx1m77rOIg8d8bGJ6jvk,5454
 prediction_market_agent_tooling/markets/manifold/utils.py,sha256=DCigEbpGWkXR-RZT_oJrvJhilmxKFAxcWMjUFi0nBBI,530
 prediction_market_agent_tooling/markets/market_fees.py,sha256=YeK3ynjYIguB0xf6sO5iyg9lOdW_HD4C6nbJfiGyRCU,1351
 prediction_market_agent_tooling/markets/markets.py,sha256=jEyYZVubrxnMvPHVO3ofxVXgzwmQ5sPlPx4QAVLy9qc,2777
 prediction_market_agent_tooling/markets/metaculus/api.py,sha256=4TRPGytQQbSdf42DCg2M_JWYPAuNjqZ3eBqaQBLkNks,2736
 prediction_market_agent_tooling/markets/metaculus/data_models.py,sha256=WjPt0MKeJNtoY-8oLQTLC8vQYYQ-dBj8UZoPq-UBYsQ,3288
-prediction_market_agent_tooling/markets/metaculus/metaculus.py,sha256=JO7hO5RdoW0qFZ3HzSMNeAivT6ddqyVTpbq6M-WmonY,5235
+prediction_market_agent_tooling/markets/metaculus/metaculus.py,sha256=kM0U0k90YfLE82ra53JAoCR9uIc9wm4B8l32hLw5imU,5312
 prediction_market_agent_tooling/markets/omen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prediction_market_agent_tooling/markets/omen/cow_contracts.py,sha256=sFaW82u_haL4nze8fjTmnQsOuV0OecunQlAhh1OAw0w,1091
 prediction_market_agent_tooling/markets/omen/data_models.py,sha256=RsBYSbM4deA6Os4kQ3egH3HvwT80tQho6T1yyoATCMs,31103
-prediction_market_agent_tooling/markets/omen/omen.py,sha256=UUeAxVRN0eSuRH2_T2qmYO7yEp3xf2_wv2aLtCtdDk0,50623
+prediction_market_agent_tooling/markets/omen/omen.py,sha256=9ynjn_UL48ach_GQt-TdcrY4ut1gCObkRCb8adePtHA,50700
 prediction_market_agent_tooling/markets/omen/omen_constants.py,sha256=XtRk4vpxwUYkTndfjlcmghA-NOIneV8zdHFdyI7tHhM,487
 prediction_market_agent_tooling/markets/omen/omen_contracts.py,sha256=27-HRngTqfk_wgvttB3GeVHhy_O2YZcz9izo9OufOI0,29991
 prediction_market_agent_tooling/markets/omen/omen_resolving.py,sha256=D-ubf_LumHs_c5rBAAntQ8wGKprtO2V1JZeedmChNIE,11035
@@ -66,15 +67,15 @@ prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py,sha256=2fL
 prediction_market_agent_tooling/markets/polymarket/api.py,sha256=-WW49xTYxzb6wBWZ8t6oOpf9puD6Vph8y2AYBocLGcM,4211
 prediction_market_agent_tooling/markets/polymarket/data_models.py,sha256=iRorGwuC9lXcWiVeOqykPip7FSx41RCAo6FOhyKlokU,4971
 prediction_market_agent_tooling/markets/polymarket/data_models_web.py,sha256=QQoFZp1hSllB9r3yaiglMbEtP3YNCCMm3VmqZ2RDnlo,10576
-prediction_market_agent_tooling/markets/polymarket/polymarket.py,sha256=sNlhjTmmsmEXpjj7vGdLrZaZOJYXTTGrXjaZBgu8P0M,6689
+prediction_market_agent_tooling/markets/polymarket/polymarket.py,sha256=ttUmiQllflT4ZCvaHVlykOx_4upsuZF4xknBDzfJMiE,6766
 prediction_market_agent_tooling/markets/polymarket/polymarket_subgraph_handler.py,sha256=bjgTQXf7HhN0Pc_oy5fFNE6v12F8rK05nq6bozgOYIg,1587
 prediction_market_agent_tooling/markets/polymarket/utils.py,sha256=A_diygWKYyp4WHbxAlAVAuC0S0ZqbEKE80wxL6mxHKQ,1275
-prediction_market_agent_tooling/markets/seer/data_models.py,sha256=CPWJptYHHCINgKIsfOO8b7anqQ_FMfyYL9tzESnnxOw,6103
+prediction_market_agent_tooling/markets/seer/data_models.py,sha256=d8J-hgUmGGhhv41u816O7xvBsi9m-TginULvj82hf_g,6681
 prediction_market_agent_tooling/markets/seer/exceptions.py,sha256=cEObdjluivD94tgOLzmimR7wgQEOt6SRakrYdhsRQtk,112
 prediction_market_agent_tooling/markets/seer/price_manager.py,sha256=PZf6-6zc6DvH1u65wHWyeD55lgG-UGnN_xzBMvrb3ug,7120
-prediction_market_agent_tooling/markets/seer/seer.py,sha256=WX8xFdxnDiXCj4AkXnnjNvb3DJ4d8pQQdNK9bb397RQ,26670
-prediction_market_agent_tooling/markets/seer/seer_contracts.py,sha256=uMzpHpI6_tgfhWxPzupLdUJlZ1P2wr0rRiYjAGClKgU,4984
-prediction_market_agent_tooling/markets/seer/seer_subgraph_handler.py,sha256=Thg9GWnA8wnLR_fdVHDo5k1X0Olk5BIiOcksRFDcvR8,12979
+prediction_market_agent_tooling/markets/seer/seer.py,sha256=MYxefiG3n7QT8R6kbjxkbJsC2iOH6iUm0ZHxCNGohJ0,28536
+prediction_market_agent_tooling/markets/seer/seer_contracts.py,sha256=JqfQNFSRWREPw6pQGpJoh-No5ZlKwmTiALJiAYEuvW8,5516
+prediction_market_agent_tooling/markets/seer/seer_subgraph_handler.py,sha256=Ta4wb-P42Q0_GJfAtHkTjRUlknxwgHmjwdUqrgfHi0s,17133
 prediction_market_agent_tooling/markets/seer/subgraph_data_models.py,sha256=7szvK5we3LF38UthWHg5V3LD6C137O5_WMEcP9Dwl8w,1763
 prediction_market_agent_tooling/markets/seer/swap_pool_handler.py,sha256=kyU65phG7eHUyFMuwQ2HQUIayjBe6PCjBJ_pS7p164s,3434
 prediction_market_agent_tooling/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -86,9 +87,9 @@ prediction_market_agent_tooling/tools/betting_strategies/utils.py,sha256=68zFWUj
 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=pdr9ZYmj4QVUfgVKdvOU6ucYdBpJGdha_FMR_LgtcEs,22912
+prediction_market_agent_tooling/tools/contract.py,sha256=37wW8bzDd5p7bxJ-uX6Zn0Vknt3RZj9dMAaH6SPGRBU,25132
 prediction_market_agent_tooling/tools/costs.py,sha256=EaAJ7v9laD4VEV3d8B44M4u3_oEO_H16jRVCdoZ93Uw,954
-prediction_market_agent_tooling/tools/cow/cow_order.py,sha256=CtCPnBcHSrWmcCuNxqqPHQwyBTL6pLqtSvNJt49nHno,13810
+prediction_market_agent_tooling/tools/cow/cow_order.py,sha256=3f4gscCTfCFoLnRXm0ivmdKnzTjy4d_j-nv-0l1geBM,13887
 prediction_market_agent_tooling/tools/cow/models.py,sha256=Z_XUAMj5AGHZE-A92oRG0DuPeXTSyWnZRIwKQ49LaXU,709
 prediction_market_agent_tooling/tools/cow/semaphore.py,sha256=IJGKRgvXnRSkEt_z1i5-eFIoVMcyhr7HK0JfIz1MXdQ,3738
 prediction_market_agent_tooling/tools/custom_exceptions.py,sha256=Fh8z1fbwONvP4-j7AmV_PuEcoqb6-QXa9PJ9m7guMcM,93
@@ -113,13 +114,13 @@ prediction_market_agent_tooling/tools/perplexity/perplexity_search.py,sha256=Ii-
 prediction_market_agent_tooling/tools/relevant_news_analysis/data_models.py,sha256=95l84aztFaxcRLLcRQ46yKJbIlOEuDAbIGLouyliDzA,1316
 prediction_market_agent_tooling/tools/relevant_news_analysis/relevant_news_analysis.py,sha256=r4MdP5uEMlaCwoa2XQZzGq3oZEqnoo9S4dg8uzXfSOY,5473
 prediction_market_agent_tooling/tools/relevant_news_analysis/relevant_news_cache.py,sha256=kNWq92T11Knb9mYBZlMiZUzOpKgCd-5adanylQUMRJA,3085
-prediction_market_agent_tooling/tools/rephrase.py,sha256=mvprYrZST2LaN3F6n5Ew3IpO_xUMnKpxSMggFcZ3IgM,2319
+prediction_market_agent_tooling/tools/rephrase.py,sha256=tqNqusaf5JeEC8ov11uL0YzUo2Dea5Ucczto77GRSQM,2319
 prediction_market_agent_tooling/tools/safe.py,sha256=o477HGPQv7X_eDoOeYoELCHryiq1_102y_JVhGEPDXw,5165
 prediction_market_agent_tooling/tools/singleton.py,sha256=CiIELUiI-OeS7U7eeHEt0rnVhtQGzwoUdAgn_7u_GBM,729
 prediction_market_agent_tooling/tools/streamlit_user_login.py,sha256=NXEqfjT9Lc9QtliwSGRASIz1opjQ7Btme43H4qJbzgE,3010
 prediction_market_agent_tooling/tools/tavily/tavily_models.py,sha256=5ldQs1pZe6uJ5eDAuP4OLpzmcqYShlIV67kttNFvGS0,342
 prediction_market_agent_tooling/tools/tavily/tavily_search.py,sha256=pPs0qZNfJ7G-1ajfz0iaWOBQyiC0TbcShfrW8T39jtg,3859
-prediction_market_agent_tooling/tools/tokens/auto_deposit.py,sha256=daGv9beH38Hvhnmdvs_FSW7YlQRM_LFxOwo43Y9FSBE,6991
+prediction_market_agent_tooling/tools/tokens/auto_deposit.py,sha256=-b3P0wye-kHEszZ3qRh7h0T0IfRJyq0QkS04sMfDhZE,9174
 prediction_market_agent_tooling/tools/tokens/auto_withdraw.py,sha256=HYLSI_MincuyRGSrSJTuf0cHMcdnyQPrXOUW3o2quBk,2811
 prediction_market_agent_tooling/tools/tokens/main_token.py,sha256=1rbwpdCusPgQIVFuo3m00nBZ_b2lCAoFVm67i-YDcEw,812
 prediction_market_agent_tooling/tools/tokens/slippage.py,sha256=ByBssRY6fBjLdxPx5bpudYfs-Z4gUa8CyOGuLDd5tQk,639
@@ -128,8 +129,8 @@ prediction_market_agent_tooling/tools/tokens/usd.py,sha256=yuW8iPPtcpP4eLH2nORMD
 prediction_market_agent_tooling/tools/transaction_cache.py,sha256=K5YKNL2_tR10Iw2TD9fuP-CTGpBbZtNdgbd0B_R7pjg,1814
 prediction_market_agent_tooling/tools/utils.py,sha256=mbOGoWKalNIm7M2K51TEPGwU9oVp1Z6SPsFaBgbn6ws,7397
 prediction_market_agent_tooling/tools/web3_utils.py,sha256=0r26snqCXGdLKCWA8jpe7DV8x2NPYWZwOy4oyKyDCYk,12615
-prediction_market_agent_tooling-0.67.2.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.67.2.dist-info/METADATA,sha256=dgNwYA_B2Ko36dTkAxhrAxKAfcrsvpIF26gN3sOiM-A,8770
-prediction_market_agent_tooling-0.67.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-prediction_market_agent_tooling-0.67.2.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.67.2.dist-info/RECORD,,
+prediction_market_agent_tooling-0.67.3.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.67.3.dist-info/METADATA,sha256=hBmjI7Am8TkC0AOrGun125L9BieiDwoBq0ifnQheHIo,8770
+prediction_market_agent_tooling-0.67.3.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+prediction_market_agent_tooling-0.67.3.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.67.3.dist-info/RECORD,,