prediction-market-agent-tooling 0.65.0__tar.gz → 0.65.2__tar.gz

{prediction_market_agent_tooling-0.65.0 → prediction_market_agent_tooling-0.65.2}/PKG-INFO

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

{prediction_market_agent_tooling-0.65.0 → prediction_market_agent_tooling-0.65.2}/prediction_market_agent_tooling/config.py

@@ -11,6 +11,7 @@ from pydantic_settings import BaseSettings, SettingsConfigDict
 from safe_eth.eth import EthereumClient
 from safe_eth.safe.safe import SafeV141
 from web3 import Account, Web3
+from web3._utils.http import construct_user_agent
 
 from prediction_market_agent_tooling.chains import ETHEREUM_ID, GNOSIS_CHAIN_ID
 from prediction_market_agent_tooling.deploy.gcp.utils import gcp_get_secret_value
@@ -280,7 +281,9 @@ class RPCConfig(BaseSettings):
     )
 
     ETHEREUM_RPC_URL: URI = Field(default=URI("https://rpc.eth.gateway.fm"))
+    ETHEREUM_RPC_BEARER: SecretStr | None = None
     GNOSIS_RPC_URL: URI = Field(default=URI("https://rpc.gnosis.gateway.fm"))
+    GNOSIS_RPC_BEARER: SecretStr | None = None
     CHAIN_ID: ChainID = Field(default=GNOSIS_CHAIN_ID)
 
     @property
@@ -300,15 +303,33 @@ class RPCConfig(BaseSettings):
         return check_not_none(self.CHAIN_ID, "CHAIN_ID missing in the environment.")
 
     def chain_id_to_rpc_url(self, chain_id: ChainID) -> URI:
-        if chain_id == ETHEREUM_ID:
-            return self.ethereum_rpc_url
-        elif chain_id == GNOSIS_CHAIN_ID:
-            return self.gnosis_rpc_url
-        else:
-            raise ValueError(f"Unsupported chain ID: {chain_id}")
+        return {
+            ETHEREUM_ID: self.ethereum_rpc_url,
+            GNOSIS_CHAIN_ID: self.gnosis_rpc_url,
+        }[chain_id]
+
+    def chain_id_to_rpc_bearer(self, chain_id: ChainID) -> SecretStr | None:
+        return {
+            ETHEREUM_ID: self.ETHEREUM_RPC_BEARER,
+            GNOSIS_CHAIN_ID: self.GNOSIS_RPC_BEARER,
+        }[chain_id]
 
     def get_web3(self) -> Web3:
-        return Web3(Web3.HTTPProvider(self.gnosis_rpc_url))
+        headers = {
+            "Content-Type": "application/json",
+            "User-Agent": construct_user_agent(str(type(self))),
+        }
+        if bearer := self.chain_id_to_rpc_bearer(self.chain_id):
+            headers["Authorization"] = f"Bearer {bearer.get_secret_value()}"
+
+        return Web3(
+            Web3.HTTPProvider(
+                self.chain_id_to_rpc_url(self.chain_id),
+                request_kwargs={
+                    "headers": headers,
+                },
+            )
+        )
 
 
 class CloudCredentials(BaseSettings):

{prediction_market_agent_tooling-0.65.0 → prediction_market_agent_tooling-0.65.2}/prediction_market_agent_tooling/loggers.py

@@ -134,6 +134,7 @@ def patch_logger(force_patch: bool = False) -> None:
             handler,
             level=config.LOG_LEVEL.value,
             colorize=False,
+            catch=False,
         )
 
     # Use logging module for warnings.

{prediction_market_agent_tooling-0.65.0 → prediction_market_agent_tooling-0.65.2}/prediction_market_agent_tooling/markets/seer/price_manager.py

@@ -122,10 +122,15 @@ class PriceManager:
         # Inspired by https://github.com/seer-pm/demo/blob/ca682153a6b4d4dd3dcc4ad8bdcbe32202fc8fe7/web/src/hooks/useMarketOdds.ts#L15
         price_data: dict[HexAddress, CollateralToken] = {}
 
-        for wrapped_token in self.seer_market.wrapped_tokens:
+        for idx, wrapped_token in enumerate(self.seer_market.wrapped_tokens):
             price = self.get_price_for_token(
                 token=Web3.to_checksum_address(wrapped_token),
             )
+            # It's okay if invalid (last) outcome has price 0, but not the other outcomes.
+            if price is None and idx != len(self.seer_market.wrapped_tokens) - 1:
+                raise ValueError(
+                    f"Couldn't get price for {wrapped_token} for market {self.seer_market.url}."
+                )
             price_data[wrapped_token] = (
                 price if price is not None else CollateralToken.zero()
             )
@@ -137,10 +142,9 @@ class PriceManager:
             sum(price_data.values(), start=CollateralToken.zero())
             == CollateralToken.zero()
         ):
-            return {
-                OutcomeStr(outcome): Probability(0)
-                for outcome in self.seer_market.outcomes
-            }
+            raise ValueError(
+                f"All prices for market {self.seer_market.url} are zero. This shouldn't happen."
+            )
 
         for outcome_token, price in price_data.items():
             old_price = price

{prediction_market_agent_tooling-0.65.0 → prediction_market_agent_tooling-0.65.2}/prediction_market_agent_tooling/markets/seer/seer_subgraph_handler.py

@@ -9,7 +9,7 @@ from prediction_market_agent_tooling.deploy.constants import (
     NO_OUTCOME_LOWERCASE_IDENTIFIER,
     YES_OUTCOME_LOWERCASE_IDENTIFIER,
 )
-from prediction_market_agent_tooling.gtypes import ChecksumAddress
+from prediction_market_agent_tooling.gtypes import ChecksumAddress, Wei
 from prediction_market_agent_tooling.loggers import logger
 from prediction_market_agent_tooling.markets.agent_market import FilterBy, SortBy
 from prediction_market_agent_tooling.markets.base_subgraph_handler import (
@@ -77,6 +77,7 @@ class SeerSubgraphHandler(BaseSubgraphHandler):
     @staticmethod
     def _build_where_statements(
         filter_by: FilterBy,
+        outcome_supply_gt_if_open: Wei,
         include_conditional_markets: bool = False,
         include_categorical_markets: bool = True,
     ) -> dict[Any, Any]:
@@ -88,6 +89,7 @@ class SeerSubgraphHandler(BaseSubgraphHandler):
             case FilterBy.OPEN:
                 and_stms["openingTs_gt"] = now
                 and_stms["hasAnswers"] = False
+                and_stms["outcomesSupply_gt"] = outcome_supply_gt_if_open.value
             case FilterBy.RESOLVED:
                 # We consider RESOLVED == CLOSED (on Seer UI)
                 and_stms["payoutReported"] = True
@@ -154,6 +156,7 @@ class SeerSubgraphHandler(BaseSubgraphHandler):
         filter_by: FilterBy,
         sort_by: SortBy = SortBy.NONE,
         limit: int | None = None,
+        outcome_supply_gt_if_open: Wei = Wei(0),
         include_conditional_markets: bool = True,
         include_categorical_markets: bool = True,
     ) -> list[SeerMarket]:
@@ -163,6 +166,7 @@ class SeerSubgraphHandler(BaseSubgraphHandler):
         # Binary markets on Seer contain 3 outcomes: OutcomeA, outcomeB and an Invalid option.
         where_stms = self._build_where_statements(
             filter_by=filter_by,
+            outcome_supply_gt_if_open=outcome_supply_gt_if_open,
             include_conditional_markets=include_conditional_markets,
             include_categorical_markets=include_categorical_markets,
         )

{prediction_market_agent_tooling-0.65.0 → prediction_market_agent_tooling-0.65.2}/pyproject.toml

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