fix yql builder as CLI (#151)

* fix yql builder as CLI * bump version
climatepolicyradar · Dec 3, 2024 · 4431ff9 · 4431ff9
1 parent f71b720
commit 4431ff9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/cpr_sdk/version.py b/src/cpr_sdk/version.py
@@ -1,6 +1,6 @@
 _MAJOR = "1"
 _MINOR = "9"
-_PATCH = "5"
+_PATCH = "6"
 _SUFFIX = ""
 
 VERSION_SHORT = "{0}.{1}".format(_MAJOR, _MINOR)

diff --git a/src/cpr_sdk/yql_builder.py b/src/cpr_sdk/yql_builder.py
@@ -1,6 +1,7 @@
 from string import Template
 from typing import Optional
 
+from rich import print as rprint
 from cpr_sdk.models.search import Filters, SearchParameters
 
 
@@ -231,7 +232,9 @@ def to_str(self) -> str:
         exact_match=False,
         limit=10,
         max_hits_per_family=10,
-        filters=Filters(**{"document_languages": "value", "family_source": "value"}),
+        filters=Filters(
+            **{"document_languages": ["value"], "family_source": ["value"]}
+        ),
         year_range=(2000, 2020),
         continuation_tokens=None,
     )
@@ -240,3 +243,5 @@ def to_str(self) -> str:
         params=params,
         sensitive=False,
     ).to_str()
+
+    rprint(yql_new)