prediction-market-agent-tooling 0.69.9__tar.gz → 0.69.10__tar.gz

{prediction_market_agent_tooling-0.69.9 → prediction_market_agent_tooling-0.69.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prediction-market-agent-tooling
-Version: 0.69.9
+Version: 0.69.10
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 License-File: LICENSE
 Author: Gnosis

{prediction_market_agent_tooling-0.69.9 → prediction_market_agent_tooling-0.69.10}/prediction_market_agent_tooling/gtypes.py

@@ -1,6 +1,6 @@
 import typing as t
 from decimal import Decimal
-from typing import NewType
+from typing import Annotated, NewType, TypeAlias
 
 from eth_typing.evm import (  # noqa: F401  # Import for the sake of easy importing with others from here.
     Address,
@@ -8,6 +8,7 @@ from eth_typing.evm import (  # noqa: F401  # Import for the sake of easy import
     HexAddress,
     HexStr,
 )
+from pydantic import BeforeValidator
 from pydantic.types import SecretStr
 from pydantic.v1.types import SecretStr as SecretStrV1
 from web3 import Web3
@@ -18,6 +19,7 @@ from web3.types import (  # noqa: F401  # Import for the sake of easy importing
 )
 from web3.types import Wei as Web3Wei
 
+from prediction_market_agent_tooling.gtypes import ChecksumAddress
 from prediction_market_agent_tooling.tools._generic_value import _GenericValue
 from prediction_market_agent_tooling.tools.datetime_utc import (  # noqa: F401  # Import for the sake of easy importing with others from here.
     DatetimeUTC,
@@ -26,6 +28,14 @@ from prediction_market_agent_tooling.tools.hexbytes_custom import (  # noqa: F40
     HexBytes,
 )
 
+VerifiedChecksumAddress: TypeAlias = Annotated[
+    ChecksumAddress, BeforeValidator(Web3.to_checksum_address)
+]
+VerifiedChecksumAddressOrNone: TypeAlias = Annotated[
+    ChecksumAddress | None,
+    BeforeValidator(lambda x: Web3.to_checksum_address(x) if x is not None else None),
+]
+
 
 class CollateralToken(_GenericValue[int | float | str | Decimal, float], parser=float):
     """

{prediction_market_agent_tooling-0.69.9 → prediction_market_agent_tooling-0.69.10}/prediction_market_agent_tooling/tools/cow/cow_order.py

@@ -399,6 +399,22 @@ def get_orders_by_owner(
     return [Order.model_validate(i) for i in items]
 
 
+@tenacity.retry(
+    stop=stop_after_attempt(3),
+    wait=wait_fixed(1),
+    after=lambda x: logger.debug(f"get_order_by_uid failed, {x.attempt_number=}."),
+)
+async def get_order_by_uid(
+    uid: HexBytes,
+) -> Order:
+    async with httpx.AsyncClient() as client:
+        response = await client.get(
+            f"https://api.cow.fi/xdai/api/v1/orders/{uid.to_0x_hex()}"
+        )
+        response.raise_for_status()
+        return Order.model_validate(response.json())
+
+
 @tenacity.retry(
     stop=stop_after_attempt(3),
     wait=wait_fixed(1),

prediction_market_agent_tooling-0.69.10/prediction_market_agent_tooling/tools/cow/models.py

@@ -0,0 +1,119 @@
+from enum import Enum
+from typing import Optional
+
+from pydantic import BaseModel, ConfigDict
+from sqlmodel import Field, SQLModel
+
+from prediction_market_agent_tooling.gtypes import (
+    HexBytes,
+    VerifiedChecksumAddress,
+    VerifiedChecksumAddressOrNone,
+    Wei,
+)
+from prediction_market_agent_tooling.tools.datetime_utc import DatetimeUTC
+from prediction_market_agent_tooling.tools.utils import utcnow
+
+
+class MinimalisticTrade(BaseModel):
+    sellToken: VerifiedChecksumAddress
+    buyToken: VerifiedChecksumAddress
+    orderUid: HexBytes
+    txHash: HexBytes
+
+
+class EthFlowData(BaseModel):
+    refundTxHash: Optional[HexBytes]
+    userValidTo: int
+
+
+class PlacementError(str, Enum):
+    QuoteNotFound = "QuoteNotFound"
+    ValidToTooFarInFuture = "ValidToTooFarInFuture"
+    PreValidationError = "PreValidationError"
+
+
+class OnchainOrderData(BaseModel):
+    sender: VerifiedChecksumAddress
+    placementError: Optional[PlacementError]
+
+
+class OrderKind(str, Enum):
+    buy = "buy"
+    sell = "sell"
+
+
+class SellTokenBalance(str, Enum):
+    external = "external"
+    internal = "internal"
+    erc20 = "erc20"
+
+
+class BuyTokenBalance(str, Enum):
+    internal = "internal"
+    erc20 = "erc20"
+
+
+class SigningScheme(str, Enum):
+    eip712 = "eip712"
+    ethsign = "ethsign"
+    presign = "presign"
+    eip1271 = "eip1271"
+
+
+class OrderClass(str, Enum):
+    limit = "limit"
+    liquidity = "liquidity"
+    market = "market"
+
+
+class OrderStatus(str, Enum):
+    presignaturePending = "presignaturePending"
+    open = "open"
+    fulfilled = "fulfilled"
+    cancelled = "cancelled"
+    expired = "expired"
+
+
+class Order(BaseModel):
+    model_config = ConfigDict(validate_by_name=True, validate_by_alias=True)
+
+    uid: str
+    quoteId: int | None = None
+    validTo: int
+    sellAmount: Wei
+    sellToken: VerifiedChecksumAddress
+    buyAmount: Wei
+    buyToken: VerifiedChecksumAddress
+    receiver: VerifiedChecksumAddressOrNone
+    feeAmount: Wei
+    creationDate: DatetimeUTC
+    kind: OrderKind
+    partiallyFillable: bool
+    sellTokenBalance: SellTokenBalance | None
+    buyTokenBalance: BuyTokenBalance | None
+    signingScheme: SigningScheme
+    signature: HexBytes
+    from_: VerifiedChecksumAddressOrNone = Field(None, alias="from")
+    appData: str
+    fullAppData: str | None
+    appDataHash: HexBytes | None = None
+    class_: str = Field(None, alias="class")
+    owner: VerifiedChecksumAddress
+    executedSellAmount: Wei
+    executedSellAmountBeforeFees: Wei
+    executedBuyAmount: Wei
+    executedFeeAmount: Wei | None
+    invalidated: bool
+    status: str
+    isLiquidityOrder: bool | None
+    ethflowData: EthFlowData | None = None
+    onchainUser: VerifiedChecksumAddressOrNone = None
+    executedFee: Wei
+    executedFeeToken: VerifiedChecksumAddressOrNone
+
+
+class RateLimit(SQLModel, table=True):
+    __tablename__ = "rate_limit"
+    __table_args__ = {"extend_existing": True}
+    id: str = Field(primary_key=True)
+    last_called_at: DatetimeUTC = Field(default_factory=utcnow)

{prediction_market_agent_tooling-0.69.9 → prediction_market_agent_tooling-0.69.10}/pyproject.toml

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

prediction_market_agent_tooling-0.69.9/prediction_market_agent_tooling/tools/cow/models.py

@@ -1,27 +0,0 @@
-from pydantic import BaseModel
-from sqlmodel import Field, SQLModel
-
-from prediction_market_agent_tooling.gtypes import ChecksumAddress, HexBytes
-from prediction_market_agent_tooling.tools.datetime_utc import DatetimeUTC
-from prediction_market_agent_tooling.tools.utils import utcnow
-
-
-class MinimalisticTrade(BaseModel):
-    sellToken: ChecksumAddress
-    buyToken: ChecksumAddress
-    orderUid: HexBytes
-    txHash: HexBytes
-
-
-class Order(BaseModel):
-    uid: str
-    sellToken: str
-    buyToken: str
-    creationDate: DatetimeUTC
-
-
-class RateLimit(SQLModel, table=True):
-    __tablename__ = "rate_limit"
-    __table_args__ = {"extend_existing": True}
-    id: str = Field(primary_key=True)
-    last_called_at: DatetimeUTC = Field(default_factory=utcnow)