prediction-market-agent-tooling 0.57.2__py3-none-any.whl → 0.57.4__py3-none-any.whl

prediction_market_agent_tooling/markets/omen/data_models.py

@@ -594,7 +594,7 @@ class RealityQuestion(BaseModel):
     updatedTimestamp: int
     contentHash: HexBytes
     questionId: HexBytes  # This is the `id` on question from omen subgraph.
-    answerFinalizedTimestamp: int
+    answerFinalizedTimestamp: int | None
     currentScheduledFinalizationTimestamp: int
 
     @property
@@ -602,8 +602,12 @@ class RealityQuestion(BaseModel):
         return DatetimeUTC.to_datetime_utc(self.updatedTimestamp)
 
     @property
-    def answer_finalized_datetime(self) -> DatetimeUTC:
-        return DatetimeUTC.to_datetime_utc(self.answerFinalizedTimestamp)
+    def answer_finalized_datetime(self) -> DatetimeUTC | None:
+        return (
+            DatetimeUTC.to_datetime_utc(self.answerFinalizedTimestamp)
+            if self.answerFinalizedTimestamp is not None
+            else None
+        )
 
     @property
     def current_scheduled_finalization_datetime(self) -> DatetimeUTC:

prediction_market_agent_tooling/markets/omen/omen.py

@@ -565,13 +565,18 @@ class OmenAgentMarket(AgentMarket):
         omen_markets: dict[HexBytes, OmenMarket] = {
             m.condition.id: m
             for m in sgh.get_omen_binary_markets(
-                limit=sys.maxsize,
+                limit=None,
                 condition_id_in=list(omen_positions_dict.keys()),
             )
         }
+
         if len(omen_markets) != len(omen_positions_dict):
+            missing_conditions_ids = set(
+                omen_position.position.condition_id for omen_position in omen_positions
+            ) - set(market.condition.id for market in omen_markets.values())
             raise ValueError(
-                f"Number of condition ids for markets {len(omen_markets)} and positions {len(omen_positions_dict)} are not equal."
+                f"Number of condition ids for markets {len(omen_markets)} and positions {len(omen_positions_dict)} are not equal. "
+                f"Missing condition ids: {missing_conditions_ids}"
             )
 
         positions = []

prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py

@@ -214,6 +214,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
         question_finalized_before: DatetimeUTC | None,
         question_finalized_after: DatetimeUTC | None,
         question_with_answers: bool | None,
+        question_pending_arbitration: bool | None,
         question_id: HexBytes | None,
         question_id_in: list[HexBytes] | None,
         question_current_answer_before: DatetimeUTC | None,
@@ -226,7 +227,6 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
         category: str | None,
     ) -> dict[str, t.Any]:
         where_stms: dict[str, t.Any] = {
-            "isPendingArbitration": False,
             "outcomes": outcomes,
             "title_not": None,
             "condition_": {},
@@ -239,6 +239,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
             finalized_before=question_finalized_before,
             finalized_after=question_finalized_after,
             with_answers=question_with_answers,
+            pending_arbitration=question_pending_arbitration,
             current_answer_before=question_current_answer_before,
             question_id_in=question_id_in,
             excluded_titles=question_excluded_titles,
@@ -377,6 +378,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
         question_finalized_before: DatetimeUTC | None = None,
         question_finalized_after: DatetimeUTC | None = None,
         question_with_answers: bool | None = None,
+        question_pending_arbitration: bool | None = None,
         question_id: HexBytes | None = None,
         question_id_in: list[HexBytes] | None = None,
         question_current_answer_before: DatetimeUTC | None = None,
@@ -406,6 +408,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
             question_finalized_before=question_finalized_before,
             question_finalized_after=question_finalized_after,
             question_with_answers=question_with_answers,
+            question_pending_arbitration=question_pending_arbitration,
             question_id=question_id,
             question_id_in=question_id_in,
             question_current_answer_before=question_current_answer_before,
@@ -654,6 +657,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
         finalized_before: DatetimeUTC | None,
         finalized_after: DatetimeUTC | None,
         with_answers: bool | None,
+        pending_arbitration: bool | None,
         question_id: HexBytes | None,
         question_id_in: list[HexBytes] | None,
         opened_before: t.Optional[DatetimeUTC],
@@ -700,9 +704,12 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
 
         if with_answers is not None:
             if with_answers:
-                where_stms["answerFinalizedTimestamp_not"] = None
+                where_stms["currentAnswer_not"] = None
             else:
-                where_stms["answerFinalizedTimestamp"] = None
+                where_stms["currentAnswer"] = None
+
+        if pending_arbitration is not None:
+            where_stms["isPendingArbitration"] = pending_arbitration
 
         if question_id_in is not None:
             # Be aware: On Omen subgraph, question's `id` represents `questionId` on reality subgraph. And `id` on reality subraph is just a weird concat of multiple things from the question.
@@ -720,6 +727,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
         finalized_before: DatetimeUTC | None,
         finalized_after: DatetimeUTC | None,
         with_answers: bool | None,
+        pending_arbitration: bool | None,
         question_id: HexBytes | None,
         question_id_in: list[HexBytes] | None,
         opened_before: t.Optional[DatetimeUTC],
@@ -732,7 +740,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
         where_stms: dict[str, t.Any] = {}
 
         if question_id is not None:
-            where_stms["questionId"] = question_id.hex()
+            where_stms["id"] = question_id.hex()
 
         if current_answer_before is not None:
             where_stms["currentAnswerTimestamp_lt"] = to_int_timestamp(
@@ -757,9 +765,12 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
 
         if with_answers is not None:
             if with_answers:
-                where_stms["answerFinalizedTimestamp_not"] = None
+                where_stms["currentAnswer_not"] = None
             else:
-                where_stms["answerFinalizedTimestamp"] = None
+                where_stms["currentAnswer"] = None
+
+        if pending_arbitration is not None:
+            where_stms["isPendingArbitration"] = pending_arbitration
 
         if question_id_in is not None:
             # Be aware: On Omen subgraph, question's `id` represents `questionId` on reality subgraph. And `id` on reality subraph is just a weird concat of multiple things from the question.
@@ -780,6 +791,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
         finalized_before: DatetimeUTC | None = None,
         finalized_after: DatetimeUTC | None = None,
         with_answers: bool | None = None,
+        pending_arbitration: bool | None = None,
         question_id_in: list[HexBytes] | None = None,
         question_id: HexBytes | None = None,
         opened_before: DatetimeUTC | None = None,
@@ -792,6 +804,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
             finalized_before=finalized_before,
             finalized_after=finalized_after,
             with_answers=with_answers,
+            pending_arbitration=pending_arbitration,
             current_answer_before=current_answer_before,
             question_id_in=question_id_in,
             question_id=question_id,
@@ -834,6 +847,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
         question_finalized_before: t.Optional[DatetimeUTC] = None,
         question_finalized_after: t.Optional[DatetimeUTC] = None,
         question_with_answers: bool | None = None,
+        question_pending_arbitration: bool | None = None,
         question_id: HexBytes | None = None,
         question_id_in: list[HexBytes] | None = None,
         question_current_answer_before: DatetimeUTC | None = None,
@@ -853,6 +867,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
             finalized_before=question_finalized_before,
             finalized_after=question_finalized_after,
             with_answers=question_with_answers,
+            pending_arbitration=question_pending_arbitration,
             current_answer_before=question_current_answer_before,
             question_id_in=question_id_in,
             excluded_titles=question_excluded_titles,

prediction_market_agent_tooling/tools/web3_utils.py

@@ -297,12 +297,15 @@ def send_xdai_to(
     from_private_key: PrivateKey,
     to_address: ChecksumAddress,
     value: Wei,
+    data_text: Optional[str] = None,
     tx_params: Optional[TxParams] = None,
     timeout: int = 180,
 ) -> TxReceipt:
     from_address = private_key_to_public_key(from_private_key)
 
     tx_params_new: TxParams = {"value": value, "to": to_address}
+    if data_text is not None:
+        tx_params_new["data"] = Web3.to_bytes(text=data_text)
     if tx_params:
         tx_params_new.update(tx_params)
     tx_params_new = _prepare_tx_params(web3, from_address, tx_params=tx_params_new)

{prediction_market_agent_tooling-0.57.2.dist-info → prediction_market_agent_tooling-0.57.4.dist-info}/METADATA

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

{prediction_market_agent_tooling-0.57.2.dist-info → prediction_market_agent_tooling-0.57.4.dist-info}/RECORD

@@ -45,11 +45,11 @@ prediction_market_agent_tooling/markets/metaculus/api.py,sha256=4TRPGytQQbSdf42D
 prediction_market_agent_tooling/markets/metaculus/data_models.py,sha256=Suxa7xELdYuFNKqvGvFh8qyfVtAg79E-vaQ6dqNZOtA,3261
 prediction_market_agent_tooling/markets/metaculus/metaculus.py,sha256=86TIx6cavEWc8Cv4KpZxSvwiSw9oFybXE3YB49pg-CA,4377
 prediction_market_agent_tooling/markets/omen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-prediction_market_agent_tooling/markets/omen/data_models.py,sha256=MBaWnUDGjYzbnDOHKPGrhAn0CP7iETVmsulRPqTf00I,28270
-prediction_market_agent_tooling/markets/omen/omen.py,sha256=jrpfeMk8SUwk1p5QWCOm3Qt9HOHsux7DIBm77WRHqW4,51582
+prediction_market_agent_tooling/markets/omen/data_models.py,sha256=javEACWZL-9y-BaFpyXu-Rl0tQJxMlOWu_vZ89nqK8w,28387
+prediction_market_agent_tooling/markets/omen/omen.py,sha256=zmERtzp1JNsd-8Bo-chnj30MDJX3rfLc3JPEHObaFFY,51851
 prediction_market_agent_tooling/markets/omen/omen_contracts.py,sha256=baXJwk-jSI3-FV-k239oCNOA4oKz6LT47juX8AKpW3A,28297
 prediction_market_agent_tooling/markets/omen/omen_resolving.py,sha256=iDWdjICGkt968exwCjY-6nsnQyrrNAg3YjnDdP430GQ,9415
-prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py,sha256=IsnWiVSmaR5I82D49MpJpRu7uE-wkMEgubsLzt0qZVI,38125
+prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py,sha256=mZ0CgKfHj0gLFq9plNpBhNqMuclb8V3qNagWfLYcpUc,38806
 prediction_market_agent_tooling/markets/polymarket/api.py,sha256=UZ4_TG8ceb9Y-qgsOKs8Qiv8zDt957QkT8IX2c83yqo,4800
 prediction_market_agent_tooling/markets/polymarket/data_models.py,sha256=Fd5PI5y3mJM8VHExBhWFWEnuuIKxQmIAXgBuoPDvNjw,4341
 prediction_market_agent_tooling/markets/polymarket/data_models_web.py,sha256=VZhVccTApygSKMmy6Au2G02JCJOKJnR_oVeKlaesuSg,12548
@@ -102,9 +102,9 @@ prediction_market_agent_tooling/tools/tavily/tavily_models.py,sha256=5ldQs1pZe6u
 prediction_market_agent_tooling/tools/tavily/tavily_search.py,sha256=Kw2mXNkMTYTEe1MBSTqhQmLoeXtgb6CkmHlcAJvhtqE,3809
 prediction_market_agent_tooling/tools/transaction_cache.py,sha256=K5YKNL2_tR10Iw2TD9fuP-CTGpBbZtNdgbd0B_R7pjg,1814
 prediction_market_agent_tooling/tools/utils.py,sha256=WvuUCHgMCiMq8_wMm5PHNwvLhcdDk2zGKaAM8OUC-qY,6438
-prediction_market_agent_tooling/tools/web3_utils.py,sha256=hWDa7D-jP-iW647t0ATPyiUgKbAq25iH97KUnha25SQ,11944
-prediction_market_agent_tooling-0.57.2.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.57.2.dist-info/METADATA,sha256=tzVRrpZtV9Jty0w9XLwxGagZiC3scgqPP6d2cBhqp1A,8188
-prediction_market_agent_tooling-0.57.2.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-prediction_market_agent_tooling-0.57.2.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.57.2.dist-info/RECORD,,
+prediction_market_agent_tooling/tools/web3_utils.py,sha256=UgD0gfsvkrh59e5si7YjE9NzGxtJo5Rk5gNoRfQ6KDE,12073
+prediction_market_agent_tooling-0.57.4.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.57.4.dist-info/METADATA,sha256=9EOjgKRas5k5LUo8vPMLdTuZQNIDtqcmM-FprXOYPFo,8188
+prediction_market_agent_tooling-0.57.4.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+prediction_market_agent_tooling-0.57.4.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.57.4.dist-info/RECORD,,