compass_api_sdk 0.9.35__py3-none-any.whl → 2.0.21rc1__py3-none-any.whl

compass_api_sdk/smart_account.py

@@ -4,18 +4,20 @@ from .basesdk import BaseSDK
 from compass_api_sdk import errors, models, utils
 from compass_api_sdk._hooks import HookContext
 from compass_api_sdk.types import OptionalNullable, UNSET
+from compass_api_sdk.utils.unmarshal_json_response import unmarshal_json_response
 from typing import Any, List, Mapping, Optional, Union
 
 
 class SmartAccount(BaseSDK):
-    def account_batched_user_operations(
+    def smart_account_batched_user_operations(
         self,
         *,
-        chain: models.Chain,
+        chain: models.BatchedUserOperationsRequestChain,
         sender: str,
         operations: Union[
             List[models.UserOperation], List[models.UserOperationTypedDict]
         ],
+        estimate_gas: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -25,9 +27,10 @@ class SmartAccount(BaseSDK):
 
         Generate a list of user operations for smart account batching.
 
-        :param chain: The chain to use.
+        :param chain:
         :param sender: The address of the transaction sender.
         :param operations: List of possible user operations
+        :param estimate_gas: Determines whether to estimate gas costs for transactions, also verifying that the transaction can be successfully executed.
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -46,12 +49,13 @@ class SmartAccount(BaseSDK):
         request = models.BatchedUserOperationsRequest(
             chain=chain,
             sender=sender,
+            estimate_gas=estimate_gas,
             operations=utils.get_pydantic_model(operations, List[models.UserOperation]),
         )
 
         req = self._build_request(
             method="POST",
-            path="/v0/smart_account/batched_user_operations",
+            path="/v1/smart_account/batched_user_operations",
             base_url=base_url,
             url_variables=url_variables,
             request=request,
@@ -65,6 +69,7 @@ class SmartAccount(BaseSDK):
             get_serialized_body=lambda: utils.serialize_request_body(
                 request, False, False, "json", models.BatchedUserOperationsRequest
             ),
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -80,8 +85,8 @@ class SmartAccount(BaseSDK):
             hook_ctx=HookContext(
                 config=self.sdk_configuration,
                 base_url=base_url or "",
-                operation_id="smart_account_batched_user_operations",
-                oauth2_scopes=[],
+                operation_id="v1_smart_account_batched_user_operations",
+                oauth2_scopes=None,
                 security_source=self.sdk_configuration.security,
             ),
             request=req,
@@ -91,42 +96,32 @@ class SmartAccount(BaseSDK):
 
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/json"):
-            return utils.unmarshal_json(
-                http_res.text, models.BatchedUserOperationsResponse
+            return unmarshal_json_response(
+                models.BatchedUserOperationsResponse, http_res
             )
         if utils.match_response(http_res, "422", "application/json"):
-            response_data = utils.unmarshal_json(
-                http_res.text, errors.HTTPValidationErrorData
+            response_data = unmarshal_json_response(
+                errors.HTTPValidationErrorData, http_res
             )
-            raise errors.HTTPValidationError(data=response_data)
+            raise errors.HTTPValidationError(response_data, http_res)
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = utils.stream_to_text(http_res)
-            raise errors.APIError(
-                "API error occurred", http_res.status_code, http_res_text, http_res
-            )
+            raise errors.APIError("API error occurred", http_res, http_res_text)
         if utils.match_response(http_res, "5XX", "*"):
             http_res_text = utils.stream_to_text(http_res)
-            raise errors.APIError(
-                "API error occurred", http_res.status_code, http_res_text, http_res
-            )
+            raise errors.APIError("API error occurred", http_res, http_res_text)
 
-        content_type = http_res.headers.get("Content-Type")
-        http_res_text = utils.stream_to_text(http_res)
-        raise errors.APIError(
-            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
-            http_res.status_code,
-            http_res_text,
-            http_res,
-        )
+        raise errors.APIError("Unexpected response received", http_res)
 
-    async def account_batched_user_operations_async(
+    async def smart_account_batched_user_operations_async(
         self,
         *,
-        chain: models.Chain,
+        chain: models.BatchedUserOperationsRequestChain,
         sender: str,
         operations: Union[
             List[models.UserOperation], List[models.UserOperationTypedDict]
         ],
+        estimate_gas: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -136,9 +131,10 @@ class SmartAccount(BaseSDK):
 
         Generate a list of user operations for smart account batching.
 
-        :param chain: The chain to use.
+        :param chain:
         :param sender: The address of the transaction sender.
         :param operations: List of possible user operations
+        :param estimate_gas: Determines whether to estimate gas costs for transactions, also verifying that the transaction can be successfully executed.
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -157,12 +153,13 @@ class SmartAccount(BaseSDK):
         request = models.BatchedUserOperationsRequest(
             chain=chain,
             sender=sender,
+            estimate_gas=estimate_gas,
             operations=utils.get_pydantic_model(operations, List[models.UserOperation]),
         )
 
         req = self._build_request_async(
             method="POST",
-            path="/v0/smart_account/batched_user_operations",
+            path="/v1/smart_account/batched_user_operations",
             base_url=base_url,
             url_variables=url_variables,
             request=request,
@@ -176,6 +173,7 @@ class SmartAccount(BaseSDK):
             get_serialized_body=lambda: utils.serialize_request_body(
                 request, False, False, "json", models.BatchedUserOperationsRequest
             ),
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -191,8 +189,8 @@ class SmartAccount(BaseSDK):
             hook_ctx=HookContext(
                 config=self.sdk_configuration,
                 base_url=base_url or "",
-                operation_id="smart_account_batched_user_operations",
-                oauth2_scopes=[],
+                operation_id="v1_smart_account_batched_user_operations",
+                oauth2_scopes=None,
                 security_source=self.sdk_configuration.security,
             ),
             request=req,
@@ -202,30 +200,19 @@ class SmartAccount(BaseSDK):
 
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/json"):
-            return utils.unmarshal_json(
-                http_res.text, models.BatchedUserOperationsResponse
+            return unmarshal_json_response(
+                models.BatchedUserOperationsResponse, http_res
             )
         if utils.match_response(http_res, "422", "application/json"):
-            response_data = utils.unmarshal_json(
-                http_res.text, errors.HTTPValidationErrorData
+            response_data = unmarshal_json_response(
+                errors.HTTPValidationErrorData, http_res
             )
-            raise errors.HTTPValidationError(data=response_data)
+            raise errors.HTTPValidationError(response_data, http_res)
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = await utils.stream_to_text_async(http_res)
-            raise errors.APIError(
-                "API error occurred", http_res.status_code, http_res_text, http_res
-            )
+            raise errors.APIError("API error occurred", http_res, http_res_text)
         if utils.match_response(http_res, "5XX", "*"):
             http_res_text = await utils.stream_to_text_async(http_res)
-            raise errors.APIError(
-                "API error occurred", http_res.status_code, http_res_text, http_res
-            )
+            raise errors.APIError("API error occurred", http_res, http_res_text)
 
-        content_type = http_res.headers.get("Content-Type")
-        http_res_text = await utils.stream_to_text_async(http_res)
-        raise errors.APIError(
-            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
-            http_res.status_code,
-            http_res_text,
-            http_res,
-        )
+        raise errors.APIError("Unexpected response received", http_res)

compass_api_sdk/swap.py

@@ -0,0 +1,248 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from .basesdk import BaseSDK
+from compass_api_sdk import errors, models, utils
+from compass_api_sdk._hooks import HookContext
+from compass_api_sdk.types import OptionalNullable, UNSET
+from compass_api_sdk.utils.unmarshal_json_response import unmarshal_json_response
+from typing import Any, Mapping, Optional, Union
+
+
+class Swap(BaseSDK):
+    def swap_odos(
+        self,
+        *,
+        token_in: str,
+        token_out: str,
+        amount_in: Union[
+            models.OdosSwapRequestAmountIn, models.OdosSwapRequestAmountInTypedDict
+        ],
+        max_slippage_percent: float,
+        chain: models.OdosSwapRequestChain,
+        sender: str,
+        estimate_gas: Optional[bool] = None,
+        retries: OptionalNullable[utils.RetryConfig] = UNSET,
+        server_url: Optional[str] = None,
+        timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
+    ) -> models.OdosTransactionResponse:
+        r"""Odos Swap
+
+        Swap between two tokens using Odos Smart Order Routing.
+        <Info>
+        **Required Allowances**
+
+        In order to make this transaction, token allowances need to be set for the following contracts.
+
+        - `OdosRouter`
+        </Info>
+
+
+        :param token_in: The symbol or address of the token that is to be sold.
+        :param token_out: The symbol or address of the token that is to be bought.
+        :param amount_in: The amount of token_in to be sold.
+        :param max_slippage_percent: The maximum slippage allowed in percent. e.g. `1` means `1%` slippage allowed.
+        :param chain:
+        :param sender: The address of the transaction sender.
+        :param estimate_gas: Determines whether to estimate gas costs for transactions, also verifying that the transaction can be successfully executed.
+        :param retries: Override the default retry configuration for this method
+        :param server_url: Override the default server URL for this method
+        :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
+        :param http_headers: Additional headers to set or replace on requests.
+        """
+        base_url = None
+        url_variables = None
+        if timeout_ms is None:
+            timeout_ms = self.sdk_configuration.timeout_ms
+
+        if server_url is not None:
+            base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
+
+        request = models.OdosSwapRequest(
+            token_in=token_in,
+            token_out=token_out,
+            amount_in=amount_in,
+            max_slippage_percent=max_slippage_percent,
+            chain=chain,
+            sender=sender,
+            estimate_gas=estimate_gas,
+        )
+
+        req = self._build_request(
+            method="POST",
+            path="/v1/swap/odos",
+            base_url=base_url,
+            url_variables=url_variables,
+            request=request,
+            request_body_required=True,
+            request_has_path_params=False,
+            request_has_query_params=True,
+            user_agent_header="user-agent",
+            accept_header_value="application/json",
+            http_headers=http_headers,
+            security=self.sdk_configuration.security,
+            get_serialized_body=lambda: utils.serialize_request_body(
+                request, False, False, "json", models.OdosSwapRequest
+            ),
+            allow_empty_value=None,
+            timeout_ms=timeout_ms,
+        )
+
+        if retries == UNSET:
+            if self.sdk_configuration.retry_config is not UNSET:
+                retries = self.sdk_configuration.retry_config
+
+        retry_config = None
+        if isinstance(retries, utils.RetryConfig):
+            retry_config = (retries, ["429", "500", "502", "503", "504"])
+
+        http_res = self.do_request(
+            hook_ctx=HookContext(
+                config=self.sdk_configuration,
+                base_url=base_url or "",
+                operation_id="v1_swap_odos",
+                oauth2_scopes=None,
+                security_source=self.sdk_configuration.security,
+            ),
+            request=req,
+            error_status_codes=["422", "4XX", "5XX"],
+            retry_config=retry_config,
+        )
+
+        response_data: Any = None
+        if utils.match_response(http_res, "200", "application/json"):
+            return unmarshal_json_response(models.OdosTransactionResponse, http_res)
+        if utils.match_response(http_res, "422", "application/json"):
+            response_data = unmarshal_json_response(
+                errors.HTTPValidationErrorData, http_res
+            )
+            raise errors.HTTPValidationError(response_data, http_res)
+        if utils.match_response(http_res, "4XX", "*"):
+            http_res_text = utils.stream_to_text(http_res)
+            raise errors.APIError("API error occurred", http_res, http_res_text)
+        if utils.match_response(http_res, "5XX", "*"):
+            http_res_text = utils.stream_to_text(http_res)
+            raise errors.APIError("API error occurred", http_res, http_res_text)
+
+        raise errors.APIError("Unexpected response received", http_res)
+
+    async def swap_odos_async(
+        self,
+        *,
+        token_in: str,
+        token_out: str,
+        amount_in: Union[
+            models.OdosSwapRequestAmountIn, models.OdosSwapRequestAmountInTypedDict
+        ],
+        max_slippage_percent: float,
+        chain: models.OdosSwapRequestChain,
+        sender: str,
+        estimate_gas: Optional[bool] = None,
+        retries: OptionalNullable[utils.RetryConfig] = UNSET,
+        server_url: Optional[str] = None,
+        timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
+    ) -> models.OdosTransactionResponse:
+        r"""Odos Swap
+
+        Swap between two tokens using Odos Smart Order Routing.
+        <Info>
+        **Required Allowances**
+
+        In order to make this transaction, token allowances need to be set for the following contracts.
+
+        - `OdosRouter`
+        </Info>
+
+
+        :param token_in: The symbol or address of the token that is to be sold.
+        :param token_out: The symbol or address of the token that is to be bought.
+        :param amount_in: The amount of token_in to be sold.
+        :param max_slippage_percent: The maximum slippage allowed in percent. e.g. `1` means `1%` slippage allowed.
+        :param chain:
+        :param sender: The address of the transaction sender.
+        :param estimate_gas: Determines whether to estimate gas costs for transactions, also verifying that the transaction can be successfully executed.
+        :param retries: Override the default retry configuration for this method
+        :param server_url: Override the default server URL for this method
+        :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
+        :param http_headers: Additional headers to set or replace on requests.
+        """
+        base_url = None
+        url_variables = None
+        if timeout_ms is None:
+            timeout_ms = self.sdk_configuration.timeout_ms
+
+        if server_url is not None:
+            base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
+
+        request = models.OdosSwapRequest(
+            token_in=token_in,
+            token_out=token_out,
+            amount_in=amount_in,
+            max_slippage_percent=max_slippage_percent,
+            chain=chain,
+            sender=sender,
+            estimate_gas=estimate_gas,
+        )
+
+        req = self._build_request_async(
+            method="POST",
+            path="/v1/swap/odos",
+            base_url=base_url,
+            url_variables=url_variables,
+            request=request,
+            request_body_required=True,
+            request_has_path_params=False,
+            request_has_query_params=True,
+            user_agent_header="user-agent",
+            accept_header_value="application/json",
+            http_headers=http_headers,
+            security=self.sdk_configuration.security,
+            get_serialized_body=lambda: utils.serialize_request_body(
+                request, False, False, "json", models.OdosSwapRequest
+            ),
+            allow_empty_value=None,
+            timeout_ms=timeout_ms,
+        )
+
+        if retries == UNSET:
+            if self.sdk_configuration.retry_config is not UNSET:
+                retries = self.sdk_configuration.retry_config
+
+        retry_config = None
+        if isinstance(retries, utils.RetryConfig):
+            retry_config = (retries, ["429", "500", "502", "503", "504"])
+
+        http_res = await self.do_request_async(
+            hook_ctx=HookContext(
+                config=self.sdk_configuration,
+                base_url=base_url or "",
+                operation_id="v1_swap_odos",
+                oauth2_scopes=None,
+                security_source=self.sdk_configuration.security,
+            ),
+            request=req,
+            error_status_codes=["422", "4XX", "5XX"],
+            retry_config=retry_config,
+        )
+
+        response_data: Any = None
+        if utils.match_response(http_res, "200", "application/json"):
+            return unmarshal_json_response(models.OdosTransactionResponse, http_res)
+        if utils.match_response(http_res, "422", "application/json"):
+            response_data = unmarshal_json_response(
+                errors.HTTPValidationErrorData, http_res
+            )
+            raise errors.HTTPValidationError(response_data, http_res)
+        if utils.match_response(http_res, "4XX", "*"):
+            http_res_text = await utils.stream_to_text_async(http_res)
+            raise errors.APIError("API error occurred", http_res, http_res_text)
+        if utils.match_response(http_res, "5XX", "*"):
+            http_res_text = await utils.stream_to_text_async(http_res)
+            raise errors.APIError("API error occurred", http_res, http_res_text)
+
+        raise errors.APIError("Unexpected response received", http_res)