compass_api_sdk 0.9.51__py3-none-any.whl → 1.0.0__py3-none-any.whl

compass_api_sdk/models/aaveborrowrequest.py

@@ -1,7 +1,6 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .chain import Chain
 from .interestratemode import InterestRateMode
 from .token_enum import TokenEnum
 from compass_api_sdk.types import (
@@ -12,6 +11,7 @@ from compass_api_sdk.types import (
     UNSET_SENTINEL,
 )
 from compass_api_sdk.utils import validate_const
+from enum import Enum
 import pydantic
 from pydantic import model_serializer
 from pydantic.functional_validators import AfterValidator
@@ -29,6 +29,12 @@ AaveBorrowRequestAmount = TypeAliasType("AaveBorrowRequestAmount", Union[float,
 r"""The amount of the asset to borrow"""
 
 
+class AaveBorrowRequestChain(str, Enum):
+    ARBITRUM = "arbitrum"
+    BASE = "base"
+    ETHEREUM = "ethereum"
+
+
 class AaveBorrowRequestTypedDict(TypedDict):
     token: TokenEnum
     r"""A class representing the token.
@@ -43,8 +49,7 @@ class AaveBorrowRequestTypedDict(TypedDict):
 
     A stable (but typically higher rate), or a variable rate.
     """
-    chain: Chain
-    r"""The chain to use."""
+    chain: AaveBorrowRequestChain
     sender: str
     r"""The address of the transaction sender."""
     action_type: Literal["AAVE_BORROW"]
@@ -69,8 +74,7 @@ class AaveBorrowRequest(BaseModel):
     A stable (but typically higher rate), or a variable rate.
     """
 
-    chain: Chain
-    r"""The chain to use."""
+    chain: AaveBorrowRequestChain
 
     sender: str
     r"""The address of the transaction sender."""

compass_api_sdk/models/aavehistoricaltransactionsresponse.py

@@ -15,8 +15,8 @@ from typing import List, Union
 from typing_extensions import Annotated, TypeAliasType, TypedDict
 
 
-TransactionTypedDict = TypeAliasType(
-    "TransactionTypedDict",
+AaveHistoricalTransactionsResponseTransactionTypedDict = TypeAliasType(
+    "AaveHistoricalTransactionsResponseTransactionTypedDict",
     Union[
         RedeemUnderlyingTypedDict,
         RepayTypedDict,
@@ -29,7 +29,7 @@ TransactionTypedDict = TypeAliasType(
 )
 
 
-Transaction = Annotated[
+AaveHistoricalTransactionsResponseTransaction = Annotated[
     Union[
         Annotated[Borrow, Tag("Borrow")],
         Annotated[LiquidationCall, Tag("LiquidationCall")],
@@ -50,7 +50,7 @@ class AaveHistoricalTransactionsResponseTypedDict(TypedDict):
     r"""Specifies how many transactions to skip before returning results, letting you choose the starting point for the data you want to receive."""
     limit: int
     r"""Sets the maximum number of transactions to include in the response, helping control the size of the returned dataset."""
-    transactions: List[TransactionTypedDict]
+    transactions: List[AaveHistoricalTransactionsResponseTransactionTypedDict]
 
 
 class AaveHistoricalTransactionsResponse(BaseModel):
@@ -62,4 +62,4 @@ class AaveHistoricalTransactionsResponse(BaseModel):
     limit: int
     r"""Sets the maximum number of transactions to include in the response, helping control the size of the returned dataset."""
 
-    transactions: List[Transaction]
+    transactions: List[AaveHistoricalTransactionsResponseTransaction]

compass_api_sdk/models/aavelooprequest.py

@@ -1,12 +1,25 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .chain import Chain
 from .signedauthorization import SignedAuthorization, SignedAuthorizationTypedDict
 from .token_enum import TokenEnum
-from compass_api_sdk.types import BaseModel
-from typing import Union
-from typing_extensions import TypeAliasType, TypedDict
+from compass_api_sdk.types import (
+    BaseModel,
+    Nullable,
+    OptionalNullable,
+    UNSET,
+    UNSET_SENTINEL,
+)
+from enum import Enum
+from pydantic import model_serializer
+from typing import Optional, Union
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
+
+
+class AaveLoopRequestChain(str, Enum):
+    ARBITRUM = "arbitrum"
+    BASE = "base"
+    ETHEREUM = "ethereum"
 
 
 InitialCollateralAmountTypedDict = TypeAliasType(
@@ -48,11 +61,9 @@ r"""Loan To Value percentage of the loop"""
 class AaveLoopRequestTypedDict(TypedDict):
     r"""Request model for executing an Aave loop strategy."""
 
-    chain: Chain
-    r"""The chain to use."""
+    chain: AaveLoopRequestChain
     sender: str
     r"""The address of the transaction sender."""
-    signed_authorization: SignedAuthorizationTypedDict
     collateral_token: TokenEnum
     r"""A class representing the token.
 
@@ -73,19 +84,20 @@ class AaveLoopRequestTypedDict(TypedDict):
     r"""Maximum allowed slippage for token swaps in percentage"""
     loan_to_value: LoanToValueTypedDict
     r"""Loan To Value percentage of the loop"""
+    signed_authorization: NotRequired[Nullable[SignedAuthorizationTypedDict]]
+    r"""EIP-7702 authorization. Required when `is_account_abstraction` is False."""
+    is_account_abstraction: NotRequired[bool]
+    r"""Whether to use account abstraction for the transaction."""
 
 
 class AaveLoopRequest(BaseModel):
     r"""Request model for executing an Aave loop strategy."""
 
-    chain: Chain
-    r"""The chain to use."""
+    chain: AaveLoopRequestChain
 
     sender: str
     r"""The address of the transaction sender."""
 
-    signed_authorization: SignedAuthorization
-
     collateral_token: TokenEnum
     r"""A class representing the token.
 
@@ -111,3 +123,39 @@ class AaveLoopRequest(BaseModel):
 
     loan_to_value: LoanToValue
     r"""Loan To Value percentage of the loop"""
+
+    signed_authorization: OptionalNullable[SignedAuthorization] = UNSET
+    r"""EIP-7702 authorization. Required when `is_account_abstraction` is False."""
+
+    is_account_abstraction: Optional[bool] = False
+    r"""Whether to use account abstraction for the transaction."""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["signed_authorization", "is_account_abstraction"]
+        nullable_fields = ["signed_authorization"]
+        null_default_fields = []
+
+        serialized = handler(self)
+
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+            serialized.pop(k, None)
+
+            optional_nullable = k in optional_fields and k in nullable_fields
+            is_set = (
+                self.__pydantic_fields_set__.intersection({n})
+                or k in null_default_fields
+            )  # pylint: disable=no-member
+
+            if val is not None and val != UNSET_SENTINEL:
+                m[k] = val
+            elif val != UNSET_SENTINEL and (
+                not k in optional_fields or (optional_nullable and is_set)
+            ):
+                m[k] = val
+
+        return m

compass_api_sdk/models/aaverepayrequest.py

@@ -1,7 +1,6 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .chain import Chain
 from .interestratemode import InterestRateMode
 from .token_enum import TokenEnum
 from compass_api_sdk.types import (
@@ -12,6 +11,7 @@ from compass_api_sdk.types import (
     UNSET_SENTINEL,
 )
 from compass_api_sdk.utils import validate_const
+from enum import Enum
 import pydantic
 from pydantic import model_serializer
 from pydantic.functional_validators import AfterValidator
@@ -29,6 +29,12 @@ AaveRepayRequestAmount = TypeAliasType("AaveRepayRequestAmount", Union[float, st
 r"""The amount of the asset to repay"""
 
 
+class AaveRepayRequestChain(str, Enum):
+    ARBITRUM = "arbitrum"
+    BASE = "base"
+    ETHEREUM = "ethereum"
+
+
 class AaveRepayRequestTypedDict(TypedDict):
     token: TokenEnum
     r"""A class representing the token.
@@ -43,8 +49,7 @@ class AaveRepayRequestTypedDict(TypedDict):
 
     A stable (but typically higher rate), or a variable rate.
     """
-    chain: Chain
-    r"""The chain to use."""
+    chain: AaveRepayRequestChain
     sender: str
     r"""The address of the transaction sender."""
     action_type: Literal["AAVE_REPAY"]
@@ -69,8 +74,7 @@ class AaveRepayRequest(BaseModel):
     A stable (but typically higher rate), or a variable rate.
     """
 
-    chain: Chain
-    r"""The chain to use."""
+    chain: AaveRepayRequestChain
 
     sender: str
     r"""The address of the transaction sender."""

compass_api_sdk/models/aavereserveoverviewresponse.py

@@ -1,7 +1,8 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from compass_api_sdk.types import BaseModel
+from compass_api_sdk.types import BaseModel, Nullable, UNSET_SENTINEL
+from pydantic import model_serializer
 from typing_extensions import TypedDict
 
 
@@ -10,7 +11,7 @@ class AaveReserveOverviewResponseTypedDict(TypedDict):
     r"""Total tokens supplied in an Aave Reserve in USD. E.G. How much WBTC has been supplied on Aave in USD."""
     total_borrowed: float
     r"""Total tokens borrowed in an Aave Reserve converted to USD. E.G. How much WBTC has been supplied on Aave (in USD)."""
-    utilization_ratio: float
+    utilization_ratio: Nullable[float]
     r"""Total borrowed divided by total supplied in an Aave Reserve. E.G. How much WBTC has been borrowed on Aaave divided by the amount supplied"""
 
 
@@ -21,5 +22,35 @@ class AaveReserveOverviewResponse(BaseModel):
     total_borrowed: float
     r"""Total tokens borrowed in an Aave Reserve converted to USD. E.G. How much WBTC has been supplied on Aave (in USD)."""
 
-    utilization_ratio: float
+    utilization_ratio: Nullable[float]
     r"""Total borrowed divided by total supplied in an Aave Reserve. E.G. How much WBTC has been borrowed on Aaave divided by the amount supplied"""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = []
+        nullable_fields = ["utilization_ratio"]
+        null_default_fields = []
+
+        serialized = handler(self)
+
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+            serialized.pop(k, None)
+
+            optional_nullable = k in optional_fields and k in nullable_fields
+            is_set = (
+                self.__pydantic_fields_set__.intersection({n})
+                or k in null_default_fields
+            )  # pylint: disable=no-member
+
+            if val is not None and val != UNSET_SENTINEL:
+                m[k] = val
+            elif val != UNSET_SENTINEL and (
+                not k in optional_fields or (optional_nullable and is_set)
+            ):
+                m[k] = val
+
+        return m

compass_api_sdk/models/aavesupplyrequest.py

@@ -1,7 +1,6 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .chain import Chain
 from .token_enum import TokenEnum
 from compass_api_sdk.types import (
     BaseModel,
@@ -11,6 +10,7 @@ from compass_api_sdk.types import (
     UNSET_SENTINEL,
 )
 from compass_api_sdk.utils import validate_const
+from enum import Enum
 import pydantic
 from pydantic import model_serializer
 from pydantic.functional_validators import AfterValidator
@@ -28,6 +28,12 @@ AaveSupplyRequestAmount = TypeAliasType("AaveSupplyRequestAmount", Union[float,
 r"""The amount of the asset to supply"""
 
 
+class AaveSupplyRequestChain(str, Enum):
+    ARBITRUM = "arbitrum"
+    BASE = "base"
+    ETHEREUM = "ethereum"
+
+
 class AaveSupplyRequestTypedDict(TypedDict):
     token: TokenEnum
     r"""A class representing the token.
@@ -37,8 +43,7 @@ class AaveSupplyRequestTypedDict(TypedDict):
     """
     amount: AaveSupplyRequestAmountTypedDict
     r"""The amount of the asset to supply"""
-    chain: Chain
-    r"""The chain to use."""
+    chain: AaveSupplyRequestChain
     sender: str
     r"""The address of the transaction sender."""
     action_type: Literal["AAVE_SUPPLY"]
@@ -57,8 +62,7 @@ class AaveSupplyRequest(BaseModel):
     amount: AaveSupplyRequestAmount
     r"""The amount of the asset to supply"""
 
-    chain: Chain
-    r"""The chain to use."""
+    chain: AaveSupplyRequestChain
 
     sender: str
     r"""The address of the transaction sender."""

compass_api_sdk/models/aavewithdrawrequest.py

@@ -1,10 +1,10 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .chain import Chain
 from .token_enum import TokenEnum
 from compass_api_sdk.types import BaseModel
 from compass_api_sdk.utils import validate_const
+from enum import Enum
 import pydantic
 from pydantic.functional_validators import AfterValidator
 from typing import Literal, Optional, Union
@@ -23,6 +23,12 @@ AaveWithdrawRequestAmount = TypeAliasType(
 r"""The amount of the asset to withdraw"""
 
 
+class AaveWithdrawRequestChain(str, Enum):
+    ARBITRUM = "arbitrum"
+    BASE = "base"
+    ETHEREUM = "ethereum"
+
+
 class AaveWithdrawRequestTypedDict(TypedDict):
     token: TokenEnum
     r"""A class representing the token.
@@ -34,8 +40,7 @@ class AaveWithdrawRequestTypedDict(TypedDict):
     r"""The amount of the asset to withdraw"""
     recipient: str
     r"""The address of the recipient of the withdrawn funds."""
-    chain: Chain
-    r"""The chain to use."""
+    chain: AaveWithdrawRequestChain
     sender: str
     r"""The address of the transaction sender."""
     action_type: Literal["AAVE_WITHDRAW"]
@@ -55,8 +60,7 @@ class AaveWithdrawRequest(BaseModel):
     recipient: str
     r"""The address of the recipient of the withdrawn funds."""
 
-    chain: Chain
-    r"""The chain to use."""
+    chain: AaveWithdrawRequestChain
 
     sender: str
     r"""The address of the transaction sender."""

compass_api_sdk/models/aerodromeslipstreambuyexactlyrequest.py

@@ -1,10 +1,10 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .chain import Chain
 from .token_enum import TokenEnum
 from compass_api_sdk.types import BaseModel
 from compass_api_sdk.utils import validate_const
+from enum import Enum
 import pydantic
 from pydantic.functional_validators import AfterValidator
 from typing import Literal, Optional, Union
@@ -35,6 +35,10 @@ AerodromeSlipstreamBuyExactlyRequestAmountInMaximum = TypeAliasType(
 r"""The maximum amount of the token to swap from"""
 
 
+class AerodromeSlipstreamBuyExactlyRequestChain(str, Enum):
+    BASE = "base"
+
+
 class AerodromeSlipstreamBuyExactlyRequestTypedDict(TypedDict):
     r"""Request model for buying exactly an amount of tokens."""
 
@@ -56,8 +60,7 @@ class AerodromeSlipstreamBuyExactlyRequestTypedDict(TypedDict):
     r"""The amount of the token to swap to"""
     amount_in_maximum: AerodromeSlipstreamBuyExactlyRequestAmountInMaximumTypedDict
     r"""The maximum amount of the token to swap from"""
-    chain: Chain
-    r"""The chain to use."""
+    chain: AerodromeSlipstreamBuyExactlyRequestChain
     sender: str
     r"""The address of the transaction sender."""
     action_type: Literal["AERODROME_SLIPSTREAM_BUY_EXACTLY"]
@@ -89,8 +92,7 @@ class AerodromeSlipstreamBuyExactlyRequest(BaseModel):
     amount_in_maximum: AerodromeSlipstreamBuyExactlyRequestAmountInMaximum
     r"""The maximum amount of the token to swap from"""
 
-    chain: Chain
-    r"""The chain to use."""
+    chain: AerodromeSlipstreamBuyExactlyRequestChain
 
     sender: str
     r"""The address of the transaction sender."""

compass_api_sdk/models/aerodromeslipstreamincreaseliquidityprovisionrequest.py

@@ -1,9 +1,9 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .chain import Chain
 from compass_api_sdk.types import BaseModel
 from compass_api_sdk.utils import validate_const
+from enum import Enum
 import pydantic
 from pydantic.functional_validators import AfterValidator
 from typing import Literal, Optional, Union
@@ -68,6 +68,10 @@ AerodromeSlipstreamIncreaseLiquidityProvisionRequestAmount1Min = TypeAliasType(
 r"""The minimum amount of the second token to deposit"""
 
 
+class AerodromeSlipstreamIncreaseLiquidityProvisionRequestChain(str, Enum):
+    BASE = "base"
+
+
 class AerodromeSlipstreamIncreaseLiquidityProvisionRequestTypedDict(TypedDict):
     token_id: int
     r"""Token ID of the NFT representing the liquidity provisioned position."""
@@ -83,8 +87,7 @@ class AerodromeSlipstreamIncreaseLiquidityProvisionRequestTypedDict(TypedDict):
     r"""The minimum amount of the first token to deposit"""
     amount1_min: AerodromeSlipstreamIncreaseLiquidityProvisionRequestAmount1MinTypedDict
     r"""The minimum amount of the second token to deposit"""
-    chain: Chain
-    r"""The chain to use."""
+    chain: AerodromeSlipstreamIncreaseLiquidityProvisionRequestChain
     sender: str
     r"""The address of the transaction sender."""
     action_type: Literal["AERODROME_SLIPSTREAM_INCREASE_LIQUIDITY_PROVISION"]
@@ -106,8 +109,7 @@ class AerodromeSlipstreamIncreaseLiquidityProvisionRequest(BaseModel):
     amount1_min: AerodromeSlipstreamIncreaseLiquidityProvisionRequestAmount1Min
     r"""The minimum amount of the second token to deposit"""
 
-    chain: Chain
-    r"""The chain to use."""
+    chain: AerodromeSlipstreamIncreaseLiquidityProvisionRequestChain
 
     sender: str
     r"""The address of the transaction sender."""

compass_api_sdk/models/aerodromeslipstreammintliquidityprovisionrequest.py

@@ -1,7 +1,6 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .chain import Chain
 from .token_enum import TokenEnum
 from compass_api_sdk.types import (
     BaseModel,
@@ -11,6 +10,7 @@ from compass_api_sdk.types import (
     UNSET_SENTINEL,
 )
 from compass_api_sdk.utils import validate_const
+from enum import Enum
 import pydantic
 from pydantic import model_serializer
 from pydantic.functional_validators import AfterValidator
@@ -70,6 +70,10 @@ AerodromeSlipstreamMintLiquidityProvisionRequestAmount1Min = TypeAliasType(
 r"""The minimum amount of the second token to deposit"""
 
 
+class AerodromeSlipstreamMintLiquidityProvisionRequestChain(str, Enum):
+    BASE = "base"
+
+
 class AerodromeSlipstreamMintLiquidityProvisionRequestTypedDict(TypedDict):
     r"""Request model for minting a new liquidity position."""
 
@@ -103,8 +107,7 @@ class AerodromeSlipstreamMintLiquidityProvisionRequestTypedDict(TypedDict):
     r"""The minimum amount of the first token to deposit"""
     amount1_min: AerodromeSlipstreamMintLiquidityProvisionRequestAmount1MinTypedDict
     r"""The minimum amount of the second token to deposit"""
-    chain: Chain
-    r"""The chain to use."""
+    chain: AerodromeSlipstreamMintLiquidityProvisionRequestChain
     sender: str
     r"""The address of the transaction sender."""
     action_type: Literal["AERODROME_SLIPSTREAM_MINT_LIQUIDITY_PROVISION"]
@@ -150,8 +153,7 @@ class AerodromeSlipstreamMintLiquidityProvisionRequest(BaseModel):
     amount1_min: AerodromeSlipstreamMintLiquidityProvisionRequestAmount1Min
     r"""The minimum amount of the second token to deposit"""
 
-    chain: Chain
-    r"""The chain to use."""
+    chain: AerodromeSlipstreamMintLiquidityProvisionRequestChain
 
     sender: str
     r"""The address of the transaction sender."""

compass_api_sdk/models/aerodromeslipstreamsellexactlyrequest.py

@@ -1,10 +1,10 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .chain import Chain
 from .token_enum import TokenEnum
 from compass_api_sdk.types import BaseModel
 from compass_api_sdk.utils import validate_const
+from enum import Enum
 import pydantic
 from pydantic.functional_validators import AfterValidator
 from typing import Literal, Optional, Union
@@ -35,6 +35,10 @@ AerodromeSlipstreamSellExactlyRequestAmountOutMinimum = TypeAliasType(
 r"""The minimum amount of the token to swap to, defaults to 0"""
 
 
+class AerodromeSlipstreamSellExactlyRequestChain(str, Enum):
+    BASE = "base"
+
+
 class AerodromeSlipstreamSellExactlyRequestTypedDict(TypedDict):
     r"""Request model for selling exactly an amount of tokens."""
 
@@ -54,8 +58,7 @@ class AerodromeSlipstreamSellExactlyRequestTypedDict(TypedDict):
     r"""The tick spacing of the pool"""
     amount_in: AerodromeSlipstreamSellExactlyRequestAmountInTypedDict
     r"""The amount of the token to swap from"""
-    chain: Chain
-    r"""The chain to use."""
+    chain: AerodromeSlipstreamSellExactlyRequestChain
     sender: str
     r"""The address of the transaction sender."""
     action_type: Literal["AERODROME_SLIPSTREAM_SELL_EXACTLY"]
@@ -88,8 +91,7 @@ class AerodromeSlipstreamSellExactlyRequest(BaseModel):
     amount_in: AerodromeSlipstreamSellExactlyRequestAmountIn
     r"""The amount of the token to swap from"""
 
-    chain: Chain
-    r"""The chain to use."""
+    chain: AerodromeSlipstreamSellExactlyRequestChain
 
     sender: str
     r"""The address of the transaction sender."""

compass_api_sdk/models/aerodromeslipstreamwithdrawliquidityprovisionrequest.py

@@ -1,9 +1,9 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .chain import Chain
 from compass_api_sdk.types import BaseModel
 from compass_api_sdk.utils import validate_const
+from enum import Enum
 import pydantic
 from pydantic.functional_validators import AfterValidator
 from typing import Literal, Optional, Union
@@ -26,13 +26,16 @@ AerodromeSlipstreamWithdrawLiquidityProvisionRequestPercentageForWithdrawal = (
 r"""How much liquidity to take out in percentage."""
 
 
+class AerodromeSlipstreamWithdrawLiquidityProvisionRequestChain(str, Enum):
+    BASE = "base"
+
+
 class AerodromeSlipstreamWithdrawLiquidityProvisionRequestTypedDict(TypedDict):
     token_id: int
     r"""Token ID of the NFT representing the liquidity provisioned position."""
     percentage_for_withdrawal: AerodromeSlipstreamWithdrawLiquidityProvisionRequestPercentageForWithdrawalTypedDict
     r"""How much liquidity to take out in percentage."""
-    chain: Chain
-    r"""The chain to use."""
+    chain: AerodromeSlipstreamWithdrawLiquidityProvisionRequestChain
     sender: str
     r"""The address of the transaction sender."""
     action_type: Literal["AERODROME_SLIPSTREAM_WITHDRAW_LIQUIDITY_PROVISION"]
@@ -47,8 +50,7 @@ class AerodromeSlipstreamWithdrawLiquidityProvisionRequest(BaseModel):
     )
     r"""How much liquidity to take out in percentage."""
 
-    chain: Chain
-    r"""The chain to use."""
+    chain: AerodromeSlipstreamWithdrawLiquidityProvisionRequestChain
 
     sender: str
     r"""The address of the transaction sender."""

compass_api_sdk/models/batcheduseroperationsrequest.py

@@ -1,18 +1,23 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .chain import Chain
 from .useroperation import UserOperation, UserOperationTypedDict
 from compass_api_sdk.types import BaseModel
+from enum import Enum
 from typing import List
 from typing_extensions import TypedDict
 
 
+class BatchedUserOperationsRequestChain(str, Enum):
+    ARBITRUM = "arbitrum"
+    BASE = "base"
+    ETHEREUM = "ethereum"
+
+
 class BatchedUserOperationsRequestTypedDict(TypedDict):
     r"""Request model for batching user operations."""
 
-    chain: Chain
-    r"""The chain to use."""
+    chain: BatchedUserOperationsRequestChain
     sender: str
     r"""The address of the transaction sender."""
     operations: List[UserOperationTypedDict]
@@ -22,8 +27,7 @@ class BatchedUserOperationsRequestTypedDict(TypedDict):
 class BatchedUserOperationsRequest(BaseModel):
     r"""Request model for batching user operations."""
 
-    chain: Chain
-    r"""The chain to use."""
+    chain: BatchedUserOperationsRequestChain
 
     sender: str
     r"""The address of the transaction sender."""

compass_api_sdk/models/chain.py

@@ -7,6 +7,6 @@ from enum import Enum
 class Chain(str, Enum):
     r"""The chain to use."""
 
-    BASE_MAINNET = "base:mainnet"
-    ETHEREUM_MAINNET = "ethereum:mainnet"
-    ARBITRUM_MAINNET = "arbitrum:mainnet"
+    BASE = "base"
+    ETHEREUM = "ethereum"
+    ARBITRUM = "arbitrum"