Fix polymarket models (#246)

gnosis · May 15, 2024 · c62786d · c62786d
1 parent 969b697
commit c62786d
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/prediction_market_agent_tooling/markets/polymarket/data_models_web.py b/prediction_market_agent_tooling/markets/polymarket/data_models_web.py
@@ -83,7 +83,7 @@ class Market(BaseModel):
     category: t.Any | None
     ammType: t.Any | None
     description: str
-    liquidity: str
+    liquidity: str | None
     startDate: datetime
     createdAt: datetime
     xAxisValue: t.Any | None
@@ -128,7 +128,7 @@ class Market(BaseModel):
     umaResolutionStatus: t.Any | None
     curationOrder: t.Any | None
     volumeNum: USDC | None
-    liquidityNum: float
+    liquidityNum: float | None
     endDateIso: datetime | None
     startDateIso: datetime | None
     umaEndDateIso: datetime | None
@@ -137,8 +137,8 @@ class Market(BaseModel):
     gameStartTime: datetime | None
     secondsDelay: int | None
     clobTokenIds: list[str]
-    liquidityAmm: float
-    liquidityClob: float
+    liquidityAmm: float | None
+    liquidityClob: float | None
     makerBaseFee: int | None
     takerBaseFee: int | None
     negRisk: t.Any | None
@@ -163,7 +163,7 @@ class Market(BaseModel):
     closed_time: t.Any | None
     wide_format: bool | None
     volume_num: USDC | None
-    liquidity_num: USDC
+    liquidity_num: USDC | None
     image_raw: str
     resolutionData: ResolutionData
 
@@ -228,7 +228,7 @@ class PolymarketFullMarket(BaseModel):
     new: bool
     featured: bool
     restricted: bool
-    liquidity: USDC
+    liquidity: USDC | None
     volume: USDC | None
     volume24hr: USDC | None
     competitive: float
@@ -239,16 +239,16 @@ class PolymarketFullMarket(BaseModel):
     disqusThread: t.Any | None
     updatedAt: datetime
     enableOrderBook: bool
-    liquidityAmm: float
-    liquidityClob: float
+    liquidityAmm: float | None
+    liquidityClob: float | None
     imageOptimized: ImageOptimized | None
     iconOptimized: IconOptimized | None
     featuredImageOptimized: str | None
     negRisk: t.Any | None
     negRiskMarketID: t.Any | None
     negRiskFeeBips: t.Any | None
     markets: list[Market]
-    categories: list[Category] | None
+    categories: list[Category] | None = None
     series: t.Any | None
     image_raw: str