qanswer_sdk 3.1213.0__py3-none-any.whl → 3.1244.0__py3-none-any.whl

qanswer_sdk/api/task_search_api.py

@@ -16,9 +16,10 @@ from pydantic import validate_call, Field, StrictFloat, StrictStr, StrictInt
 from typing import Any, Dict, List, Optional, Tuple, Union
 from typing_extensions import Annotated
 
-from pydantic import StrictStr, field_validator
-from typing import Dict, List, Optional
+from pydantic import StrictStr
+from typing import List, Optional
 from qanswer_sdk.models.aggregations_group import AggregationsGroup
+from qanswer_sdk.models.available_aggregation import AvailableAggregation
 from qanswer_sdk.models.rag_payload import RAGPayload
 from qanswer_sdk.models.rag_response import RAGResponse
 from qanswer_sdk.models.search_payload import SearchPayload
@@ -642,7 +643,7 @@ class TaskSearchApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> Dict[str, str]:
+    ) -> List[AvailableAggregation]:
         """Check which aggregations are available for the dataset
 
 
@@ -685,7 +686,7 @@ class TaskSearchApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "Dict[str, str]",
+            '200': "List[AvailableAggregation]",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -716,7 +717,7 @@ class TaskSearchApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> ApiResponse[Dict[str, str]]:
+    ) -> ApiResponse[List[AvailableAggregation]]:
         """Check which aggregations are available for the dataset
 
 
@@ -759,7 +760,7 @@ class TaskSearchApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "Dict[str, str]",
+            '200': "List[AvailableAggregation]",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -833,7 +834,7 @@ class TaskSearchApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "Dict[str, str]",
+            '200': "List[AvailableAggregation]",
         }
         response_data = self.api_client.call_api(
             *_param,