model config for json correctness

ComposableFi · Apr 16, 2024 · 03d2de5 · 03d2de5
1 parent 5ebf738
commit 03d2de5
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/mantis/simulation/routers/data.py b/mantis/simulation/routers/data.py
@@ -325,6 +325,8 @@ class Input(
     this is what user asks for
     """
 
+    model_config = ConfigDict(coerce_numbers_to_str=True)
+
     # natural set key is ordered pair (`in_asset_id``, `out_asset_id`)
     in_asset_id: TId = Field(example="158456325028528675187087900673")
     out_asset_id: TId = Field(example="158456325028528675187087900674")

diff --git a/mantis/simulation/routers/test_data.py b/mantis/simulation/routers/test_data.py
@@ -20,7 +20,12 @@ def test_big_numbers_api():
     """_summary_
     Ensures that amounts are strings on the wire but big integers in memory    
     """
-    input : Input[str, str] = Input(in_asset_id="1", out_asset_id="2", in_asset_amount="100", out_asset_amount="100")
+    input_api_raw : Input[str, str] = Input(in_asset_id="1", out_asset_id="2", in_asset_amount="100", out_asset_amount="100")
+    input_memory : Input[str, int] = Input(in_asset_id="1", out_asset_id="2", in_asset_amount=100, out_asset_amount=100)
+    input_api_converted : Input[str, str] = Input[str, str](**input_memory.model_dump())
+    assert input_api_converted.model_dump_json() == input_api_raw.model_dump_json()
+    # assert input_api_raw == input_api_converted
+    # input_memory : Input[str, int] = Input[str, int].model_validate(input_api)
 
 def test_token_price_in_usd():
     pica_usd = new_pair(