prediction-market-agent-tooling 0.62.0.dev466__tar.gz → 0.62.0.dev468__tar.gz

{prediction_market_agent_tooling-0.62.0.dev466 → prediction_market_agent_tooling-0.62.0.dev468}/PKG-INFO

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

{prediction_market_agent_tooling-0.62.0.dev466 → prediction_market_agent_tooling-0.62.0.dev468}/prediction_market_agent_tooling/markets/agent_market.py

@@ -17,7 +17,6 @@ from prediction_market_agent_tooling.markets.data_models import (
     Bet,
     ExistingPosition,
     PlacedTrade,
-    Position,
     ProbabilisticAnswer,
     Resolution,
     ResolvedBet,

{prediction_market_agent_tooling-0.62.0.dev466 → prediction_market_agent_tooling-0.62.0.dev468}/prediction_market_agent_tooling/tools/_generic_value.py

@@ -255,3 +255,7 @@ class _GenericValue(
     @classmethod
     def zero(cls: type[GenericValueType]) -> GenericValueType:
         return cls(0)  # type: ignore[arg-type]
+
+    @property
+    def symbol(self) -> str:
+        return self.__class__.__name__

{prediction_market_agent_tooling-0.62.0.dev466 → prediction_market_agent_tooling-0.62.0.dev468}/prediction_market_agent_tooling/tools/cow/cow_order.py

@@ -14,6 +14,8 @@ from cowdao_cowpy.order_book.generated.model import (
     OrderMetaData,
     OrderQuoteRequest,
     OrderQuoteSide1,
+    OrderQuoteSide3,
+    OrderQuoteSideKindBuy,
     OrderQuoteSideKindSell,
     OrderStatus,
     TokenAmount,
@@ -34,13 +36,41 @@ def get_order_book_api(env: Envs, chain: Chain) -> OrderBookApi:
     return OrderBookApi(OrderBookAPIConfigFactory.get_config(env, chain_id))
 
 
+def get_sell_token_amount(
+    buy_amount: Wei,
+    sell_token: ChecksumAddress,
+    buy_token: ChecksumAddress,
+    chain: Chain = Chain.GNOSIS,
+    env: Envs = "prod",
+) -> Wei:
+    """
+    Calculate how much of the sell_token is needed to obtain a specified amount of buy_token.
+    """
+    order_book_api = get_order_book_api(env, chain)
+    order_quote_request = OrderQuoteRequest(
+        sellToken=Address(sell_token),
+        buyToken=Address(buy_token),
+        from_=Address(
+            "0x1234567890abcdef1234567890abcdef12345678"
+        ),  # Just random address, doesn't matter.
+    )
+    order_side = OrderQuoteSide3(
+        kind=OrderQuoteSideKindBuy.buy,
+        buyAmountAfterFee=TokenAmount(str(buy_amount)),
+    )
+    order_quote = asyncio.run(
+        order_book_api.post_quote(order_quote_request, order_side)
+    )
+    return Wei(order_quote.quote.sellAmount.root)
+
+
 @tenacity.retry(
     stop=tenacity.stop_after_attempt(3),
     wait=tenacity.wait_fixed(1),
     after=lambda x: logger.debug(f"get_buy_token_amount failed, {x.attempt_number=}."),
 )
 def get_buy_token_amount(
-    amount_wei: Wei,
+    sell_amount: Wei,
     sell_token: ChecksumAddress,
     buy_token: ChecksumAddress,
     chain: Chain = Chain.GNOSIS,
@@ -56,7 +86,7 @@ def get_buy_token_amount(
     )
     order_side = OrderQuoteSide1(
         kind=OrderQuoteSideKindSell.sell,
-        sellAmountBeforeFee=TokenAmount(str(amount_wei)),
+        sellAmountBeforeFee=TokenAmount(str(sell_amount)),
     )
     order_quote = asyncio.run(
         order_book_api.post_quote(order_quote_request, order_side)

{prediction_market_agent_tooling-0.62.0.dev466 → prediction_market_agent_tooling-0.62.0.dev468}/prediction_market_agent_tooling/tools/tokens/usd.py

@@ -55,7 +55,7 @@ def get_single_token_unit_to_usd_rate(token_address: ChecksumAddress) -> USD:
             .as_token.value
         )
     in_wei = get_buy_token_amount(
-        amount_wei=Token(1).as_wei,
+        sell_amount=Token(1).as_wei,
         sell_token=token_address,
         buy_token=WRAPPED_XDAI_CONTRACT_ADDRESS,
     )

{prediction_market_agent_tooling-0.62.0.dev466 → prediction_market_agent_tooling-0.62.0.dev468}/pyproject.toml

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