compass_api_sdk 0.5.4__py3-none-any.whl → 0.6.1__py3-none-any.whl

compass_api_sdk/models/token_balanceop.py

@@ -2,10 +2,17 @@
 
 from __future__ import annotations
 from .token_enum import TokenEnum
-from compass_api_sdk.types import BaseModel
+from compass_api_sdk.types import (
+    BaseModel,
+    Nullable,
+    OptionalNullable,
+    UNSET,
+    UNSET_SENTINEL,
+)
 from compass_api_sdk.utils import FieldMetadata, QueryParamMetadata
 from enum import Enum
-from typing import Optional, Union
+from pydantic import model_serializer
+from typing import Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
@@ -17,16 +24,6 @@ class TokenBalanceChain(str, Enum):
     ARBITRUM_MAINNET = "arbitrum:mainnet"
 
 
-TokenBalanceBlockTypedDict = TypeAliasType(
-    "TokenBalanceBlockTypedDict", Union[int, str]
-)
-r"""The block number you want to get this data at."""
-
-
-TokenBalanceBlock = TypeAliasType("TokenBalanceBlock", Union[int, str])
-r"""The block number you want to get this data at."""
-
-
 TokenBalanceTokenTypedDict = TypeAliasType(
     "TokenBalanceTokenTypedDict", Union[TokenEnum, str]
 )
@@ -42,8 +39,8 @@ class TokenBalanceRequestTypedDict(TypedDict):
     r"""The symbol of the token for which the balance is checked.."""
     chain: TokenBalanceChain
     r"""The chain to use."""
-    block: NotRequired[TokenBalanceBlockTypedDict]
-    r"""The block number you want to get this data at."""
+    block: NotRequired[Nullable[int]]
+    r"""Optional block number (defaults to latest)."""
     user: str
     r"""The user to get the ERC20 balance of."""
 
@@ -62,12 +59,42 @@ class TokenBalanceRequest(BaseModel):
     r"""The chain to use."""
 
     block: Annotated[
-        Optional[TokenBalanceBlock],
+        OptionalNullable[int],
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
-    ] = None
-    r"""The block number you want to get this data at."""
+    ] = UNSET
+    r"""Optional block number (defaults to latest)."""
 
     user: Annotated[
         str, FieldMetadata(query=QueryParamMetadata(style="form", explode=True))
     ] = "0x29F20a192328eF1aD35e1564aBFf4Be9C5ce5f7B"
     r"""The user to get the ERC20 balance of."""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["block"]
+        nullable_fields = ["block"]
+        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/uniswap_liquidity_provision_in_rangeop.py

@@ -1,11 +1,17 @@
 """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,
+    OptionalNullable,
+    UNSET,
+    UNSET_SENTINEL,
+)
 from compass_api_sdk.utils import FieldMetadata, QueryParamMetadata
 from enum import Enum
-from typing import Optional, Union
-from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
+from pydantic import model_serializer
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 class UniswapLiquidityProvisionInRangeChain(str, Enum):
@@ -16,23 +22,11 @@ class UniswapLiquidityProvisionInRangeChain(str, Enum):
     ARBITRUM_MAINNET = "arbitrum:mainnet"
 
 
-UniswapLiquidityProvisionInRangeBlockTypedDict = TypeAliasType(
-    "UniswapLiquidityProvisionInRangeBlockTypedDict", Union[int, str]
-)
-r"""The block number you want to get this data at."""
-
-
-UniswapLiquidityProvisionInRangeBlock = TypeAliasType(
-    "UniswapLiquidityProvisionInRangeBlock", Union[int, str]
-)
-r"""The block number you want to get this data at."""
-
-
 class UniswapLiquidityProvisionInRangeRequestTypedDict(TypedDict):
     chain: UniswapLiquidityProvisionInRangeChain
     r"""The chain to use."""
-    block: NotRequired[UniswapLiquidityProvisionInRangeBlockTypedDict]
-    r"""The block number you want to get this data at."""
+    block: NotRequired[Nullable[int]]
+    r"""Optional block number (defaults to latest)."""
     token_id: int
     r"""Token ID of the NFT representing the liquidity provisioned position."""
 
@@ -45,12 +39,42 @@ class UniswapLiquidityProvisionInRangeRequest(BaseModel):
     r"""The chain to use."""
 
     block: Annotated[
-        Optional[UniswapLiquidityProvisionInRangeBlock],
+        OptionalNullable[int],
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
-    ] = None
-    r"""The block number you want to get this data at."""
+    ] = UNSET
+    r"""Optional block number (defaults to latest)."""
 
     token_id: Annotated[
         int, FieldMetadata(query=QueryParamMetadata(style="form", explode=True))
     ] = 4318185
     r"""Token ID of the NFT representing the liquidity provisioned position."""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["block"]
+        nullable_fields = ["block"]
+        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/uniswap_liquidity_provision_positionsop.py

@@ -1,11 +1,18 @@
 """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,
+    OptionalNullable,
+    UNSET,
+    UNSET_SENTINEL,
+)
 from compass_api_sdk.utils import FieldMetadata, QueryParamMetadata
 from enum import Enum
-from typing import Optional, Union
-from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
+from pydantic import model_serializer
+from typing import Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 class UniswapLiquidityProvisionPositionsChain(str, Enum):
@@ -16,23 +23,11 @@ class UniswapLiquidityProvisionPositionsChain(str, Enum):
     ARBITRUM_MAINNET = "arbitrum:mainnet"
 
 
-UniswapLiquidityProvisionPositionsBlockTypedDict = TypeAliasType(
-    "UniswapLiquidityProvisionPositionsBlockTypedDict", Union[int, str]
-)
-r"""The block number you want to get this data at."""
-
-
-UniswapLiquidityProvisionPositionsBlock = TypeAliasType(
-    "UniswapLiquidityProvisionPositionsBlock", Union[int, str]
-)
-r"""The block number you want to get this data at."""
-
-
 class UniswapLiquidityProvisionPositionsRequestTypedDict(TypedDict):
     chain: UniswapLiquidityProvisionPositionsChain
     r"""The chain to use."""
-    block: NotRequired[UniswapLiquidityProvisionPositionsBlockTypedDict]
-    r"""The block number you want to get this data at."""
+    block: NotRequired[Nullable[int]]
+    r"""Optional block number (defaults to latest)."""
     user: NotRequired[str]
     r"""The user to get positions for."""
 
@@ -45,13 +40,43 @@ class UniswapLiquidityProvisionPositionsRequest(BaseModel):
     r"""The chain to use."""
 
     block: Annotated[
-        Optional[UniswapLiquidityProvisionPositionsBlock],
+        OptionalNullable[int],
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
-    ] = None
-    r"""The block number you want to get this data at."""
+    ] = UNSET
+    r"""Optional block number (defaults to latest)."""
 
     user: Annotated[
         Optional[str],
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = "0x29F20a192328eF1aD35e1564aBFf4Be9C5ce5f7B"
     r"""The user to get positions for."""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["block", "user"]
+        nullable_fields = ["block"]
+        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/uniswap_pool_priceop.py

@@ -1,11 +1,17 @@
 """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,
+    OptionalNullable,
+    UNSET,
+    UNSET_SENTINEL,
+)
 from compass_api_sdk.utils import FieldMetadata, QueryParamMetadata
 from enum import Enum
-from typing import Optional, Union
-from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
+from pydantic import model_serializer
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 class UniswapPoolPriceChain(str, Enum):
@@ -16,16 +22,6 @@ class UniswapPoolPriceChain(str, Enum):
     ARBITRUM_MAINNET = "arbitrum:mainnet"
 
 
-UniswapPoolPriceBlockTypedDict = TypeAliasType(
-    "UniswapPoolPriceBlockTypedDict", Union[int, str]
-)
-r"""The block number you want to get this data at."""
-
-
-UniswapPoolPriceBlock = TypeAliasType("UniswapPoolPriceBlock", Union[int, str])
-r"""The block number you want to get this data at."""
-
-
 class UniswapPoolPriceTokenInToken(str, Enum):
     r"""A class representing the token.
 
@@ -147,8 +143,8 @@ class UniswapPoolPriceFeeEnum(str, Enum):
 class UniswapPoolPriceRequestTypedDict(TypedDict):
     chain: UniswapPoolPriceChain
     r"""The chain to use."""
-    block: NotRequired[UniswapPoolPriceBlockTypedDict]
-    r"""The block number you want to get this data at."""
+    block: NotRequired[Nullable[int]]
+    r"""Optional block number (defaults to latest)."""
     token_in: UniswapPoolPriceTokenInToken
     r"""The symbol of a token in the pool"""
     token_out: UniswapPoolPriceTokenOutToken
@@ -165,10 +161,10 @@ class UniswapPoolPriceRequest(BaseModel):
     r"""The chain to use."""
 
     block: Annotated[
-        Optional[UniswapPoolPriceBlock],
+        OptionalNullable[int],
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
-    ] = None
-    r"""The block number you want to get this data at."""
+    ] = UNSET
+    r"""Optional block number (defaults to latest)."""
 
     token_in: Annotated[
         UniswapPoolPriceTokenInToken,
@@ -187,3 +183,33 @@ class UniswapPoolPriceRequest(BaseModel):
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = UniswapPoolPriceFeeEnum.ZERO_DOT_01
     r"""The fee of the pool"""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["block"]
+        nullable_fields = ["block"]
+        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/uniswap_quote_buy_exactlyop.py

@@ -1,10 +1,17 @@
 """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,
+    OptionalNullable,
+    UNSET,
+    UNSET_SENTINEL,
+)
 from compass_api_sdk.utils import FieldMetadata, QueryParamMetadata
 from enum import Enum
-from typing import Optional, Union
+from pydantic import model_serializer
+from typing import Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
@@ -16,18 +23,6 @@ class UniswapQuoteBuyExactlyChain(str, Enum):
     ARBITRUM_MAINNET = "arbitrum:mainnet"
 
 
-UniswapQuoteBuyExactlyBlockTypedDict = TypeAliasType(
-    "UniswapQuoteBuyExactlyBlockTypedDict", Union[int, str]
-)
-r"""The block number you want to get this data at."""
-
-
-UniswapQuoteBuyExactlyBlock = TypeAliasType(
-    "UniswapQuoteBuyExactlyBlock", Union[int, str]
-)
-r"""The block number you want to get this data at."""
-
-
 class UniswapQuoteBuyExactlyTokenInToken(str, Enum):
     r"""A class representing the token.
 
@@ -163,8 +158,8 @@ class UniswapQuoteBuyExactlyRequestTypedDict(TypedDict):
     r"""The amount of the token to swap to"""
     chain: UniswapQuoteBuyExactlyChain
     r"""The chain to use."""
-    block: NotRequired[UniswapQuoteBuyExactlyBlockTypedDict]
-    r"""The block number you want to get this data at."""
+    block: NotRequired[Nullable[int]]
+    r"""Optional block number (defaults to latest)."""
     token_in: UniswapQuoteBuyExactlyTokenInToken
     r"""The symbol of the token to swap from."""
     token_out: UniswapQuoteBuyExactlyTokenOutToken
@@ -187,10 +182,10 @@ class UniswapQuoteBuyExactlyRequest(BaseModel):
     r"""The chain to use."""
 
     block: Annotated[
-        Optional[UniswapQuoteBuyExactlyBlock],
+        OptionalNullable[int],
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
-    ] = None
-    r"""The block number you want to get this data at."""
+    ] = UNSET
+    r"""Optional block number (defaults to latest)."""
 
     token_in: Annotated[
         UniswapQuoteBuyExactlyTokenInToken,
@@ -209,3 +204,33 @@ class UniswapQuoteBuyExactlyRequest(BaseModel):
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = UniswapQuoteBuyExactlyFeeEnum.ZERO_DOT_01
     r"""The fee to pay for the swap"""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["block"]
+        nullable_fields = ["block"]
+        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/uniswap_quote_sell_exactlyop.py

@@ -1,10 +1,17 @@
 """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,
+    OptionalNullable,
+    UNSET,
+    UNSET_SENTINEL,
+)
 from compass_api_sdk.utils import FieldMetadata, QueryParamMetadata
 from enum import Enum
-from typing import Optional, Union
+from pydantic import model_serializer
+from typing import Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
@@ -16,18 +23,6 @@ class UniswapQuoteSellExactlyChain(str, Enum):
     ARBITRUM_MAINNET = "arbitrum:mainnet"
 
 
-UniswapQuoteSellExactlyBlockTypedDict = TypeAliasType(
-    "UniswapQuoteSellExactlyBlockTypedDict", Union[int, str]
-)
-r"""The block number you want to get this data at."""
-
-
-UniswapQuoteSellExactlyBlock = TypeAliasType(
-    "UniswapQuoteSellExactlyBlock", Union[int, str]
-)
-r"""The block number you want to get this data at."""
-
-
 class UniswapQuoteSellExactlyTokenInToken(str, Enum):
     r"""A class representing the token.
 
@@ -163,8 +158,8 @@ class UniswapQuoteSellExactlyRequestTypedDict(TypedDict):
     r"""The amount of the token to swap from"""
     chain: UniswapQuoteSellExactlyChain
     r"""The chain to use."""
-    block: NotRequired[UniswapQuoteSellExactlyBlockTypedDict]
-    r"""The block number you want to get this data at."""
+    block: NotRequired[Nullable[int]]
+    r"""Optional block number (defaults to latest)."""
     token_in: UniswapQuoteSellExactlyTokenInToken
     r"""The symbol of the token to swap from."""
     token_out: UniswapQuoteSellExactlyTokenOutToken
@@ -187,10 +182,10 @@ class UniswapQuoteSellExactlyRequest(BaseModel):
     r"""The chain to use."""
 
     block: Annotated[
-        Optional[UniswapQuoteSellExactlyBlock],
+        OptionalNullable[int],
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
-    ] = None
-    r"""The block number you want to get this data at."""
+    ] = UNSET
+    r"""Optional block number (defaults to latest)."""
 
     token_in: Annotated[
         UniswapQuoteSellExactlyTokenInToken,
@@ -209,3 +204,33 @@ class UniswapQuoteSellExactlyRequest(BaseModel):
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = UniswapQuoteSellExactlyFeeEnum.ZERO_DOT_01
     r"""The fee to pay for the swap"""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["block"]
+        nullable_fields = ["block"]
+        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