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

compass_api_sdk/models/generic_supported_tokensop.py

@@ -4,7 +4,8 @@ from __future__ import annotations
 from compass_api_sdk.types import BaseModel
 from compass_api_sdk.utils import FieldMetadata, QueryParamMetadata
 from enum import Enum
-from typing_extensions import Annotated, TypedDict
+from typing import Optional, Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
 class GenericSupportedTokensChain(str, Enum):
@@ -15,9 +16,23 @@ class GenericSupportedTokensChain(str, Enum):
     ARBITRUM_MAINNET = "arbitrum:mainnet"
 
 
+GenericSupportedTokensBlockTypedDict = TypeAliasType(
+    "GenericSupportedTokensBlockTypedDict", Union[int, str]
+)
+r"""The block number you want to get this data at."""
+
+
+GenericSupportedTokensBlock = TypeAliasType(
+    "GenericSupportedTokensBlock", Union[int, str]
+)
+r"""The block number you want to get this data at."""
+
+
 class GenericSupportedTokensRequestTypedDict(TypedDict):
     chain: GenericSupportedTokensChain
     r"""The chain to use."""
+    block: NotRequired[GenericSupportedTokensBlockTypedDict]
+    r"""The block number you want to get this data at."""
 
 
 class GenericSupportedTokensRequest(BaseModel):
@@ -26,3 +41,9 @@ class GenericSupportedTokensRequest(BaseModel):
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = GenericSupportedTokensChain.ARBITRUM_MAINNET
     r"""The chain to use."""
+
+    block: Annotated[
+        Optional[GenericSupportedTokensBlock],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = None
+    r"""The block number you want to get this data at."""

compass_api_sdk/models/generic_visualize_portfolioop.py

@@ -4,8 +4,8 @@ from __future__ import annotations
 from compass_api_sdk.types import BaseModel
 from compass_api_sdk.utils import FieldMetadata, QueryParamMetadata
 from enum import Enum
-from typing import Optional
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing import Optional, Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
 class GenericVisualizePortfolioChain(str, Enum):
@@ -16,9 +16,23 @@ class GenericVisualizePortfolioChain(str, Enum):
     ARBITRUM_MAINNET = "arbitrum:mainnet"
 
 
+GenericVisualizePortfolioBlockTypedDict = TypeAliasType(
+    "GenericVisualizePortfolioBlockTypedDict", Union[int, str]
+)
+r"""The block number you want to get this data at."""
+
+
+GenericVisualizePortfolioBlock = TypeAliasType(
+    "GenericVisualizePortfolioBlock", Union[int, str]
+)
+r"""The block number you want to get this data at."""
+
+
 class GenericVisualizePortfolioRequestTypedDict(TypedDict):
     chain: GenericVisualizePortfolioChain
     r"""The chain to use."""
+    block: NotRequired[GenericVisualizePortfolioBlockTypedDict]
+    r"""The block number you want to get this data at."""
     user: NotRequired[str]
     r"""The user to get portfolio for."""
 
@@ -30,6 +44,12 @@ class GenericVisualizePortfolioRequest(BaseModel):
     ] = GenericVisualizePortfolioChain.ARBITRUM_MAINNET
     r"""The chain to use."""
 
+    block: Annotated[
+        Optional[GenericVisualizePortfolioBlock],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = None
+    r"""The block number you want to get this data at."""
+
     user: Annotated[
         Optional[str],
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),

compass_api_sdk/models/morpho_market_positionop.py

@@ -4,7 +4,8 @@ from __future__ import annotations
 from compass_api_sdk.types import BaseModel
 from compass_api_sdk.utils import FieldMetadata, QueryParamMetadata
 from enum import Enum
-from typing_extensions import Annotated, TypedDict
+from typing import Optional, Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
 class MorphoMarketPositionChain(str, Enum):
@@ -12,8 +13,20 @@ class MorphoMarketPositionChain(str, Enum):
     BASE_MAINNET = "base:mainnet"
 
 
+MorphoMarketPositionBlockTypedDict = TypeAliasType(
+    "MorphoMarketPositionBlockTypedDict", Union[int, str]
+)
+r"""The block number you want to get this data at."""
+
+
+MorphoMarketPositionBlock = TypeAliasType("MorphoMarketPositionBlock", Union[int, str])
+r"""The block number you want to get this data at."""
+
+
 class MorphoMarketPositionRequestTypedDict(TypedDict):
     chain: MorphoMarketPositionChain
+    block: NotRequired[MorphoMarketPositionBlockTypedDict]
+    r"""The block number you want to get this data at."""
     user_address: str
     r"""The user address of the desired market position."""
     unique_market_key: str
@@ -26,6 +39,12 @@ class MorphoMarketPositionRequest(BaseModel):
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = MorphoMarketPositionChain.ETHEREUM_MAINNET
 
+    block: Annotated[
+        Optional[MorphoMarketPositionBlock],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = None
+    r"""The block number you want to get this data at."""
+
     user_address: Annotated[
         str, FieldMetadata(query=QueryParamMetadata(style="form", explode=True))
     ] = "0xa829B388A3DF7f581cE957a95edbe419dd146d1B"

compass_api_sdk/models/morpho_marketop.py

@@ -4,7 +4,8 @@ from __future__ import annotations
 from compass_api_sdk.types import BaseModel
 from compass_api_sdk.utils import FieldMetadata, QueryParamMetadata
 from enum import Enum
-from typing_extensions import Annotated, TypedDict
+from typing import Optional, Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
 class MorphoMarketChain(str, Enum):
@@ -12,8 +13,20 @@ class MorphoMarketChain(str, Enum):
     BASE_MAINNET = "base:mainnet"
 
 
+MorphoMarketBlockTypedDict = TypeAliasType(
+    "MorphoMarketBlockTypedDict", Union[int, str]
+)
+r"""The block number you want to get this data at."""
+
+
+MorphoMarketBlock = TypeAliasType("MorphoMarketBlock", Union[int, str])
+r"""The block number you want to get this data at."""
+
+
 class MorphoMarketRequestTypedDict(TypedDict):
     chain: MorphoMarketChain
+    block: NotRequired[MorphoMarketBlockTypedDict]
+    r"""The block number you want to get this data at."""
     unique_market_key: str
     r"""The key that uniquely identifies the market. This can be found using the 'Get Markets' endpoint."""
 
@@ -24,6 +37,12 @@ class MorphoMarketRequest(BaseModel):
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = MorphoMarketChain.ETHEREUM_MAINNET
 
+    block: Annotated[
+        Optional[MorphoMarketBlock],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = None
+    r"""The block number you want to get this data at."""
+
     unique_market_key: Annotated[
         str, FieldMetadata(query=QueryParamMetadata(style="form", explode=True))
     ] = "0x83b7ad16905809ea36482f4fbf6cfee9c9f316d128de9a5da1952607d5e4df5e"

compass_api_sdk/models/morpho_marketsop.py

@@ -11,7 +11,8 @@ from compass_api_sdk.types import (
 from compass_api_sdk.utils import FieldMetadata, QueryParamMetadata
 from enum import Enum
 from pydantic import model_serializer
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing import Optional, Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
 class MorphoMarketsChain(str, Enum):
@@ -19,8 +20,20 @@ class MorphoMarketsChain(str, Enum):
     BASE_MAINNET = "base:mainnet"
 
 
+MorphoMarketsBlockTypedDict = TypeAliasType(
+    "MorphoMarketsBlockTypedDict", Union[int, str]
+)
+r"""The block number you want to get this data at"""
+
+
+MorphoMarketsBlock = TypeAliasType("MorphoMarketsBlock", Union[int, str])
+r"""The block number you want to get this data at"""
+
+
 class MorphoMarketsRequestTypedDict(TypedDict):
     chain: MorphoMarketsChain
+    block: NotRequired[MorphoMarketsBlockTypedDict]
+    r"""The block number you want to get this data at"""
     collateral_token: NotRequired[Nullable[str]]
     r"""Collateral token to identify the market."""
     loan_token: NotRequired[Nullable[str]]
@@ -33,6 +46,12 @@ class MorphoMarketsRequest(BaseModel):
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = MorphoMarketsChain.ETHEREUM_MAINNET
 
+    block: Annotated[
+        Optional[MorphoMarketsBlock],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = None
+    r"""The block number you want to get this data at"""
+
     collateral_token: Annotated[
         OptionalNullable[str],
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
@@ -47,7 +66,7 @@ class MorphoMarketsRequest(BaseModel):
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["collateral_token", "loan_token"]
+        optional_fields = ["block", "collateral_token", "loan_token"]
         nullable_fields = ["collateral_token", "loan_token"]
         null_default_fields = []
 

compass_api_sdk/models/morpho_user_positionop.py

@@ -4,7 +4,8 @@ from __future__ import annotations
 from compass_api_sdk.types import BaseModel
 from compass_api_sdk.utils import FieldMetadata, QueryParamMetadata
 from enum import Enum
-from typing_extensions import Annotated, TypedDict
+from typing import Optional, Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
 class MorphoUserPositionChain(str, Enum):
@@ -12,8 +13,20 @@ class MorphoUserPositionChain(str, Enum):
     BASE_MAINNET = "base:mainnet"
 
 
+MorphoUserPositionBlockTypedDict = TypeAliasType(
+    "MorphoUserPositionBlockTypedDict", Union[int, str]
+)
+r"""The block number you want to get this data at."""
+
+
+MorphoUserPositionBlock = TypeAliasType("MorphoUserPositionBlock", Union[int, str])
+r"""The block number you want to get this data at."""
+
+
 class MorphoUserPositionRequestTypedDict(TypedDict):
     chain: MorphoUserPositionChain
+    block: NotRequired[MorphoUserPositionBlockTypedDict]
+    r"""The block number you want to get this data at."""
     user_address: str
     r"""The user wallet address of the desired user position."""
 
@@ -24,6 +37,12 @@ class MorphoUserPositionRequest(BaseModel):
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = MorphoUserPositionChain.ETHEREUM_MAINNET
 
+    block: Annotated[
+        Optional[MorphoUserPositionBlock],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = None
+    r"""The block number you want to get this data at."""
+
     user_address: Annotated[
         str, FieldMetadata(query=QueryParamMetadata(style="form", explode=True))
     ] = "0xa829B388A3DF7f581cE957a95edbe419dd146d1B"

compass_api_sdk/models/morpho_vault_positionop.py

@@ -4,7 +4,8 @@ from __future__ import annotations
 from compass_api_sdk.types import BaseModel
 from compass_api_sdk.utils import FieldMetadata, QueryParamMetadata
 from enum import Enum
-from typing_extensions import Annotated, TypedDict
+from typing import Optional, Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
 class MorphoVaultPositionChain(str, Enum):
@@ -12,8 +13,20 @@ class MorphoVaultPositionChain(str, Enum):
     BASE_MAINNET = "base:mainnet"
 
 
+MorphoVaultPositionBlockTypedDict = TypeAliasType(
+    "MorphoVaultPositionBlockTypedDict", Union[int, str]
+)
+r"""The block number you want to get this data at."""
+
+
+MorphoVaultPositionBlock = TypeAliasType("MorphoVaultPositionBlock", Union[int, str])
+r"""The block number you want to get this data at."""
+
+
 class MorphoVaultPositionRequestTypedDict(TypedDict):
     chain: MorphoVaultPositionChain
+    block: NotRequired[MorphoVaultPositionBlockTypedDict]
+    r"""The block number you want to get this data at."""
     user_address: str
     r"""The user address of the desired vault position."""
     vault_address: str
@@ -26,6 +39,12 @@ class MorphoVaultPositionRequest(BaseModel):
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = MorphoVaultPositionChain.ETHEREUM_MAINNET
 
+    block: Annotated[
+        Optional[MorphoVaultPositionBlock],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = None
+    r"""The block number you want to get this data at."""
+
     user_address: Annotated[
         str, FieldMetadata(query=QueryParamMetadata(style="form", explode=True))
     ] = "0xa829B388A3DF7f581cE957a95edbe419dd146d1B"

compass_api_sdk/models/morpho_vaultop.py

@@ -4,7 +4,8 @@ from __future__ import annotations
 from compass_api_sdk.types import BaseModel
 from compass_api_sdk.utils import FieldMetadata, QueryParamMetadata
 from enum import Enum
-from typing_extensions import Annotated, TypedDict
+from typing import Optional, Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
 class MorphoVaultChain(str, Enum):
@@ -12,8 +13,18 @@ class MorphoVaultChain(str, Enum):
     BASE_MAINNET = "base:mainnet"
 
 
+MorphoVaultBlockTypedDict = TypeAliasType("MorphoVaultBlockTypedDict", Union[int, str])
+r"""The block number you want to get this data at."""
+
+
+MorphoVaultBlock = TypeAliasType("MorphoVaultBlock", Union[int, str])
+r"""The block number you want to get this data at."""
+
+
 class MorphoVaultRequestTypedDict(TypedDict):
     chain: MorphoVaultChain
+    block: NotRequired[MorphoVaultBlockTypedDict]
+    r"""The block number you want to get this data at."""
     vault_address: str
     r"""The vault address of the desired vault data & metrics."""
 
@@ -24,6 +35,12 @@ class MorphoVaultRequest(BaseModel):
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = MorphoVaultChain.ETHEREUM_MAINNET
 
+    block: Annotated[
+        Optional[MorphoVaultBlock],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = None
+    r"""The block number you want to get this data at."""
+
     vault_address: Annotated[
         str, FieldMetadata(query=QueryParamMetadata(style="form", explode=True))
     ] = "0xbEef047a543E45807105E51A8BBEFCc5950fcfBa"

compass_api_sdk/models/morpho_vaultsop.py

@@ -11,7 +11,8 @@ from compass_api_sdk.types import (
 from compass_api_sdk.utils import FieldMetadata, QueryParamMetadata
 from enum import Enum
 from pydantic import model_serializer
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing import Optional, Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
 class MorphoVaultsChain(str, Enum):
@@ -19,8 +20,20 @@ class MorphoVaultsChain(str, Enum):
     BASE_MAINNET = "base:mainnet"
 
 
+MorphoVaultsBlockTypedDict = TypeAliasType(
+    "MorphoVaultsBlockTypedDict", Union[int, str]
+)
+r"""The block number you want to get this data at."""
+
+
+MorphoVaultsBlock = TypeAliasType("MorphoVaultsBlock", Union[int, str])
+r"""The block number you want to get this data at."""
+
+
 class MorphoVaultsRequestTypedDict(TypedDict):
     chain: MorphoVaultsChain
+    block: NotRequired[MorphoVaultsBlockTypedDict]
+    r"""The block number you want to get this data at."""
     deposit_token: NotRequired[Nullable[str]]
     r"""Token address that will filter vaults by this deposit token."""
 
@@ -31,6 +44,12 @@ class MorphoVaultsRequest(BaseModel):
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = MorphoVaultsChain.ETHEREUM_MAINNET
 
+    block: Annotated[
+        Optional[MorphoVaultsBlock],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = None
+    r"""The block number you want to get this data at."""
+
     deposit_token: Annotated[
         OptionalNullable[str],
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
@@ -39,7 +58,7 @@ class MorphoVaultsRequest(BaseModel):
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["deposit_token"]
+        optional_fields = ["block", "deposit_token"]
         nullable_fields = ["deposit_token"]
         null_default_fields = []
 

compass_api_sdk/models/multicallauthorizationrequest.py

@@ -3,7 +3,16 @@
 from __future__ import annotations
 from .chain import Chain
 from compass_api_sdk.types import BaseModel
-from typing_extensions import TypedDict
+from typing import Optional, Union
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
+
+
+BlockTypedDict = TypeAliasType("BlockTypedDict", Union[int, str])
+r"""The block number you want to get this data at."""
+
+
+Block = TypeAliasType("Block", Union[int, str])
+r"""The block number you want to get this data at."""
 
 
 class MulticallAuthorizationRequestTypedDict(TypedDict):
@@ -20,6 +29,8 @@ class MulticallAuthorizationRequestTypedDict(TypedDict):
     r"""The chain to use."""
     sender: str
     r"""The Ethereum address to use for authorization"""
+    block: NotRequired[BlockTypedDict]
+    r"""The block number you want to get this data at."""
 
 
 class MulticallAuthorizationRequest(BaseModel):
@@ -37,3 +48,6 @@ class MulticallAuthorizationRequest(BaseModel):
 
     sender: str
     r"""The Ethereum address to use for authorization"""
+
+    block: Optional[Block] = None
+    r"""The block number you want to get this data at."""

compass_api_sdk/models/pendle_marketop.py

@@ -4,7 +4,8 @@ from __future__ import annotations
 from compass_api_sdk.types import BaseModel
 from compass_api_sdk.utils import FieldMetadata, QueryParamMetadata
 from enum import Enum
-from typing_extensions import Annotated, TypedDict
+from typing import Optional, Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
 class PendleMarketChain(str, Enum):
@@ -15,9 +16,21 @@ class PendleMarketChain(str, Enum):
     ARBITRUM_MAINNET = "arbitrum:mainnet"
 
 
+PendleMarketBlockTypedDict = TypeAliasType(
+    "PendleMarketBlockTypedDict", Union[int, str]
+)
+r"""The block number you want to get this data at."""
+
+
+PendleMarketBlock = TypeAliasType("PendleMarketBlock", Union[int, str])
+r"""The block number you want to get this data at."""
+
+
 class PendleMarketRequestTypedDict(TypedDict):
     chain: PendleMarketChain
     r"""The chain to use."""
+    block: NotRequired[PendleMarketBlockTypedDict]
+    r"""The block number you want to get this data at."""
     user_address: str
     r"""The user address of the desired position."""
     market_address: str
@@ -31,6 +44,12 @@ class PendleMarketRequest(BaseModel):
     ] = PendleMarketChain.ETHEREUM_MAINNET
     r"""The chain to use."""
 
+    block: Annotated[
+        Optional[PendleMarketBlock],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = None
+    r"""The block number you want to get this data at."""
+
     user_address: Annotated[
         str, FieldMetadata(query=QueryParamMetadata(style="form", explode=True))
     ] = "0xa829B388A3DF7f581cE957a95edbe419dd146d1B"

compass_api_sdk/models/pendle_positionop.py

@@ -4,7 +4,8 @@ from __future__ import annotations
 from compass_api_sdk.types import BaseModel
 from compass_api_sdk.utils import FieldMetadata, QueryParamMetadata
 from enum import Enum
-from typing_extensions import Annotated, TypedDict
+from typing import Optional, Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
 class PendlePositionChain(str, Enum):
@@ -15,9 +16,21 @@ class PendlePositionChain(str, Enum):
     ARBITRUM_MAINNET = "arbitrum:mainnet"
 
 
+PendlePositionBlockTypedDict = TypeAliasType(
+    "PendlePositionBlockTypedDict", Union[int, str]
+)
+r"""The block number you want to get this data at."""
+
+
+PendlePositionBlock = TypeAliasType("PendlePositionBlock", Union[int, str])
+r"""The block number you want to get this data at."""
+
+
 class PendlePositionRequestTypedDict(TypedDict):
     chain: PendlePositionChain
     r"""The chain to use."""
+    block: NotRequired[PendlePositionBlockTypedDict]
+    r"""The block number you want to get this data at."""
     user_address: str
     r"""The user address of the desired position."""
     market_address: str
@@ -31,6 +44,12 @@ class PendlePositionRequest(BaseModel):
     ] = PendlePositionChain.ETHEREUM_MAINNET
     r"""The chain to use."""
 
+    block: Annotated[
+        Optional[PendlePositionBlock],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = None
+    r"""The block number you want to get this data at."""
+
     user_address: Annotated[
         str, FieldMetadata(query=QueryParamMetadata(style="form", explode=True))
     ] = "0xa829B388A3DF7f581cE957a95edbe419dd146d1B"

compass_api_sdk/models/sky_positionop.py

@@ -5,12 +5,22 @@ from compass_api_sdk.types import BaseModel
 from compass_api_sdk.utils import FieldMetadata, QueryParamMetadata, validate_const
 import pydantic
 from pydantic.functional_validators import AfterValidator
-from typing import Literal, Optional
-from typing_extensions import Annotated, TypedDict
+from typing import Literal, Optional, Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
+
+
+SkyPositionBlockTypedDict = TypeAliasType("SkyPositionBlockTypedDict", Union[int, str])
+r"""The block number you want to get this data at."""
+
+
+SkyPositionBlock = TypeAliasType("SkyPositionBlock", Union[int, str])
+r"""The block number you want to get this data at."""
 
 
 class SkyPositionRequestTypedDict(TypedDict):
     chain: Literal["ethereum:mainnet"]
+    block: NotRequired[SkyPositionBlockTypedDict]
+    r"""The block number you want to get this data at."""
     user_address: str
     r"""The user address of the desired position."""
 
@@ -25,6 +35,12 @@ class SkyPositionRequest(BaseModel):
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = "ethereum:mainnet"
 
+    block: Annotated[
+        Optional[SkyPositionBlock],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = None
+    r"""The block number you want to get this data at."""
+
     user_address: Annotated[
         str, FieldMetadata(query=QueryParamMetadata(style="form", explode=True))
     ] = "0xa829B388A3DF7f581cE957a95edbe419dd146d1B"

compass_api_sdk/models/token_addressop.py

@@ -4,7 +4,8 @@ from __future__ import annotations
 from compass_api_sdk.types import BaseModel
 from compass_api_sdk.utils import FieldMetadata, QueryParamMetadata
 from enum import Enum
-from typing_extensions import Annotated, TypedDict
+from typing import Optional, Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
 class TokenAddressChain(str, Enum):
@@ -15,6 +16,16 @@ class TokenAddressChain(str, Enum):
     ARBITRUM_MAINNET = "arbitrum:mainnet"
 
 
+TokenAddressBlockTypedDict = TypeAliasType(
+    "TokenAddressBlockTypedDict", Union[int, str]
+)
+r"""The block number you want to get this data at."""
+
+
+TokenAddressBlock = TypeAliasType("TokenAddressBlock", Union[int, str])
+r"""The block number you want to get this data at."""
+
+
 class TokenAddressToken(str, Enum):
     r"""A class representing the token.
 
@@ -71,6 +82,8 @@ class TokenAddressToken(str, Enum):
 class TokenAddressRequestTypedDict(TypedDict):
     chain: TokenAddressChain
     r"""The chain to use."""
+    block: NotRequired[TokenAddressBlockTypedDict]
+    r"""The block number you want to get this data at."""
     token: TokenAddressToken
     r"""The token symbol to get the address for.."""
 
@@ -82,6 +95,12 @@ class TokenAddressRequest(BaseModel):
     ] = TokenAddressChain.ARBITRUM_MAINNET
     r"""The chain to use."""
 
+    block: Annotated[
+        Optional[TokenAddressBlock],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = None
+    r"""The block number you want to get this data at."""
+
     token: Annotated[
         TokenAddressToken,
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),

compass_api_sdk/models/token_balanceop.py

@@ -5,8 +5,8 @@ from .token_enum import TokenEnum
 from compass_api_sdk.types import BaseModel
 from compass_api_sdk.utils import FieldMetadata, QueryParamMetadata
 from enum import Enum
-from typing import Union
-from typing_extensions import Annotated, TypeAliasType, TypedDict
+from typing import Optional, Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
 class TokenBalanceChain(str, Enum):
@@ -17,6 +17,16 @@ 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]
 )
@@ -32,6 +42,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."""
     user: str
     r"""The user to get the ERC20 balance of."""
 
@@ -49,6 +61,12 @@ class TokenBalanceRequest(BaseModel):
     ] = TokenBalanceChain.ARBITRUM_MAINNET
     r"""The chain to use."""
 
+    block: Annotated[
+        Optional[TokenBalanceBlock],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = None
+    r"""The block number you want to get this data at."""
+
     user: Annotated[
         str, FieldMetadata(query=QueryParamMetadata(style="form", explode=True))
     ] = "0x29F20a192328eF1aD35e1564aBFf4Be9C5ce5f7B"