compass_api_sdk 0.9.52__py3-none-any.whl → 1.0.1__py3-none-any.whl

compass_api_sdk/pendle.py

@@ -20,7 +20,7 @@ class Pendle(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.CompassAPIBackendModelsPendleReadResponseMarketPendleGetMarketResponse:
+    ) -> models.PendleGetMarketResponse:
         r"""Get Market & User Position
 
         Get the market's implied APY, maturity date and the associated token data.
@@ -92,10 +92,7 @@ class Pendle(BaseSDK):
 
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/json"):
-            return unmarshal_json_response(
-                models.CompassAPIBackendModelsPendleReadResponseMarketPendleGetMarketResponse,
-                http_res,
-            )
+            return unmarshal_json_response(models.PendleGetMarketResponse, http_res)
         if utils.match_response(http_res, "422", "application/json"):
             response_data = unmarshal_json_response(
                 errors.HTTPValidationErrorData, http_res
@@ -121,7 +118,7 @@ class Pendle(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.CompassAPIBackendModelsPendleReadResponseMarketPendleGetMarketResponse:
+    ) -> models.PendleGetMarketResponse:
         r"""Get Market & User Position
 
         Get the market's implied APY, maturity date and the associated token data.
@@ -193,10 +190,7 @@ class Pendle(BaseSDK):
 
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/json"):
-            return unmarshal_json_response(
-                models.CompassAPIBackendModelsPendleReadResponseMarketPendleGetMarketResponse,
-                http_res,
-            )
+            return unmarshal_json_response(models.PendleGetMarketResponse, http_res)
         if utils.match_response(http_res, "422", "application/json"):
             response_data = unmarshal_json_response(
                 errors.HTTPValidationErrorData, http_res
@@ -220,7 +214,7 @@ class Pendle(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.CompassAPIBackendModelsPendleReadResponsePositionsPendleListUserPositionsResponse:
+    ) -> models.PendleListUserPositionsResponse:
         r"""List User's Market Positions
 
         List the user's SY, PT, YT and LP positions for all markets on a given chain.
@@ -287,8 +281,7 @@ class Pendle(BaseSDK):
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/json"):
             return unmarshal_json_response(
-                models.CompassAPIBackendModelsPendleReadResponsePositionsPendleListUserPositionsResponse,
-                http_res,
+                models.PendleListUserPositionsResponse, http_res
             )
         if utils.match_response(http_res, "422", "application/json"):
             response_data = unmarshal_json_response(
@@ -313,7 +306,7 @@ class Pendle(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.CompassAPIBackendModelsPendleReadResponsePositionsPendleListUserPositionsResponse:
+    ) -> models.PendleListUserPositionsResponse:
         r"""List User's Market Positions
 
         List the user's SY, PT, YT and LP positions for all markets on a given chain.
@@ -380,8 +373,7 @@ class Pendle(BaseSDK):
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/json"):
             return unmarshal_json_response(
-                models.CompassAPIBackendModelsPendleReadResponsePositionsPendleListUserPositionsResponse,
-                http_res,
+                models.PendleListUserPositionsResponse, http_res
             )
         if utils.match_response(http_res, "422", "application/json"):
             response_data = unmarshal_json_response(
@@ -405,9 +397,7 @@ class Pendle(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> (
-        models.CompassAPIBackendModelsPendleReadResponseMarketsPendleListMarketsResponse
-    ):
+    ) -> models.PendleListMarketsResponse:
         r"""List Market Data
 
         Get a list of active markets.
@@ -471,10 +461,7 @@ class Pendle(BaseSDK):
 
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/json"):
-            return unmarshal_json_response(
-                models.CompassAPIBackendModelsPendleReadResponseMarketsPendleListMarketsResponse,
-                http_res,
-            )
+            return unmarshal_json_response(models.PendleListMarketsResponse, http_res)
         if utils.match_response(http_res, "422", "application/json"):
             response_data = unmarshal_json_response(
                 errors.HTTPValidationErrorData, http_res
@@ -497,9 +484,7 @@ class Pendle(BaseSDK):
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> (
-        models.CompassAPIBackendModelsPendleReadResponseMarketsPendleListMarketsResponse
-    ):
+    ) -> models.PendleListMarketsResponse:
         r"""List Market Data
 
         Get a list of active markets.
@@ -563,10 +548,7 @@ class Pendle(BaseSDK):
 
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/json"):
-            return unmarshal_json_response(
-                models.CompassAPIBackendModelsPendleReadResponseMarketsPendleListMarketsResponse,
-                http_res,
-            )
+            return unmarshal_json_response(models.PendleListMarketsResponse, http_res)
         if utils.match_response(http_res, "422", "application/json"):
             response_data = unmarshal_json_response(
                 errors.HTTPValidationErrorData, http_res
@@ -609,8 +591,6 @@ class Pendle(BaseSDK):
 
         A sufficient allowance for the Pendle Router on the appropriate token contract must be set
         beforehand. For `action` set to `BUY`, this is the `token` contract. For `action` set to `SELL`, this is the PT contract.
-
-        A quote is included with the response.
         <Info>
         **Required Allowances**
 
@@ -737,8 +717,6 @@ class Pendle(BaseSDK):
 
         A sufficient allowance for the Pendle Router on the appropriate token contract must be set
         beforehand. For `action` set to `BUY`, this is the `token` contract. For `action` set to `SELL`, this is the PT contract.
-
-        A quote is included with the response.
         <Info>
         **Required Allowances**
 
@@ -865,8 +843,6 @@ class Pendle(BaseSDK):
 
         A sufficient allowance for the Pendle Router on the appropriate token contract must be set
         beforehand. For `action` set to `BUY`, this is the `token` contract. For `action` set to `SELL`, this is the YT contract.
-
-        A quote is included with the response.
         <Info>
         **Required Allowances**
 
@@ -993,8 +969,6 @@ class Pendle(BaseSDK):
 
         A sufficient allowance for the Pendle Router on the appropriate token contract must be set
         beforehand. For `action` set to `BUY`, this is the `token` contract. For `action` set to `SELL`, this is the YT contract.
-
-        A quote is included with the response.
         <Info>
         **Required Allowances**
 
@@ -1125,8 +1099,6 @@ class Pendle(BaseSDK):
 
         A sufficient allowance for the Pendle Router on the appropriate token contract must be set
         beforehand. For `action` set to `SUPPLY`, this is the `token` contract. For `action` set to `WTIHDRAW`, this is the market contract (LP).
-
-        A quote is included with the response.
         <Info>
         **Required Allowances**
 
@@ -1257,8 +1229,6 @@ class Pendle(BaseSDK):
 
         A sufficient allowance for the Pendle Router on the appropriate token contract must be set
         beforehand. For `action` set to `SUPPLY`, this is the `token` contract. For `action` set to `WTIHDRAW`, this is the market contract (LP).
-
-        A quote is included with the response.
         <Info>
         **Required Allowances**
 
@@ -1361,7 +1331,7 @@ class Pendle(BaseSDK):
         self,
         *,
         market_address: str,
-        chain: models.CompassAPIBackendModelsPendleTransactRequestRedeemYieldPendleRedeemYieldRequestChain,
+        chain: models.PendleRedeemYieldRequestChain,
         sender: str,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -1398,7 +1368,7 @@ class Pendle(BaseSDK):
         else:
             base_url = self._get_url(base_url, url_variables)
 
-        request = models.CompassAPIBackendModelsPendleTransactRequestRedeemYieldPendleRedeemYieldRequest(
+        request = models.PendleRedeemYieldRequest(
             market_address=market_address,
             chain=chain,
             sender=sender,
@@ -1418,11 +1388,7 @@ class Pendle(BaseSDK):
             http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
-                request,
-                False,
-                False,
-                "json",
-                models.CompassAPIBackendModelsPendleTransactRequestRedeemYieldPendleRedeemYieldRequest,
+                request, False, False, "json", models.PendleRedeemYieldRequest
             ),
             timeout_ms=timeout_ms,
         )
@@ -1469,7 +1435,7 @@ class Pendle(BaseSDK):
         self,
         *,
         market_address: str,
-        chain: models.CompassAPIBackendModelsPendleTransactRequestRedeemYieldPendleRedeemYieldRequestChain,
+        chain: models.PendleRedeemYieldRequestChain,
         sender: str,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -1506,7 +1472,7 @@ class Pendle(BaseSDK):
         else:
             base_url = self._get_url(base_url, url_variables)
 
-        request = models.CompassAPIBackendModelsPendleTransactRequestRedeemYieldPendleRedeemYieldRequest(
+        request = models.PendleRedeemYieldRequest(
             market_address=market_address,
             chain=chain,
             sender=sender,
@@ -1526,11 +1492,7 @@ class Pendle(BaseSDK):
             http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
-                request,
-                False,
-                False,
-                "json",
-                models.CompassAPIBackendModelsPendleTransactRequestRedeemYieldPendleRedeemYieldRequest,
+                request, False, False, "json", models.PendleRedeemYieldRequest
             ),
             timeout_ms=timeout_ms,
         )

compass_api_sdk/sdk.py

@@ -17,6 +17,7 @@ if TYPE_CHECKING:
     from compass_api_sdk.aave_v3 import AaveV3
     from compass_api_sdk.aerodrome_slipstream import AerodromeSlipstream
     from compass_api_sdk.erc_4626_vaults import ERC4626Vaults
+    from compass_api_sdk.ethena import Ethena
     from compass_api_sdk.morpho import Morpho
     from compass_api_sdk.pendle import Pendle
     from compass_api_sdk.sky import Sky
@@ -43,6 +44,7 @@ class CompassAPI(BaseSDK):
     transaction_bundler: "TransactionBundler"
     smart_account: "SmartAccount"
     erc_4626_vaults: "ERC4626Vaults"
+    ethena: "Ethena"
     _sub_sdk_map = {
         "aave_v3": ("compass_api_sdk.aave_v3", "AaveV3"),
         "aerodrome_slipstream": (
@@ -62,6 +64,7 @@ class CompassAPI(BaseSDK):
         ),
         "smart_account": ("compass_api_sdk.smart_account", "SmartAccount"),
         "erc_4626_vaults": ("compass_api_sdk.erc_4626_vaults", "ERC4626Vaults"),
+        "ethena": ("compass_api_sdk.ethena", "Ethena"),
     }
 
     def __init__(

compass_api_sdk/sky.py

@@ -192,12 +192,9 @@ class Sky(BaseSDK):
     def sky_buy(
         self,
         *,
-        token_in: models.CompassAPIBackendModelsSkyTransactRequestBuySkyBuyRequestTokenIn,
-        amount: Union[
-            models.CompassAPIBackendModelsSkyTransactRequestBuySkyBuyRequestAmount,
-            models.CompassAPIBackendModelsSkyTransactRequestBuySkyBuyRequestAmountTypedDict,
-        ],
-        chain: models.CompassAPIBackendModelsSkyTransactRequestBuySkyBuyRequestChain,
+        token_in: models.SkyBuyRequestTokenIn,
+        amount: Union[models.SkyBuyRequestAmount, models.SkyBuyRequestAmountTypedDict],
+        chain: models.SkyBuyRequestChain,
         sender: str,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -241,7 +238,7 @@ class Sky(BaseSDK):
         else:
             base_url = self._get_url(base_url, url_variables)
 
-        request = models.CompassAPIBackendModelsSkyTransactRequestBuySkyBuyRequest(
+        request = models.SkyBuyRequest(
             token_in=token_in,
             amount=amount,
             chain=chain,
@@ -262,11 +259,7 @@ class Sky(BaseSDK):
             http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
-                request,
-                False,
-                False,
-                "json",
-                models.CompassAPIBackendModelsSkyTransactRequestBuySkyBuyRequest,
+                request, False, False, "json", models.SkyBuyRequest
             ),
             timeout_ms=timeout_ms,
         )
@@ -312,12 +305,9 @@ class Sky(BaseSDK):
     async def sky_buy_async(
         self,
         *,
-        token_in: models.CompassAPIBackendModelsSkyTransactRequestBuySkyBuyRequestTokenIn,
-        amount: Union[
-            models.CompassAPIBackendModelsSkyTransactRequestBuySkyBuyRequestAmount,
-            models.CompassAPIBackendModelsSkyTransactRequestBuySkyBuyRequestAmountTypedDict,
-        ],
-        chain: models.CompassAPIBackendModelsSkyTransactRequestBuySkyBuyRequestChain,
+        token_in: models.SkyBuyRequestTokenIn,
+        amount: Union[models.SkyBuyRequestAmount, models.SkyBuyRequestAmountTypedDict],
+        chain: models.SkyBuyRequestChain,
         sender: str,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -361,7 +351,7 @@ class Sky(BaseSDK):
         else:
             base_url = self._get_url(base_url, url_variables)
 
-        request = models.CompassAPIBackendModelsSkyTransactRequestBuySkyBuyRequest(
+        request = models.SkyBuyRequest(
             token_in=token_in,
             amount=amount,
             chain=chain,
@@ -382,11 +372,7 @@ class Sky(BaseSDK):
             http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
-                request,
-                False,
-                False,
-                "json",
-                models.CompassAPIBackendModelsSkyTransactRequestBuySkyBuyRequest,
+                request, False, False, "json", models.SkyBuyRequest
             ),
             timeout_ms=timeout_ms,
         )
@@ -432,15 +418,14 @@ class Sky(BaseSDK):
     def sky_sell(
         self,
         *,
-        token_out: models.CompassAPIBackendModelsSkyTransactRequestSellSkySellRequestTokenOut,
+        token_out: models.SkySellRequestTokenOut,
         amount: Union[
-            models.CompassAPIBackendModelsSkyTransactRequestSellSkySellRequestAmount,
-            models.CompassAPIBackendModelsSkyTransactRequestSellSkySellRequestAmountTypedDict,
+            models.SkySellRequestAmount, models.SkySellRequestAmountTypedDict
         ],
         sender: str,
         chain: Optional[
-            models.CompassAPIBackendModelsSkyTransactRequestSellSkySellRequestChain
-        ] = models.CompassAPIBackendModelsSkyTransactRequestSellSkySellRequestChain.ETHEREUM,
+            models.SkySellRequestChain
+        ] = models.SkySellRequestChain.ETHEREUM,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -483,7 +468,7 @@ class Sky(BaseSDK):
         else:
             base_url = self._get_url(base_url, url_variables)
 
-        request = models.CompassAPIBackendModelsSkyTransactRequestSellSkySellRequest(
+        request = models.SkySellRequest(
             token_out=token_out,
             amount=amount,
             chain=chain,
@@ -504,11 +489,7 @@ class Sky(BaseSDK):
             http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
-                request,
-                False,
-                False,
-                "json",
-                models.CompassAPIBackendModelsSkyTransactRequestSellSkySellRequest,
+                request, False, False, "json", models.SkySellRequest
             ),
             timeout_ms=timeout_ms,
         )
@@ -554,15 +535,14 @@ class Sky(BaseSDK):
     async def sky_sell_async(
         self,
         *,
-        token_out: models.CompassAPIBackendModelsSkyTransactRequestSellSkySellRequestTokenOut,
+        token_out: models.SkySellRequestTokenOut,
         amount: Union[
-            models.CompassAPIBackendModelsSkyTransactRequestSellSkySellRequestAmount,
-            models.CompassAPIBackendModelsSkyTransactRequestSellSkySellRequestAmountTypedDict,
+            models.SkySellRequestAmount, models.SkySellRequestAmountTypedDict
         ],
         sender: str,
         chain: Optional[
-            models.CompassAPIBackendModelsSkyTransactRequestSellSkySellRequestChain
-        ] = models.CompassAPIBackendModelsSkyTransactRequestSellSkySellRequestChain.ETHEREUM,
+            models.SkySellRequestChain
+        ] = models.SkySellRequestChain.ETHEREUM,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -605,7 +585,7 @@ class Sky(BaseSDK):
         else:
             base_url = self._get_url(base_url, url_variables)
 
-        request = models.CompassAPIBackendModelsSkyTransactRequestSellSkySellRequest(
+        request = models.SkySellRequest(
             token_out=token_out,
             amount=amount,
             chain=chain,
@@ -626,11 +606,7 @@ class Sky(BaseSDK):
             http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
-                request,
-                False,
-                False,
-                "json",
-                models.CompassAPIBackendModelsSkyTransactRequestSellSkySellRequest,
+                request, False, False, "json", models.SkySellRequest
             ),
             timeout_ms=timeout_ms,
         )
@@ -677,14 +653,13 @@ class Sky(BaseSDK):
         self,
         *,
         amount: Union[
-            models.CompassAPIBackendModelsSkyTransactRequestDepositSkyDepositRequestAmount,
-            models.CompassAPIBackendModelsSkyTransactRequestDepositSkyDepositRequestAmountTypedDict,
+            models.SkyDepositRequestAmount, models.SkyDepositRequestAmountTypedDict
         ],
         sender: str,
         receiver: OptionalNullable[str] = UNSET,
         chain: Optional[
-            models.CompassAPIBackendModelsSkyTransactRequestDepositSkyDepositRequestChain
-        ] = models.CompassAPIBackendModelsSkyTransactRequestDepositSkyDepositRequestChain.ETHEREUM,
+            models.SkyDepositRequestChain
+        ] = models.SkyDepositRequestChain.ETHEREUM,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -725,13 +700,11 @@ class Sky(BaseSDK):
         else:
             base_url = self._get_url(base_url, url_variables)
 
-        request = (
-            models.CompassAPIBackendModelsSkyTransactRequestDepositSkyDepositRequest(
-                amount=amount,
-                receiver=receiver,
-                chain=chain,
-                sender=sender,
-            )
+        request = models.SkyDepositRequest(
+            amount=amount,
+            receiver=receiver,
+            chain=chain,
+            sender=sender,
         )
 
         req = self._build_request(
@@ -748,11 +721,7 @@ class Sky(BaseSDK):
             http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
-                request,
-                False,
-                False,
-                "json",
-                models.CompassAPIBackendModelsSkyTransactRequestDepositSkyDepositRequest,
+                request, False, False, "json", models.SkyDepositRequest
             ),
             timeout_ms=timeout_ms,
         )
@@ -799,14 +768,13 @@ class Sky(BaseSDK):
         self,
         *,
         amount: Union[
-            models.CompassAPIBackendModelsSkyTransactRequestDepositSkyDepositRequestAmount,
-            models.CompassAPIBackendModelsSkyTransactRequestDepositSkyDepositRequestAmountTypedDict,
+            models.SkyDepositRequestAmount, models.SkyDepositRequestAmountTypedDict
         ],
         sender: str,
         receiver: OptionalNullable[str] = UNSET,
         chain: Optional[
-            models.CompassAPIBackendModelsSkyTransactRequestDepositSkyDepositRequestChain
-        ] = models.CompassAPIBackendModelsSkyTransactRequestDepositSkyDepositRequestChain.ETHEREUM,
+            models.SkyDepositRequestChain
+        ] = models.SkyDepositRequestChain.ETHEREUM,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -847,13 +815,11 @@ class Sky(BaseSDK):
         else:
             base_url = self._get_url(base_url, url_variables)
 
-        request = (
-            models.CompassAPIBackendModelsSkyTransactRequestDepositSkyDepositRequest(
-                amount=amount,
-                receiver=receiver,
-                chain=chain,
-                sender=sender,
-            )
+        request = models.SkyDepositRequest(
+            amount=amount,
+            receiver=receiver,
+            chain=chain,
+            sender=sender,
         )
 
         req = self._build_request_async(
@@ -870,11 +836,7 @@ class Sky(BaseSDK):
             http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
-                request,
-                False,
-                False,
-                "json",
-                models.CompassAPIBackendModelsSkyTransactRequestDepositSkyDepositRequest,
+                request, False, False, "json", models.SkyDepositRequest
             ),
             timeout_ms=timeout_ms,
         )
@@ -924,8 +886,8 @@ class Sky(BaseSDK):
         sender: str,
         receiver: OptionalNullable[str] = UNSET,
         chain: Optional[
-            models.CompassAPIBackendModelsSkyTransactRequestWithdrawSkyWithdrawRequestChain
-        ] = models.CompassAPIBackendModelsSkyTransactRequestWithdrawSkyWithdrawRequestChain.ETHEREUM,
+            models.SkyWithdrawRequestChain
+        ] = models.SkyWithdrawRequestChain.ETHEREUM,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -967,13 +929,11 @@ class Sky(BaseSDK):
         else:
             base_url = self._get_url(base_url, url_variables)
 
-        request = (
-            models.CompassAPIBackendModelsSkyTransactRequestWithdrawSkyWithdrawRequest(
-                amount=amount,
-                receiver=receiver,
-                chain=chain,
-                sender=sender,
-            )
+        request = models.SkyWithdrawRequest(
+            amount=amount,
+            receiver=receiver,
+            chain=chain,
+            sender=sender,
         )
 
         req = self._build_request(
@@ -990,11 +950,7 @@ class Sky(BaseSDK):
             http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
-                request,
-                False,
-                False,
-                "json",
-                models.CompassAPIBackendModelsSkyTransactRequestWithdrawSkyWithdrawRequest,
+                request, False, False, "json", models.SkyWithdrawRequest
             ),
             timeout_ms=timeout_ms,
         )
@@ -1044,8 +1000,8 @@ class Sky(BaseSDK):
         sender: str,
         receiver: OptionalNullable[str] = UNSET,
         chain: Optional[
-            models.CompassAPIBackendModelsSkyTransactRequestWithdrawSkyWithdrawRequestChain
-        ] = models.CompassAPIBackendModelsSkyTransactRequestWithdrawSkyWithdrawRequestChain.ETHEREUM,
+            models.SkyWithdrawRequestChain
+        ] = models.SkyWithdrawRequestChain.ETHEREUM,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -1087,13 +1043,11 @@ class Sky(BaseSDK):
         else:
             base_url = self._get_url(base_url, url_variables)
 
-        request = (
-            models.CompassAPIBackendModelsSkyTransactRequestWithdrawSkyWithdrawRequest(
-                amount=amount,
-                receiver=receiver,
-                chain=chain,
-                sender=sender,
-            )
+        request = models.SkyWithdrawRequest(
+            amount=amount,
+            receiver=receiver,
+            chain=chain,
+            sender=sender,
         )
 
         req = self._build_request_async(
@@ -1110,11 +1064,7 @@ class Sky(BaseSDK):
             http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
-                request,
-                False,
-                False,
-                "json",
-                models.CompassAPIBackendModelsSkyTransactRequestWithdrawSkyWithdrawRequest,
+                request, False, False, "json", models.SkyWithdrawRequest
             ),
             timeout_ms=timeout_ms,
         )