async-hyperliquid 0.4.4__tar.gz → 0.4.6__tar.gz

{async_hyperliquid-0.4.4 → async_hyperliquid-0.4.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: async-hyperliquid
-Version: 0.4.4
+Version: 0.4.6
 Summary: Async Hyperliquid client using aiohttp
 Keywords: dex,hyperliquid,async,aiohttp,trading,cryptocurrency,defi
 Author: Yuki

{async_hyperliquid-0.4.4 → async_hyperliquid-0.4.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "async-hyperliquid"
-version = "0.4.4"
+version = "0.4.6"
 description = "Async Hyperliquid client using aiohttp"
 authors = [{ name = "Yuki", email = "yuqi.lyle@gmail.com" }]
 readme = "README.md"

{async_hyperliquid-0.4.4 → async_hyperliquid-0.4.6}/src/async_hyperliquid/_async_hyperliquid/core.py

@@ -34,6 +34,9 @@ class AsyncHyperliquidCore(AsyncAPI):
     coin_symbols: dict[str, str]
     asset_sz_decimals: dict[int, int]
     spot_tokens: dict[str, SpotTokenMeta]
+    _meta_init_lock: asyncio.Lock
+    _meta_init_task: asyncio.Task[None] | None
+    _metas_initialized: bool
 
     perp_dexs: list[str]
 
@@ -79,6 +82,9 @@ class AsyncHyperliquidCore(AsyncAPI):
         self.coin_symbols = {}
         self.asset_sz_decimals = {}
         self.spot_tokens = {}
+        self._meta_init_lock = asyncio.Lock()
+        self._meta_init_task = None
+        self._metas_initialized = False
 
         self.vault = vault
         self.expires: int | None = None
@@ -124,23 +130,50 @@ class AsyncHyperliquidCore(AsyncAPI):
 
         self.evm_exchange = EVMExchange(rpc_url, private_key)
 
-    def _init_perp_meta(self, meta: PerpMeta, offset: int) -> None:
+    def _init_perp_meta(
+        self,
+        meta: PerpMeta,
+        offset: int,
+        *,
+        coin_assets: dict[str, int] | None = None,
+        coin_names: dict[str, str] | None = None,
+        asset_sz_decimals: dict[int, int] | None = None,
+    ) -> None:
+        coin_assets = self.coin_assets if coin_assets is None else coin_assets
+        coin_names = self.coin_names if coin_names is None else coin_names
+        asset_sz_decimals = (
+            self.asset_sz_decimals if asset_sz_decimals is None else asset_sz_decimals
+        )
         for asset, info in enumerate(meta["universe"]):
             asset += offset
             asset_name = info["name"]
-            self.coin_assets[asset_name] = asset
-            self.coin_names[asset_name] = asset_name
-            self.asset_sz_decimals[asset] = info["szDecimals"]
+            coin_assets[asset_name] = asset
+            coin_names[asset_name] = asset_name
+            asset_sz_decimals[asset] = info["szDecimals"]
 
-    def _init_spot_meta(self, meta: SpotMeta) -> None:
+    def _init_spot_meta(
+        self,
+        meta: SpotMeta,
+        *,
+        coin_assets: dict[str, int] | None = None,
+        coin_names: dict[str, str] | None = None,
+        asset_sz_decimals: dict[int, int] | None = None,
+        spot_tokens: dict[str, SpotTokenMeta] | None = None,
+    ) -> None:
+        coin_assets = self.coin_assets if coin_assets is None else coin_assets
+        coin_names = self.coin_names if coin_names is None else coin_names
+        asset_sz_decimals = (
+            self.asset_sz_decimals if asset_sz_decimals is None else asset_sz_decimals
+        )
+        spot_tokens = self.spot_tokens if spot_tokens is None else spot_tokens
         tokens = meta["tokens"]
         total_tokens = len(tokens)
         for info in meta["universe"]:
             asset = info["index"] + SPOT_OFFSET
             asset_name = info["name"]
 
-            self.coin_assets[asset_name] = asset
-            self.coin_names[asset_name] = asset_name
+            coin_assets[asset_name] = asset
+            coin_names[asset_name] = asset_name
 
             base, quote = info["tokens"]
             if not 0 <= base < total_tokens or not 0 <= quote < total_tokens:
@@ -151,19 +184,65 @@ class AsyncHyperliquidCore(AsyncAPI):
             quote_info = tokens[quote]
             quote_name = quote_info["name"]
             name = f"{base_name}/{quote_name}"
-            self.coin_names.setdefault(name, asset_name)
-            self.coin_names.setdefault(quote_name, quote_name)
-
-            self.asset_sz_decimals[asset] = base_info["szDecimals"]
-            self.spot_tokens[asset_name] = base_info
-            self.spot_tokens.setdefault(quote_name, quote_info)
-
-    def _update_coin_symbols(self) -> None:
-        self.coin_symbols = {
-            v: k for k, v in self.coin_names.items() if not k.startswith("@")
-        }
-
-    async def init_metas(self) -> None:
+            coin_names.setdefault(name, asset_name)
+            coin_names.setdefault(quote_name, quote_name)
+
+            asset_sz_decimals[asset] = base_info["szDecimals"]
+            spot_tokens[asset_name] = base_info
+            spot_tokens.setdefault(quote_name, quote_info)
+
+    def _build_coin_symbols(self, coin_names: dict[str, str]) -> dict[str, str]:
+        return {v: k for k, v in coin_names.items() if not k.startswith("@")}
+
+    def _get_meta_init_lock(self) -> asyncio.Lock:
+        lock = getattr(self, "_meta_init_lock", None)
+        if lock is None:
+            lock = asyncio.Lock()
+            self._meta_init_lock = lock
+        return lock
+
+    def _lookup_cached_coin_name(self, coin: str) -> str | None:
+        coin_names = getattr(self, "coin_names", {})
+        if coin in coin_names:
+            return coin_names[coin]
+
+        coin_assets = getattr(self, "coin_assets", {})
+        if coin in coin_assets:
+            return coin
+
+        return None
+
+    def _lookup_cached_asset_id(self, coin: str) -> int | None:
+        coin_assets = getattr(self, "coin_assets", {})
+        asset = coin_assets.get(coin)
+        if asset is not None:
+            return asset
+
+        coin_name = getattr(self, "coin_names", {}).get(coin)
+        if coin_name is None:
+            return None
+
+        return coin_assets.get(coin_name)
+
+    def _lookup_cached_asset(self, coin: str) -> tuple[str, int] | None:
+        asset = self._lookup_cached_asset_id(coin)
+        if asset is None:
+            return None
+
+        coin_name = self._lookup_cached_coin_name(coin)
+        if coin_name is None:
+            return None
+
+        return coin_name, asset
+
+    async def _refresh_metas(self) -> None:
+        # TODO: Add HIP-4 outcome meta initialization from the spot info
+        # `outcomeMeta` endpoint once outcomes move beyond testnet-only rollout.
+        # Outcome asset IDs do not follow the current perp/spot offset scheme:
+        # `encoding = 10 * outcome + side`, coin names use `#{encoding}`, token
+        # names use `+{encoding}`, and `asset_id = 100_000_000 + encoding`, so
+        # this path will need dedicated outcome mappings instead of reusing the
+        # existing perp/spot logic.
         meta_task = self.info.get_perp_meta()
         spot_meta_task = self.info.get_spot_meta()
         all_dex_names_task = self.get_all_dex_name()
@@ -172,17 +251,36 @@ class AsyncHyperliquidCore(AsyncAPI):
             meta_task, spot_meta_task, all_dex_names_task
         )
 
-        self._init_perp_meta(meta, 0)
-        self._init_spot_meta(spot_meta)
+        coin_assets: dict[str, int] = {}
+        coin_names: dict[str, str] = {}
+        asset_sz_decimals: dict[int, int] = {}
+        spot_tokens: dict[str, SpotTokenMeta] = {}
+
+        self._init_perp_meta(
+            meta,
+            0,
+            coin_assets=coin_assets,
+            coin_names=coin_names,
+            asset_sz_decimals=asset_sz_decimals,
+        )
+        self._init_spot_meta(
+            spot_meta,
+            coin_assets=coin_assets,
+            coin_names=coin_names,
+            asset_sz_decimals=asset_sz_decimals,
+            spot_tokens=spot_tokens,
+        )
 
+        dex_index_by_name = {
+            name: idx for idx, name in enumerate(all_dex_names) if name is not None
+        }
         dex_meta_tasks = []
         dex_indices = []
         for dex in self.perp_dexs:
             if dex == "":
                 continue
-            try:
-                idx = all_dex_names.index(dex)
-            except ValueError:
+            idx = dex_index_by_name.get(dex)
+            if idx is None:
                 continue
 
             if idx > 0:
@@ -193,9 +291,49 @@ class AsyncHyperliquidCore(AsyncAPI):
             dex_metas = await asyncio.gather(*dex_meta_tasks)
             for idx, dex_meta in zip(dex_indices, dex_metas):
                 dex_asset_offset = PERP_DEX_OFFSET + (idx - 1) * 10000
-                self._init_perp_meta(dex_meta, dex_asset_offset)
+                self._init_perp_meta(
+                    dex_meta,
+                    dex_asset_offset,
+                    coin_assets=coin_assets,
+                    coin_names=coin_names,
+                    asset_sz_decimals=asset_sz_decimals,
+                )
+
+        self.coin_assets = coin_assets
+        self.coin_names = coin_names
+        self.asset_sz_decimals = asset_sz_decimals
+        self.spot_tokens = spot_tokens
+        self.coin_symbols = self._build_coin_symbols(coin_names)
+        self._metas_initialized = True
+
+    async def _run_meta_refresh(self, *, only_if_missing: bool) -> None:
+        if only_if_missing and getattr(self, "_metas_initialized", False):
+            return
+
+        task = getattr(self, "_meta_init_task", None)
+        if task is not None and not task.done():
+            await task
+            return
+
+        async with self._get_meta_init_lock():
+            task = getattr(self, "_meta_init_task", None)
+            if task is None or task.done():
+                if only_if_missing and getattr(self, "_metas_initialized", False):
+                    return
+                task = asyncio.create_task(self._refresh_metas())
+                self._meta_init_task = task
+
+        try:
+            await task
+        finally:
+            if getattr(self, "_meta_init_task", None) is task and task.done():
+                self._meta_init_task = None
+
+    async def _ensure_metas_initialized(self) -> None:
+        await self._run_meta_refresh(only_if_missing=True)
 
-        self._update_coin_symbols()
+    async def init_metas(self) -> None:
+        await self._run_meta_refresh(only_if_missing=False)
 
     async def get_metas(self, perp_only: bool = False) -> Metas:
         metas: Metas = {"perp": {}, "spot": [], "dexs": {}}  # type: ignore
@@ -237,31 +375,45 @@ class AsyncHyperliquidCore(AsyncAPI):
         return names
 
     async def get_coin_name(self, coin: str) -> str:
-        if not hasattr(self, "coin_names") or coin not in self.coin_names:
-            await self.init_metas()
+        coin_name = self._lookup_cached_coin_name(coin)
+        if coin_name is not None:
+            return coin_name
 
-        if coin not in self.coin_names:
+        if getattr(self, "_metas_initialized", False):
+            await self.init_metas()
+        else:
+            await self._ensure_metas_initialized()
+        coin_name = self._lookup_cached_coin_name(coin)
+        if coin_name is None:
             raise ValueError(f"Coin {coin} not found")
 
-        return self.coin_names[coin]
+        return coin_name
 
     async def get_coin_asset(self, coin: str) -> int:
+        cached = self._lookup_cached_asset(coin)
+        if cached is not None:
+            return cached[1]
+
         coin_name = await self.get_coin_name(coin)
-        if coin_name not in self.coin_assets:
+        asset = self.coin_assets.get(coin_name)
+        if asset is None:
             raise ValueError(f"Coin {coin}({coin_name}) not found")
 
-        return self.coin_assets[coin_name]
+        return asset
 
     async def get_coin_symbol(self, coin: str) -> str:
         coin_name = await self.get_coin_name(coin)
         return self.coin_symbols[coin_name]
 
     async def get_coin_sz_decimals(self, coin: str) -> int:
-        coin_name = await self.get_coin_name(coin)
-        if coin_name not in self.coin_assets:
-            raise ValueError(f"Coin {coin}({coin_name}) not found")
-
-        asset = self.coin_assets[coin_name]
+        cached = self._lookup_cached_asset(coin)
+        if cached is not None:
+            _, asset = cached
+        else:
+            coin_name = await self.get_coin_name(coin)
+            asset = self.coin_assets.get(coin_name)
+            if asset is None:
+                raise ValueError(f"Coin {coin}({coin_name}) not found")
         return self.asset_sz_decimals[asset]
 
     async def get_token_info(self, coin: str) -> SpotTokenMeta:

{async_hyperliquid-0.4.4 → async_hyperliquid-0.4.6}/src/async_hyperliquid/_async_hyperliquid/info.py

@@ -1,6 +1,6 @@
 import asyncio
 import warnings
-from typing import Literal
+from typing import Literal, cast
 
 from async_hyperliquid.utils.constants import ONE_HOUR_MS, PERP_DEX_OFFSET, SPOT_OFFSET
 from async_hyperliquid.utils.miscs import get_coin_dex, get_timestamp_ms
@@ -9,9 +9,13 @@ from async_hyperliquid.utils.types import (
     AccountState,
     ClearinghouseState,
     OrderWithStatus,
+    PerpMeta,
+    PerpMetaCtxItem,
     Portfolio,
     Position,
+    SpotMeta,
     SpotClearinghouseState,
+    SpotMetaCtxItem,
     UserDeposit,
     UserNonFundingDelta,
     UserOpenOrders,
@@ -23,13 +27,96 @@ from .core import AsyncHyperliquidCore
 
 
 class AsyncHyperliquidInfoClient(AsyncHyperliquidCore):
+    def _get_cached_perp_ctx_index(self, coin_name: str) -> tuple[str, int] | None:
+        asset = self._lookup_cached_asset_id(coin_name)
+        if asset is None:
+            return None
+
+        if asset < SPOT_OFFSET:
+            return get_coin_dex(coin_name), asset
+
+        if asset >= PERP_DEX_OFFSET:
+            return get_coin_dex(coin_name), (asset - PERP_DEX_OFFSET) % 10000
+
+        return None
+
+    def _get_cached_spot_ctx_index(self, coin_name: str) -> int | None:
+        asset = self._lookup_cached_asset_id(coin_name)
+        if asset is None:
+            return None
+
+        if SPOT_OFFSET <= asset < PERP_DEX_OFFSET:
+            return asset - SPOT_OFFSET
+
+        return None
+
+    async def _get_perp_mark_price(
+        self, coin_name: str, dex: str = "", asset_index: int | None = None
+    ) -> float:
+        meta_ctx = await self.info.get_perp_meta_ctx(dex)
+        meta = cast(PerpMeta, meta_ctx[0])
+        asset_ctxs = cast(list[PerpMetaCtxItem], meta_ctx[1])
+
+        if asset_index is not None and 0 <= asset_index < len(asset_ctxs):
+            return float(asset_ctxs[asset_index]["markPx"])
+
+        for idx, info in enumerate(meta["universe"]):
+            if info["name"] == coin_name:
+                return float(asset_ctxs[idx]["markPx"])
+
+        raise ValueError(f"Coin {coin_name} not found in perp dex '{dex}'")
+
+    async def _get_spot_mark_price(
+        self, coin_name: str, asset_index: int | None = None
+    ) -> float:
+        meta_ctx = await self.info.get_spot_meta_ctx()
+        meta = cast(SpotMeta, meta_ctx[0])
+        asset_ctxs = cast(list[SpotMetaCtxItem], meta_ctx[1])
+
+        if asset_index is not None and 0 <= asset_index < len(asset_ctxs):
+            return float(asset_ctxs[asset_index]["markPx"])
+
+        for info in meta["universe"]:
+            if info["name"] == coin_name:
+                asset_index = info["index"]
+                return float(asset_ctxs[asset_index]["markPx"])
+
+        raise ValueError(f"Coin {coin_name} not found in spot metadata")
+
+    async def get_mark_price(self, coin: str) -> float:
+        if ":" in coin:
+            cached = self._get_cached_perp_ctx_index(coin)
+            if cached is not None:
+                dex, asset_index = cached
+                return await self._get_perp_mark_price(
+                    coin, dex, asset_index=asset_index
+                )
+            return await self._get_perp_mark_price(coin, get_coin_dex(coin))
+
+        coin_name = await self.get_coin_name(coin)
+        asset = self._lookup_cached_asset_id(coin_name)
+        if asset is None:
+            raise ValueError(f"Coin {coin}({coin_name}) not found")
+
+        is_spot_asset = SPOT_OFFSET <= asset < PERP_DEX_OFFSET
+        if is_spot_asset:
+            return await self._get_spot_mark_price(
+                coin_name, asset_index=asset - SPOT_OFFSET
+            )
+
+        return await self._get_perp_mark_price(
+            coin_name,
+            get_coin_dex(coin_name),
+            asset_index=(
+                asset if asset < SPOT_OFFSET else (asset - PERP_DEX_OFFSET) % 10000
+            ),
+        )
+
     async def get_market_price(self, coin: str) -> float:
         warnings.warn(
             "get_market_price is deprecated and will remove in the future, use get_mid_price instead"
         )
-        coin_name = await self.get_coin_name(coin)
-        market_prices = await self.get_all_market_prices()
-        return market_prices[coin_name]
+        return await self.get_mark_price(coin)
 
     async def get_all_market_prices(
         self, market: Literal["spot", "perp", "all"] = "all"

{async_hyperliquid-0.4.4 → async_hyperliquid-0.4.6}/src/async_hyperliquid/_async_hyperliquid/orders.py

@@ -20,14 +20,33 @@ from .info import AsyncHyperliquidInfoClient
 
 
 class AsyncHyperliquidOrdersClient(AsyncHyperliquidInfoClient):
+    def _round_sz_px_cached(
+        self, coin: str, sz: float, px: float
+    ) -> tuple[int, float, float | int] | None:
+        asset = self._lookup_cached_asset_id(coin)
+        if asset is None:
+            return None
+
+        sz_decimals = self.asset_sz_decimals.get(asset)
+        if sz_decimals is None:
+            return None
+
+        is_spot = SPOT_OFFSET <= asset < PERP_DEX_OFFSET
+        px_decimals = (6 if not is_spot else 8) - sz_decimals
+        return asset, round_float(sz, sz_decimals), round_px(px, px_decimals)
+
     async def _round_sz_px(self, coin: str, sz: float, px: float):
+        cached = self._round_sz_px_cached(coin, sz, px)
+        if cached is not None:
+            return cached
+
         coin_name = await self.get_coin_name(coin)
-        if coin_name not in self.coin_assets:
+        asset = self.coin_assets.get(coin_name)
+        if asset is None:
             raise ValueError(f"Coin {coin}({coin_name}) not found")
 
-        asset = self.coin_assets[coin_name]
         sz_decimals = self.asset_sz_decimals[asset]
-        is_spot = asset >= SPOT_OFFSET and asset < PERP_DEX_OFFSET
+        is_spot = SPOT_OFFSET <= asset < PERP_DEX_OFFSET
         px_decimals = (6 if not is_spot else 8) - sz_decimals
         return asset, round_float(sz, sz_decimals), round_px(px, px_decimals)
 
@@ -155,12 +174,21 @@ class AsyncHyperliquidOrdersClient(AsyncHyperliquidInfoClient):
         return await self.place_orders(reqs, grouping=grouping, builder=builder)
 
     async def _get_batch_limit_orders(self, orders: BatchPlaceOrderRequest):
+        rounded_orders = [
+            self._round_sz_px_cached(o["coin"], o["sz"], o["px"]) for o in orders
+        ]
+        if all(rounded_order is not None for rounded_order in rounded_orders):
+            return [
+                {**order, "asset": asset, "sz": sz, "px": px}
+                for order, (asset, sz, px) in zip(orders, rounded_orders, strict=True)
+            ]
+
         rounded_orders = await asyncio.gather(
             *(self._round_sz_px(o["coin"], o["sz"], o["px"]) for o in orders)
         )
         return [
             {**order, "asset": asset, "sz": sz, "px": px}
-            for order, (asset, sz, px) in zip(orders, rounded_orders)
+            for order, (asset, sz, px) in zip(orders, rounded_orders, strict=True)
         ]
 
     async def _get_batch_market_orders(
@@ -175,15 +203,25 @@ class AsyncHyperliquidOrdersClient(AsyncHyperliquidInfoClient):
             slippage_factor = (1 + slippage) if order["is_buy"] else (1 - slippage)
             quoted_prices.append(market_price * slippage_factor)
 
+        rounded_orders = [
+            self._round_sz_px_cached(order["coin"], order["sz"], quoted_price)
+            for order, quoted_price in zip(orders, quoted_prices, strict=True)
+        ]
+        if all(rounded_order is not None for rounded_order in rounded_orders):
+            return [
+                {**order, "asset": asset, "sz": sz, "px": px, "order_type": order_type}
+                for order, (asset, sz, px) in zip(orders, rounded_orders, strict=True)
+            ]
+
         rounded_orders = await asyncio.gather(
             *(
                 self._round_sz_px(order["coin"], order["sz"], quoted_price)
-                for order, quoted_price in zip(orders, quoted_prices)
+                for order, quoted_price in zip(orders, quoted_prices, strict=True)
             )
         )
         return [
             {**order, "asset": asset, "sz": sz, "px": px, "order_type": order_type}
-            for order, (asset, sz, px) in zip(orders, rounded_orders)
+            for order, (asset, sz, px) in zip(orders, rounded_orders, strict=True)
         ]
 
     async def cancel_order(self, coin: str, oid: int):
@@ -193,9 +231,15 @@ class AsyncHyperliquidOrdersClient(AsyncHyperliquidInfoClient):
         return await self.cancel_orders(cancels)
 
     async def cancel_orders(self, cancels: BatchCancelRequest):
-        assets = await asyncio.gather(
-            *(self.get_coin_asset(coin) for coin, _ in cancels)
-        )
+        assets = [
+            asset
+            for coin, _ in cancels
+            if (asset := self._lookup_cached_asset_id(coin)) is not None
+        ]
+        if len(assets) != len(cancels):
+            assets = await asyncio.gather(
+                *(self.get_coin_asset(coin) for coin, _ in cancels)
+            )
         action = {
             "type": "cancel",
             "cancels": [
@@ -212,9 +256,15 @@ class AsyncHyperliquidOrdersClient(AsyncHyperliquidInfoClient):
         return await self.batch_cancel_by_cloid([(coin, cloid)])
 
     async def batch_cancel_by_cloid(self, cancels: list[tuple[str, Cloid]]):
-        assets = await asyncio.gather(
-            *(self.get_coin_asset(coin) for coin, _ in cancels)
-        )
+        assets = [
+            asset
+            for coin, _ in cancels
+            if (asset := self._lookup_cached_asset_id(coin)) is not None
+        ]
+        if len(assets) != len(cancels):
+            assets = await asyncio.gather(
+                *(self.get_coin_asset(coin) for coin, _ in cancels)
+            )
         action = {
             "type": "cancelByCloid",
             "cancels": [

{async_hyperliquid-0.4.4 → async_hyperliquid-0.4.6}/src/async_hyperliquid/info.py

@@ -190,8 +190,8 @@ class InfoAPI(AsyncAPI):
         payload = {"type": "meta", "dex": dex}
         return await self.post(payload)
 
-    async def get_perp_meta_ctx(self) -> PerpMetaCtx:
-        payload = {"type": "metaAndAssetCtxs"}
+    async def get_perp_meta_ctx(self, dex: str = "") -> PerpMetaCtx:
+        payload = {"type": "metaAndAssetCtxs", "dex": dex}
         return await self.post(payload)
 
     async def get_perp_dexs(self) -> PerpDexs:

{async_hyperliquid-0.4.4 → async_hyperliquid-0.4.6}/src/async_hyperliquid/utils/signing.py

@@ -1,5 +1,5 @@
+import math
 from typing import Any, List
-from decimal import Decimal
 
 import msgpack
 from eth_utils.crypto import keccak
@@ -174,14 +174,17 @@ def sign_action(
 
 
 def round_float(x: float) -> str:
+    if not math.isfinite(x):
+        raise ValueError("round_float requires finite number", x)
+
     rounded = f"{x:.8f}"
     if abs(float(rounded) - x) >= 1e-12:
         raise ValueError("round_float causes rounding", x)
 
-    if rounded == "-0":
-        rounded = "0"
-    normalized = Decimal(rounded).normalize()
-    return f"{normalized:f}"
+    trimmed = rounded.rstrip("0").rstrip(".")
+    if trimmed in {"", "-0"}:
+        return "0"
+    return trimmed
 
 
 def ensure_order_type(order_type: OrderType) -> OrderType: