architect-py 5.0.0b3__py3-none-any.whl → 5.1.0b1__py3-none-any.whl

architect_py/graphql_client/client.py

@@ -1,60 +1,22 @@
 # Generated by ariadne-codegen
 # Source: queries.graphql
 
-from datetime import datetime
-from decimal import Decimal
-from typing import TYPE_CHECKING, Any, AsyncIterator, Dict, List, Optional, Union
-from uuid import UUID
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
 
-from architect_py.common_types import OrderDir, TradableProduct
-from architect_py.common_types.order_dir import graphql_serialize_order_dir
-from architect_py.common_types.scalars import convert_datetime_to_utc_str
+from architect_py.common_types import TradableProduct
 
 from .base_model import UNSET
 from .juniper_base_client import JuniperBaseClient
 
 if TYPE_CHECKING:
     from .base_model import UnsetType
-    from .cancel_all_orders_mutation import CancelAllOrdersMutationOms
-    from .cancel_order_mutation import CancelOrderMutationOms
-    from .create_jwt import CreateJwtUser
-    from .enums import CandleWidth, OrderType, TimeInForce
-    from .get_account_history_query import GetAccountHistoryQueryFolio
-    from .get_account_query import GetAccountQueryUser
-    from .get_account_summaries_query import GetAccountSummariesQueryFolio
-    from .get_account_summary_query import GetAccountSummaryQueryFolio
-    from .get_candle_snapshot_query import GetCandleSnapshotQueryMarketdata
     from .get_execution_info_query import GetExecutionInfoQuerySymbology
     from .get_execution_infos_query import GetExecutionInfosQuerySymbology
-    from .get_fills_query import GetFillsQueryFolio
     from .get_first_notice_date_query import GetFirstNoticeDateQuerySymbology
     from .get_future_series_query import GetFutureSeriesQuerySymbology
-    from .get_historical_orders_query import GetHistoricalOrdersQueryFolio
-    from .get_l_1_book_snapshot_query import GetL1BookSnapshotQueryMarketdata
-    from .get_l_1_book_snapshots_query import GetL1BookSnapshotsQueryMarketdata
-    from .get_l_2_book_snapshot_query import GetL2BookSnapshotQueryMarketdata
-    from .get_market_status_query import GetMarketStatusQueryMarketdata
-    from .get_open_orders_query import GetOpenOrdersQueryOms
     from .get_product_info_query import GetProductInfoQuerySymbology
     from .get_product_infos_query import GetProductInfosQuerySymbology
-    from .list_accounts_query import ListAccountsQueryUser
-    from .place_order_mutation import PlaceOrderMutationOms
     from .search_symbols_query import SearchSymbolsQuerySymbology
-    from .subscribe_candles import SubscribeCandlesCandles
-    from .subscribe_orderflow import (
-        SubscribeOrderflowOrderflowAberrantFill,
-        SubscribeOrderflowOrderflowCancel,
-        SubscribeOrderflowOrderflowCancelReject,
-        SubscribeOrderflowOrderflowFill,
-        SubscribeOrderflowOrderflowGqlOrderReject,
-        SubscribeOrderflowOrderflowOrder,
-        SubscribeOrderflowOrderflowOrderAck,
-        SubscribeOrderflowOrderflowOrderCanceled,
-        SubscribeOrderflowOrderflowOrderCanceling,
-        SubscribeOrderflowOrderflowOrderOut,
-        SubscribeOrderflowOrderflowOrderStale,
-    )
-    from .subscribe_trades import SubscribeTradesTrades
     from .user_id_query import UserIdQueryUser
 
 
@@ -348,1006 +310,3 @@ class GraphQLClient(JuniperBaseClient):
         )
         data = self.get_data(response)
         return GetExecutionInfosQuery.model_validate(data).symbology
-
-    async def get_candle_snapshot_query(
-        self,
-        venue: str,
-        symbol: str,
-        candle_width: "CandleWidth",
-        start: datetime,
-        end: datetime,
-        **kwargs: Any
-    ) -> "GetCandleSnapshotQueryMarketdata":
-        from .get_candle_snapshot_query import GetCandleSnapshotQuery
-
-        query = gql(
-            """
-            query GetCandleSnapshotQuery($venue: MarketdataVenue!, $symbol: String!, $candleWidth: CandleWidth!, $start: DateTime!, $end: DateTime!) {
-              marketdata {
-                historicalCandles(
-                  venue: $venue
-                  symbol: $symbol
-                  candleWidth: $candleWidth
-                  start: $start
-                  end: $end
-                ) {
-                  ...CandleFields
-                }
-              }
-            }
-
-            fragment CandleFields on Candle {
-              timestamp
-              width
-              open
-              high
-              low
-              close
-              volume
-            }
-            """
-        )
-        variables: Dict[str, object] = {
-            "venue": venue,
-            "symbol": symbol,
-            "candleWidth": candle_width,
-            "start": convert_datetime_to_utc_str(start),
-            "end": convert_datetime_to_utc_str(end),
-        }
-        response = await self.execute(
-            query=query,
-            operation_name="GetCandleSnapshotQuery",
-            variables=variables,
-            **kwargs
-        )
-        data = self.get_data(response)
-        return GetCandleSnapshotQuery.model_validate(data).marketdata
-
-    async def get_l_2_book_snapshot_query(
-        self,
-        symbol: str,
-        venue: Union[Optional[str], "UnsetType"] = UNSET,
-        **kwargs: Any
-    ) -> "GetL2BookSnapshotQueryMarketdata":
-        from .get_l_2_book_snapshot_query import GetL2BookSnapshotQuery
-
-        query = gql(
-            """
-            query GetL2BookSnapshotQuery($venue: MarketdataVenue, $symbol: String!) {
-              marketdata {
-                l2BookSnapshot(venue: $venue, symbol: $symbol) {
-                  ...L2BookFields
-                }
-              }
-            }
-
-            fragment L2BookFields on L2Book {
-              timestamp
-              bids {
-                ...L2BookLevelFields
-              }
-              asks {
-                ...L2BookLevelFields
-              }
-            }
-
-            fragment L2BookLevelFields on L2BookLevel {
-              price
-              size
-            }
-            """
-        )
-        variables: Dict[str, object] = {"venue": venue, "symbol": symbol}
-        response = await self.execute(
-            query=query,
-            operation_name="GetL2BookSnapshotQuery",
-            variables=variables,
-            **kwargs
-        )
-        data = self.get_data(response)
-        return GetL2BookSnapshotQuery.model_validate(data).marketdata
-
-    async def get_l_1_book_snapshot_query(
-        self,
-        symbol: str,
-        venue: Union[Optional[str], "UnsetType"] = UNSET,
-        **kwargs: Any
-    ) -> "GetL1BookSnapshotQueryMarketdata":
-        from .get_l_1_book_snapshot_query import GetL1BookSnapshotQuery
-
-        query = gql(
-            """
-            query GetL1BookSnapshotQuery($venue: MarketdataVenue, $symbol: String!) {
-              marketdata {
-                ticker(venue: $venue, symbol: $symbol) {
-                  ...MarketTickerFields
-                }
-              }
-            }
-
-            fragment MarketTickerFields on Ticker {
-              symbol
-              timestamp
-              bidPrice
-              bidSize
-              askPrice
-              askSize
-              lastPrice
-              lastSize
-            }
-            """
-        )
-        variables: Dict[str, object] = {"venue": venue, "symbol": symbol}
-        response = await self.execute(
-            query=query,
-            operation_name="GetL1BookSnapshotQuery",
-            variables=variables,
-            **kwargs
-        )
-        data = self.get_data(response)
-        return GetL1BookSnapshotQuery.model_validate(data).marketdata
-
-    async def get_l_1_book_snapshots_query(
-        self,
-        venue: str,
-        symbols: Union[Optional[List[str]], "UnsetType"] = UNSET,
-        **kwargs: Any
-    ) -> "GetL1BookSnapshotsQueryMarketdata":
-        from .get_l_1_book_snapshots_query import GetL1BookSnapshotsQuery
-
-        query = gql(
-            """
-            query GetL1BookSnapshotsQuery($venue: MarketdataVenue!, $symbols: [String!]) {
-              marketdata {
-                tickers(venue: $venue, symbols: $symbols) {
-                  ...MarketTickerFields
-                }
-              }
-            }
-
-            fragment MarketTickerFields on Ticker {
-              symbol
-              timestamp
-              bidPrice
-              bidSize
-              askPrice
-              askSize
-              lastPrice
-              lastSize
-            }
-            """
-        )
-        variables: Dict[str, object] = {"venue": venue, "symbols": symbols}
-        response = await self.execute(
-            query=query,
-            operation_name="GetL1BookSnapshotsQuery",
-            variables=variables,
-            **kwargs
-        )
-        data = self.get_data(response)
-        return GetL1BookSnapshotsQuery.model_validate(data).marketdata
-
-    async def get_market_status_query(
-        self,
-        symbol: str,
-        venue: Union[Optional[str], "UnsetType"] = UNSET,
-        **kwargs: Any
-    ) -> "GetMarketStatusQueryMarketdata":
-        from .get_market_status_query import GetMarketStatusQuery
-
-        query = gql(
-            """
-            query GetMarketStatusQuery($venue: MarketdataVenue, $symbol: String!) {
-              marketdata {
-                marketStatus(venue: $venue, symbol: $symbol) {
-                  ...MarketStatusFields
-                }
-              }
-            }
-
-            fragment MarketStatusFields on MarketStatus {
-              symbol
-              isTrading
-              isQuoting
-            }
-            """
-        )
-        variables: Dict[str, object] = {"venue": venue, "symbol": symbol}
-        response = await self.execute(
-            query=query,
-            operation_name="GetMarketStatusQuery",
-            variables=variables,
-            **kwargs
-        )
-        data = self.get_data(response)
-        return GetMarketStatusQuery.model_validate(data).marketdata
-
-    async def get_account_query(
-        self,
-        id: Union[Optional[UUID], "UnsetType"] = UNSET,
-        name: Union[Optional[str], "UnsetType"] = UNSET,
-        **kwargs: Any
-    ) -> "GetAccountQueryUser":
-        from .get_account_query import GetAccountQuery
-
-        query = gql(
-            """
-            query GetAccountQuery($id: Uuid, $name: AccountName) {
-              user {
-                account(id: $id, name: $name) {
-                  ...AccountWithPermissionsFields
-                }
-              }
-            }
-
-            fragment AccountWithPermissionsFields on AccountWithPermissions {
-              account {
-                id
-                name
-              }
-              trader
-              permissions {
-                list
-                view
-                trade
-                reduceOrClose
-                setLimits
-              }
-            }
-            """
-        )
-        variables: Dict[str, object] = {"id": id, "name": name}
-        response = await self.execute(
-            query=query, operation_name="GetAccountQuery", variables=variables, **kwargs
-        )
-        data = self.get_data(response)
-        return GetAccountQuery.model_validate(data).user
-
-    async def list_accounts_query(self, **kwargs: Any) -> "ListAccountsQueryUser":
-        from .list_accounts_query import ListAccountsQuery
-
-        query = gql(
-            """
-            query ListAccountsQuery {
-              user {
-                accounts {
-                  ...AccountWithPermissionsFields
-                }
-              }
-            }
-
-            fragment AccountWithPermissionsFields on AccountWithPermissions {
-              account {
-                id
-                name
-              }
-              trader
-              permissions {
-                list
-                view
-                trade
-                reduceOrClose
-                setLimits
-              }
-            }
-            """
-        )
-        variables: Dict[str, object] = {}
-        response = await self.execute(
-            query=query,
-            operation_name="ListAccountsQuery",
-            variables=variables,
-            **kwargs
-        )
-        data = self.get_data(response)
-        return ListAccountsQuery.model_validate(data).user
-
-    async def get_account_summary_query(
-        self, account: str, **kwargs: Any
-    ) -> "GetAccountSummaryQueryFolio":
-        from .get_account_summary_query import GetAccountSummaryQuery
-
-        query = gql(
-            """
-            query GetAccountSummaryQuery($account: String!) {
-              folio {
-                accountSummary(account: $account) {
-                  ...AccountSummaryFields
-                }
-              }
-            }
-
-            fragment AccountSummaryFields on AccountSummary {
-              account
-              timestamp
-              balances {
-                product
-                balance
-              }
-              positions {
-                symbol
-                quantity
-                tradeTime
-                costBasis
-                breakEvenPrice
-                liquidationPrice
-              }
-              unrealizedPnl
-              realizedPnl
-              equity
-              yesterdayEquity
-              cashExcess
-              purchasingPower
-              totalMargin
-              positionMargin
-            }
-            """
-        )
-        variables: Dict[str, object] = {"account": account}
-        response = await self.execute(
-            query=query,
-            operation_name="GetAccountSummaryQuery",
-            variables=variables,
-            **kwargs
-        )
-        data = self.get_data(response)
-        return GetAccountSummaryQuery.model_validate(data).folio
-
-    async def get_account_summaries_query(
-        self,
-        trader: Union[Optional[str], "UnsetType"] = UNSET,
-        accounts: Union[Optional[List[str]], "UnsetType"] = UNSET,
-        **kwargs: Any
-    ) -> "GetAccountSummariesQueryFolio":
-        from .get_account_summaries_query import GetAccountSummariesQuery
-
-        query = gql(
-            """
-            query GetAccountSummariesQuery($trader: String, $accounts: [String!]) {
-              folio {
-                accountSummaries(trader: $trader, accounts: $accounts) {
-                  ...AccountSummaryFields
-                }
-              }
-            }
-
-            fragment AccountSummaryFields on AccountSummary {
-              account
-              timestamp
-              balances {
-                product
-                balance
-              }
-              positions {
-                symbol
-                quantity
-                tradeTime
-                costBasis
-                breakEvenPrice
-                liquidationPrice
-              }
-              unrealizedPnl
-              realizedPnl
-              equity
-              yesterdayEquity
-              cashExcess
-              purchasingPower
-              totalMargin
-              positionMargin
-            }
-            """
-        )
-        variables: Dict[str, object] = {"trader": trader, "accounts": accounts}
-        response = await self.execute(
-            query=query,
-            operation_name="GetAccountSummariesQuery",
-            variables=variables,
-            **kwargs
-        )
-        data = self.get_data(response)
-        return GetAccountSummariesQuery.model_validate(data).folio
-
-    async def get_account_history_query(
-        self,
-        account: str,
-        from_inclusive: Union[Optional[datetime], "UnsetType"] = UNSET,
-        to_exclusive: Union[Optional[datetime], "UnsetType"] = UNSET,
-        **kwargs: Any
-    ) -> "GetAccountHistoryQueryFolio":
-        from .get_account_history_query import GetAccountHistoryQuery
-
-        query = gql(
-            """
-            query GetAccountHistoryQuery($account: String!, $fromInclusive: DateTime, $toExclusive: DateTime) {
-              folio {
-                accountHistory(
-                  account: $account
-                  fromInclusive: $fromInclusive
-                  toExclusive: $toExclusive
-                ) {
-                  ...AccountSummaryFields
-                }
-              }
-            }
-
-            fragment AccountSummaryFields on AccountSummary {
-              account
-              timestamp
-              balances {
-                product
-                balance
-              }
-              positions {
-                symbol
-                quantity
-                tradeTime
-                costBasis
-                breakEvenPrice
-                liquidationPrice
-              }
-              unrealizedPnl
-              realizedPnl
-              equity
-              yesterdayEquity
-              cashExcess
-              purchasingPower
-              totalMargin
-              positionMargin
-            }
-            """
-        )
-        variables: Dict[str, object] = {
-            "account": account,
-            "fromInclusive": convert_datetime_to_utc_str(from_inclusive),
-            "toExclusive": convert_datetime_to_utc_str(to_exclusive),
-        }
-        response = await self.execute(
-            query=query,
-            operation_name="GetAccountHistoryQuery",
-            variables=variables,
-            **kwargs
-        )
-        data = self.get_data(response)
-        return GetAccountHistoryQuery.model_validate(data).folio
-
-    async def get_open_orders_query(
-        self,
-        venue: Union[Optional[str], "UnsetType"] = UNSET,
-        account: Union[Optional[str], "UnsetType"] = UNSET,
-        trader: Union[Optional[str], "UnsetType"] = UNSET,
-        symbol: Union[Optional[str], "UnsetType"] = UNSET,
-        parent_order_id: Union[Optional[str], "UnsetType"] = UNSET,
-        order_ids: Union[Optional[List[str]], "UnsetType"] = UNSET,
-        **kwargs: Any
-    ) -> "GetOpenOrdersQueryOms":
-        from .get_open_orders_query import GetOpenOrdersQuery
-
-        query = gql(
-            """
-            query GetOpenOrdersQuery($venue: ExecutionVenue, $account: String, $trader: String, $symbol: String, $parentOrderId: OrderId, $orderIds: [OrderId!]) {
-              oms {
-                openOrders(
-                  venue: $venue
-                  account: $account
-                  trader: $trader
-                  symbol: $symbol
-                  parentOrderId: $parentOrderId
-                  orderIds: $orderIds
-                ) {
-                  ...OrderFields
-                }
-              }
-            }
-
-            fragment OrderFields on Order {
-              id
-              parentId
-              recvTime
-              status
-              rejectReason
-              rejectMessage
-              symbol
-              trader
-              account
-              dir
-              quantity
-              filledQuantity
-              averageFillPrice
-              orderType
-              limitPrice
-              postOnly
-              triggerPrice
-              timeInForce
-              goodTilDate
-              source
-              executionVenue
-            }
-            """
-        )
-        variables: Dict[str, object] = {
-            "venue": venue,
-            "account": account,
-            "trader": trader,
-            "symbol": symbol,
-            "parentOrderId": parent_order_id,
-            "orderIds": order_ids,
-        }
-        response = await self.execute(
-            query=query,
-            operation_name="GetOpenOrdersQuery",
-            variables=variables,
-            **kwargs
-        )
-        data = self.get_data(response)
-        return GetOpenOrdersQuery.model_validate(data).oms
-
-    async def get_historical_orders_query(
-        self,
-        order_ids: Union[Optional[List[str]], "UnsetType"] = UNSET,
-        venue: Union[Optional[str], "UnsetType"] = UNSET,
-        account: Union[Optional[str], "UnsetType"] = UNSET,
-        parent_order_id: Union[Optional[str], "UnsetType"] = UNSET,
-        from_inclusive: Union[Optional[datetime], "UnsetType"] = UNSET,
-        to_exclusive: Union[Optional[datetime], "UnsetType"] = UNSET,
-        **kwargs: Any
-    ) -> "GetHistoricalOrdersQueryFolio":
-        from .get_historical_orders_query import GetHistoricalOrdersQuery
-
-        query = gql(
-            """
-            query GetHistoricalOrdersQuery($orderIds: [OrderId!], $venue: ExecutionVenue, $account: String, $parentOrderId: OrderId, $fromInclusive: DateTime, $toExclusive: DateTime) {
-              folio {
-                historicalOrders(
-                  orderIds: $orderIds
-                  venue: $venue
-                  account: $account
-                  parentOrderId: $parentOrderId
-                  fromInclusive: $fromInclusive
-                  toExclusive: $toExclusive
-                ) {
-                  ...OrderFields
-                }
-              }
-            }
-
-            fragment OrderFields on Order {
-              id
-              parentId
-              recvTime
-              status
-              rejectReason
-              rejectMessage
-              symbol
-              trader
-              account
-              dir
-              quantity
-              filledQuantity
-              averageFillPrice
-              orderType
-              limitPrice
-              postOnly
-              triggerPrice
-              timeInForce
-              goodTilDate
-              source
-              executionVenue
-            }
-            """
-        )
-        variables: Dict[str, object] = {
-            "orderIds": order_ids,
-            "venue": venue,
-            "account": account,
-            "parentOrderId": parent_order_id,
-            "fromInclusive": convert_datetime_to_utc_str(from_inclusive),
-            "toExclusive": convert_datetime_to_utc_str(to_exclusive),
-        }
-        response = await self.execute(
-            query=query,
-            operation_name="GetHistoricalOrdersQuery",
-            variables=variables,
-            **kwargs
-        )
-        data = self.get_data(response)
-        return GetHistoricalOrdersQuery.model_validate(data).folio
-
-    async def get_fills_query(
-        self,
-        venue: Union[Optional[str], "UnsetType"] = UNSET,
-        account: Union[Optional[str], "UnsetType"] = UNSET,
-        order_id: Union[Optional[str], "UnsetType"] = UNSET,
-        from_inclusive: Union[Optional[datetime], "UnsetType"] = UNSET,
-        to_exclusive: Union[Optional[datetime], "UnsetType"] = UNSET,
-        **kwargs: Any
-    ) -> "GetFillsQueryFolio":
-        from .get_fills_query import GetFillsQuery
-
-        query = gql(
-            """
-            query GetFillsQuery($venue: ExecutionVenue, $account: String, $orderId: OrderId, $fromInclusive: DateTime, $toExclusive: DateTime) {
-              folio {
-                historicalFills(
-                  venue: $venue
-                  account: $account
-                  orderId: $orderId
-                  fromInclusive: $fromInclusive
-                  toExclusive: $toExclusive
-                ) {
-                  fills {
-                    fillId
-                    fillKind
-                    executionVenue
-                    exchangeFillId
-                    orderId
-                    trader
-                    account
-                    symbol
-                    dir
-                    quantity
-                    price
-                    recvTime
-                    tradeTime
-                  }
-                  aberrantFills {
-                    fillId
-                    fillKind
-                    executionVenue
-                    exchangeFillId
-                    orderId
-                    trader
-                    account
-                    symbol
-                    dir
-                    quantity
-                    price
-                    recvTime
-                    tradeTime
-                  }
-                }
-              }
-            }
-            """
-        )
-        variables: Dict[str, object] = {
-            "venue": venue,
-            "account": account,
-            "orderId": order_id,
-            "fromInclusive": convert_datetime_to_utc_str(from_inclusive),
-            "toExclusive": convert_datetime_to_utc_str(to_exclusive),
-        }
-        response = await self.execute(
-            query=query, operation_name="GetFillsQuery", variables=variables, **kwargs
-        )
-        data = self.get_data(response)
-        return GetFillsQuery.model_validate(data).folio
-
-    async def subscribe_trades(
-        self, venue: str, symbol: str, **kwargs: Any
-    ) -> AsyncIterator["SubscribeTradesTrades"]:
-        from .subscribe_trades import SubscribeTrades
-
-        query = gql(
-            """
-            subscription SubscribeTrades($venue: MarketdataVenue!, $symbol: String!) {
-              trades(venue: $venue, symbol: $symbol) {
-                timestamp
-                direction
-                price
-                size
-              }
-            }
-            """
-        )
-        variables: Dict[str, object] = {"venue": venue, "symbol": symbol}
-        async for data in self.execute_ws(
-            query=query, operation_name="SubscribeTrades", variables=variables, **kwargs
-        ):
-            yield SubscribeTrades.model_validate(data).trades
-
-    async def subscribe_candles(
-        self,
-        venue: str,
-        symbol: str,
-        widths: Union[Optional[List["CandleWidth"]], "UnsetType"] = UNSET,
-        **kwargs: Any
-    ) -> AsyncIterator["SubscribeCandlesCandles"]:
-        from .subscribe_candles import SubscribeCandles
-
-        query = gql(
-            """
-            subscription SubscribeCandles($venue: MarketdataVenue!, $symbol: String!, $widths: [CandleWidth!]) {
-              candles(venue: $venue, symbol: $symbol, candleWidths: $widths) {
-                ...CandleFields
-              }
-            }
-
-            fragment CandleFields on Candle {
-              timestamp
-              width
-              open
-              high
-              low
-              close
-              volume
-            }
-            """
-        )
-        variables: Dict[str, object] = {
-            "venue": venue,
-            "symbol": symbol,
-            "widths": widths,
-        }
-        async for data in self.execute_ws(
-            query=query,
-            operation_name="SubscribeCandles",
-            variables=variables,
-            **kwargs
-        ):
-            yield SubscribeCandles.model_validate(data).candles
-
-    async def cancel_order_mutation(
-        self, order_id: str, **kwargs: Any
-    ) -> "CancelOrderMutationOms":
-        from .cancel_order_mutation import CancelOrderMutation
-
-        query = gql(
-            """
-            mutation CancelOrderMutation($orderId: OrderId!) {
-              oms {
-                cancelOrder(orderId: $orderId) {
-                  ...CancelFields
-                }
-              }
-            }
-
-            fragment CancelFields on Cancel {
-              cancelId
-              orderId
-              recvTime
-              status
-              rejectReason
-            }
-            """
-        )
-        variables: Dict[str, object] = {"orderId": order_id}
-        response = await self.execute(
-            query=query,
-            operation_name="CancelOrderMutation",
-            variables=variables,
-            **kwargs
-        )
-        data = self.get_data(response)
-        return CancelOrderMutation.model_validate(data).oms
-
-    async def cancel_all_orders_mutation(
-        self, **kwargs: Any
-    ) -> "CancelAllOrdersMutationOms":
-        from .cancel_all_orders_mutation import CancelAllOrdersMutation
-
-        query = gql(
-            """
-            mutation CancelAllOrdersMutation {
-              oms {
-                cancelAllOrders
-              }
-            }
-            """
-        )
-        variables: Dict[str, object] = {}
-        response = await self.execute(
-            query=query,
-            operation_name="CancelAllOrdersMutation",
-            variables=variables,
-            **kwargs
-        )
-        data = self.get_data(response)
-        return CancelAllOrdersMutation.model_validate(data).oms
-
-    async def place_order_mutation(
-        self,
-        symbol: str,
-        dir: OrderDir,
-        quantity: Decimal,
-        order_type: "OrderType",
-        time_in_force: "TimeInForce",
-        id: Union[Optional[str], "UnsetType"] = UNSET,
-        trader: Union[Optional[str], "UnsetType"] = UNSET,
-        account: Union[Optional[str], "UnsetType"] = UNSET,
-        limit_price: Union[Optional[Decimal], "UnsetType"] = UNSET,
-        post_only: Union[Optional[bool], "UnsetType"] = UNSET,
-        trigger_price: Union[Optional[Decimal], "UnsetType"] = UNSET,
-        good_til_date: Union[Optional[datetime], "UnsetType"] = UNSET,
-        execution_venue: Union[Optional[str], "UnsetType"] = UNSET,
-        **kwargs: Any
-    ) -> "PlaceOrderMutationOms":
-        from .place_order_mutation import PlaceOrderMutation
-
-        query = gql(
-            """
-            mutation PlaceOrderMutation($id: OrderId, $symbol: String!, $dir: Dir!, $quantity: Decimal!, $trader: String, $account: String, $orderType: OrderType!, $limitPrice: Decimal, $postOnly: Boolean, $triggerPrice: Decimal, $timeInForce: TimeInForce!, $goodTilDate: DateTime, $executionVenue: ExecutionVenue) {
-              oms {
-                placeOrder(
-                  id: $id
-                  symbol: $symbol
-                  dir: $dir
-                  quantity: $quantity
-                  trader: $trader
-                  account: $account
-                  orderType: $orderType
-                  limitPrice: $limitPrice
-                  postOnly: $postOnly
-                  triggerPrice: $triggerPrice
-                  timeInForce: $timeInForce
-                  goodTilDate: $goodTilDate
-                  executionVenue: $executionVenue
-                ) {
-                  ...OrderFields
-                }
-              }
-            }
-
-            fragment OrderFields on Order {
-              id
-              parentId
-              recvTime
-              status
-              rejectReason
-              rejectMessage
-              symbol
-              trader
-              account
-              dir
-              quantity
-              filledQuantity
-              averageFillPrice
-              orderType
-              limitPrice
-              postOnly
-              triggerPrice
-              timeInForce
-              goodTilDate
-              source
-              executionVenue
-            }
-            """
-        )
-        variables: Dict[str, object] = {
-            "id": id,
-            "symbol": symbol,
-            "dir": graphql_serialize_order_dir(dir),
-            "quantity": quantity,
-            "trader": trader,
-            "account": account,
-            "orderType": order_type,
-            "limitPrice": limit_price,
-            "postOnly": post_only,
-            "triggerPrice": trigger_price,
-            "timeInForce": time_in_force,
-            "goodTilDate": convert_datetime_to_utc_str(good_til_date),
-            "executionVenue": execution_venue,
-        }
-        response = await self.execute(
-            query=query,
-            operation_name="PlaceOrderMutation",
-            variables=variables,
-            **kwargs
-        )
-        data = self.get_data(response)
-        return PlaceOrderMutation.model_validate(data).oms
-
-    async def subscribe_orderflow(self, **kwargs: Any) -> AsyncIterator[
-        Union[
-            "SubscribeOrderflowOrderflowOrder",
-            "SubscribeOrderflowOrderflowOrderAck",
-            "SubscribeOrderflowOrderflowGqlOrderReject",
-            "SubscribeOrderflowOrderflowOrderOut",
-            "SubscribeOrderflowOrderflowOrderStale",
-            "SubscribeOrderflowOrderflowCancel",
-            "SubscribeOrderflowOrderflowCancelReject",
-            "SubscribeOrderflowOrderflowOrderCanceling",
-            "SubscribeOrderflowOrderflowOrderCanceled",
-            "SubscribeOrderflowOrderflowFill",
-            "SubscribeOrderflowOrderflowAberrantFill",
-        ]
-    ]:
-        from .subscribe_orderflow import SubscribeOrderflow
-
-        query = gql(
-            """
-            subscription SubscribeOrderflow {
-              orderflow {
-                __typename
-                ... on Order {
-                  ...OrderFields
-                }
-                ... on OrderAck {
-                  orderId
-                }
-                ... on OrderCanceled {
-                  orderId
-                  cancelId
-                }
-                ... on GqlOrderReject {
-                  orderId
-                  reason
-                  message
-                }
-                ... on CancelReject {
-                  orderId
-                  message
-                }
-                ... on Fill {
-                  fillOrderId: orderId
-                  fillId
-                  fillKind
-                  executionVenue
-                  exchangeFillId
-                  symbol
-                  dir
-                  quantity
-                  price
-                  recvTime
-                  tradeTime
-                }
-                ... on OrderOut {
-                  orderId
-                }
-                ... on OrderStale {
-                  orderId
-                }
-              }
-            }
-
-            fragment OrderFields on Order {
-              id
-              parentId
-              recvTime
-              status
-              rejectReason
-              rejectMessage
-              symbol
-              trader
-              account
-              dir
-              quantity
-              filledQuantity
-              averageFillPrice
-              orderType
-              limitPrice
-              postOnly
-              triggerPrice
-              timeInForce
-              goodTilDate
-              source
-              executionVenue
-            }
-            """
-        )
-        variables: Dict[str, object] = {}
-        async for data in self.execute_ws(
-            query=query,
-            operation_name="SubscribeOrderflow",
-            variables=variables,
-            **kwargs
-        ):
-            yield SubscribeOrderflow.model_validate(data).orderflow
-
-    async def create_jwt(self, **kwargs: Any) -> "CreateJwtUser":
-        from .create_jwt import CreateJwt
-
-        query = gql(
-            """
-            mutation CreateJwt {
-              user {
-                createJwt
-              }
-            }
-            """
-        )
-        variables: Dict[str, object] = {}
-        response = await self.execute(
-            query=query, operation_name="CreateJwt", variables=variables, **kwargs
-        )
-        data = self.get_data(response)
-        return CreateJwt.model_validate(data).user