prediction-market-agent-tooling 0.25.0__tar.gz → 0.26.0__tar.gz

{prediction_market_agent_tooling-0.25.0 → prediction_market_agent_tooling-0.26.0}/PKG-INFO

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

{prediction_market_agent_tooling-0.25.0 → prediction_market_agent_tooling-0.26.0}/prediction_market_agent_tooling/markets/agent_market.py

@@ -2,10 +2,12 @@ import typing as t
 from datetime import datetime
 from enum import Enum
 
+from eth_typing import ChecksumAddress
 from pydantic import BaseModel, field_validator
 
 from prediction_market_agent_tooling.gtypes import Probability
 from prediction_market_agent_tooling.markets.data_models import (
+    Bet,
     BetAmount,
     Currency,
     Position,
@@ -139,6 +141,12 @@ class AgentMarket(BaseModel):
     def get_binary_market(id: str) -> "AgentMarket":
         raise NotImplementedError("Subclasses must implement this method")
 
+    @staticmethod
+    def get_bets_made_since(
+        better_address: ChecksumAddress, start_time: datetime
+    ) -> list[Bet]:
+        raise NotImplementedError("Subclasses must implement this method")
+
     def is_resolved(self) -> bool:
         return self.resolution is not None
 

{prediction_market_agent_tooling-0.25.0 → prediction_market_agent_tooling-0.26.0}/prediction_market_agent_tooling/markets/omen/data_models.py

@@ -15,6 +15,7 @@ from prediction_market_agent_tooling.gtypes import (
     xDai,
 )
 from prediction_market_agent_tooling.markets.data_models import (
+    Bet,
     BetAmount,
     Currency,
     ProfitAmount,
@@ -379,6 +380,14 @@ class OmenBet(BaseModel):
             currency=Currency.xDai,
         )
 
+    def to_bet(self) -> Bet:
+        return Bet(
+            amount=BetAmount(amount=self.collateralAmountUSD, currency=Currency.xDai),
+            outcome=self.boolean_outcome,
+            created_time=self.creation_datetime,
+            market_question=self.title,
+        )
+
     def to_generic_resolved_bet(self) -> ResolvedBet:
         if not self.fpmm.is_resolved_with_valid_answer:
             raise ValueError(

{prediction_market_agent_tooling-0.25.0 → prediction_market_agent_tooling-0.26.0}/prediction_market_agent_tooling/markets/omen/omen.py

@@ -24,6 +24,7 @@ from prediction_market_agent_tooling.markets.agent_market import (
     SortBy,
 )
 from prediction_market_agent_tooling.markets.data_models import (
+    Bet,
     BetAmount,
     Currency,
     Position,
@@ -265,6 +266,19 @@ class OmenAgentMarket(AgentMarket):
             )
         )
 
+    @staticmethod
+    def get_bets_made_since(
+        better_address: ChecksumAddress, start_time: datetime
+    ) -> list[Bet]:
+        bets = OmenSubgraphHandler().get_bets(
+            better_address=better_address, start_time=start_time
+        )
+        # get unique titles
+        seen_titles = {bet.title: bet for bet in bets}
+        bets = list(seen_titles.values())
+        bets.sort(key=lambda x: x.creation_datetime)
+        return [b.to_bet() for b in bets]
+
     def get_contract(
         self,
     ) -> OmenFixedProductMarketMakerContract:

{prediction_market_agent_tooling-0.25.0 → prediction_market_agent_tooling-0.26.0}/pyproject.toml

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