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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prediction-market-agent-tooling
-Version: 0.16.0
+Version: 0.18.0
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.12
@@ -22,6 +22,7 @@ Requires-Dist: isort (>=5.13.2,<6.0.0)
 Requires-Dist: langchain (>=0.1.9,<0.2.0) ; extra == "langchain"
 Requires-Dist: langchain-community (>=0.0.19)
 Requires-Dist: langchain-openai (>=0.0.5,<0.0.6) ; extra == "langchain"
+Requires-Dist: langfuse (>=2.27.1,<3.0.0)
 Requires-Dist: loguru (>=0.7.2,<0.8.0)
 Requires-Dist: mech-client (>=0.2.13,<0.3.0)
 Requires-Dist: numpy (>=1.26.4,<2.0.0)

{prediction_market_agent_tooling-0.16.0 → prediction_market_agent_tooling-0.18.0}/prediction_market_agent_tooling/benchmark/agents.py

@@ -6,6 +6,7 @@ from prediction_market_agent_tooling.benchmark.utils import (
     OutcomePrediction,
     Prediction,
 )
+from prediction_market_agent_tooling.gtypes import Probability
 
 
 class AbstractBenchmarkedAgent:
@@ -86,7 +87,8 @@ class RandomAgent(AbstractBenchmarkedAgent):
         p_yes, confidence = random.random(), random.random()
         return Prediction(
             outcome_prediction=OutcomePrediction(
-                p_yes=p_yes,
+                decision=p_yes > 0.5,
+                p_yes=Probability(p_yes),
                 confidence=confidence,
                 info_utility=None,
             ),
@@ -109,7 +111,8 @@ class FixedAgent(AbstractBenchmarkedAgent):
         p_yes, confidence = 1.0 if self.fixed_answer else 0.0, 1.0
         return Prediction(
             outcome_prediction=OutcomePrediction(
-                p_yes=p_yes,
+                decision=self.fixed_answer,
+                p_yes=Probability(p_yes),
                 confidence=confidence,
                 info_utility=None,
             ),

{prediction_market_agent_tooling-0.16.0 → prediction_market_agent_tooling-0.18.0}/prediction_market_agent_tooling/benchmark/utils.py

@@ -3,12 +3,11 @@ import typing as t
 
 from pydantic import BaseModel
 
+from prediction_market_agent_tooling.deploy.agent import Answer
 from prediction_market_agent_tooling.markets.data_models import Resolution
 
 
-class OutcomePrediction(BaseModel):
-    p_yes: float
-    confidence: float
+class OutcomePrediction(Answer):
     info_utility: t.Optional[float]
 
     @property

{prediction_market_agent_tooling-0.16.0 → prediction_market_agent_tooling-0.18.0}/prediction_market_agent_tooling/config.py

@@ -1,5 +1,7 @@
 import typing as t
 
+from gnosis.eth import EthereumClient
+from gnosis.safe import Safe
 from pydantic import BaseModel
 from pydantic.types import SecretStr
 from pydantic_settings import BaseSettings, SettingsConfigDict
@@ -30,6 +32,10 @@ class APIKeys(BaseSettings):
     GOOGLE_SEARCH_API_KEY: t.Optional[SecretStr] = None
     GOOGLE_SEARCH_ENGINE_ID: t.Optional[SecretStr] = None
 
+    LANGFUSE_SECRET_KEY: t.Optional[SecretStr] = None
+    LANGFUSE_PUBLIC_KEY: t.Optional[SecretStr] = None
+    LANGFUSE_HOST: t.Optional[str] = None
+
     ENABLE_CACHE: bool = True
     CACHE_DIR: str = "./.cache"
 
@@ -72,6 +78,24 @@ class APIKeys(BaseSettings):
             "GOOGLE_SEARCH_ENGINE_ID missing in the environment.",
         )
 
+    @property
+    def langfuse_secret_key(self) -> SecretStr:
+        return check_not_none(
+            self.LANGFUSE_SECRET_KEY, "LANGFUSE_SECRET_KEY missing in the environment."
+        )
+
+    @property
+    def langfuse_public_key(self) -> SecretStr:
+        return check_not_none(
+            self.LANGFUSE_PUBLIC_KEY, "LANGFUSE_PUBLIC_KEY missing in the environment."
+        )
+
+    @property
+    def langfuse_host(self) -> str:
+        return check_not_none(
+            self.LANGFUSE_HOST, "LANGFUSE_HOST missing in the environment."
+        )
+
     def model_dump_public(self) -> dict[str, t.Any]:
         return {
             k: v
@@ -110,3 +134,11 @@ class PrivateCredentials(BaseModel):
             private_key=api_keys.bet_from_private_key,
             safe_address=api_keys.SAFE_ADDRESS,
         )
+
+    def check_if_is_safe_owner(self, ethereum_client: EthereumClient) -> bool:
+        if not self.safe_address:
+            raise ValueError("Cannot check ownership if safe_address is not defined.")
+
+        s = Safe(self.safe_address, ethereum_client)  # type: ignore[abstract]
+        public_key_from_signer = private_key_to_public_key(self.private_key)
+        return s.retrieve_is_owner(public_key_from_signer)

{prediction_market_agent_tooling-0.16.0 → prediction_market_agent_tooling-0.18.0}/prediction_market_agent_tooling/deploy/agent.py

@@ -6,6 +6,8 @@ import typing as t
 from datetime import datetime
 
 from loguru import logger
+from pydantic import BaseModel, BeforeValidator
+from typing_extensions import Annotated
 
 from prediction_market_agent_tooling.config import APIKeys, PrivateCredentials
 from prediction_market_agent_tooling.deploy.constants import (
@@ -21,6 +23,7 @@ from prediction_market_agent_tooling.deploy.gcp.utils import (
     gcp_function_is_active,
     gcp_resolve_api_keys_secrets,
 )
+from prediction_market_agent_tooling.gtypes import Probability
 from prediction_market_agent_tooling.markets.agent_market import (
     AgentMarket,
     FilterBy,
@@ -31,6 +34,9 @@ from prediction_market_agent_tooling.markets.markets import MarketType
 from prediction_market_agent_tooling.markets.omen.omen import (
     redeem_from_all_user_positions,
 )
+from prediction_market_agent_tooling.monitor.langfuse.langfuse_wrapper import (
+    LangfuseWrapper,
+)
 from prediction_market_agent_tooling.monitor.monitor_app import (
     MARKET_TYPE_TO_DEPLOYED_AGENT,
 )
@@ -39,8 +45,43 @@ from prediction_market_agent_tooling.tools.utils import DatetimeWithTimezone, ut
 MAX_AVAILABLE_MARKETS = 20
 
 
+def to_boolean_outcome(value: str | bool) -> bool:
+    if isinstance(value, bool):
+        return value
+
+    elif isinstance(value, str):
+        value = value.lower().strip()
+
+        if value in {"true", "yes", "y", "1"}:
+            return True
+
+        elif value in {"false", "no", "n", "0"}:
+            return False
+
+        else:
+            raise ValueError(f"Expected a boolean string, but got {value}")
+
+    else:
+        raise ValueError(f"Expected a boolean or a string, but got {value}")
+
+
+Decision = Annotated[bool, BeforeValidator(to_boolean_outcome)]
+
+
+class Answer(BaseModel):
+    decision: Decision  # Warning: p_yes > 0.5 doesn't necessarily mean decision is True! For example, if our p_yes is 55%, but market's p_yes is 80%, then it might be profitable to bet on False.
+    p_yes: Probability
+    confidence: float
+    reasoning: str | None = None
+
+    @property
+    def p_no(self) -> Probability:
+        return Probability(1 - self.p_yes)
+
+
 class DeployableAgent:
     def __init__(self) -> None:
+        self.langfuse_wrapper = LangfuseWrapper(agent_name=self.__class__.__name__)
         self.load()
 
     def __init_subclass__(cls, **kwargs: t.Any) -> None:
@@ -58,7 +99,7 @@ class DeployableAgent:
         """
         return markets[:1]
 
-    def answer_binary_market(self, market: AgentMarket) -> bool | None:
+    def answer_binary_market(self, market: AgentMarket) -> Answer | None:
         """
         Answer the binary market. This method must be implemented by the subclass.
         """
@@ -157,7 +198,7 @@ def {entrypoint_function_name}(request) -> str:
         if cron_schedule:
             schedule_deployed_gcp_function(fname, cron_schedule=cron_schedule)
 
-    def calculate_bet_amount(self, answer: bool, market: AgentMarket) -> BetAmount:
+    def calculate_bet_amount(self, answer: Answer, market: AgentMarket) -> BetAmount:
         """
         Calculate the bet amount. By default, it returns the minimum bet amount.
         """
@@ -205,7 +246,7 @@ def {entrypoint_function_name}(request) -> str:
                 )
                 market.place_bet(
                     amount=amount,
-                    outcome=result,
+                    outcome=result.decision,
                 )
 
     def after(self, market_type: MarketType) -> None:

prediction_market_agent_tooling-0.18.0/prediction_market_agent_tooling/deploy/agent_example.py

@@ -0,0 +1,28 @@
+import random
+import typing as t
+
+from prediction_market_agent_tooling.deploy.agent import (
+    Answer,
+    DeployableAgent,
+    Probability,
+)
+from prediction_market_agent_tooling.markets.agent_market import AgentMarket
+
+
+class DeployableCoinFlipAgent(DeployableAgent):
+    def pick_markets(self, markets: t.Sequence[AgentMarket]) -> t.Sequence[AgentMarket]:
+        return random.sample(markets, 1)
+
+    def answer_binary_market(self, market: AgentMarket) -> Answer | None:
+        decision = random.choice([True, False])
+        return Answer(
+            decision=decision,
+            p_yes=Probability(float(decision)),
+            confidence=0.5,
+            reasoning="I flipped a coin to decide.",
+        )
+
+
+class DeployableAlwaysRaiseAgent(DeployableAgent):
+    def answer_binary_market(self, market: AgentMarket) -> Answer | None:
+        raise RuntimeError("I always raise!")

{prediction_market_agent_tooling-0.16.0 → prediction_market_agent_tooling-0.18.0}/prediction_market_agent_tooling/markets/omen/omen.py

@@ -432,6 +432,7 @@ def omen_sell_outcome_tx(
     market: OmenAgentMarket,
     outcome: str,
     auto_withdraw: bool,
+    web3: Web3 | None = None,
 ) -> None:
     """
     Sells the given xDai value of shares corresponding to the given outcome in
@@ -447,7 +448,10 @@ def omen_sell_outcome_tx(
     collateral_token = OmenCollateralTokenContract()
 
     # Verify, that markets uses conditional tokens that we expect.
-    if market_contract.conditionalTokens() != conditional_token_contract.address:
+    if (
+        market_contract.conditionalTokens(web3=web3)
+        != conditional_token_contract.address
+    ):
         raise ValueError(
             f"Market {market.id} uses conditional token that we didn't expect, {market_contract.conditionalTokens()} != {conditional_token_contract.address=}"
         )
@@ -457,7 +461,7 @@ def omen_sell_outcome_tx(
 
     # Calculate the amount of shares we will sell for the given selling amount of xdai.
     max_outcome_tokens_to_sell = market_contract.calcSellAmount(
-        amount_wei, outcome_index
+        amount_wei, outcome_index, web3=web3
     )
     # Allow 1% slippage.
     max_outcome_tokens_to_sell = add_fraction(max_outcome_tokens_to_sell, 0.01)
@@ -467,6 +471,7 @@ def omen_sell_outcome_tx(
         private_credentials=private_credentials,
         for_address=market_contract.address,
         approve=True,
+        web3=web3,
     )
     # Sell the shares.
     market_contract.sell(
@@ -474,12 +479,12 @@ def omen_sell_outcome_tx(
         amount_wei,
         outcome_index,
         max_outcome_tokens_to_sell,
+        web3=web3,
     )
     if auto_withdraw:
         # Optionally, withdraw from the collateral token back to the `from_address` wallet.
         collateral_token.withdraw(
-            private_credentials=private_credentials,
-            amount_wei=amount_wei,
+            private_credentials=private_credentials, amount_wei=amount_wei, web3=web3
         )
 
 
@@ -489,6 +494,7 @@ def binary_omen_sell_outcome_tx(
     market: OmenAgentMarket,
     binary_outcome: bool,
     auto_withdraw: bool,
+    web3: Web3 | None = None,
 ) -> None:
     omen_sell_outcome_tx(
         private_credentials=private_credentials,
@@ -496,6 +502,7 @@ def binary_omen_sell_outcome_tx(
         market=market,
         outcome=OMEN_TRUE_OUTCOME if binary_outcome else OMEN_FALSE_OUTCOME,
         auto_withdraw=auto_withdraw,
+        web3=web3,
     )
 
 
@@ -509,6 +516,7 @@ def omen_create_market_tx(
     outcomes: list[str],
     auto_deposit: bool,
     fee: float = OMEN_DEFAULT_MARKET_FEE,
+    web3: Web3 | None = None,
 ) -> ChecksumAddress:
     """
     Based on omen-exchange TypeScript code: https://github.com/protofire/omen-exchange/blob/b0b9a3e71b415d6becf21fe428e1c4fc0dad2e80/app/src/services/cpk/cpk.ts#L308
@@ -539,19 +547,22 @@ def omen_create_market_tx(
         private_credentials=private_credentials,
         for_address=factory_contract.address,
         amount_wei=initial_funds_wei,
+        web3=web3,
     )
 
     # Deposit xDai to the collateral token,
     # this can be skipped, if we know we already have enough collateral tokens.
     collateral_token_balance = collateral_token_contract.balanceOf(
-        for_address=from_address,
+        for_address=from_address, web3=web3
     )
     if (
         auto_deposit
         and initial_funds_wei > 0
         and collateral_token_balance < initial_funds_wei
     ):
-        collateral_token_contract.deposit(private_credentials, initial_funds_wei)
+        collateral_token_contract.deposit(
+            private_credentials, initial_funds_wei, web3=web3
+        )
 
     # Create the question on Realitio.
     question_id = realitio_contract.askQuestion(
@@ -562,6 +573,7 @@ def omen_create_market_tx(
         language=language,
         arbitrator=Arbitrator.KLEROS,
         opening=closing_time,  # The question is opened at the closing time of the market.
+        web3=web3,
     )
 
     # Construct the condition id.
@@ -569,13 +581,15 @@ def omen_create_market_tx(
         question_id=question_id,
         oracle_address=oracle_contract.address,
         outcomes_slot_count=len(outcomes),
+        web3=web3,
     )
-    if not conditional_token_contract.does_condition_exists(condition_id):
+    if not conditional_token_contract.does_condition_exists(condition_id, web3=web3):
         conditional_token_contract.prepareCondition(
             private_credentials=private_credentials,
             question_id=question_id,
             oracle_address=oracle_contract.address,
             outcomes_slot_count=len(outcomes),
+            web3=web3,
         )
 
     # Create the market.
@@ -584,6 +598,7 @@ def omen_create_market_tx(
         condition_id=condition_id,
         fee=fee,
         initial_funds_wei=initial_funds_wei,
+        web3=web3,
     )
 
     # Note: In the Omen's Typescript code, there is futher a creation of `stakingRewardsFactoryAddress`,
@@ -602,6 +617,7 @@ def omen_fund_market_tx(
     market: OmenAgentMarket,
     funds: Wei,
     auto_deposit: bool,
+    web3: Web3 | None = None,
 ) -> None:
     from_address = private_credentials.public_key
     market_contract = market.get_contract()
@@ -611,20 +627,19 @@ def omen_fund_market_tx(
     # this can be skipped, if we know we already have enough collateral tokens.
     if (
         auto_deposit
-        and collateral_token_contract.balanceOf(
-            for_address=from_address,
-        )
+        and collateral_token_contract.balanceOf(for_address=from_address, web3=web3)
         < funds
     ):
-        collateral_token_contract.deposit(private_credentials, funds)
+        collateral_token_contract.deposit(private_credentials, funds, web3=web3)
 
     collateral_token_contract.approve(
         private_credentials=private_credentials,
         for_address=market_contract.address,
         amount_wei=funds,
+        web3=web3,
     )
 
-    market_contract.addFunding(private_credentials, funds)
+    market_contract.addFunding(private_credentials, funds, web3=web3)
 
 
 def build_parent_collection_id() -> HexStr:
@@ -727,7 +742,7 @@ def omen_remove_fund_market_tx(
     """
     from_address = private_credentials.public_key
     market_contract = market.get_contract()
-    original_balances = get_balances(from_address)
+    original_balances = get_balances(from_address, web3=web3)
 
     total_shares = market_contract.balanceOf(from_address, web3=web3)
     if total_shares == 0:
@@ -763,7 +778,8 @@ def omen_remove_fund_market_tx(
         amount=amount_to_merge,
         web3=web3,
     )
-    new_balances = get_balances(from_address)
+
+    new_balances = get_balances(from_address, web3)
 
     logger.debug(f"Result from merge positions {result}")
     logger.info(
@@ -800,7 +816,7 @@ def redeem_from_all_user_positions(
             f"[{index+1} / {len(user_positions)}] Processing redeem from {user_position.id=}."
         )
 
-        original_balances = get_balances(public_key)
+        original_balances = get_balances(public_key, web3)
         conditional_token_contract.redeemPositions(
             private_credentials=private_credentials,
             collateral_token_address=user_position.position.collateral_token_contract_address_checksummed,
@@ -809,7 +825,7 @@ def redeem_from_all_user_positions(
             index_sets=user_position.position.indexSets,
             web3=web3,
         )
-        new_balances = get_balances(public_key)
+        new_balances = get_balances(public_key, web3)
 
         logger.info(
             f"Redeemed {new_balances.wxdai - original_balances.wxdai} wxDai from position {user_position.id=}."

{prediction_market_agent_tooling-0.16.0 → prediction_market_agent_tooling-0.18.0}/prediction_market_agent_tooling/markets/omen/omen_contracts.py

@@ -56,6 +56,7 @@ class OmenOracleContract(ContractOnGnosisChain):
         template_id: int,
         question_raw: str,
         n_outcomes: int,
+        web3: Web3 | None = None,
     ) -> TxReceipt:
         return self.send(
             private_credentials=private_credentials,
@@ -66,6 +67,7 @@ class OmenOracleContract(ContractOnGnosisChain):
                 question=question_raw,
                 numOutcomes=n_outcomes,
             ),
+            web3=web3,
         )
 
 
@@ -86,11 +88,13 @@ class OmenConditionalTokenContract(ContractOnGnosisChain):
         question_id: HexBytes,
         oracle_address: ChecksumAddress,
         outcomes_slot_count: int,
+        web3: Web3 | None = None,
     ) -> HexBytes:
         id_ = HexBytes(
             self.call(
                 "getConditionId",
                 [oracle_address, question_id, outcomes_slot_count],
+                web3=web3,
             )
         )
         return id_
@@ -177,34 +181,29 @@ class OmenConditionalTokenContract(ContractOnGnosisChain):
         )
 
     def getOutcomeSlotCount(
-        self,
-        condition_id: HexBytes,
+        self, condition_id: HexBytes, web3: Web3 | None = None
     ) -> int:
-        count: int = self.call(
-            "getOutcomeSlotCount",
-            [condition_id],
-        )
+        count: int = self.call("getOutcomeSlotCount", [condition_id], web3=web3)
         return count
 
     def does_condition_exists(
-        self,
-        condition_id: HexBytes,
+        self, condition_id: HexBytes, web3: Web3 | None = None
     ) -> bool:
-        return self.getOutcomeSlotCount(condition_id) > 0
+        return self.getOutcomeSlotCount(condition_id, web3=web3) > 0
 
     def is_condition_resolved(
-        self,
-        condition_id: HexBytes,
+        self, condition_id: HexBytes, web3: Web3 | None = None
     ) -> bool:
         # from ConditionalTokens.redeemPositions:
         # uint den = payoutDenominator[conditionId]; require(den > 0, "result for condition not received yet");
-        payout_for_condition = self.payoutDenominator(condition_id)
+        payout_for_condition = self.payoutDenominator(condition_id, web3=web3)
         return payout_for_condition > 0
 
-    def payoutDenominator(self, condition_id: HexBytes) -> int:
+    def payoutDenominator(
+        self, condition_id: HexBytes, web3: Web3 | None = None
+    ) -> int:
         payoutForCondition: int = self.call(
-            "payoutDenominator",
-            [condition_id],
+            "payoutDenominator", [condition_id], web3=web3
         )
         return payoutForCondition
 
@@ -214,6 +213,7 @@ class OmenConditionalTokenContract(ContractOnGnosisChain):
         for_address: ChecksumAddress,
         approve: bool,
         tx_params: t.Optional[TxParams] = None,
+        web3: Web3 | None = None,
     ) -> TxReceipt:
         return self.send(
             private_credentials=private_credentials,
@@ -223,6 +223,7 @@ class OmenConditionalTokenContract(ContractOnGnosisChain):
                 approve,
             ],
             tx_params=tx_params,
+            web3=web3,
         )
 
     def prepareCondition(
@@ -232,6 +233,7 @@ class OmenConditionalTokenContract(ContractOnGnosisChain):
         question_id: HexBytes,
         outcomes_slot_count: int,
         tx_params: t.Optional[TxParams] = None,
+        web3: Web3 | None = None,
     ) -> TxReceipt:
         return self.send(
             private_credentials=private_credentials,
@@ -242,6 +244,7 @@ class OmenConditionalTokenContract(ContractOnGnosisChain):
                 outcomes_slot_count,
             ],
             tx_params=tx_params,
+            web3=web3,
         )
 
 
@@ -272,25 +275,18 @@ class OmenFixedProductMarketMakerContract(ContractOnGnosisChain):
         return calculated_shares
 
     def calcSellAmount(
-        self,
-        return_amount: Wei,
-        outcome_index: int,
+        self, return_amount: Wei, outcome_index: int, web3: Web3 | None = None
     ) -> OmenOutcomeToken:
         """
         Returns amount of shares we will sell for the requested wei.
         """
         calculated_shares: OmenOutcomeToken = self.call(
-            "calcSellAmount",
-            [return_amount, outcome_index],
+            "calcSellAmount", [return_amount, outcome_index], web3=web3
         )
         return calculated_shares
 
-    def conditionalTokens(
-        self,
-    ) -> HexAddress:
-        address: HexAddress = self.call(
-            "conditionalTokens",
-        )
+    def conditionalTokens(self, web3: Web3 | None = None) -> HexAddress:
+        address: HexAddress = self.call("conditionalTokens", web3=web3)
         return address
 
     def buy(
@@ -321,6 +317,7 @@ class OmenFixedProductMarketMakerContract(ContractOnGnosisChain):
         outcome_index: int,
         max_outcome_tokens_to_sell: OmenOutcomeToken,
         tx_params: t.Optional[TxParams] = None,
+        web3: Web3 | None = None,
     ) -> TxReceipt:
         return self.send(
             private_credentials=private_credentials,
@@ -331,6 +328,7 @@ class OmenFixedProductMarketMakerContract(ContractOnGnosisChain):
                 max_outcome_tokens_to_sell,
             ],
             tx_params=tx_params,
+            web3=web3,
         )
 
     def addFunding(
@@ -338,6 +336,7 @@ class OmenFixedProductMarketMakerContract(ContractOnGnosisChain):
         private_credentials: PrivateCredentials,
         add_funding: Wei,
         tx_params: t.Optional[TxParams] = None,
+        web3: Web3 | None = None,
     ) -> TxReceipt:
         """
         Funding is added in Weis (xDai) and then converted to shares.
@@ -349,6 +348,7 @@ class OmenFixedProductMarketMakerContract(ContractOnGnosisChain):
             function_name="addFunding",
             function_params=[add_funding, distribution_hint],
             tx_params=tx_params,
+            web3=web3,
         )
 
     def removeFunding(
@@ -369,9 +369,9 @@ class OmenFixedProductMarketMakerContract(ContractOnGnosisChain):
             web3=web3,
         )
 
-    def totalSupply(self) -> Wei:
+    def totalSupply(self, web3: Web3 | None = None) -> Wei:
         # This is the liquidity you seen on the Omen website (but in Wei).
-        total_supply: Wei = self.call("totalSupply")
+        total_supply: Wei = self.call("totalSupply", web3=web3)
         return total_supply
 
 
@@ -412,6 +412,7 @@ class OmenFixedProductMarketMakerFactoryContract(ContractOnGnosisChain):
         initial_funds_wei: Wei,
         fee: float = OMEN_DEFAULT_MARKET_FEE,
         tx_params: t.Optional[TxParams] = None,
+        web3: Web3 | None = None,
     ) -> TxReceipt:
         fee_wei = xdai_to_wei(
             xdai_type(fee)
@@ -431,6 +432,7 @@ class OmenFixedProductMarketMakerFactoryContract(ContractOnGnosisChain):
                 distributionHint=[],
             ),
             tx_params=tx_params,
+            web3=web3,
         )
 
 
@@ -498,6 +500,7 @@ class OmenRealitioContract(ContractOnGnosisChain):
         opening: datetime,
         nonce: int | None = None,
         tx_params: t.Optional[TxParams] = None,
+        web3: Web3 | None = None,
     ) -> HexBytes:
         """
         After the question is created, you can find it at https://reality.eth.link/app/#!/creator/{from_address}.
@@ -528,6 +531,7 @@ class OmenRealitioContract(ContractOnGnosisChain):
                 ),  # Two equal questions need to have different nonces.
             ),
             tx_params=tx_params,
+            web3=web3,
         )
         question_id = HexBytes(
             receipt_tx["logs"][0]["topics"][1]
@@ -542,6 +546,7 @@ class OmenRealitioContract(ContractOnGnosisChain):
         outcomes: list[str],
         bond: Wei,
         max_previous: Wei | None = None,
+        web3: Web3 | None = None,
     ) -> TxReceipt:
         if max_previous is None:
             # If not provided, defaults to 0, which means no checking,
@@ -563,6 +568,7 @@ class OmenRealitioContract(ContractOnGnosisChain):
                 max_previous=max_previous,
             ),
             amount_wei=bond,
+            web3=web3,
         )
 
     def claimWinnings(
@@ -574,6 +580,7 @@ class OmenRealitioContract(ContractOnGnosisChain):
         bonds: list[Wei],
         answers: list[HexBytes],
         tx_params: t.Optional[TxParams] = None,
+        web3: Web3 | None = None,
     ) -> TxReceipt:
         return self.send(
             private_credentials=private_credentials,
@@ -586,17 +593,22 @@ class OmenRealitioContract(ContractOnGnosisChain):
                 answers=answers,
             ),
             tx_params=tx_params,
+            web3=web3,
         )
 
-    def balanceOf(self, from_address: ChecksumAddress) -> Wei:
-        balance = wei_type(self.call("balanceOf", [from_address]))
+    def balanceOf(
+        self,
+        from_address: ChecksumAddress,
+        web3: Web3 | None = None,
+    ) -> Wei:
+        balance = wei_type(self.call("balanceOf", [from_address], web3=web3))
         return balance
 
     def withdraw(
         self,
         private_credentials: PrivateCredentials,
+        web3: Web3 | None = None,
     ) -> TxReceipt:
         return self.send(
-            private_credentials=private_credentials,
-            function_name="withdraw",
+            private_credentials=private_credentials, function_name="withdraw", web3=web3
         )