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

shaped/autogen/api_client.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.
@@ -21,6 +21,7 @@ import mimetypes
 import os
 import re
 import tempfile
+import uuid
 
 from urllib.parse import quote
 from typing import Tuple, Optional, List, Dict, Union
@@ -356,6 +357,8 @@ class ApiClient:
             return obj.get_secret_value()
         elif isinstance(obj, self.PRIMITIVE_TYPES):
             return obj
+        elif isinstance(obj, uuid.UUID):
+            return str(obj)
         elif isinstance(obj, list):
             return [
                 self.sanitize_for_serialization(sub_obj) for sub_obj in obj
@@ -382,6 +385,10 @@ class ApiClient:
             else:
                 obj_dict = obj.__dict__
 
+        if isinstance(obj_dict, list):
+            # here we handle instances that can either be a list or something else, and only became a real list by calling to_dict()
+            return self.sanitize_for_serialization(obj_dict)
+
         return {
             key: self.sanitize_for_serialization(val)
             for key, val in obj_dict.items()
@@ -404,7 +411,7 @@ class ApiClient:
                 data = json.loads(response_text)
             except ValueError:
                 data = response_text
-        elif re.match(r'^application/(json|[\w!#$&.+-^_]+\+json)\s*(;|$)', content_type, re.IGNORECASE):
+        elif re.match(r'^application/(json|[\w!#$&.+\-^_]+\+json)\s*(;|$)', content_type, re.IGNORECASE):
             if response_text == "":
                 data = ""
             else:
@@ -453,13 +460,13 @@ class ApiClient:
 
         if klass in self.PRIMITIVE_TYPES:
             return self.__deserialize_primitive(data, klass)
-        elif klass == object:
+        elif klass is object:
             return self.__deserialize_object(data)
-        elif klass == datetime.date:
+        elif klass is datetime.date:
             return self.__deserialize_date(data)
-        elif klass == datetime.datetime:
+        elif klass is datetime.datetime:
             return self.__deserialize_datetime(data)
-        elif klass == decimal.Decimal:
+        elif klass is decimal.Decimal:
             return decimal.Decimal(data)
         elif issubclass(klass, Enum):
             return self.__deserialize_enum(data, klass)
@@ -517,7 +524,7 @@ class ApiClient:
             if k in collection_formats:
                 collection_format = collection_formats[k]
                 if collection_format == 'multi':
-                    new_params.extend((k, str(value)) for value in v)
+                    new_params.extend((k, quote(str(value))) for value in v)
                 else:
                     if collection_format == 'ssv':
                         delimiter = ' '

shaped/autogen/configuration.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.
@@ -18,7 +18,7 @@ import logging
 from logging import FileHandler
 import multiprocessing
 import sys
-from typing import Any, ClassVar, Dict, List, Literal, Optional, TypedDict
+from typing import Any, ClassVar, Dict, List, Literal, Optional, TypedDict, Union
 from typing_extensions import NotRequired, Self
 
 import urllib3
@@ -160,6 +160,10 @@ class Configuration:
     :param ssl_ca_cert: str - the path to a file of concatenated CA certificates
       in PEM format.
     :param retries: Number of retries for API requests.
+    :param ca_cert_data: verify the peer using concatenated CA certificate data
+      in PEM (str) or DER (bytes) format.
+    :param cert_file: the path to a client certificate file, for mTLS.
+    :param key_file: the path to a client key file, for mTLS. 
 
     """
 
@@ -173,19 +177,22 @@ class Configuration:
         username: Optional[str]=None,
         password: Optional[str]=None,
         access_token: Optional[str]=None,
-        server_index: Optional[int]=None, 
+        server_index: Optional[int]=None,
         server_variables: Optional[ServerVariablesT]=None,
         server_operation_index: Optional[Dict[int, int]]=None,
         server_operation_variables: Optional[Dict[int, ServerVariablesT]]=None,
         ignore_operation_servers: bool=False,
         ssl_ca_cert: Optional[str]=None,
         retries: Optional[int] = None,
+        ca_cert_data: Optional[Union[str, bytes]] = None,
+        cert_file: Optional[str]=None,
+        key_file: Optional[str]=None,
         *,
         debug: Optional[bool] = None,
     ) -> None:
         """Constructor
         """
-        self._base_path = "https://api.shaped.ai/v1" if host is None else host
+        self._base_path = "https://api.shaped.ai/v2" if host is None else host
         """Default Base url
         """
         self.server_index = 0 if server_index is None and host is None else server_index
@@ -257,10 +264,14 @@ class Configuration:
         self.ssl_ca_cert = ssl_ca_cert
         """Set this to customize the certificate file to verify the peer.
         """
-        self.cert_file = None
+        self.ca_cert_data = ca_cert_data
+        """Set this to verify the peer using PEM (str) or DER (bytes)
+           certificate data.
+        """
+        self.cert_file = cert_file
         """client certificate file
         """
-        self.key_file = None
+        self.key_file = key_file
         """client key file
         """
         self.assert_hostname = None
@@ -493,7 +504,7 @@ class Configuration:
         return "Python SDK Debug Report:\n"\
                "OS: {env}\n"\
                "Python Version: {pyversion}\n"\
-               "Version of the API: 1.0.1\n"\
+               "Version of the API: 2.0.0\n"\
                "SDK Package Version: 1.0.0".\
                format(env=sys.platform, pyversion=sys.version)
 
@@ -504,7 +515,7 @@ class Configuration:
         """
         return [
             {
-                'url': "https://api.shaped.ai/v1",
+                'url': "https://api.shaped.ai/v2",
                 'description': "No description provided",
             }
         ]

shaped/autogen/exceptions.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.
@@ -150,6 +150,13 @@ class ApiException(OpenApiException):
         if http_resp.status == 404:
             raise NotFoundException(http_resp=http_resp, body=body, data=data)
 
+        # Added new conditions for 409 and 422
+        if http_resp.status == 409:
+            raise ConflictException(http_resp=http_resp, body=body, data=data)
+
+        if http_resp.status == 422:
+            raise UnprocessableEntityException(http_resp=http_resp, body=body, data=data)
+
         if 500 <= http_resp.status <= 599:
             raise ServiceException(http_resp=http_resp, body=body, data=data)
         raise ApiException(http_resp=http_resp, body=body, data=data)
@@ -188,6 +195,16 @@ class ServiceException(ApiException):
     pass
 
 
+class ConflictException(ApiException):
+    """Exception for HTTP 409 Conflict."""
+    pass
+
+
+class UnprocessableEntityException(ApiException):
+    """Exception for HTTP 422 Unprocessable Entity."""
+    pass
+
+
 def render_path(path_to_item):
     """Returns a string representation of a path"""
     result = ""

shaped/autogen/models/__init__.py

@@ -4,65 +4,267 @@
 """
     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.
 """  # noqa: E501
 
-
 # import models into model package
-from shaped.autogen.models.aws_pinpoint_dataset_config import AWSPinpointDatasetConfig
-from shaped.autogen.models.amplitude_dataset_config import AmplitudeDatasetConfig
-from shaped.autogen.models.big_query_dataset_config import BigQueryDatasetConfig
-from shaped.autogen.models.complement_items_request import ComplementItemsRequest
-from shaped.autogen.models.complement_items_response import ComplementItemsResponse
-from shaped.autogen.models.connectors_inner import ConnectorsInner
-from shaped.autogen.models.create_dataset_arguments import CreateDatasetArguments
-from shaped.autogen.models.create_embedding_response import CreateEmbeddingResponse
-from shaped.autogen.models.create_item_embedding_request import CreateItemEmbeddingRequest
-from shaped.autogen.models.create_model_arguments import CreateModelArguments
-from shaped.autogen.models.create_model_response import CreateModelResponse
-from shaped.autogen.models.create_user_embedding_request import CreateUserEmbeddingRequest
-from shaped.autogen.models.custom_dataset_config import CustomDatasetConfig
-from shaped.autogen.models.dataset_config import DatasetConfig
-from shaped.autogen.models.dataset_schema_type import DatasetSchemaType
-from shaped.autogen.models.datasets_inner import DatasetsInner
-from shaped.autogen.models.delete_model_response import DeleteModelResponse
-from shaped.autogen.models.fetch_config import FetchConfig
-from shaped.autogen.models.file_config import FileConfig
-from shaped.autogen.models.file_source_config import FileSourceConfig
+from shaped.autogen.models.ai_enrichment_view_config import AIEnrichmentViewConfig
+from shaped.autogen.models.aws_pinpoint_table_config import AWSPinpointTableConfig
+from shaped.autogen.models.algorithm import Algorithm
+from shaped.autogen.models.amplitude_table_config import AmplitudeTableConfig
+from shaped.autogen.models.ascending import Ascending
+from shaped.autogen.models.attn_dropout_prob import AttnDropoutProb
+from shaped.autogen.models.attribute_journey import AttributeJourney
+from shaped.autogen.models.attribute_value import AttributeValue
+from shaped.autogen.models.autoscaling_config import AutoscalingConfig
+from shaped.autogen.models.batch_size import BatchSize
+from shaped.autogen.models.batch_size1 import BatchSize1
+from shaped.autogen.models.batch_size2 import BatchSize2
+from shaped.autogen.models.big_query_table_config import BigQueryTableConfig
+from shaped.autogen.models.bm25 import Bm25
+from shaped.autogen.models.boosted_reorder_step import BoostedReorderStep
+from shaped.autogen.models.canary_rollout import CanaryRollout
+from shaped.autogen.models.candidate_attributes_retrieve_step import CandidateAttributesRetrieveStep
+from shaped.autogen.models.candidate_ids_retrieve_step import CandidateIdsRetrieveStep
+from shaped.autogen.models.candidate_retrieval_strategy import CandidateRetrievalStrategy
+from shaped.autogen.models.clickhouse_table_config import ClickhouseTableConfig
+from shaped.autogen.models.column_order_retrieve_step import ColumnOrderRetrieveStep
+from shaped.autogen.models.column_ordering import ColumnOrdering
+from shaped.autogen.models.create_table_response import CreateTableResponse
+from shaped.autogen.models.create_view_response import CreateViewResponse
+from shaped.autogen.models.custom_table_config import CustomTableConfig
+from shaped.autogen.models.data_compute_config import DataComputeConfig
+from shaped.autogen.models.data_config import DataConfig
+from shaped.autogen.models.data_config_interaction_table import DataConfigInteractionTable
+from shaped.autogen.models.data_split_config import DataSplitConfig
+from shaped.autogen.models.data_split_strategy import DataSplitStrategy
+from shaped.autogen.models.data_tier import DataTier
+from shaped.autogen.models.default import Default
+from shaped.autogen.models.delete_engine_response import DeleteEngineResponse
+from shaped.autogen.models.delete_table_response import DeleteTableResponse
+from shaped.autogen.models.delete_view_response import DeleteViewResponse
+from shaped.autogen.models.deployment_config import DeploymentConfig
+from shaped.autogen.models.distance_function import DistanceFunction
+from shaped.autogen.models.diversity_reorder_step import DiversityReorderStep
+from shaped.autogen.models.dropout_rate import DropoutRate
+from shaped.autogen.models.dynamo_db_table_config import DynamoDBTableConfig
+from shaped.autogen.models.dynamo_db_table_config_scan_kwargs_value import DynamoDBTableConfigScanKwargsValue
+from shaped.autogen.models.embedder_batch_size import EmbedderBatchSize
+from shaped.autogen.models.embedding_config import EmbeddingConfig
+from shaped.autogen.models.embedding_dim import EmbeddingDim
+from shaped.autogen.models.embedding_dims import EmbeddingDims
+from shaped.autogen.models.embedding_size import EmbeddingSize
+from shaped.autogen.models.encoder import Encoder
+from shaped.autogen.models.encoding_pooling_strategy import EncodingPoolingStrategy
+from shaped.autogen.models.engine import Engine
+from shaped.autogen.models.engine_config_v2 import EngineConfigV2
+from shaped.autogen.models.engine_details_response import EngineDetailsResponse
+from shaped.autogen.models.engine_schema import EngineSchema
+from shaped.autogen.models.engine_schema_user_inner import EngineSchemaUserInner
+from shaped.autogen.models.entity_config import EntityConfig
+from shaped.autogen.models.entity_journey import EntityJourney
+from shaped.autogen.models.entity_type import EntityType
+from shaped.autogen.models.evaluation_config import EvaluationConfig
+from shaped.autogen.models.exploration_reorder_step import ExplorationReorderStep
+from shaped.autogen.models.expression_filter_step import ExpressionFilterStep
+from shaped.autogen.models.factors import Factors
+from shaped.autogen.models.factors1 import Factors1
+from shaped.autogen.models.feature import Feature
+from shaped.autogen.models.feature_type import FeatureType
+from shaped.autogen.models.file_table_config import FileTableConfig
+from shaped.autogen.models.filter_config import FilterConfig
+from shaped.autogen.models.filter_dataset import FilterDataset
+from shaped.autogen.models.filter_index_type import FilterIndexType
+from shaped.autogen.models.filter_retrieve_step import FilterRetrieveStep
+from shaped.autogen.models.global_filter import GlobalFilter
 from shaped.autogen.models.http_validation_error import HTTPValidationError
-from shaped.autogen.models.inference_config import InferenceConfig
-from shaped.autogen.models.insert_model_response import InsertModelResponse
-from shaped.autogen.models.interaction import Interaction
-from shaped.autogen.models.list_datasets_response import ListDatasetsResponse
-from shaped.autogen.models.list_models_response import ListModelsResponse
-from shaped.autogen.models.model_config import ModelConfig
-from shaped.autogen.models.model_response import ModelResponse
-from shaped.autogen.models.mongo_db_dataset_config import MongoDBDatasetConfig
-from shaped.autogen.models.my_sql_dataset_config import MySQLDatasetConfig
-from shaped.autogen.models.path import Path
-from shaped.autogen.models.post_rank_request import PostRankRequest
-from shaped.autogen.models.postgres_dataset_config import PostgresDatasetConfig
-from shaped.autogen.models.rank_attribute_response import RankAttributeResponse
-from shaped.autogen.models.rank_grid_attribute_request import RankGridAttributeRequest
-from shaped.autogen.models.rank_grid_attribute_request1 import RankGridAttributeRequest1
-from shaped.autogen.models.rank_grid_attribute_response import RankGridAttributeResponse
-from shaped.autogen.models.rank_response import RankResponse
-from shaped.autogen.models.redshift_dataset_config import RedshiftDatasetConfig
-from shaped.autogen.models.retrieve_request import RetrieveRequest
-from shaped.autogen.models.retrieve_response import RetrieveResponse
-from shaped.autogen.models.retriever_top_k_override import RetrieverTopKOverride
-from shaped.autogen.models.rudder_stack_dataset_config import RudderStackDatasetConfig
-from shaped.autogen.models.segment_dataset_config import SegmentDatasetConfig
-from shaped.autogen.models.similar_item_request import SimilarItemRequest
-from shaped.autogen.models.similar_response import SimilarResponse
-from shaped.autogen.models.similar_users_request import SimilarUsersRequest
-from shaped.autogen.models.snowflake_dataset_config import SnowflakeDatasetConfig
-from shaped.autogen.models.successful_response import SuccessfulResponse
+from shaped.autogen.models.hidden_dropout_prob import HiddenDropoutProb
+from shaped.autogen.models.hidden_size import HiddenSize
+from shaped.autogen.models.hidden_size1 import HiddenSize1
+from shaped.autogen.models.http_problem_response import HttpProblemResponse
+from shaped.autogen.models.hugging_face_encoder import HuggingFaceEncoder
+from shaped.autogen.models.iceberg_table_config import IcebergTableConfig
+from shaped.autogen.models.index_config import IndexConfig
+from shaped.autogen.models.inner_size import InnerSize
+from shaped.autogen.models.inner_size1 import InnerSize1
+from shaped.autogen.models.interaction_config import InteractionConfig
+from shaped.autogen.models.interaction_pooling_encoder import InteractionPoolingEncoder
+from shaped.autogen.models.interaction_round_robin_encoder import InteractionRoundRobinEncoder
+from shaped.autogen.models.item_attribute_pooling_encoder import ItemAttributePoolingEncoder
+from shaped.autogen.models.journey import Journey
+from shaped.autogen.models.kafka_table_config import KafkaTableConfig
+from shaped.autogen.models.kinesis_table_config import KinesisTableConfig
+from shaped.autogen.models.kinesis_table_config_column_schema_value import KinesisTableConfigColumnSchemaValue
+from shaped.autogen.models.label import Label
+from shaped.autogen.models.label_type import LabelType
+from shaped.autogen.models.laplace_smoothing import LaplaceSmoothing
+from shaped.autogen.models.latency_scaling_policy import LatencyScalingPolicy
+from shaped.autogen.models.learning_rate import LearningRate
+from shaped.autogen.models.learning_rate1 import LearningRate1
+from shaped.autogen.models.learning_rate2 import LearningRate2
+from shaped.autogen.models.learning_rate3 import LearningRate3
+from shaped.autogen.models.lexical_search_mode import LexicalSearchMode
+from shaped.autogen.models.list_engines_response import ListEnginesResponse
+from shaped.autogen.models.list_tables_response import ListTablesResponse
+from shaped.autogen.models.list_views_response import ListViewsResponse
+from shaped.autogen.models.loss_types import LossTypes
+from shaped.autogen.models.lr import Lr
+from shaped.autogen.models.lr1 import Lr1
+from shaped.autogen.models.lr2 import Lr2
+from shaped.autogen.models.mssql_table_config import MSSQLTableConfig
+from shaped.autogen.models.max_depth import MaxDepth
+from shaped.autogen.models.max_leaves import MaxLeaves
+from shaped.autogen.models.max_seq_length import MaxSeqLength
+from shaped.autogen.models.max_seq_length1 import MaxSeqLength1
+from shaped.autogen.models.max_seq_length2 import MaxSeqLength2
+from shaped.autogen.models.mode import Mode
+from shaped.autogen.models.mode1 import Mode1
+from shaped.autogen.models.mode2 import Mode2
+from shaped.autogen.models.mongo_db_table_config import MongoDBTableConfig
+from shaped.autogen.models.my_sql_table_config import MySQLTableConfig
+from shaped.autogen.models.n_epochs import NEpochs
+from shaped.autogen.models.n_epochs1 import NEpochs1
+from shaped.autogen.models.n_epochs2 import NEpochs2
+from shaped.autogen.models.n_estimators import NEstimators
+from shaped.autogen.models.n_heads import NHeads
+from shaped.autogen.models.n_layers import NLayers
+from shaped.autogen.models.neg_per_positive import NegPerPositive
+from shaped.autogen.models.negative_samples_count import NegativeSamplesCount
+from shaped.autogen.models.ngram_tokenizer import NgramTokenizer
+from shaped.autogen.models.no_op_config import NoOpConfig
+from shaped.autogen.models.num_blocks import NumBlocks
+from shaped.autogen.models.num_heads import NumHeads
+from shaped.autogen.models.num_leaves import NumLeaves
+from shaped.autogen.models.objective import Objective
+from shaped.autogen.models.objective1 import Objective1
+from shaped.autogen.models.online_store_config import OnlineStoreConfig
+from shaped.autogen.models.pagination_config import PaginationConfig
+from shaped.autogen.models.parameter_definition import ParameterDefinition
+from shaped.autogen.models.parameters_value import ParametersValue
+from shaped.autogen.models.passthrough_score import PassthroughScore
+from shaped.autogen.models.personal_filter import PersonalFilter
+from shaped.autogen.models.pipeline_stage_explanation import PipelineStageExplanation
+from shaped.autogen.models.policy import Policy
+from shaped.autogen.models.pool_fn import PoolFn
+from shaped.autogen.models.pooling_function import PoolingFunction
+from shaped.autogen.models.postgres_table_config import PostgresTableConfig
+from shaped.autogen.models.posthog_table_config import PosthogTableConfig
+from shaped.autogen.models.prebuilt_filter_step import PrebuiltFilterStep
+from shaped.autogen.models.precomputed_item_embedding import PrecomputedItemEmbedding
+from shaped.autogen.models.precomputed_user_embedding import PrecomputedUserEmbedding
+from shaped.autogen.models.query import Query
+from shaped.autogen.models.query1 import Query1
+from shaped.autogen.models.query_any_of import QueryAnyOf
+from shaped.autogen.models.query_definition import QueryDefinition
+from shaped.autogen.models.query_encoder import QueryEncoder
+from shaped.autogen.models.query_explanation import QueryExplanation
+from shaped.autogen.models.query_request import QueryRequest
+from shaped.autogen.models.query_result import QueryResult
+from shaped.autogen.models.query_table_config import QueryTableConfig
+from shaped.autogen.models.rank_item_attribute_values_query_config import RankItemAttributeValuesQueryConfig
+from shaped.autogen.models.rank_query_config import RankQueryConfig
+from shaped.autogen.models.rank_query_config_filter_inner import RankQueryConfigFilterInner
+from shaped.autogen.models.rank_query_config_reorder_inner import RankQueryConfigReorderInner
+from shaped.autogen.models.rank_query_config_retrieve_inner import RankQueryConfigRetrieveInner
+from shaped.autogen.models.recreate_rollout import RecreateRollout
+from shaped.autogen.models.redshift_table_config import RedshiftTableConfig
+from shaped.autogen.models.reference_table_config import ReferenceTableConfig
+from shaped.autogen.models.regularization import Regularization
+from shaped.autogen.models.request import Request
+from shaped.autogen.models.request1 import Request1
+from shaped.autogen.models.requests_per_second_scaling_policy import RequestsPerSecondScalingPolicy
+from shaped.autogen.models.response_get_view_details_views_view_name_get import ResponseGetViewDetailsViewsViewNameGet
+from shaped.autogen.models.result import Result
+from shaped.autogen.models.result_embeddings_value import ResultEmbeddingsValue
+from shaped.autogen.models.retriever import Retriever
+from shaped.autogen.models.retriever1 import Retriever1
+from shaped.autogen.models.rollout_config import RolloutConfig
+from shaped.autogen.models.rudderstack_table_config import RudderstackTableConfig
+from shaped.autogen.models.sql_transform_type import SQLTransformType
+from shaped.autogen.models.sql_view_config import SQLViewConfig
+from shaped.autogen.models.sampling_strategy import SamplingStrategy
+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.models.schema_config import SchemaConfig
+from shaped.autogen.models.score import Score
+from shaped.autogen.models.score_ensemble import ScoreEnsemble
+from shaped.autogen.models.score_ensemble_policy_config import ScoreEnsemblePolicyConfig
+from shaped.autogen.models.score_ensemble_policy_config_policies_inner import ScoreEnsemblePolicyConfigPoliciesInner
+from shaped.autogen.models.search_config import SearchConfig
+from shaped.autogen.models.segment_table_config import SegmentTableConfig
+from shaped.autogen.models.sequence_length import SequenceLength
+from shaped.autogen.models.server_config import ServerConfig
+from shaped.autogen.models.setup_engine_response import SetupEngineResponse
+from shaped.autogen.models.shaped_internal_recsys_policies_als_model_policy_als_model_policy_config import ShapedInternalRecsysPoliciesAlsModelPolicyALSModelPolicyConfig
+from shaped.autogen.models.shaped_internal_recsys_policies_beeformer_model_policy_beeformer_model_policy_beeformer_model_policy_config import ShapedInternalRecsysPoliciesBeeformerModelPolicyBeeformerModelPolicyBeeformerModelPolicyConfig
+from shaped.autogen.models.shaped_internal_recsys_policies_bert_model_policy_bert_model_policy_bert_model_policy_config import ShapedInternalRecsysPoliciesBertModelPolicyBertModelPolicyBERTModelPolicyConfig
+from shaped.autogen.models.shaped_internal_recsys_policies_chronological_model_policy_chronological_model_policy_config import ShapedInternalRecsysPoliciesChronologicalModelPolicyChronologicalModelPolicyConfig
+from shaped.autogen.models.shaped_internal_recsys_policies_elsa_model_policy_elsa_model_policy_elsa_model_policy_config import ShapedInternalRecsysPoliciesElsaModelPolicyElsaModelPolicyELSAModelPolicyConfig
+from shaped.autogen.models.shaped_internal_recsys_policies_gsasrec_model_policy_gsasrec_model_policy_gsas_rec_model_policy_config import ShapedInternalRecsysPoliciesGsasrecModelPolicyGsasrecModelPolicyGSASRecModelPolicyConfig
+from shaped.autogen.models.shaped_internal_recsys_policies_item2vec_model_policy_item2_vec_model_policy_config import ShapedInternalRecsysPoliciesItem2vecModelPolicyItem2VecModelPolicyConfig
+from shaped.autogen.models.shaped_internal_recsys_policies_item_content_similarity_model_policy_item_content_similarity_model_policy_config import ShapedInternalRecsysPoliciesItemContentSimilarityModelPolicyItemContentSimilarityModelPolicyConfig
+from shaped.autogen.models.shaped_internal_recsys_policies_lightgbm_model_policy_lightgbm_model_policy_light_gbm_model_policy_config import ShapedInternalRecsysPoliciesLightgbmModelPolicyLightgbmModelPolicyLightGBMModelPolicyConfig
+from shaped.autogen.models.shaped_internal_recsys_policies_ngram_model_policy_ngram_model_policy_config import ShapedInternalRecsysPoliciesNgramModelPolicyNgramModelPolicyConfig
+from shaped.autogen.models.shaped_internal_recsys_policies_popular_model_policy_popular_model_policy_config import ShapedInternalRecsysPoliciesPopularModelPolicyPopularModelPolicyConfig
+from shaped.autogen.models.shaped_internal_recsys_policies_random_model_policy_random_model_policy_config import ShapedInternalRecsysPoliciesRandomModelPolicyRandomModelPolicyConfig
+from shaped.autogen.models.shaped_internal_recsys_policies_recently_popular_policy_recently_popular_policy_config import ShapedInternalRecsysPoliciesRecentlyPopularPolicyRecentlyPopularPolicyConfig
+from shaped.autogen.models.shaped_internal_recsys_policies_rising_popular_policy_rising_popular_policy_config import ShapedInternalRecsysPoliciesRisingPopularPolicyRisingPopularPolicyConfig
+from shaped.autogen.models.shaped_internal_recsys_policies_sasrec_model_policy_sasrec_model_policy_sas_rec_model_policy_config import ShapedInternalRecsysPoliciesSasrecModelPolicySasrecModelPolicySASRecModelPolicyConfig
+from shaped.autogen.models.shaped_internal_recsys_policies_svd_model_policy_svd_model_policy_config import ShapedInternalRecsysPoliciesSvdModelPolicySVDModelPolicyConfig
+from shaped.autogen.models.shaped_internal_recsys_policies_two_tower_model_policy_two_tower_model_policy_two_tower_model_policy_config import ShapedInternalRecsysPoliciesTwoTowerModelPolicyTwoTowerModelPolicyTwoTowerModelPolicyConfig
+from shaped.autogen.models.shaped_internal_recsys_policies_user_item_content_similarity_model_policy_user_item_content_similarity_model_policy_config import ShapedInternalRecsysPoliciesUserItemContentSimilarityModelPolicyUserItemContentSimilarityModelPolicyConfig
+from shaped.autogen.models.shaped_internal_recsys_policies_widedeep_model_policy_wide_deep_model_policy_config import ShapedInternalRecsysPoliciesWidedeepModelPolicyWideDeepModelPolicyConfig
+from shaped.autogen.models.shaped_internal_recsys_policies_xgboost_model_policy_xg_boost_model_policy_config import ShapedInternalRecsysPoliciesXgboostModelPolicyXGBoostModelPolicyConfig
+from shaped.autogen.models.shopify_table_config import ShopifyTableConfig
+from shaped.autogen.models.similarity_retrieve_step import SimilarityRetrieveStep
+from shaped.autogen.models.snowflake_table_config import SnowflakeTableConfig
+from shaped.autogen.models.stemmer_tokenizer import StemmerTokenizer
+from shaped.autogen.models.step_explanation import StepExplanation
+from shaped.autogen.models.strategy import Strategy
+from shaped.autogen.models.table import Table
+from shaped.autogen.models.table_deployment_type import TableDeploymentType
+from shaped.autogen.models.table_insert_arguments import TableInsertArguments
+from shaped.autogen.models.table_insert_response import TableInsertResponse
+from shaped.autogen.models.text_encoding import TextEncoding
+from shaped.autogen.models.text_search_retrieve_step import TextSearchRetrieveStep
+from shaped.autogen.models.time_frequency import TimeFrequency
+from shaped.autogen.models.time_window import TimeWindow
+from shaped.autogen.models.time_window_in_days import TimeWindowInDays
+from shaped.autogen.models.tokenizer import Tokenizer
+from shaped.autogen.models.trained_model_encoder import TrainedModelEncoder
+from shaped.autogen.models.training_compute_config import TrainingComputeConfig
+from shaped.autogen.models.training_config import TrainingConfig
+from shaped.autogen.models.training_config_models_inner import TrainingConfigModelsInner
+from shaped.autogen.models.training_strategy import TrainingStrategy
+from shaped.autogen.models.trending_mode import TrendingMode
+from shaped.autogen.models.truncate_filter_step import TruncateFilterStep
+from shaped.autogen.models.tunable_bool import TunableBool
+from shaped.autogen.models.tunable_float import TunableFloat
+from shaped.autogen.models.tunable_int import TunableInt
+from shaped.autogen.models.tunable_int_categorical import TunableIntCategorical
+from shaped.autogen.models.tunable_string import TunableString
+from shaped.autogen.models.tuning_config import TuningConfig
+from shaped.autogen.models.type import Type
+from shaped.autogen.models.update_table_response import UpdateTableResponse
+from shaped.autogen.models.update_view_response import UpdateViewResponse
+from shaped.autogen.models.user_attribute_pooling_encoder import UserAttributePoolingEncoder
+from shaped.autogen.models.val_split import ValSplit
 from shaped.autogen.models.validation_error import ValidationError
+from shaped.autogen.models.validation_error_loc_inner import ValidationErrorLocInner
 from shaped.autogen.models.value_type import ValueType
-from shaped.autogen.models.view_model_response import ViewModelResponse
+from shaped.autogen.models.vector_search_mode import VectorSearchMode
+from shaped.autogen.models.view import View
+from shaped.autogen.models.view_details_ai import ViewDetailsAI
+from shaped.autogen.models.view_details_ai_schema_value import ViewDetailsAISchemaValue
+from shaped.autogen.models.view_details_sql import ViewDetailsSQL
+from shaped.autogen.models.view_status import ViewStatus
+from shaped.autogen.models.weight_decay import WeightDecay
+from shaped.autogen.models.whitespace_tokenizer import WhitespaceTokenizer
+from shaped.autogen.models.window_size import WindowSize
+