frank/prediction markets (#197)

* most of prediction markets * add market constants * remaining prediction markets stuff * remove dbg prints * rename worst_case_quote_asset_amount * rm code not needed for driftpy --------- Co-authored-by: Chris Heaney <[email protected]>
drift-labs · Aug 15, 2024 · 4d9bcef · 4d9bcef
1 parent 003e1b8
commit 4d9bcef
Show file tree

Hide file tree

Showing 4 changed files with 422 additions and 138 deletions.
diff --git a/src/driftpy/dlob/orderbook_levels.py b/src/driftpy/dlob/orderbook_levels.py
@@ -21,6 +21,7 @@
     PerpMarketAccount,
     PositionDirection,
     SwapDirection,
+    is_variant,
 )
 
 LiquiditySource = ["serum", "vamm", "dlob", "phoenix"]
@@ -168,7 +169,10 @@ def get_vamm_l2_generator(
 
     now = now or int(datetime.now().timestamp())
     bid_reserves, ask_reserves = calculate_spread_reserves(
-        updated_amm, oracle_price_data, now
+        updated_amm,
+        oracle_price_data,
+        now,
+        is_variant(market_account.contract_type, "Prediction"),
     )
 
     num_bids = 0