prediction-market-agent-tooling 0.18.0__tar.gz → 0.20.0__tar.gz

{prediction_market_agent_tooling-0.18.0 → prediction_market_agent_tooling-0.20.0}/PKG-INFO

@@ -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 → prediction_market_agent_tooling-0.20.0}/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-0.18.0 → prediction_market_agent_tooling-0.20.0}/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-0.18.0 → prediction_market_agent_tooling-0.20.0}/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 → prediction_market_agent_tooling-0.20.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "prediction-market-agent-tooling"
-version = "0.18.0"
+version = "0.20.0"
 description = "Tools to benchmark, deploy and monitor prediction market agents."
 authors = ["Gnosis"]
 readme = "README.md"