compass_api_sdk 1.1.1__py3-none-any.whl → 2.2.1rc3__py3-none-any.whl

compass_api_sdk/models/positionpnl.py

@@ -0,0 +1,52 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from compass_api_sdk.types import BaseModel
+from typing_extensions import TypedDict
+
+
+class PositionPnLTypedDict(TypedDict):
+    r"""Simplified PnL metrics for API response.
+
+    Historical events (deposits/withdrawals) are now at the position level, not inside
+    PnL.
+    """
+
+    total_deposited: str
+    r"""Total assets deposited over all time"""
+    current_value: str
+    r"""Current position value in underlying"""
+    unrealized_pnl: str
+    r"""current_value - cost_basis_remaining"""
+    realized_pnl: str
+    r"""Sum of profits/losses from all withdrawals"""
+    total_pnl: str
+    r"""unrealized_pnl + realized_pnl"""
+    total_pnl_percent: str
+    r"""(total_pnl / total_deposited) * 100"""
+
+
+class PositionPnL(BaseModel):
+    r"""Simplified PnL metrics for API response.
+
+    Historical events (deposits/withdrawals) are now at the position level, not inside
+    PnL.
+    """
+
+    total_deposited: str
+    r"""Total assets deposited over all time"""
+
+    current_value: str
+    r"""Current position value in underlying"""
+
+    unrealized_pnl: str
+    r"""current_value - cost_basis_remaining"""
+
+    realized_pnl: str
+    r"""Sum of profits/losses from all withdrawals"""
+
+    total_pnl: str
+    r"""unrealized_pnl + realized_pnl"""
+
+    total_pnl_percent: str
+    r"""(total_pnl / total_deposited) * 100"""

compass_api_sdk/models/reserve.py

@@ -1,7 +1,6 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .token_enum import TokenEnum
 from compass_api_sdk.types import (
     BaseModel,
     Nullable,
@@ -15,14 +14,10 @@ from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 class ReserveTypedDict(TypedDict):
-    symbol: TokenEnum
-    r"""A class representing the token.
-
-    This class is used to represent the token in the system. Notice individual
-    endpoints' documentation where per chain tokens are presented.
-    """
+    symbol: str
+    r"""Symbol of token."""
     name: str
-    r"""Name of Token"""
+    r"""Name of token"""
     underlying_asset: str
     r"""Checksum address of Token."""
     decimals: NotRequired[Nullable[int]]
@@ -30,15 +25,11 @@ class ReserveTypedDict(TypedDict):
 
 
 class Reserve(BaseModel):
-    symbol: TokenEnum
-    r"""A class representing the token.
-
-    This class is used to represent the token in the system. Notice individual
-    endpoints' documentation where per chain tokens are presented.
-    """
+    symbol: str
+    r"""Symbol of token."""
 
     name: str
-    r"""Name of Token"""
+    r"""Name of token"""
 
     underlying_asset: Annotated[str, pydantic.Field(alias="underlyingAsset")]
     r"""Checksum address of Token."""

compass_api_sdk/models/safetransactionparams.py

@@ -0,0 +1,78 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from compass_api_sdk.types import BaseModel
+import pydantic
+from typing_extensions import Annotated, TypedDict
+
+
+class SafeTransactionParamsTypedDict(TypedDict):
+    r"""Parameters for proposing a transaction to a Safe multi-sig wallet.
+
+    This model is used when the Safe has threshold > 1 or when the user wants to propose
+    a transaction for other owners to sign via the Safe Transaction Service.
+    """
+
+    to: str
+    r"""Destination address (MultiSend contract for bundled txs)"""
+    value: int
+    r"""ETH value to send"""
+    data: str
+    r"""Encoded transaction data"""
+    operation: int
+    r"""0 = Call, 1 = DelegateCall"""
+    safe_tx_gas: int
+    r"""Gas for the Safe transaction execution"""
+    base_gas: int
+    r"""Base gas for the transaction"""
+    gas_price: int
+    r"""Gas price for refund calculation"""
+    gas_token: str
+    r"""Token address for gas refund (0x0 for ETH)"""
+    refund_receiver: str
+    r"""Address to receive gas refund"""
+    nonce: int
+    r"""Safe transaction nonce"""
+    safe_tx_hash: str
+    r"""EIP-712 hash of the Safe transaction (sign this with your wallet)"""
+
+
+class SafeTransactionParams(BaseModel):
+    r"""Parameters for proposing a transaction to a Safe multi-sig wallet.
+
+    This model is used when the Safe has threshold > 1 or when the user wants to propose
+    a transaction for other owners to sign via the Safe Transaction Service.
+    """
+
+    to: str
+    r"""Destination address (MultiSend contract for bundled txs)"""
+
+    value: int
+    r"""ETH value to send"""
+
+    data: str
+    r"""Encoded transaction data"""
+
+    operation: int
+    r"""0 = Call, 1 = DelegateCall"""
+
+    safe_tx_gas: Annotated[int, pydantic.Field(alias="safeTxGas")]
+    r"""Gas for the Safe transaction execution"""
+
+    base_gas: Annotated[int, pydantic.Field(alias="baseGas")]
+    r"""Base gas for the transaction"""
+
+    gas_price: Annotated[int, pydantic.Field(alias="gasPrice")]
+    r"""Gas price for refund calculation"""
+
+    gas_token: Annotated[str, pydantic.Field(alias="gasToken")]
+    r"""Token address for gas refund (0x0 for ETH)"""
+
+    refund_receiver: Annotated[str, pydantic.Field(alias="refundReceiver")]
+    r"""Address to receive gas refund"""
+
+    nonce: int
+    r"""Safe transaction nonce"""
+
+    safe_tx_hash: Annotated[str, pydantic.Field(alias="safeTxHash")]
+    r"""EIP-712 hash of the Safe transaction (sign this with your wallet)"""

compass_api_sdk/models/safetransactionresponse.py

@@ -0,0 +1,28 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .safetransactionparams import SafeTransactionParams, SafeTransactionParamsTypedDict
+from compass_api_sdk.types import BaseModel
+from typing_extensions import TypedDict
+
+
+class SafeTransactionResponseTypedDict(TypedDict):
+    r"""Response containing Safe transaction parameters for multi-sig proposal."""
+
+    transaction: SafeTransactionParamsTypedDict
+    r"""Parameters for proposing a transaction to a Safe multi-sig wallet.
+
+    This model is used when the Safe has threshold > 1 or when the user wants to propose
+    a transaction for other owners to sign via the Safe Transaction Service.
+    """
+
+
+class SafeTransactionResponse(BaseModel):
+    r"""Response containing Safe transaction parameters for multi-sig proposal."""
+
+    transaction: SafeTransactionParams
+    r"""Parameters for proposing a transaction to a Safe multi-sig wallet.
+
+    This model is used when the Safe has threshold > 1 or when the user wants to propose
+    a transaction for other owners to sign via the Safe Transaction Service.
+    """

compass_api_sdk/models/safetxfield.py

@@ -0,0 +1,20 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from compass_api_sdk.types import BaseModel
+from typing_extensions import TypedDict
+
+
+class SafeTxFieldTypedDict(TypedDict):
+    r"""A field in the SafeTx type definition."""
+
+    name: str
+    type: str
+
+
+class SafeTxField(BaseModel):
+    r"""A field in the SafeTx type definition."""
+
+    name: str
+
+    type: str

compass_api_sdk/models/safetxmessage.py

@@ -0,0 +1,66 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .operationtype import OperationType
+from compass_api_sdk.types import BaseModel
+import pydantic
+from typing_extensions import Annotated, TypedDict
+
+
+class SafeTxMessageTypedDict(TypedDict):
+    r"""The message data for the transaction."""
+
+    to: str
+    r"""Destination address"""
+    value: str
+    r"""Value in wei as a string"""
+    data: str
+    r"""Transaction data as hex string"""
+    operation: OperationType
+    r"""Safe operation types."""
+    safe_tx_gas: str
+    r"""Gas for the transaction"""
+    base_gas: str
+    r"""Base gas costs"""
+    gas_price: str
+    r"""Gas price"""
+    gas_token: str
+    r"""Token address for gas payment"""
+    refund_receiver: str
+    r"""Address to receive gas refund"""
+    nonce: str
+    r"""Transaction nonce"""
+
+
+class SafeTxMessage(BaseModel):
+    r"""The message data for the transaction."""
+
+    to: str
+    r"""Destination address"""
+
+    value: str
+    r"""Value in wei as a string"""
+
+    data: str
+    r"""Transaction data as hex string"""
+
+    operation: OperationType
+    r"""Safe operation types."""
+
+    safe_tx_gas: Annotated[str, pydantic.Field(alias="safeTxGas")]
+    r"""Gas for the transaction"""
+
+    base_gas: Annotated[str, pydantic.Field(alias="baseGas")]
+    r"""Base gas costs"""
+
+    gas_price: Annotated[str, pydantic.Field(alias="gasPrice")]
+    r"""Gas price"""
+
+    gas_token: Annotated[str, pydantic.Field(alias="gasToken")]
+    r"""Token address for gas payment"""
+
+    refund_receiver: Annotated[str, pydantic.Field(alias="refundReceiver")]
+    r"""Address to receive gas refund"""
+
+    nonce: str
+    r"""Transaction nonce"""

compass_api_sdk/models/setallowanceparams.py

@@ -1,7 +1,6 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .token_enum import TokenEnum
 from compass_api_sdk.types import BaseModel
 from compass_api_sdk.utils import validate_const
 from enum import Enum
@@ -11,18 +10,6 @@ from typing import Literal, Optional, Union
 from typing_extensions import Annotated, TypeAliasType, TypedDict
 
 
-SetAllowanceParamsTokenTypedDict = TypeAliasType(
-    "SetAllowanceParamsTokenTypedDict", Union[TokenEnum, str]
-)
-r"""The symbol or address of the token for which the allowance is set.."""
-
-
-SetAllowanceParamsToken = TypeAliasType(
-    "SetAllowanceParamsToken", Union[TokenEnum, str]
-)
-r"""The symbol or address of the token for which the allowance is set.."""
-
-
 class SetAllowanceParamsContractEnum(str, Enum):
     AAVE_V3_POOL = "AaveV3Pool"
     AERODROME_BASIC_ROUTER = "AerodromeBasicRouter"
@@ -32,7 +19,7 @@ class SetAllowanceParamsContractEnum(str, Enum):
     )
     UNISWAP_V3_ROUTER = "UniswapV3Router"
     UNISWAP_V3_NFT_POSITION_MANAGER = "UniswapV3NFTPositionManager"
-    MORPHO = "Morpho"
+    MORPHO_MARKET = "MorphoMarket"
     SKY_DAI_USDS_CONVERTER = "SkyDaiUsdsConverter"
     SKY_USDC_USDS_CONVERTER = "SkyUsdcUsdsConverter"
     SKY_USDS_VAULT = "SkyUsdsVault"
@@ -67,7 +54,7 @@ r"""The amount to set the allowance to."""
 class SetAllowanceParamsTypedDict(TypedDict):
     r"""Parameters model for setting the token allowance for a given contract."""
 
-    token: SetAllowanceParamsTokenTypedDict
+    token: str
     r"""The symbol or address of the token for which the allowance is set.."""
     contract: SetAllowanceParamsContractUnionTypedDict
     r"""The name or address of the contract to set spending allowance for."""
@@ -79,7 +66,7 @@ class SetAllowanceParamsTypedDict(TypedDict):
 class SetAllowanceParams(BaseModel):
     r"""Parameters model for setting the token allowance for a given contract."""
 
-    token: SetAllowanceParamsToken
+    token: str
     r"""The symbol or address of the token for which the allowance is set.."""
 
     contract: SetAllowanceParamsContractUnion

compass_api_sdk/models/setallowancerequest.py

@@ -1,26 +1,13 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-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
-from typing_extensions import Annotated, TypeAliasType, TypedDict
-
-
-SetAllowanceRequestTokenTypedDict = TypeAliasType(
-    "SetAllowanceRequestTokenTypedDict", Union[TokenEnum, str]
-)
-r"""The symbol or address of the token for which the allowance is set.."""
-
-
-SetAllowanceRequestToken = TypeAliasType(
-    "SetAllowanceRequestToken", Union[TokenEnum, str]
-)
-r"""The symbol or address of the token for which the allowance is set.."""
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
 class SetAllowanceRequestContractEnum(str, Enum):
@@ -32,7 +19,7 @@ class SetAllowanceRequestContractEnum(str, Enum):
     )
     UNISWAP_V3_ROUTER = "UniswapV3Router"
     UNISWAP_V3_NFT_POSITION_MANAGER = "UniswapV3NFTPositionManager"
-    MORPHO = "Morpho"
+    MORPHO_MARKET = "MorphoMarket"
     SKY_DAI_USDS_CONVERTER = "SkyDaiUsdsConverter"
     SKY_USDC_USDS_CONVERTER = "SkyUsdcUsdsConverter"
     SKY_USDS_VAULT = "SkyUsdsVault"
@@ -75,7 +62,7 @@ class SetAllowanceRequestChain(str, Enum):
 class SetAllowanceRequestTypedDict(TypedDict):
     r"""Request model for increasing token allowance for a contract."""
 
-    token: SetAllowanceRequestTokenTypedDict
+    token: str
     r"""The symbol or address of the token for which the allowance is set.."""
     contract: SetAllowanceRequestContractUnionTypedDict
     r"""The name or address of the contract to set spending allowance for."""
@@ -85,12 +72,14 @@ class SetAllowanceRequestTypedDict(TypedDict):
     sender: str
     r"""The address of the transaction sender."""
     action_type: Literal["SET_ALLOWANCE"]
+    estimate_gas: NotRequired[bool]
+    r"""Determines whether to estimate gas costs for transactions, also verifying that the transaction can be successfully executed."""
 
 
 class SetAllowanceRequest(BaseModel):
     r"""Request model for increasing token allowance for a contract."""
 
-    token: SetAllowanceRequestToken
+    token: str
     r"""The symbol or address of the token for which the allowance is set.."""
 
     contract: SetAllowanceRequestContractUnion
@@ -111,3 +100,6 @@ class SetAllowanceRequest(BaseModel):
         ],
         pydantic.Field(alias="action_type"),
     ] = "SET_ALLOWANCE"
+
+    estimate_gas: Optional[bool] = None
+    r"""Determines whether to estimate gas costs for transactions, also verifying that the transaction can be successfully executed."""

compass_api_sdk/models/skybuyrequest.py

@@ -7,7 +7,7 @@ from enum import Enum
 import pydantic
 from pydantic.functional_validators import AfterValidator
 from typing import Literal, Optional, Union
-from typing_extensions import Annotated, TypeAliasType, TypedDict
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
 class SkyBuyRequestTokenIn(str, Enum):
@@ -40,6 +40,8 @@ class SkyBuyRequestTypedDict(TypedDict):
     sender: str
     r"""The address of the transaction sender."""
     action_type: Literal["SKY_BUY"]
+    estimate_gas: NotRequired[bool]
+    r"""Determines whether to estimate gas costs for transactions, also verifying that the transaction can be successfully executed."""
 
 
 class SkyBuyRequest(BaseModel):
@@ -60,3 +62,6 @@ class SkyBuyRequest(BaseModel):
         ],
         pydantic.Field(alias="action_type"),
     ] = "SKY_BUY"
+
+    estimate_gas: Optional[bool] = None
+    r"""Determines whether to estimate gas costs for transactions, also verifying that the transaction can be successfully executed."""

compass_api_sdk/models/skydepositrequest.py

@@ -40,6 +40,8 @@ class SkyDepositRequestTypedDict(TypedDict):
     receiver: NotRequired[Nullable[str]]
     r"""The address which will receive the sUSDS. Defaults to the sender."""
     chain: NotRequired[SkyDepositRequestChain]
+    estimate_gas: NotRequired[bool]
+    r"""Determines whether to estimate gas costs for transactions, also verifying that the transaction can be successfully executed."""
 
 
 class SkyDepositRequest(BaseModel):
@@ -60,11 +62,14 @@ class SkyDepositRequest(BaseModel):
     receiver: OptionalNullable[str] = UNSET
     r"""The address which will receive the sUSDS. Defaults to the sender."""
 
-    chain: Optional[SkyDepositRequestChain] = SkyDepositRequestChain.ETHEREUM
+    chain: Optional[SkyDepositRequestChain] = None
+
+    estimate_gas: Optional[bool] = None
+    r"""Determines whether to estimate gas costs for transactions, also verifying that the transaction can be successfully executed."""
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["action_type", "receiver", "chain"]
+        optional_fields = ["action_type", "receiver", "chain", "estimate_gas"]
         nullable_fields = ["receiver"]
         null_default_fields = []
 

compass_api_sdk/models/skysellrequest.py

@@ -36,10 +36,12 @@ class SkySellRequestTypedDict(TypedDict):
     r"""The token you would like to swap 1:1 with USDS. Choose from DAI or USDC."""
     amount: SkySellRequestAmountTypedDict
     r"""The amount of USDS you would like to sell 1:1 for 'token_out'."""
+    chain: SkySellRequestChain
     sender: str
     r"""The address of the transaction sender."""
     action_type: Literal["SKY_SELL"]
-    chain: NotRequired[SkySellRequestChain]
+    estimate_gas: NotRequired[bool]
+    r"""Determines whether to estimate gas costs for transactions, also verifying that the transaction can be successfully executed."""
 
 
 class SkySellRequest(BaseModel):
@@ -49,6 +51,8 @@ class SkySellRequest(BaseModel):
     amount: SkySellRequestAmount
     r"""The amount of USDS you would like to sell 1:1 for 'token_out'."""
 
+    chain: SkySellRequestChain
+
     sender: str
     r"""The address of the transaction sender."""
 
@@ -59,4 +63,5 @@ class SkySellRequest(BaseModel):
         pydantic.Field(alias="action_type"),
     ] = "SKY_SELL"
 
-    chain: Optional[SkySellRequestChain] = SkySellRequestChain.ETHEREUM
+    estimate_gas: Optional[bool] = None
+    r"""Determines whether to estimate gas costs for transactions, also verifying that the transaction can be successfully executed."""

compass_api_sdk/models/skywithdrawrequest.py

@@ -30,6 +30,8 @@ class SkyWithdrawRequestTypedDict(TypedDict):
     receiver: NotRequired[Nullable[str]]
     r"""The address which will receive the withdrawn USDS. Defaults to the sender."""
     chain: NotRequired[SkyWithdrawRequestChain]
+    estimate_gas: NotRequired[bool]
+    r"""Determines whether to estimate gas costs for transactions, also verifying that the transaction can be successfully executed."""
 
 
 class SkyWithdrawRequest(BaseModel):
@@ -50,11 +52,14 @@ class SkyWithdrawRequest(BaseModel):
     receiver: OptionalNullable[str] = UNSET
     r"""The address which will receive the withdrawn USDS. Defaults to the sender."""
 
-    chain: Optional[SkyWithdrawRequestChain] = SkyWithdrawRequestChain.ETHEREUM
+    chain: Optional[SkyWithdrawRequestChain] = None
+
+    estimate_gas: Optional[bool] = None
+    r"""Determines whether to estimate gas costs for transactions, also verifying that the transaction can be successfully executed."""
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["action_type", "receiver", "chain"]
+        optional_fields = ["action_type", "receiver", "chain", "estimate_gas"]
         nullable_fields = ["receiver"]
         null_default_fields = []
 

compass_api_sdk/models/sponsorgasrequest.py

@@ -0,0 +1,71 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .batchedsafeoperationsresponse_input import (
+    BatchedSafeOperationsResponseInput,
+    BatchedSafeOperationsResponseInputTypedDict,
+)
+from .chain import Chain
+from .permit2typeddata_input import (
+    Permit2TypedDataInput,
+    Permit2TypedDataInputTypedDict,
+)
+from .permittypeddataresponse_input import (
+    PermitTypedDataResponseInput,
+    PermitTypedDataResponseInputTypedDict,
+)
+from compass_api_sdk.types import BaseModel
+from typing import Union
+from typing_extensions import TypeAliasType, TypedDict
+
+
+SponsorGasRequestEip712TypedDict = TypeAliasType(
+    "SponsorGasRequestEip712TypedDict",
+    Union[
+        BatchedSafeOperationsResponseInputTypedDict,
+        PermitTypedDataResponseInputTypedDict,
+        Permit2TypedDataInputTypedDict,
+    ],
+)
+r"""The EIP-712 typed data that was signed."""
+
+
+SponsorGasRequestEip712 = TypeAliasType(
+    "SponsorGasRequestEip712",
+    Union[
+        BatchedSafeOperationsResponseInput,
+        PermitTypedDataResponseInput,
+        Permit2TypedDataInput,
+    ],
+)
+r"""The EIP-712 typed data that was signed."""
+
+
+class SponsorGasRequestTypedDict(TypedDict):
+    owner: str
+    r"""The wallet address that owns the Earn Account."""
+    chain: Chain
+    r"""The chain to use."""
+    eip_712: SponsorGasRequestEip712TypedDict
+    r"""The EIP-712 typed data that was signed."""
+    signature: str
+    r"""The EIP-712 signed typed data signature."""
+    sender: str
+    r"""The address of the wallet which will send the transaction."""
+
+
+class SponsorGasRequest(BaseModel):
+    owner: str
+    r"""The wallet address that owns the Earn Account."""
+
+    chain: Chain
+    r"""The chain to use."""
+
+    eip_712: SponsorGasRequestEip712
+    r"""The EIP-712 typed data that was signed."""
+
+    signature: str
+    r"""The EIP-712 signed typed data signature."""
+
+    sender: str
+    r"""The address of the wallet which will send the transaction."""

compass_api_sdk/models/sponsorgasresponse.py

@@ -0,0 +1,14 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .unsignedtransaction import UnsignedTransaction, UnsignedTransactionTypedDict
+from compass_api_sdk.types import BaseModel
+from typing_extensions import TypedDict
+
+
+class SponsorGasResponseTypedDict(TypedDict):
+    transaction: UnsignedTransactionTypedDict
+
+
+class SponsorGasResponse(BaseModel):
+    transaction: UnsignedTransaction

compass_api_sdk/models/tokenbalanceresponse.py

@@ -1,18 +1,8 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .token_enum import TokenEnum
 from compass_api_sdk.types import BaseModel
-from typing import Union
-from typing_extensions import TypeAliasType, TypedDict
-
-
-TokenSymbolTypedDict = TypeAliasType("TokenSymbolTypedDict", Union[TokenEnum, str])
-r"""Symbol of the token."""
-
-
-TokenSymbol = TypeAliasType("TokenSymbol", Union[TokenEnum, str])
-r"""Symbol of the token."""
+from typing_extensions import TypedDict
 
 
 class TokenBalanceResponseTypedDict(TypedDict):
@@ -22,8 +12,8 @@ class TokenBalanceResponseTypedDict(TypedDict):
     r"""Amount of tokens a particular address holds"""
     decimals: int
     r"""Number of decimals of the token"""
-    token_symbol: TokenSymbolTypedDict
-    r"""Symbol of the token."""
+    token_symbol: str
+    r"""Symbol of the token"""
     token_address: str
     r"""Address of the token"""
 
@@ -37,8 +27,8 @@ class TokenBalanceResponse(BaseModel):
     decimals: int
     r"""Number of decimals of the token"""
 
-    token_symbol: TokenSymbol
-    r"""Symbol of the token."""
+    token_symbol: str
+    r"""Symbol of the token"""
 
     token_address: str
     r"""Address of the token"""