prediction-market-agent-tooling 0.18.0__py3-none-any.whl → 0.20.0__py3-none-any.whl

prediction_market_agent_tooling/deploy/agent.py

@@ -3,7 +3,7 @@ import os
 import tempfile
 import time
 import typing as t
-from datetime import datetime
+from datetime import datetime, timedelta
 
 from loguru import logger
 from pydantic import BaseModel, BeforeValidator
@@ -30,7 +30,10 @@ from prediction_market_agent_tooling.markets.agent_market import (
     SortBy,
 )
 from prediction_market_agent_tooling.markets.data_models import BetAmount
-from prediction_market_agent_tooling.markets.markets import MarketType
+from prediction_market_agent_tooling.markets.markets import (
+    MarketType,
+    have_bet_on_market_since,
+)
 from prediction_market_agent_tooling.markets.omen.omen import (
     redeem_from_all_user_positions,
 )
@@ -40,6 +43,7 @@ from prediction_market_agent_tooling.monitor.langfuse.langfuse_wrapper import (
 from prediction_market_agent_tooling.monitor.monitor_app import (
     MARKET_TYPE_TO_DEPLOYED_AGENT,
 )
+from prediction_market_agent_tooling.tools.is_predictable import is_predictable_binary
 from prediction_market_agent_tooling.tools.utils import DatetimeWithTimezone, utcnow
 
 MAX_AVAILABLE_MARKETS = 20
@@ -80,6 +84,8 @@ class Answer(BaseModel):
 
 
 class DeployableAgent:
+    bet_on_n_markets_per_run: int = 1
+
     def __init__(self) -> None:
         self.langfuse_wrapper = LangfuseWrapper(agent_name=self.__class__.__name__)
         self.load()
@@ -93,11 +99,30 @@ class DeployableAgent:
     def load(self) -> None:
         pass
 
+    def have_bet_on_market_since(self, market: AgentMarket, since: timedelta) -> bool:
+        return have_bet_on_market_since(keys=APIKeys(), market=market, since=since)
+
     def pick_markets(self, markets: t.Sequence[AgentMarket]) -> t.Sequence[AgentMarket]:
         """
-        This method should be implemented by the subclass to pick the markets to bet on. By default, it picks only the first market.
+        Subclasses can implement their own logic instead of this one, or on top of this one.
+        By default, it picks only the first {n_markets_per_run} markets where user didn't bet recently and it's a reasonable question.
         """
-        return markets[:1]
+        picked: list[AgentMarket] = []
+
+        for market in markets:
+            if len(picked) >= self.bet_on_n_markets_per_run:
+                break
+
+            if self.have_bet_on_market_since(market, since=timedelta(hours=24)):
+                continue
+
+            # Do as a last check, as it uses paid OpenAI API.
+            if not is_predictable_binary(market.question):
+                continue
+
+            picked.append(market)
+
+        return picked
 
     def answer_binary_market(self, market: AgentMarket) -> Answer | None:
         """

prediction_market_agent_tooling/markets/markets.py

@@ -1,19 +1,29 @@
 import typing as t
-from datetime import datetime
+from datetime import datetime, timedelta
 from enum import Enum
 
+from prediction_market_agent_tooling.config import APIKeys, PrivateCredentials
 from prediction_market_agent_tooling.markets.agent_market import (
     AgentMarket,
     FilterBy,
     SortBy,
 )
+from prediction_market_agent_tooling.markets.manifold.api import (
+    get_authenticated_user,
+    get_manifold_bets,
+    get_manifold_market,
+)
 from prediction_market_agent_tooling.markets.manifold.manifold import (
     ManifoldAgentMarket,
 )
 from prediction_market_agent_tooling.markets.omen.omen import OmenAgentMarket
+from prediction_market_agent_tooling.markets.omen.omen_subgraph_handler import (
+    OmenSubgraphHandler,
+)
 from prediction_market_agent_tooling.markets.polymarket.polymarket import (
     PolymarketAgentMarket,
 )
+from prediction_market_agent_tooling.tools.utils import should_not_happen, utcnow
 
 
 class MarketType(str, Enum):
@@ -52,3 +62,35 @@ def get_binary_markets(
         excluded_questions=excluded_questions,
     )
     return markets
+
+
+def have_bet_on_market_since(
+    keys: APIKeys, market: AgentMarket, since: timedelta
+) -> bool:
+    start_time = utcnow() - since
+    credentials = PrivateCredentials.from_api_keys(keys)
+    recently_betted_questions = (
+        set(
+            get_manifold_market(b.contractId).question
+            for b in get_manifold_bets(
+                user_id=get_authenticated_user(
+                    keys.manifold_api_key.get_secret_value()
+                ).id,
+                start_time=start_time,
+                end_time=None,
+            )
+        )
+        if isinstance(market, ManifoldAgentMarket)
+        else (
+            set(
+                b.title
+                for b in OmenSubgraphHandler().get_bets(
+                    better_address=credentials.public_key,
+                    start_time=start_time,
+                )
+            )
+            if isinstance(market, OmenAgentMarket)
+            else should_not_happen(f"Uknown market: {market}")
+        )
+    )
+    return market.question in recently_betted_questions

prediction_market_agent_tooling/tools/balances.py

@@ -14,7 +14,7 @@ class Balances(BaseModel):
     wxdai: xDai
 
 
-def get_balances(address: ChecksumAddress, web3: Web3 | None) -> Balances:
+def get_balances(address: ChecksumAddress, web3: Web3 | None = None) -> Balances:
     if not web3:
         web3 = WrappedxDaiContract().get_web3()
     xdai_balance = Wei(web3.eth.get_balance(address))

{prediction_market_agent_tooling-0.18.0.dist-info → prediction_market_agent_tooling-0.20.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prediction-market-agent-tooling
-Version: 0.18.0
+Version: 0.20.0
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.12

{prediction_market_agent_tooling-0.18.0.dist-info → prediction_market_agent_tooling-0.20.0.dist-info}/RECORD

@@ -12,7 +12,7 @@ prediction_market_agent_tooling/benchmark/agents.py,sha256=HPIFJvackW110ch3Ukktb
 prediction_market_agent_tooling/benchmark/benchmark.py,sha256=xiHKzZx5GHSsDerFHMZ9j_LXAXnSaITSvv67iPe3MEU,21095
 prediction_market_agent_tooling/benchmark/utils.py,sha256=iS1BzyXcCMfMm1Rx--1QCH0pHvBTblTndcDQFbTUJ2s,2897
 prediction_market_agent_tooling/config.py,sha256=FI8feNLmA0QYmN-llvyDh3c6tN2ZGl7Eerw1e27v51A,4757
-prediction_market_agent_tooling/deploy/agent.py,sha256=zFO9mIlDSdsbIEIm_3f-q0uW_BNKOHIerc_9MWtjgGE,9103
+prediction_market_agent_tooling/deploy/agent.py,sha256=aXlrfOymjvLdpxgF88XsFBsPnCG0Q3-EqreV1WYqxro,9985
 prediction_market_agent_tooling/deploy/agent_example.py,sha256=KaTJm43cwbBZR2BmTEyB1ohChAqwYlBUqaaVumsWQe4,891
 prediction_market_agent_tooling/deploy/constants.py,sha256=M5ty8URipYMGe_G-RzxRydK3AFL6CyvmqCraJUrLBnE,82
 prediction_market_agent_tooling/deploy/gcp/deploy.py,sha256=CYUgnfy-9XVk04kkxA_5yp0GE9Mw5caYqlFUZQ2j3ks,3739
@@ -27,7 +27,7 @@ prediction_market_agent_tooling/markets/manifold/api.py,sha256=K79zWSqBs3rpkZwpp
 prediction_market_agent_tooling/markets/manifold/data_models.py,sha256=2DZIxwtDp-PH0UWTGiktMFIGAAQoVutI07UsxjNyTyE,5296
 prediction_market_agent_tooling/markets/manifold/manifold.py,sha256=DJZ88r5BGtAugUw5SIyDfzK1S70titba_fwT7OYXuAY,3896
 prediction_market_agent_tooling/markets/manifold/utils.py,sha256=cPPFWXm3vCYH1jy7_ctJZuQH9ZDaPL4_AgAYzGWkoow,513
-prediction_market_agent_tooling/markets/markets.py,sha256=pIYIlTAvXGrGh1XxNlHJQPEGi36_DHQ-WN2m0TCjEn0,1575
+prediction_market_agent_tooling/markets/markets.py,sha256=TXo2qNmjUXhbTK0d-V7hW_aaJDpqP9RWHxm7mayhI0o,3042
 prediction_market_agent_tooling/markets/omen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prediction_market_agent_tooling/markets/omen/data_models.py,sha256=g6A_DQ54m-_3XQO02AM4NliDycsgYvg8hgpt8Dri5cg,14111
 prediction_market_agent_tooling/markets/omen/omen.py,sha256=TbyZ39rcMsM1xpsW1CtHJSKBN3CIufN-O0RwGKW1IyQ,31823
@@ -47,7 +47,7 @@ prediction_market_agent_tooling/monitor/monitor.py,sha256=zTluK6Ha_pis6JWd8SF8Pb
 prediction_market_agent_tooling/monitor/monitor_app.py,sha256=rDxgdDJqSK0ARx0TJFMkS76npkHZJz0__Rp0xTpiRok,4611
 prediction_market_agent_tooling/monitor/monitor_settings.py,sha256=Xiozs3AsufuJ04JOe1vjUri-IAMWHjjmc2ugGGiHNH4,947
 prediction_market_agent_tooling/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-prediction_market_agent_tooling/tools/balances.py,sha256=Tdh8tkbNDqHrsNlvds0pBsa1v-yC0EOfiPXsA8chL_w,742
+prediction_market_agent_tooling/tools/balances.py,sha256=efDKEe-SsuSmGWKuGNiuHEWCGuCrcwjrCdTz1iKFtqw,749
 prediction_market_agent_tooling/tools/betting_strategies/kelly_criterion.py,sha256=IbhQPoKsQnjnag_n_wAL12n8QdCe7tAMRNV2QS8yYxY,3520
 prediction_market_agent_tooling/tools/betting_strategies/market_moving.py,sha256=0qPFA0hNzgUKt49xovwwFaSYrunFgpWZ0Di1Lo3g5qw,4367
 prediction_market_agent_tooling/tools/betting_strategies/minimum_bet_to_win.py,sha256=-FUSuQQgjcWSSnoFxnlAyTeilY6raJABJVM2QKkFqAY,438
@@ -64,8 +64,8 @@ prediction_market_agent_tooling/tools/safe.py,sha256=gNgcY6ugckx3DbupR8VBezDgh1d
 prediction_market_agent_tooling/tools/singleton.py,sha256=CiIELUiI-OeS7U7eeHEt0rnVhtQGzwoUdAgn_7u_GBM,729
 prediction_market_agent_tooling/tools/utils.py,sha256=6MRfLr5xGPEQGobLz_eCsE_p-XaAeOCcmygwgNCIvqE,5044
 prediction_market_agent_tooling/tools/web3_utils.py,sha256=-IAt12kudrx0BaW_IFQu2nAsJJXVGn8hNH43xjQeIw4,9822
-prediction_market_agent_tooling-0.18.0.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.18.0.dist-info/METADATA,sha256=oGAkzPkL-ciNt1KrjWxajZTZs-RCMgNImsAn2LRso6Y,5510
-prediction_market_agent_tooling-0.18.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-prediction_market_agent_tooling-0.18.0.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.18.0.dist-info/RECORD,,
+prediction_market_agent_tooling-0.20.0.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.20.0.dist-info/METADATA,sha256=AY6jt-QfmhXK25mX52MacYm0y667hC1hdpgO2ymMX-k,5510
+prediction_market_agent_tooling-0.20.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+prediction_market_agent_tooling-0.20.0.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.20.0.dist-info/RECORD,,