shaped 2.0.1__py3-none-any.whl → 2.0.2__py3-none-any.whl

shaped/autogen/api/{dataset_api.py → query_api.py}

@@ -3,9 +3,9 @@
 """
     Shaped API
 
-    Welcome to Shaped's API reference docs. These provide a detailed view of the endpoints and CLI commands that Shaped provides and brief explanations of how they should be used.   Shaped's API is composed of 3 components:   1. **Dataset** - used to provision and manage 'Shaped Datasets', which are persisted     data views of external data. Shaped Datasets can be created from any of our     'Shaped connectors' (e.g. S3, Segment, Snowflake, etc.) and support both batch     ingestion (up to a 15min delay) and stream ingestion (up to a 30 second     delay) depending on the specific connector used. Shaped datasets can also be     created from local files, which is particularly useful for getting started     with a snapshot of data.    2. **Model Management** - used to provision and manage 'Shaped Models', which     represent a system of data pipelines, training and serving infrastructure for     your ranking use-case.    3. **Model Inference** - a high performance API that's used to make     user-understanding requests or ranking inferences to your 'Shaped Models'. For     example, the 'rank' endpoint can be used to determine for a given user id     query, what is the content that is most engaging to that user.    The recommended workflow to interact with the Shaped API is as follows:   1. First create 'Shaped Datasets' to sync over data that your Shaped     understanding models will need. The models at the minimum need interaction     data to understand behavior of your users, so start with that and add your item     and user catalog data later.   2. Then create 'Shaped Models' that use your created 'Shaped Datasets' as     input. Your Shaped Model will will start streaming, processing and training     from your connected data immediately. After a few hours your model will have     tuned all parameters based on your data and will deploy an active model.   3. You can now use the 'Model Inference' endpoints to make real-time     inferences to your model based on your use-case. 
+    Welcome to Shaped's API reference docs. These provide a  detailed view of the endpoints and CLI commands that Shaped provides and brief  explanations of how they should be used.  The Shaped API has four main endpoints:  **Tables** - Provision and manage batch and real-time data connectors.   **Views** - Configure SQL transformations and AI enrichment on your input data. Use SQL to combine multiple data sources or use an LLM to add new categories, extract specific attributes from descriptions, etc.  **Engines** - Deploy and manage your relevance engines. The Engine API exposes configuration for indexing logic, input datasets, externam embeddings, and more.  **Query** - Execute queries against your engines, to return data based on an input query or rerank an existing list. The Query API exposes the retrieve, filter, score, and ranking steps of the 4-stage ranking architecture.  The base URL for each endpoint is: `https://api.shaped.ai/v2`
 
-    The version of the OpenAPI document: 1.0.1
+    The version of the OpenAPI document: 2.0.0
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.
@@ -16,19 +16,19 @@ 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 Field, StrictStr
-from typing_extensions import Annotated
-from shaped.autogen.models.create_dataset_arguments import CreateDatasetArguments
-from shaped.autogen.models.insert_model_response import InsertModelResponse
-from shaped.autogen.models.list_datasets_response import ListDatasetsResponse
-from shaped.autogen.models.successful_response import SuccessfulResponse
+from pydantic import StrictStr
+from shaped.autogen.models.query_request import QueryRequest
+from shaped.autogen.models.query_result import QueryResult
+from shaped.autogen.models.saved_query_info_response import SavedQueryInfoResponse
+from shaped.autogen.models.saved_query_list_response import SavedQueryListResponse
+from shaped.autogen.models.saved_query_request import SavedQueryRequest
 
 from shaped.autogen.api_client import ApiClient, RequestSerialized
 from shaped.autogen.api_response import ApiResponse
 from shaped.autogen.rest import RESTResponseType
 
 
-class DatasetApi:
+class QueryApi:
     """NOTE: This class is auto generated by OpenAPI Generator
     Ref: https://openapi-generator.tech
 
@@ -42,10 +42,10 @@ class DatasetApi:
 
 
     @validate_call
-    def datasets_create_dataset_post(
+    def execute_ad_hoc_query_query_post(
         self,
-        x_api_key: Annotated[str, Field(min_length=40, strict=True, max_length=40)],
-        create_dataset_arguments: CreateDatasetArguments,
+        engine_name: StrictStr,
+        query_request: QueryRequest,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -58,15 +58,15 @@ class DatasetApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> SuccessfulResponse:
-        """Create Dataset
+    ) -> QueryResult:
+        """Execute Query
 
-        Create Dataset is used to provision a 'Shaped Dataset', which is a persisted view of external data. A 'Shaped Dataset' can be hydrated with data using the 'dataset_insert' endpoint. Or you can connect to a batch or streaming connector and we'll ingest the data continuously for you. Batch connectors ingest data at 15minute intervals, stream connectors ingest in real-time.  Here are the supported data connectors and their respective data ingestion type: * CUSTOM (batch or stream) * BIGQUERY (batch) * MONGODB (batch) * SNOWFLAKE (batch) * POSTGRES (batch) * MYSQL (batch) * REDSHIFT (batch) * SEGMENT (stream) * RUDDERSTACK (stream) * AMPLITUDE (stream) * AWS_PINPOINT (stream) * KINESIS (stream) 
+        Execute a query with the provided QueryConfig. Use this endpoint to execute on-demand queries for development or experimentation.
 
-        :param x_api_key: (required)
-        :type x_api_key: str
-        :param create_dataset_arguments: (required)
-        :type create_dataset_arguments: CreateDatasetArguments
+        :param engine_name: (required)
+        :type engine_name: str
+        :param query_request: (required)
+        :type query_request: QueryRequest
         :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
@@ -89,9 +89,9 @@ class DatasetApi:
         :return: Returns the result object.
         """ # noqa: E501
 
-        _param = self._datasets_create_dataset_post_serialize(
-            x_api_key=x_api_key,
-            create_dataset_arguments=create_dataset_arguments,
+        _param = self._execute_ad_hoc_query_query_post_serialize(
+            engine_name=engine_name,
+            query_request=query_request,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -99,7 +99,7 @@ class DatasetApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "SuccessfulResponse",
+            '200': "QueryResult",
             '422': "HTTPValidationError",
         }
         response_data = self.api_client.call_api(
@@ -114,10 +114,10 @@ class DatasetApi:
 
 
     @validate_call
-    def datasets_create_dataset_post_with_http_info(
+    def execute_ad_hoc_query_query_post_with_http_info(
         self,
-        x_api_key: Annotated[str, Field(min_length=40, strict=True, max_length=40)],
-        create_dataset_arguments: CreateDatasetArguments,
+        engine_name: StrictStr,
+        query_request: QueryRequest,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -130,15 +130,15 @@ class DatasetApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> ApiResponse[SuccessfulResponse]:
-        """Create Dataset
+    ) -> ApiResponse[QueryResult]:
+        """Execute Query
 
-        Create Dataset is used to provision a 'Shaped Dataset', which is a persisted view of external data. A 'Shaped Dataset' can be hydrated with data using the 'dataset_insert' endpoint. Or you can connect to a batch or streaming connector and we'll ingest the data continuously for you. Batch connectors ingest data at 15minute intervals, stream connectors ingest in real-time.  Here are the supported data connectors and their respective data ingestion type: * CUSTOM (batch or stream) * BIGQUERY (batch) * MONGODB (batch) * SNOWFLAKE (batch) * POSTGRES (batch) * MYSQL (batch) * REDSHIFT (batch) * SEGMENT (stream) * RUDDERSTACK (stream) * AMPLITUDE (stream) * AWS_PINPOINT (stream) * KINESIS (stream) 
+        Execute a query with the provided QueryConfig. Use this endpoint to execute on-demand queries for development or experimentation.
 
-        :param x_api_key: (required)
-        :type x_api_key: str
-        :param create_dataset_arguments: (required)
-        :type create_dataset_arguments: CreateDatasetArguments
+        :param engine_name: (required)
+        :type engine_name: str
+        :param query_request: (required)
+        :type query_request: QueryRequest
         :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
@@ -161,9 +161,9 @@ class DatasetApi:
         :return: Returns the result object.
         """ # noqa: E501
 
-        _param = self._datasets_create_dataset_post_serialize(
-            x_api_key=x_api_key,
-            create_dataset_arguments=create_dataset_arguments,
+        _param = self._execute_ad_hoc_query_query_post_serialize(
+            engine_name=engine_name,
+            query_request=query_request,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -171,7 +171,7 @@ class DatasetApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "SuccessfulResponse",
+            '200': "QueryResult",
             '422': "HTTPValidationError",
         }
         response_data = self.api_client.call_api(
@@ -186,10 +186,10 @@ class DatasetApi:
 
 
     @validate_call
-    def datasets_create_dataset_post_without_preload_content(
+    def execute_ad_hoc_query_query_post_without_preload_content(
         self,
-        x_api_key: Annotated[str, Field(min_length=40, strict=True, max_length=40)],
-        create_dataset_arguments: CreateDatasetArguments,
+        engine_name: StrictStr,
+        query_request: QueryRequest,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -203,14 +203,14 @@ class DatasetApi:
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
     ) -> RESTResponseType:
-        """Create Dataset
+        """Execute Query
 
-        Create Dataset is used to provision a 'Shaped Dataset', which is a persisted view of external data. A 'Shaped Dataset' can be hydrated with data using the 'dataset_insert' endpoint. Or you can connect to a batch or streaming connector and we'll ingest the data continuously for you. Batch connectors ingest data at 15minute intervals, stream connectors ingest in real-time.  Here are the supported data connectors and their respective data ingestion type: * CUSTOM (batch or stream) * BIGQUERY (batch) * MONGODB (batch) * SNOWFLAKE (batch) * POSTGRES (batch) * MYSQL (batch) * REDSHIFT (batch) * SEGMENT (stream) * RUDDERSTACK (stream) * AMPLITUDE (stream) * AWS_PINPOINT (stream) * KINESIS (stream) 
+        Execute a query with the provided QueryConfig. Use this endpoint to execute on-demand queries for development or experimentation.
 
-        :param x_api_key: (required)
-        :type x_api_key: str
-        :param create_dataset_arguments: (required)
-        :type create_dataset_arguments: CreateDatasetArguments
+        :param engine_name: (required)
+        :type engine_name: str
+        :param query_request: (required)
+        :type query_request: QueryRequest
         :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
@@ -233,9 +233,9 @@ class DatasetApi:
         :return: Returns the result object.
         """ # noqa: E501
 
-        _param = self._datasets_create_dataset_post_serialize(
-            x_api_key=x_api_key,
-            create_dataset_arguments=create_dataset_arguments,
+        _param = self._execute_ad_hoc_query_query_post_serialize(
+            engine_name=engine_name,
+            query_request=query_request,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -243,7 +243,7 @@ class DatasetApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "SuccessfulResponse",
+            '200': "QueryResult",
             '422': "HTTPValidationError",
         }
         response_data = self.api_client.call_api(
@@ -253,10 +253,10 @@ class DatasetApi:
         return response_data.response
 
 
-    def _datasets_create_dataset_post_serialize(
+    def _execute_ad_hoc_query_query_post_serialize(
         self,
-        x_api_key,
-        create_dataset_arguments,
+        engine_name,
+        query_request,
         _request_auth,
         _content_type,
         _headers,
@@ -278,14 +278,14 @@ class DatasetApi:
         _body_params: Optional[bytes] = None
 
         # process the path parameters
+        if engine_name is not None:
+            _path_params['engine_name'] = engine_name
         # process the query parameters
         # process the header parameters
-        if x_api_key is not None:
-            _header_params['x-api-key'] = x_api_key
         # process the form parameters
         # process the body parameter
-        if create_dataset_arguments is not None:
-            _body_params = create_dataset_arguments
+        if query_request is not None:
+            _body_params = query_request
 
 
         # set the HTTP header `Accept`
@@ -316,7 +316,7 @@ class DatasetApi:
 
         return self.api_client.param_serialize(
             method='POST',
-            resource_path='/datasets/',
+            resource_path='/engines/{engine_name}/query',
             path_params=_path_params,
             query_params=_query_params,
             header_params=_header_params,
@@ -333,10 +333,11 @@ class DatasetApi:
 
 
     @validate_call
-    def datasets_dataset_insert_post(
+    def execute_saved_query_queries_query_name_post(
         self,
-        x_api_key: Annotated[str, Field(min_length=40, strict=True, max_length=40)],
-        dataset_name: StrictStr,
+        engine_name: StrictStr,
+        query_name: StrictStr,
+        saved_query_request: SavedQueryRequest,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -349,15 +350,17 @@ class DatasetApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> InsertModelResponse:
-        """Dataset Insert
+    ) -> QueryResult:
+        """Execute Saved Query
 
-        Dataset insert allows you to push data payloads like events, users or items directly to a Dataset. This is useful for uploading local data to Shaped or inserting real-time data directly from your application. 
+        Execute a saved query by name. The `parameters` field defines inputs for the saved query, such as a search string or user ID.
 
-        :param x_api_key: (required)
-        :type x_api_key: str
-        :param dataset_name: (required)
-        :type dataset_name: str
+        :param engine_name: (required)
+        :type engine_name: str
+        :param query_name: (required)
+        :type query_name: str
+        :param saved_query_request: (required)
+        :type saved_query_request: SavedQueryRequest
         :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
@@ -380,9 +383,10 @@ class DatasetApi:
         :return: Returns the result object.
         """ # noqa: E501
 
-        _param = self._datasets_dataset_insert_post_serialize(
-            x_api_key=x_api_key,
-            dataset_name=dataset_name,
+        _param = self._execute_saved_query_queries_query_name_post_serialize(
+            engine_name=engine_name,
+            query_name=query_name,
+            saved_query_request=saved_query_request,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -390,7 +394,7 @@ class DatasetApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "InsertModelResponse",
+            '200': "QueryResult",
             '422': "HTTPValidationError",
         }
         response_data = self.api_client.call_api(
@@ -405,10 +409,11 @@ class DatasetApi:
 
 
     @validate_call
-    def datasets_dataset_insert_post_with_http_info(
+    def execute_saved_query_queries_query_name_post_with_http_info(
         self,
-        x_api_key: Annotated[str, Field(min_length=40, strict=True, max_length=40)],
-        dataset_name: StrictStr,
+        engine_name: StrictStr,
+        query_name: StrictStr,
+        saved_query_request: SavedQueryRequest,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -421,15 +426,17 @@ class DatasetApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> ApiResponse[InsertModelResponse]:
-        """Dataset Insert
+    ) -> ApiResponse[QueryResult]:
+        """Execute Saved Query
 
-        Dataset insert allows you to push data payloads like events, users or items directly to a Dataset. This is useful for uploading local data to Shaped or inserting real-time data directly from your application. 
+        Execute a saved query by name. The `parameters` field defines inputs for the saved query, such as a search string or user ID.
 
-        :param x_api_key: (required)
-        :type x_api_key: str
-        :param dataset_name: (required)
-        :type dataset_name: str
+        :param engine_name: (required)
+        :type engine_name: str
+        :param query_name: (required)
+        :type query_name: str
+        :param saved_query_request: (required)
+        :type saved_query_request: SavedQueryRequest
         :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
@@ -452,9 +459,10 @@ class DatasetApi:
         :return: Returns the result object.
         """ # noqa: E501
 
-        _param = self._datasets_dataset_insert_post_serialize(
-            x_api_key=x_api_key,
-            dataset_name=dataset_name,
+        _param = self._execute_saved_query_queries_query_name_post_serialize(
+            engine_name=engine_name,
+            query_name=query_name,
+            saved_query_request=saved_query_request,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -462,7 +470,7 @@ class DatasetApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "InsertModelResponse",
+            '200': "QueryResult",
             '422': "HTTPValidationError",
         }
         response_data = self.api_client.call_api(
@@ -477,10 +485,11 @@ class DatasetApi:
 
 
     @validate_call
-    def datasets_dataset_insert_post_without_preload_content(
+    def execute_saved_query_queries_query_name_post_without_preload_content(
         self,
-        x_api_key: Annotated[str, Field(min_length=40, strict=True, max_length=40)],
-        dataset_name: StrictStr,
+        engine_name: StrictStr,
+        query_name: StrictStr,
+        saved_query_request: SavedQueryRequest,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -494,14 +503,16 @@ class DatasetApi:
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
     ) -> RESTResponseType:
-        """Dataset Insert
+        """Execute Saved Query
 
-        Dataset insert allows you to push data payloads like events, users or items directly to a Dataset. This is useful for uploading local data to Shaped or inserting real-time data directly from your application. 
+        Execute a saved query by name. The `parameters` field defines inputs for the saved query, such as a search string or user ID.
 
-        :param x_api_key: (required)
-        :type x_api_key: str
-        :param dataset_name: (required)
-        :type dataset_name: str
+        :param engine_name: (required)
+        :type engine_name: str
+        :param query_name: (required)
+        :type query_name: str
+        :param saved_query_request: (required)
+        :type saved_query_request: SavedQueryRequest
         :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
@@ -524,9 +535,10 @@ class DatasetApi:
         :return: Returns the result object.
         """ # noqa: E501
 
-        _param = self._datasets_dataset_insert_post_serialize(
-            x_api_key=x_api_key,
-            dataset_name=dataset_name,
+        _param = self._execute_saved_query_queries_query_name_post_serialize(
+            engine_name=engine_name,
+            query_name=query_name,
+            saved_query_request=saved_query_request,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -534,7 +546,7 @@ class DatasetApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "InsertModelResponse",
+            '200': "QueryResult",
             '422': "HTTPValidationError",
         }
         response_data = self.api_client.call_api(
@@ -544,10 +556,11 @@ class DatasetApi:
         return response_data.response
 
 
-    def _datasets_dataset_insert_post_serialize(
+    def _execute_saved_query_queries_query_name_post_serialize(
         self,
-        x_api_key,
-        dataset_name,
+        engine_name,
+        query_name,
+        saved_query_request,
         _request_auth,
         _content_type,
         _headers,
@@ -569,14 +582,16 @@ class DatasetApi:
         _body_params: Optional[bytes] = None
 
         # process the path parameters
-        if dataset_name is not None:
-            _path_params['dataset_name'] = dataset_name
+        if engine_name is not None:
+            _path_params['engine_name'] = engine_name
+        if query_name is not None:
+            _path_params['query_name'] = query_name
         # process the query parameters
         # process the header parameters
-        if x_api_key is not None:
-            _header_params['x-api-key'] = x_api_key
         # process the form parameters
         # process the body parameter
+        if saved_query_request is not None:
+            _body_params = saved_query_request
 
 
         # set the HTTP header `Accept`
@@ -587,6 +602,19 @@ class DatasetApi:
                 ]
             )
 
+        # set the HTTP header `Content-Type`
+        if _content_type:
+            _header_params['Content-Type'] = _content_type
+        else:
+            _default_content_type = (
+                self.api_client.select_header_content_type(
+                    [
+                        'application/json'
+                    ]
+                )
+            )
+            if _default_content_type is not None:
+                _header_params['Content-Type'] = _default_content_type
 
         # authentication setting
         _auth_settings: List[str] = [
@@ -594,7 +622,7 @@ class DatasetApi:
 
         return self.api_client.param_serialize(
             method='POST',
-            resource_path='/datasets/{dataset_name}/insert',
+            resource_path='/engines/{engine_name}/queries/{query_name}',
             path_params=_path_params,
             query_params=_query_params,
             header_params=_header_params,
@@ -611,10 +639,10 @@ class DatasetApi:
 
 
     @validate_call
-    def datasets_dataset_name_delete_dataset(
+    def get_saved_query_info_queries_query_name_get(
         self,
-        x_api_key: Annotated[str, Field(min_length=40, strict=True, max_length=40)],
-        dataset_name: StrictStr,
+        engine_name: StrictStr,
+        query_name: StrictStr,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -627,15 +655,15 @@ class DatasetApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> SuccessfulResponse:
-        """Delete Dataset
+    ) -> SavedQueryInfoResponse:
+        """Get Saved Query Details
 
-        Delete the dataset with identifier: {dataset_name}. 
+        Get information about a specific saved query.
 
-        :param x_api_key: (required)
-        :type x_api_key: str
-        :param dataset_name: (required)
-        :type dataset_name: str
+        :param engine_name: (required)
+        :type engine_name: str
+        :param query_name: (required)
+        :type query_name: str
         :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
@@ -658,9 +686,9 @@ class DatasetApi:
         :return: Returns the result object.
         """ # noqa: E501
 
-        _param = self._datasets_dataset_name_delete_dataset_serialize(
-            x_api_key=x_api_key,
-            dataset_name=dataset_name,
+        _param = self._get_saved_query_info_queries_query_name_get_serialize(
+            engine_name=engine_name,
+            query_name=query_name,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -668,7 +696,7 @@ class DatasetApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "SuccessfulResponse",
+            '200': "SavedQueryInfoResponse",
             '422': "HTTPValidationError",
         }
         response_data = self.api_client.call_api(
@@ -683,10 +711,10 @@ class DatasetApi:
 
 
     @validate_call
-    def datasets_dataset_name_delete_dataset_with_http_info(
+    def get_saved_query_info_queries_query_name_get_with_http_info(
         self,
-        x_api_key: Annotated[str, Field(min_length=40, strict=True, max_length=40)],
-        dataset_name: StrictStr,
+        engine_name: StrictStr,
+        query_name: StrictStr,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -699,15 +727,15 @@ class DatasetApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> ApiResponse[SuccessfulResponse]:
-        """Delete Dataset
+    ) -> ApiResponse[SavedQueryInfoResponse]:
+        """Get Saved Query Details
 
-        Delete the dataset with identifier: {dataset_name}. 
+        Get information about a specific saved query.
 
-        :param x_api_key: (required)
-        :type x_api_key: str
-        :param dataset_name: (required)
-        :type dataset_name: str
+        :param engine_name: (required)
+        :type engine_name: str
+        :param query_name: (required)
+        :type query_name: str
         :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
@@ -730,9 +758,9 @@ class DatasetApi:
         :return: Returns the result object.
         """ # noqa: E501
 
-        _param = self._datasets_dataset_name_delete_dataset_serialize(
-            x_api_key=x_api_key,
-            dataset_name=dataset_name,
+        _param = self._get_saved_query_info_queries_query_name_get_serialize(
+            engine_name=engine_name,
+            query_name=query_name,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -740,7 +768,7 @@ class DatasetApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "SuccessfulResponse",
+            '200': "SavedQueryInfoResponse",
             '422': "HTTPValidationError",
         }
         response_data = self.api_client.call_api(
@@ -755,10 +783,10 @@ class DatasetApi:
 
 
     @validate_call
-    def datasets_dataset_name_delete_dataset_without_preload_content(
+    def get_saved_query_info_queries_query_name_get_without_preload_content(
         self,
-        x_api_key: Annotated[str, Field(min_length=40, strict=True, max_length=40)],
-        dataset_name: StrictStr,
+        engine_name: StrictStr,
+        query_name: StrictStr,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -772,14 +800,14 @@ class DatasetApi:
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
     ) -> RESTResponseType:
-        """Delete Dataset
+        """Get Saved Query Details
 
-        Delete the dataset with identifier: {dataset_name}. 
+        Get information about a specific saved query.
 
-        :param x_api_key: (required)
-        :type x_api_key: str
-        :param dataset_name: (required)
-        :type dataset_name: str
+        :param engine_name: (required)
+        :type engine_name: str
+        :param query_name: (required)
+        :type query_name: str
         :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
@@ -802,9 +830,9 @@ class DatasetApi:
         :return: Returns the result object.
         """ # noqa: E501
 
-        _param = self._datasets_dataset_name_delete_dataset_serialize(
-            x_api_key=x_api_key,
-            dataset_name=dataset_name,
+        _param = self._get_saved_query_info_queries_query_name_get_serialize(
+            engine_name=engine_name,
+            query_name=query_name,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -812,7 +840,7 @@ class DatasetApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "SuccessfulResponse",
+            '200': "SavedQueryInfoResponse",
             '422': "HTTPValidationError",
         }
         response_data = self.api_client.call_api(
@@ -822,10 +850,10 @@ class DatasetApi:
         return response_data.response
 
 
-    def _datasets_dataset_name_delete_dataset_serialize(
+    def _get_saved_query_info_queries_query_name_get_serialize(
         self,
-        x_api_key,
-        dataset_name,
+        engine_name,
+        query_name,
         _request_auth,
         _content_type,
         _headers,
@@ -847,12 +875,12 @@ class DatasetApi:
         _body_params: Optional[bytes] = None
 
         # process the path parameters
-        if dataset_name is not None:
-            _path_params['dataset_name'] = dataset_name
+        if engine_name is not None:
+            _path_params['engine_name'] = engine_name
+        if query_name is not None:
+            _path_params['query_name'] = query_name
         # process the query parameters
         # process the header parameters
-        if x_api_key is not None:
-            _header_params['x-api-key'] = x_api_key
         # process the form parameters
         # process the body parameter
 
@@ -871,8 +899,8 @@ class DatasetApi:
         ]
 
         return self.api_client.param_serialize(
-            method='DELETE',
-            resource_path='/datasets/{dataset_name}',
+            method='GET',
+            resource_path='/engines/{engine_name}/queries/{query_name}',
             path_params=_path_params,
             query_params=_query_params,
             header_params=_header_params,
@@ -889,9 +917,9 @@ class DatasetApi:
 
 
     @validate_call
-    def datasets_list_datasets_get(
+    def list_saved_queries_queries_get(
         self,
-        x_api_key: Annotated[str, Field(min_length=40, strict=True, max_length=40)],
+        engine_name: StrictStr,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -904,13 +932,13 @@ class DatasetApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> ListDatasetsResponse:
-        """List Datasets
+    ) -> SavedQueryListResponse:
+        """List Saved Queries
 
-        List Datasets returns information about all account datasets including schema and provisioning status. 
+        List all available saved queries for an engine.
 
-        :param x_api_key: (required)
-        :type x_api_key: str
+        :param engine_name: (required)
+        :type engine_name: str
         :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
@@ -933,8 +961,8 @@ class DatasetApi:
         :return: Returns the result object.
         """ # noqa: E501
 
-        _param = self._datasets_list_datasets_get_serialize(
-            x_api_key=x_api_key,
+        _param = self._list_saved_queries_queries_get_serialize(
+            engine_name=engine_name,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -942,8 +970,7 @@ class DatasetApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "ListDatasetsResponse",
-            '422': "HTTPValidationError",
+            '200': "SavedQueryListResponse",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -957,9 +984,9 @@ class DatasetApi:
 
 
     @validate_call
-    def datasets_list_datasets_get_with_http_info(
+    def list_saved_queries_queries_get_with_http_info(
         self,
-        x_api_key: Annotated[str, Field(min_length=40, strict=True, max_length=40)],
+        engine_name: StrictStr,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -972,13 +999,13 @@ class DatasetApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> ApiResponse[ListDatasetsResponse]:
-        """List Datasets
+    ) -> ApiResponse[SavedQueryListResponse]:
+        """List Saved Queries
 
-        List Datasets returns information about all account datasets including schema and provisioning status. 
+        List all available saved queries for an engine.
 
-        :param x_api_key: (required)
-        :type x_api_key: str
+        :param engine_name: (required)
+        :type engine_name: str
         :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
@@ -1001,8 +1028,8 @@ class DatasetApi:
         :return: Returns the result object.
         """ # noqa: E501
 
-        _param = self._datasets_list_datasets_get_serialize(
-            x_api_key=x_api_key,
+        _param = self._list_saved_queries_queries_get_serialize(
+            engine_name=engine_name,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -1010,8 +1037,7 @@ class DatasetApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "ListDatasetsResponse",
-            '422': "HTTPValidationError",
+            '200': "SavedQueryListResponse",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -1025,9 +1051,9 @@ class DatasetApi:
 
 
     @validate_call
-    def datasets_list_datasets_get_without_preload_content(
+    def list_saved_queries_queries_get_without_preload_content(
         self,
-        x_api_key: Annotated[str, Field(min_length=40, strict=True, max_length=40)],
+        engine_name: StrictStr,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -1041,12 +1067,12 @@ class DatasetApi:
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
     ) -> RESTResponseType:
-        """List Datasets
+        """List Saved Queries
 
-        List Datasets returns information about all account datasets including schema and provisioning status. 
+        List all available saved queries for an engine.
 
-        :param x_api_key: (required)
-        :type x_api_key: str
+        :param engine_name: (required)
+        :type engine_name: str
         :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
@@ -1069,8 +1095,8 @@ class DatasetApi:
         :return: Returns the result object.
         """ # noqa: E501
 
-        _param = self._datasets_list_datasets_get_serialize(
-            x_api_key=x_api_key,
+        _param = self._list_saved_queries_queries_get_serialize(
+            engine_name=engine_name,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -1078,8 +1104,7 @@ class DatasetApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "ListDatasetsResponse",
-            '422': "HTTPValidationError",
+            '200': "SavedQueryListResponse",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -1088,9 +1113,9 @@ class DatasetApi:
         return response_data.response
 
 
-    def _datasets_list_datasets_get_serialize(
+    def _list_saved_queries_queries_get_serialize(
         self,
-        x_api_key,
+        engine_name,
         _request_auth,
         _content_type,
         _headers,
@@ -1112,10 +1137,10 @@ class DatasetApi:
         _body_params: Optional[bytes] = None
 
         # process the path parameters
+        if engine_name is not None:
+            _path_params['engine_name'] = engine_name
         # process the query parameters
         # process the header parameters
-        if x_api_key is not None:
-            _header_params['x-api-key'] = x_api_key
         # process the form parameters
         # process the body parameter
 
@@ -1135,7 +1160,7 @@ class DatasetApi:
 
         return self.api_client.param_serialize(
             method='GET',
-            resource_path='/datasets/',
+            resource_path='/engines/{engine_name}/queries',
             path_params=_path_params,
             query_params=_query_params,
             header_params=_header_params,