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

qanswer_sdk/api/ai_assistant_api.py

@@ -20,14 +20,15 @@ from pydantic import Field, StrictBool, StrictBytes, StrictFloat, StrictInt, Str
 from typing import List, Optional, Tuple, Union
 from typing_extensions import Annotated
 from qanswer_sdk.models.ai_assistant_dto import AiAssistantDto
-from qanswer_sdk.models.api_response import ApiResponse
+from qanswer_sdk.models.ai_assistant_filter_dto import AiAssistantFilterDto
+from qanswer_sdk.models.ai_assistant_list import AiAssistantList
 from qanswer_sdk.models.available_space_quota_for_dataset import AvailableSpaceQuotaForDataset
-from qanswer_sdk.models.dataset_description import DatasetDescription
 from qanswer_sdk.models.dataset_detail import DatasetDetail
 from qanswer_sdk.models.dataset_schema import DatasetSchema
 from qanswer_sdk.models.dataset_update_object import DatasetUpdateObject
 from qanswer_sdk.models.dataset_users_response import DatasetUsersResponse
 from qanswer_sdk.models.default_prompt import DefaultPrompt
+from qanswer_sdk.models.entity_description_dto import EntityDescriptionDto
 
 from qanswer_sdk.api_client import ApiClient, RequestSerialized
 from qanswer_sdk.api_response import ApiResponse
@@ -1846,7 +1847,7 @@ class AiAssistantApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> DatasetDescription:
+    ) -> EntityDescriptionDto:
         """To store the description of a dataset
 
 
@@ -1889,7 +1890,7 @@ class AiAssistantApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "DatasetDescription",
+            '200': "EntityDescriptionDto",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -1920,7 +1921,7 @@ class AiAssistantApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> ApiResponse[DatasetDescription]:
+    ) -> ApiResponse[EntityDescriptionDto]:
         """To store the description of a dataset
 
 
@@ -1963,7 +1964,7 @@ class AiAssistantApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "DatasetDescription",
+            '200': "EntityDescriptionDto",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -2037,7 +2038,7 @@ class AiAssistantApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "DatasetDescription",
+            '200': "EntityDescriptionDto",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -2659,7 +2660,7 @@ class AiAssistantApi:
 
 
     @validate_call
-    def get_logo1(
+    def get_logo2(
         self,
         username: StrictStr,
         dataset: StrictStr,
@@ -2708,7 +2709,7 @@ class AiAssistantApi:
         :return: Returns the result object.
         """ # noqa: E501
 
-        _param = self._get_logo1_serialize(
+        _param = self._get_logo2_serialize(
             username=username,
             dataset=dataset,
             q_answer_api_key=q_answer_api_key,
@@ -2733,7 +2734,7 @@ class AiAssistantApi:
 
 
     @validate_call
-    def get_logo1_with_http_info(
+    def get_logo2_with_http_info(
         self,
         username: StrictStr,
         dataset: StrictStr,
@@ -2782,7 +2783,7 @@ class AiAssistantApi:
         :return: Returns the result object.
         """ # noqa: E501
 
-        _param = self._get_logo1_serialize(
+        _param = self._get_logo2_serialize(
             username=username,
             dataset=dataset,
             q_answer_api_key=q_answer_api_key,
@@ -2807,7 +2808,7 @@ class AiAssistantApi:
 
 
     @validate_call
-    def get_logo1_without_preload_content(
+    def get_logo2_without_preload_content(
         self,
         username: StrictStr,
         dataset: StrictStr,
@@ -2856,7 +2857,7 @@ class AiAssistantApi:
         :return: Returns the result object.
         """ # noqa: E501
 
-        _param = self._get_logo1_serialize(
+        _param = self._get_logo2_serialize(
             username=username,
             dataset=dataset,
             q_answer_api_key=q_answer_api_key,
@@ -2876,7 +2877,7 @@ class AiAssistantApi:
         return response_data.response
 
 
-    def _get_logo1_serialize(
+    def _get_logo2_serialize(
         self,
         username,
         dataset,
@@ -2951,6 +2952,265 @@ class AiAssistantApi:
 
 
 
+    @validate_call
+    def get_logo_by_id(
+        self,
+        ai_assistant_id: StrictInt,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> bytearray:
+        """Get the logo of an ai assistant
+
+
+        :param ai_assistant_id: (required)
+        :type ai_assistant_id: int
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._get_logo_by_id_serialize(
+            ai_assistant_id=ai_assistant_id,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "bytearray",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        ).data
+
+
+    @validate_call
+    def get_logo_by_id_with_http_info(
+        self,
+        ai_assistant_id: StrictInt,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> ApiResponse[bytearray]:
+        """Get the logo of an ai assistant
+
+
+        :param ai_assistant_id: (required)
+        :type ai_assistant_id: int
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._get_logo_by_id_serialize(
+            ai_assistant_id=ai_assistant_id,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "bytearray",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        )
+
+
+    @validate_call
+    def get_logo_by_id_without_preload_content(
+        self,
+        ai_assistant_id: StrictInt,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> RESTResponseType:
+        """Get the logo of an ai assistant
+
+
+        :param ai_assistant_id: (required)
+        :type ai_assistant_id: int
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._get_logo_by_id_serialize(
+            ai_assistant_id=ai_assistant_id,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "bytearray",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        return response_data.response
+
+
+    def _get_logo_by_id_serialize(
+        self,
+        ai_assistant_id,
+        _request_auth,
+        _content_type,
+        _headers,
+        _host_index,
+    ) -> RequestSerialized:
+
+        _host = None
+
+        _collection_formats: Dict[str, str] = {
+        }
+
+        _path_params: Dict[str, str] = {}
+        _query_params: List[Tuple[str, str]] = []
+        _header_params: Dict[str, Optional[str]] = _headers or {}
+        _form_params: List[Tuple[str, str]] = []
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
+        _body_params: Optional[bytes] = None
+
+        # process the path parameters
+        if ai_assistant_id is not None:
+            _path_params['ai_assistant_id'] = ai_assistant_id
+        # process the query parameters
+        # process the header parameters
+        # process the form parameters
+        # process the body parameter
+
+
+        # set the HTTP header `Accept`
+        if 'Accept' not in _header_params:
+            _header_params['Accept'] = self.api_client.select_header_accept(
+                [
+                    'image/png'
+                ]
+            )
+
+
+        # authentication setting
+        _auth_settings: List[str] = [
+            'QAnswer-Api-Key', 
+            'Bearer'
+        ]
+
+        return self.api_client.param_serialize(
+            method='GET',
+            resource_path='/api/ai-assistant/{ai_assistant_id}/logo',
+            path_params=_path_params,
+            query_params=_query_params,
+            header_params=_header_params,
+            body=_body_params,
+            post_params=_form_params,
+            files=_files,
+            auth_settings=_auth_settings,
+            collection_formats=_collection_formats,
+            _host=_host,
+            _request_auth=_request_auth
+        )
+
+
+
+
     @validate_call
     def remove2(
         self,
@@ -2967,7 +3227,7 @@ class AiAssistantApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> ApiResponse:
+    ) -> str:
         """To remove a specific dataset
 
 
@@ -3004,7 +3264,7 @@ class AiAssistantApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "ApiResponse",
+            '200': "str",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -3033,7 +3293,7 @@ class AiAssistantApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> ApiResponse[ApiResponse]:
+    ) -> ApiResponse[str]:
         """To remove a specific dataset
 
 
@@ -3070,7 +3330,7 @@ class AiAssistantApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "ApiResponse",
+            '200': "str",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -3136,7 +3396,7 @@ class AiAssistantApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "ApiResponse",
+            '200': "str",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -3498,12 +3758,273 @@ class AiAssistantApi:
 
 
 
+    @validate_call
+    def search3(
+        self,
+        filter: AiAssistantFilterDto,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> AiAssistantList:
+        """Search over all ai assistants
+
+
+        :param filter: (required)
+        :type filter: AiAssistantFilterDto
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._search3_serialize(
+            filter=filter,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "AiAssistantList",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        ).data
+
+
+    @validate_call
+    def search3_with_http_info(
+        self,
+        filter: AiAssistantFilterDto,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> ApiResponse[AiAssistantList]:
+        """Search over all ai assistants
+
+
+        :param filter: (required)
+        :type filter: AiAssistantFilterDto
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._search3_serialize(
+            filter=filter,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "AiAssistantList",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        )
+
+
+    @validate_call
+    def search3_without_preload_content(
+        self,
+        filter: AiAssistantFilterDto,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> RESTResponseType:
+        """Search over all ai assistants
+
+
+        :param filter: (required)
+        :type filter: AiAssistantFilterDto
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._search3_serialize(
+            filter=filter,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "AiAssistantList",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        return response_data.response
+
+
+    def _search3_serialize(
+        self,
+        filter,
+        _request_auth,
+        _content_type,
+        _headers,
+        _host_index,
+    ) -> RequestSerialized:
+
+        _host = None
+
+        _collection_formats: Dict[str, str] = {
+        }
+
+        _path_params: Dict[str, str] = {}
+        _query_params: List[Tuple[str, str]] = []
+        _header_params: Dict[str, Optional[str]] = _headers or {}
+        _form_params: List[Tuple[str, str]] = []
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
+        _body_params: Optional[bytes] = None
+
+        # process the path parameters
+        # process the query parameters
+        if filter is not None:
+            
+            _query_params.append(('filter', filter))
+            
+        # process the header parameters
+        # process the form parameters
+        # process the body parameter
+
+
+        # set the HTTP header `Accept`
+        if 'Accept' not in _header_params:
+            _header_params['Accept'] = self.api_client.select_header_accept(
+                [
+                    '*/*'
+                ]
+            )
+
+
+        # authentication setting
+        _auth_settings: List[str] = [
+            'QAnswer-Api-Key', 
+            'Bearer'
+        ]
+
+        return self.api_client.param_serialize(
+            method='GET',
+            resource_path='/api/ai-assistant/search',
+            path_params=_path_params,
+            query_params=_query_params,
+            header_params=_header_params,
+            body=_body_params,
+            post_params=_form_params,
+            files=_files,
+            auth_settings=_auth_settings,
+            collection_formats=_collection_formats,
+            _host=_host,
+            _request_auth=_request_auth
+        )
+
+
+
+
     @validate_call
     def set_description(
         self,
         username: StrictStr,
         dataset: StrictStr,
-        dataset_description: DatasetDescription,
+        entity_description_dto: EntityDescriptionDto,
         q_answer_api_key: Optional[StrictStr] = None,
         _request_timeout: Union[
             None,
@@ -3517,7 +4038,7 @@ class AiAssistantApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> ApiResponse:
+    ) -> str:
         """To store the description of a dataset
 
 
@@ -3525,8 +4046,8 @@ class AiAssistantApi:
         :type username: str
         :param dataset: (required)
         :type dataset: str
-        :param dataset_description: (required)
-        :type dataset_description: DatasetDescription
+        :param entity_description_dto: (required)
+        :type entity_description_dto: EntityDescriptionDto
         :param q_answer_api_key:
         :type q_answer_api_key: str
         :param _request_timeout: timeout setting for this request. If one
@@ -3554,7 +4075,7 @@ class AiAssistantApi:
         _param = self._set_description_serialize(
             username=username,
             dataset=dataset,
-            dataset_description=dataset_description,
+            entity_description_dto=entity_description_dto,
             q_answer_api_key=q_answer_api_key,
             _request_auth=_request_auth,
             _content_type=_content_type,
@@ -3563,7 +4084,7 @@ class AiAssistantApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "ApiResponse",
+            '200': "str",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -3581,7 +4102,7 @@ class AiAssistantApi:
         self,
         username: StrictStr,
         dataset: StrictStr,
-        dataset_description: DatasetDescription,
+        entity_description_dto: EntityDescriptionDto,
         q_answer_api_key: Optional[StrictStr] = None,
         _request_timeout: Union[
             None,
@@ -3595,7 +4116,7 @@ class AiAssistantApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> ApiResponse[ApiResponse]:
+    ) -> ApiResponse[str]:
         """To store the description of a dataset
 
 
@@ -3603,8 +4124,8 @@ class AiAssistantApi:
         :type username: str
         :param dataset: (required)
         :type dataset: str
-        :param dataset_description: (required)
-        :type dataset_description: DatasetDescription
+        :param entity_description_dto: (required)
+        :type entity_description_dto: EntityDescriptionDto
         :param q_answer_api_key:
         :type q_answer_api_key: str
         :param _request_timeout: timeout setting for this request. If one
@@ -3632,7 +4153,7 @@ class AiAssistantApi:
         _param = self._set_description_serialize(
             username=username,
             dataset=dataset,
-            dataset_description=dataset_description,
+            entity_description_dto=entity_description_dto,
             q_answer_api_key=q_answer_api_key,
             _request_auth=_request_auth,
             _content_type=_content_type,
@@ -3641,7 +4162,7 @@ class AiAssistantApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "ApiResponse",
+            '200': "str",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -3659,7 +4180,7 @@ class AiAssistantApi:
         self,
         username: StrictStr,
         dataset: StrictStr,
-        dataset_description: DatasetDescription,
+        entity_description_dto: EntityDescriptionDto,
         q_answer_api_key: Optional[StrictStr] = None,
         _request_timeout: Union[
             None,
@@ -3681,8 +4202,8 @@ class AiAssistantApi:
         :type username: str
         :param dataset: (required)
         :type dataset: str
-        :param dataset_description: (required)
-        :type dataset_description: DatasetDescription
+        :param entity_description_dto: (required)
+        :type entity_description_dto: EntityDescriptionDto
         :param q_answer_api_key:
         :type q_answer_api_key: str
         :param _request_timeout: timeout setting for this request. If one
@@ -3710,7 +4231,7 @@ class AiAssistantApi:
         _param = self._set_description_serialize(
             username=username,
             dataset=dataset,
-            dataset_description=dataset_description,
+            entity_description_dto=entity_description_dto,
             q_answer_api_key=q_answer_api_key,
             _request_auth=_request_auth,
             _content_type=_content_type,
@@ -3719,7 +4240,7 @@ class AiAssistantApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "ApiResponse",
+            '200': "str",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -3732,7 +4253,7 @@ class AiAssistantApi:
         self,
         username,
         dataset,
-        dataset_description,
+        entity_description_dto,
         q_answer_api_key,
         _request_auth,
         _content_type,
@@ -3769,8 +4290,8 @@ class AiAssistantApi:
             _header_params['QAnswer-Api-Key'] = q_answer_api_key
         # process the form parameters
         # process the body parameter
-        if dataset_description is not None:
-            _body_params = dataset_description
+        if entity_description_dto is not None:
+            _body_params = entity_description_dto
 
 
         # set the HTTP header `Accept`
@@ -3820,7 +4341,7 @@ class AiAssistantApi:
 
 
     @validate_call
-    def set_logo1(
+    def set_logo2(
         self,
         username: StrictStr,
         dataset: StrictStr,
@@ -3872,7 +4393,7 @@ class AiAssistantApi:
         :return: Returns the result object.
         """ # noqa: E501
 
-        _param = self._set_logo1_serialize(
+        _param = self._set_logo2_serialize(
             username=username,
             dataset=dataset,
             file=file,
@@ -3898,7 +4419,7 @@ class AiAssistantApi:
 
 
     @validate_call
-    def set_logo1_with_http_info(
+    def set_logo2_with_http_info(
         self,
         username: StrictStr,
         dataset: StrictStr,
@@ -3950,7 +4471,7 @@ class AiAssistantApi:
         :return: Returns the result object.
         """ # noqa: E501
 
-        _param = self._set_logo1_serialize(
+        _param = self._set_logo2_serialize(
             username=username,
             dataset=dataset,
             file=file,
@@ -3976,7 +4497,7 @@ class AiAssistantApi:
 
 
     @validate_call
-    def set_logo1_without_preload_content(
+    def set_logo2_without_preload_content(
         self,
         username: StrictStr,
         dataset: StrictStr,
@@ -4028,7 +4549,7 @@ class AiAssistantApi:
         :return: Returns the result object.
         """ # noqa: E501
 
-        _param = self._set_logo1_serialize(
+        _param = self._set_logo2_serialize(
             username=username,
             dataset=dataset,
             file=file,
@@ -4049,7 +4570,7 @@ class AiAssistantApi:
         return response_data.response
 
 
-    def _set_logo1_serialize(
+    def _set_logo2_serialize(
         self,
         username,
         dataset,