fix: fix checks

app/_types.py

@@ -214,8 +214,8 @@ class SearchParameters(BaseModel):
         list[str] | None,
         Query(
             description="""Name of vega representations to return in the response.
-            If None (default) no charts are returrned."""
-        )
+            If None (default) no charts are returned."""
+        ),
     ] = None
     sort_params: Annotated[
         JSONType | None,

app/charts.py

@@ -1,16 +1,20 @@
-from ._types import SuccessSearchResponse
+from ._types import ChartsInfos, SuccessSearchResponse
 
 
 def build_charts(
     search_result: SuccessSearchResponse,
     charts_names: list[str] | None,
-) -> str:
-    charts = {}
-    if charts_names is None:
+) -> ChartsInfos:
+    charts: ChartsInfos = {}
+    aggregations = search_result.aggregations
+
+    if charts_names is None or aggregations is None:
         return charts
 
     for chart_name in charts_names:
-        buckets = search_result.aggregations[chart_name]["buckets"]
+        agg_data = aggregations.get(chart_name, {})
+
+        buckets = agg_data.get("buckets", []) if agg_data else []
 
         # Filter unknown values
         values = [

app/query.py

@@ -281,7 +281,7 @@ def parse_sort_by_script(
 
 
 def create_aggregation_clauses(
-    config: IndexConfig, fields: list[str] | None
+    config: IndexConfig, fields: set[str] | list[str] | None
 ) -> dict[str, Agg]:
     """Create term bucket aggregation clauses
     for all fields corresponding to facets,