prediction-market-agent-tooling 0.36.0__py3-none-any.whl → 0.37.0__py3-none-any.whl

prediction_market_agent_tooling/config.py

@@ -27,6 +27,7 @@ class APIKeys(BaseSettings):
     BET_FROM_PRIVATE_KEY: t.Optional[PrivateKey] = None
     SAFE_ADDRESS: t.Optional[ChecksumAddress] = None
     OPENAI_API_KEY: t.Optional[SecretStr] = None
+    GRAPH_API_KEY: t.Optional[SecretStr] = None
 
     GOOGLE_SEARCH_API_KEY: t.Optional[SecretStr] = None
     GOOGLE_SEARCH_ENGINE_ID: t.Optional[SecretStr] = None
@@ -72,6 +73,12 @@ class APIKeys(BaseSettings):
             self.OPENAI_API_KEY, "OPENAI_API_KEY missing in the environment."
         )
 
+    @property
+    def graph_api_key(self) -> SecretStr:
+        return check_not_none(
+            self.GRAPH_API_KEY, "GRAPH_API_KEY missing in the environment."
+        )
+
     @property
     def google_search_api_key(self) -> SecretStr:
         return check_not_none(

prediction_market_agent_tooling/markets/omen/omen.py

@@ -10,7 +10,6 @@ from prediction_market_agent_tooling.gtypes import (
     ChecksumAddress,
     HexAddress,
     HexStr,
-    OmenOutcomeToken,
     OutcomeStr,
     Probability,
     Wei,
@@ -84,7 +83,6 @@ class OmenAgentMarket(AgentMarket):
     finalized_time: datetime | None
     created_time: datetime
     close_time: datetime
-    outcome_token_amounts: list[OmenOutcomeToken]
     fee: float  # proportion, from 0 to 1
 
     INVALID_MARKET_ANSWER: HexStr = HexStr(
@@ -144,6 +142,7 @@ class OmenAgentMarket(AgentMarket):
         amount: BetAmount,
         omen_auto_deposit: bool = True,
         web3: Web3 | None = None,
+        api_keys: APIKeys | None = None,
     ) -> None:
         if not self.can_be_traded():
             raise ValueError(
@@ -153,7 +152,7 @@ class OmenAgentMarket(AgentMarket):
             raise ValueError(f"Omen bets are made in xDai. Got {amount.currency}.")
         amount_xdai = xDai(amount.amount)
         binary_omen_buy_outcome_tx(
-            api_keys=APIKeys(),
+            api_keys=api_keys if api_keys is not None else APIKeys(),
             amount=amount_xdai,
             market=self,
             binary_outcome=outcome,
@@ -161,20 +160,34 @@ class OmenAgentMarket(AgentMarket):
             web3=web3,
         )
 
+    def buy_tokens(
+        self,
+        outcome: bool,
+        amount: TokenAmount,
+        web3: Web3 | None = None,
+        api_keys: APIKeys | None = None,
+    ) -> None:
+        return self.place_bet(
+            outcome=outcome,
+            amount=amount,
+            web3=web3,
+            api_keys=api_keys,
+        )
+
     def calculate_sell_amount_in_collateral(
-        self, amount: TokenAmount, outcome: bool
+        self, amount: TokenAmount, outcome: bool, web3: Web3 | None = None
     ) -> xDai:
-        if len(self.outcome_token_amounts) != 2:
-            raise ValueError(
-                f"Market {self.id} has {len(self.outcome_token_amounts)} "
-                f"outcomes. This method only supports binary markets."
-            )
-        sell_index = self.yes_index if outcome else self.no_index
-        other_index = self.no_index if outcome else self.yes_index
+        pool_balance = get_conditional_tokens_balance_for_market(
+            self, self.market_maker_contract_address_checksummed, web3=web3
+        )
+
+        sell_str = self.outcomes[self.yes_index if outcome else self.no_index]
+        other_str = self.outcomes[self.no_index if outcome else self.yes_index]
+
         collateral = calculate_sell_amount_in_collateral(
             shares_to_sell=amount.amount,
-            holdings=wei_to_xdai(Wei(self.outcome_token_amounts[sell_index])),
-            other_holdings=wei_to_xdai(Wei(self.outcome_token_amounts[other_index])),
+            holdings=wei_to_xdai(pool_balance[self.get_index_set(sell_str)]),
+            other_holdings=wei_to_xdai(pool_balance[self.get_index_set(other_str)]),
             fee=self.fee,
         )
         return xDai(collateral)
@@ -197,6 +210,7 @@ class OmenAgentMarket(AgentMarket):
         collateral = self.calculate_sell_amount_in_collateral(
             amount=amount,
             outcome=outcome,
+            web3=web3,
         )
         binary_omen_sell_outcome_tx(
             amount=collateral,
@@ -282,7 +296,6 @@ class OmenAgentMarket(AgentMarket):
             url=model.url,
             volume=wei_to_xdai(model.collateralVolume),
             close_time=model.close_time,
-            outcome_token_amounts=model.outcomeTokenAmounts,
             fee=float(wei_to_xdai(model.fee)) if model.fee is not None else 0.0,
         )
 
@@ -462,8 +475,9 @@ def omen_buy_outcome_tx(
         for_address=from_address_checksummed, web3=web3
     )
     if auto_deposit and collateral_token_balance < amount_wei:
+        deposit_amount_wei = Wei(amount_wei - collateral_token_balance)
         collateral_token_contract.deposit(
-            api_keys=api_keys, amount_wei=amount_wei, web3=web3
+            api_keys=api_keys, amount_wei=deposit_amount_wei, web3=web3
         )
     # Buy shares using the deposited xDai in the collateral token.
     market_contract.buy(

prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py

@@ -6,6 +6,7 @@ import tenacity
 from eth_typing import ChecksumAddress
 from subgrounds import FieldPath, Subgrounds
 
+from prediction_market_agent_tooling.config import APIKeys
 from prediction_market_agent_tooling.gtypes import HexAddress, HexBytes, Wei, wei_type
 from prediction_market_agent_tooling.loggers import logger
 from prediction_market_agent_tooling.markets.agent_market import FilterBy, SortBy
@@ -29,13 +30,11 @@ class OmenSubgraphHandler(metaclass=SingletonMeta):
     Class responsible for handling interactions with Omen subgraphs (trades, conditionalTokens).
     """
 
-    OMEN_TRADES_SUBGRAPH = "https://api.thegraph.com/subgraphs/name/protofire/omen-xdai"
-    CONDITIONAL_TOKENS_SUBGRAPH = (
-        "https://api.thegraph.com/subgraphs/name/gnosis/conditional-tokens-gc"
-    )
-    REALITYETH_GRAPH_URL = (
-        "https://api.thegraph.com/subgraphs/name/realityeth/realityeth-gnosis"
-    )
+    OMEN_TRADES_SUBGRAPH = "https://gateway-arbitrum.network.thegraph.com/api/{graph_api_key}/subgraphs/id/9fUVQpFwzpdWS9bq5WkAnmKbNNcoBwatMR4yZq81pbbz"
+
+    CONDITIONAL_TOKENS_SUBGRAPH = "https://gateway-arbitrum.network.thegraph.com/api/{graph_api_key}/subgraphs/id/7s9rGBffUTL8kDZuxvvpuc46v44iuDarbrADBFw5uVp2"
+
+    REALITYETH_GRAPH_URL = "https://gateway-arbitrum.network.thegraph.com/api/{graph_api_key}/subgraphs/id/E7ymrCnNcQdAAgLbdFWzGE5mvr5Mb5T9VfT43FqA7bNh"
 
     INVALID_ANSWER = "ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff"
 
@@ -49,12 +48,24 @@ class OmenSubgraphHandler(metaclass=SingletonMeta):
             after=lambda x: logger.debug(f"query_json failed, {x.attempt_number=}."),
         )(self.sg.query_json)
 
+        keys = APIKeys()
+
         # Load the subgraph
-        self.trades_subgraph = self.sg.load_subgraph(self.OMEN_TRADES_SUBGRAPH)
+        self.trades_subgraph = self.sg.load_subgraph(
+            self.OMEN_TRADES_SUBGRAPH.format(
+                graph_api_key=keys.graph_api_key.get_secret_value()
+            )
+        )
         self.conditional_tokens_subgraph = self.sg.load_subgraph(
-            self.CONDITIONAL_TOKENS_SUBGRAPH
+            self.CONDITIONAL_TOKENS_SUBGRAPH.format(
+                graph_api_key=keys.graph_api_key.get_secret_value()
+            )
+        )
+        self.realityeth_subgraph = self.sg.load_subgraph(
+            self.REALITYETH_GRAPH_URL.format(
+                graph_api_key=keys.graph_api_key.get_secret_value()
+            )
         )
-        self.realityeth_subgraph = self.sg.load_subgraph(self.REALITYETH_GRAPH_URL)
 
     def _get_fields_for_bets(self, bets_field: FieldPath) -> list[FieldPath]:
         markets = bets_field.fpmm

prediction_market_agent_tooling/tools/balances.py

@@ -13,6 +13,10 @@ class Balances(BaseModel):
     xdai: xDai
     wxdai: xDai
 
+    @property
+    def total(self) -> xDai:
+        return xDai(self.xdai + self.wxdai)
+
 
 def get_balances(address: ChecksumAddress, web3: Web3 | None = None) -> Balances:
     if not web3:

prediction_market_agent_tooling/tools/image_gen/image_gen.py

@@ -0,0 +1,49 @@
+import base64
+import io
+import typing as t
+
+from PIL import Image
+from PIL.Image import Image as ImageType
+
+from prediction_market_agent_tooling.config import APIKeys
+from prediction_market_agent_tooling.tools.utils import check_not_none
+
+
+def generate_image(
+    prompt: str,
+    model: str = "dall-e-3",
+    size: t.Literal[
+        "256x256", "512x512", "1024x1024", "1792x1024", "1024x1792"
+    ] = "1024x1024",
+    quality: t.Literal["standard", "hd"] = "standard",
+) -> ImageType:
+    try:
+        from openai import OpenAI
+    except ImportError:
+        raise ImportError(
+            "openai not installed, please install extras `openai` to use this function."
+        )
+    response = (
+        OpenAI(
+            api_key=APIKeys().openai_api_key.get_secret_value(),
+        )
+        .images.generate(
+            model=model,
+            prompt=prompt,
+            size=size,
+            quality=quality,
+            response_format="b64_json",
+            n=1,
+        )
+        .data[0]
+    )
+    image = Image.open(
+        io.BytesIO(
+            base64.b64decode(
+                check_not_none(
+                    response.b64_json, "Can't be none if response_format is b64_json."
+                )
+            )
+        )
+    )
+    return image

prediction_market_agent_tooling/tools/image_gen/market_thumbnail_gen.py

@@ -0,0 +1,30 @@
+from PIL.Image import Image as ImageType
+
+from prediction_market_agent_tooling.config import APIKeys
+from prediction_market_agent_tooling.tools.image_gen.image_gen import generate_image
+
+
+def rewrite_question_into_image_generation_prompt(question: str) -> str:
+    try:
+        from langchain_openai import ChatOpenAI
+    except ImportError:
+        raise ImportError(
+            "openai not installed, please install extras `langchain` to use this function."
+        )
+    llm = ChatOpenAI(
+        model="gpt-4-turbo",
+        temperature=0.0,
+        api_key=APIKeys().openai_api_key.get_secret_value(),
+    )
+    rewritten = str(
+        llm.invoke(
+            f"Rewrite this prediction market question '{question}' into a form that will generate nice thumbnail with DALL-E-3."
+            "The thumbnail should be catchy and visually appealing. With a large object in the center of the image."
+        ).content
+    )
+    return rewritten
+
+
+def generate_image_for_market(question: str) -> ImageType:
+    prompt = rewrite_question_into_image_generation_prompt(question)
+    return generate_image(prompt)

{prediction_market_agent_tooling-0.36.0.dist-info → prediction_market_agent_tooling-0.37.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prediction-market-agent-tooling
-Version: 0.36.0
+Version: 0.37.0
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.12
@@ -9,6 +9,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Provides-Extra: google
 Provides-Extra: langchain
+Provides-Extra: openai
 Requires-Dist: autoflake (>=2.2.1,<3.0.0)
 Requires-Dist: cron-validator (>=1.0.8,<2.0.0)
 Requires-Dist: eth-account (>=0.8.0,<0.9.0)
@@ -25,6 +26,7 @@ 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: numpy (>=1.26.4,<2.0.0)
+Requires-Dist: openai (>=1.0.0,<2.0.0) ; extra == "openai"
 Requires-Dist: prompt-toolkit (>=3.0.43,<4.0.0)
 Requires-Dist: pydantic (>=2.6.1,<3.0.0)
 Requires-Dist: pydantic-settings (>=2.1.0,<3.0.0)

{prediction_market_agent_tooling-0.36.0.dist-info → prediction_market_agent_tooling-0.37.0.dist-info}/RECORD

@@ -11,7 +11,7 @@ prediction_market_agent_tooling/benchmark/__init__.py,sha256=47DEQpj8HBSa-_TImW-
 prediction_market_agent_tooling/benchmark/agents.py,sha256=HPIFJvackW110ch3UkktbxhU48gMRVo4gQse84Klhdc,4000
 prediction_market_agent_tooling/benchmark/benchmark.py,sha256=xiHKzZx5GHSsDerFHMZ9j_LXAXnSaITSvv67iPe3MEU,21095
 prediction_market_agent_tooling/benchmark/utils.py,sha256=iS1BzyXcCMfMm1Rx--1QCH0pHvBTblTndcDQFbTUJ2s,2897
-prediction_market_agent_tooling/config.py,sha256=yELIlzAm2yBwZzGRvHtHBZZV3NZy5CllJfo3chMQMo0,4297
+prediction_market_agent_tooling/config.py,sha256=pTeV4mbvPmrS_MzmiBklg05_gtsR3pa9Id5HCS-AzKQ,4519
 prediction_market_agent_tooling/deploy/agent.py,sha256=h33uU97x3DjBy8INONtS1Hc5B-NnNDV6MENeeCp0qoA,10185
 prediction_market_agent_tooling/deploy/agent_example.py,sha256=tqXVA2HSFK3pdZ49tMmta8aKdJFT8JN8WeJ1akjrpBk,909
 prediction_market_agent_tooling/deploy/constants.py,sha256=M5ty8URipYMGe_G-RzxRydK3AFL6CyvmqCraJUrLBnE,82
@@ -31,10 +31,10 @@ prediction_market_agent_tooling/markets/manifold/utils.py,sha256=cPPFWXm3vCYH1jy
 prediction_market_agent_tooling/markets/markets.py,sha256=w05Oo7yCA2axpCw69Q9y4i9Gcdpht0u5bZGbWqld3rU,2964
 prediction_market_agent_tooling/markets/omen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prediction_market_agent_tooling/markets/omen/data_models.py,sha256=EXtjmcujx68Xu50BVkYXvLuf_Asx5o65RvFS3ZS6HGs,14405
-prediction_market_agent_tooling/markets/omen/omen.py,sha256=aJ5CCste61B7q0OD3EjLP8H7AjD_RaBBNyUfnZGh5KM,33597
+prediction_market_agent_tooling/markets/omen/omen.py,sha256=fS19WoDG_N8WIa5FY-YRF21Np46y6d93QBJOpNKXi7Q,33939
 prediction_market_agent_tooling/markets/omen/omen_contracts.py,sha256=eDS8vN4Klv_-Y1wwfIeLDt3twhl9U_AJjPQov0JScb0,19888
 prediction_market_agent_tooling/markets/omen/omen_resolving.py,sha256=g77QsQ5WnSI2rzBlX87L_EhWMwobkyXyfRhHQmpAdzo,9012
-prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py,sha256=RlaQMDF75lo2fos8LUb_OAdrw_ILLOXQLdejzE7tOmA,23053
+prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py,sha256=2y89FJlDrJ_hpeNTFZU1W30xCpt8QXjssEW8bNZnEpU,23596
 prediction_market_agent_tooling/markets/polymarket/api.py,sha256=HXmA1akA0qDj0m3e-GEvWG8x75pm6BX4H7YJPQcST7I,4767
 prediction_market_agent_tooling/markets/polymarket/data_models.py,sha256=9CJzakyEcsn6DQBK2nOXjOMzTZBLAmK_KqevXvW17DI,4292
 prediction_market_agent_tooling/markets/polymarket/data_models_web.py,sha256=f8SRQy0Rn-gIHSEMrJJAI8H3J7l8lzOLj3aCMe0vJv8,11324
@@ -48,7 +48,7 @@ prediction_market_agent_tooling/monitor/monitor.py,sha256=uJdaEpGWp4VcejoQLF0uOx
 prediction_market_agent_tooling/monitor/monitor_app.py,sha256=rDxgdDJqSK0ARx0TJFMkS76npkHZJz0__Rp0xTpiRok,4611
 prediction_market_agent_tooling/monitor/monitor_settings.py,sha256=Xiozs3AsufuJ04JOe1vjUri-IAMWHjjmc2ugGGiHNH4,947
 prediction_market_agent_tooling/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-prediction_market_agent_tooling/tools/balances.py,sha256=efDKEe-SsuSmGWKuGNiuHEWCGuCrcwjrCdTz1iKFtqw,749
+prediction_market_agent_tooling/tools/balances.py,sha256=nR8_dSfbm3yTOOmMAwhGlurftEiNo1w1WIVzbskjdmM,837
 prediction_market_agent_tooling/tools/betting_strategies/kelly_criterion.py,sha256=IbhQPoKsQnjnag_n_wAL12n8QdCe7tAMRNV2QS8yYxY,3520
 prediction_market_agent_tooling/tools/betting_strategies/market_moving.py,sha256=wtrHVQRuA0uDx06z0OxQLYbswuOpHQ1UyCWwLCrD_oM,4400
 prediction_market_agent_tooling/tools/betting_strategies/minimum_bet_to_win.py,sha256=-FUSuQQgjcWSSnoFxnlAyTeilY6raJABJVM2QKkFqAY,438
@@ -59,6 +59,8 @@ prediction_market_agent_tooling/tools/costs.py,sha256=EaAJ7v9laD4VEV3d8B44M4u3_o
 prediction_market_agent_tooling/tools/gnosis_rpc.py,sha256=_MYSoyOR2MgAJkop1ERf8RhLum-M8S6OjaAsaqUW41w,203
 prediction_market_agent_tooling/tools/google.py,sha256=SfVDxb3oEOUK8mpd0l3mTX9ybrdrTPNM6HjfJ7kfNjA,1794
 prediction_market_agent_tooling/tools/hexbytes_custom.py,sha256=Bp94qgPjvjWf1Vb4lNzGFDXRdThw1rJ91vL6r2PWq5E,2096
+prediction_market_agent_tooling/tools/image_gen/image_gen.py,sha256=HzRwBx62hOXBOmrtpkXaP9Qq1Ku03uUGdREocyjLQ_k,1266
+prediction_market_agent_tooling/tools/image_gen/market_thumbnail_gen.py,sha256=svEu-oJNVuqpPpA7JjPEz3s1yM1_DgVf4M6gu90YKMQ,1110
 prediction_market_agent_tooling/tools/is_predictable.py,sha256=yRSkmu4lZqrSje9QGpnyWTFmdCnRvClfaFDHE2Zf9G4,2936
 prediction_market_agent_tooling/tools/parallelism.py,sha256=8mgkF5sBwFGS5GMvlpzam82Y3p2swPYuNsywpQuy-a4,1508
 prediction_market_agent_tooling/tools/safe.py,sha256=h0xOO0eNtitClf0fPkn-0oTc6A_bflDTee98V_aiV-A,5195
@@ -66,8 +68,8 @@ prediction_market_agent_tooling/tools/singleton.py,sha256=CiIELUiI-OeS7U7eeHEt0r
 prediction_market_agent_tooling/tools/streamlit_user_login.py,sha256=NXEqfjT9Lc9QtliwSGRASIz1opjQ7Btme43H4qJbzgE,3010
 prediction_market_agent_tooling/tools/utils.py,sha256=-G22UEbCRm59bm1RWFdeF55hRsaxgwZVAHvK32-Ye1g,6190
 prediction_market_agent_tooling/tools/web3_utils.py,sha256=cboATXNmEdn5RmPbVblHOwOdUMKBYrUK3GiI6i6Vzxo,9855
-prediction_market_agent_tooling-0.36.0.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.36.0.dist-info/METADATA,sha256=qmWuCtdncca0d6UJlarbVMkbgW15JBQojVPh1ZKz1bY,7513
-prediction_market_agent_tooling-0.36.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-prediction_market_agent_tooling-0.36.0.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.36.0.dist-info/RECORD,,
+prediction_market_agent_tooling-0.37.0.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.37.0.dist-info/METADATA,sha256=CaelrPuw-JpZ4VVfKQTy8v8pdKTeWPLrqnEfhqRFHJg,7595
+prediction_market_agent_tooling-0.37.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+prediction_market_agent_tooling-0.37.0.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.37.0.dist-info/RECORD,,