Add sort argument to get_markets (#22)

gnosis · Feb 21, 2024 · bc88190 · bc88190
1 parent f52a2d2
commit bc88190
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/prediction_market_agent_tooling/benchmark/utils.py b/prediction_market_agent_tooling/benchmark/utils.py
@@ -22,6 +22,11 @@ class MarketFilter(str, Enum):
     resolved = "resolved"
 
 
+class MarketSort(str, Enum):
+    liquidity = "liquidity"
+    newest = "newest"
+
+
 class MarketResolution(str, Enum):
     YES = "yes"
     NO = "no"
@@ -305,13 +310,19 @@ def get_markets(
     number: int,
     source: MarketSource,
     filter_: MarketFilter = MarketFilter.open,
+    sort: MarketSort | None = None,
     excluded_questions: set[str] | None = None,
 ) -> t.List[Market]:
     if source == MarketSource.MANIFOLD:
         return get_manifold_markets_paged(
-            number=number, excluded_questions=excluded_questions, filter_=filter_.value
+            number=number,
+            excluded_questions=excluded_questions,
+            filter_=filter_.value,
+            sort=(sort or MarketSort.liquidity).value,
         )
     elif source == MarketSource.POLYMARKET:
+        if sort is not None:
+            raise ValueError(f"Polymarket doesn't support sorting.")
         return get_polymarket_markets(
             limit=number,
             excluded_questions=excluded_questions,