prediction-market-agent-tooling 0.36.0__tar.gz → 0.37.0__tar.gz

{prediction_market_agent_tooling-0.36.0 → prediction_market_agent_tooling-0.37.0}/PKG-INFO

@@ -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 → prediction_market_agent_tooling-0.37.0}/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-0.36.0 → prediction_market_agent_tooling-0.37.0}/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-0.36.0 → prediction_market_agent_tooling-0.37.0}/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-0.36.0 → prediction_market_agent_tooling-0.37.0}/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-0.37.0/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-0.37.0/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 → prediction_market_agent_tooling-0.37.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "prediction-market-agent-tooling"
-version = "0.36.0"
+version = "0.37.0"
 description = "Tools to benchmark, deploy and monitor prediction market agents."
 authors = ["Gnosis"]
 readme = "README.md"
@@ -41,8 +41,10 @@ eth-account = "^0.8.0"
 prompt-toolkit = "^3.0.43"
 safe-cli = "^1.0.0"
 langfuse = "^2.27.1"
+openai = { version = "^1.0.0", optional = true}
 
 [tool.poetry.extras]
+openai = ["openai"]
 langchain = ["langchain", "langchain-openai"]
 google = ["google-api-python-client"]