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

shaped/autogen/models/__init__.py

@@ -6,7 +6,7 @@
 
     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: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.
@@ -17,7 +17,6 @@ from shaped.autogen.models.ai_enrichment_view_config import AIEnrichmentViewConf
 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
@@ -49,7 +48,6 @@ 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
@@ -60,12 +58,10 @@ 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
@@ -78,9 +74,11 @@ 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.filter_step_explanation import FilterStepExplanation
+from shaped.autogen.models.filter_table import FilterTable
+from shaped.autogen.models.filter_type import FilterType
 from shaped.autogen.models.global_filter import GlobalFilter
 from shaped.autogen.models.http_validation_error import HTTPValidationError
 from shaped.autogen.models.hidden_dropout_prob import HiddenDropoutProb
@@ -90,8 +88,10 @@ 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_entity_id import InnerEntityId
 from shaped.autogen.models.inner_size import InnerSize
 from shaped.autogen.models.inner_size1 import InnerSize1
+from shaped.autogen.models.inner_uid import InnerUid
 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
@@ -112,6 +112,7 @@ 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.location_inner import LocationInner
 from shaped.autogen.models.loss_types import LossTypes
 from shaped.autogen.models.lr import Lr
 from shaped.autogen.models.lr1 import Lr1
@@ -124,7 +125,7 @@ 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.models_inner import ModelsInner
 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
@@ -150,7 +151,6 @@ 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
@@ -159,7 +159,7 @@ from shaped.autogen.models.precomputed_item_embedding import PrecomputedItemEmbe
 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_config import QueryConfig
 from shaped.autogen.models.query_definition import QueryDefinition
 from shaped.autogen.models.query_encoder import QueryEncoder
 from shaped.autogen.models.query_explanation import QueryExplanation
@@ -169,18 +169,22 @@ 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.reorder_inner import ReorderInner
+from shaped.autogen.models.reorder_step_explanation import ReorderStepExplanation
 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.resource_config import ResourceConfig
 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.retrieval_scores_value import RetrievalScoresValue
+from shaped.autogen.models.retrieve_inner import RetrieveInner
+from shaped.autogen.models.retrieve_step_explanation import RetrieveStepExplanation
 from shaped.autogen.models.retriever import Retriever
 from shaped.autogen.models.retriever1 import Retriever1
 from shaped.autogen.models.rollout_config import RolloutConfig
@@ -194,8 +198,7 @@ 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.score_step_explanation import ScoreStepExplanation
 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
@@ -204,21 +207,15 @@ 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
@@ -226,23 +223,21 @@ from shaped.autogen.models.similarity_retrieve_step import SimilarityRetrieveSte
 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.steps_inner import StepsInner
 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.transform_status import TransformStatus
 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
@@ -250,20 +245,18 @@ 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.user_inner import UserInner
 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.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

shaped/autogen/models/ai_enrichment_view_config.py

@@ -5,7 +5,7 @@
 
     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: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.

shaped/autogen/models/algorithm.py

@@ -5,7 +5,7 @@
 
     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: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.

shaped/autogen/models/amplitude_table_config.py

@@ -5,7 +5,7 @@
 
     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: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.

shaped/autogen/models/ascending.py

@@ -13,19 +13,19 @@
 
 
 from __future__ import annotations
-from inspect import getfullargspec
+
 import json
 import pprint
 import re  # noqa: F401
-from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr, ValidationError, field_validator
-from typing import Optional
+from typing import TYPE_CHECKING, Any, Dict, Optional, Set, Union
+
+from pydantic import BaseModel, StrictBool, ValidationError, field_validator
 from shaped.autogen.models.tunable_bool import TunableBool
-from typing import Union, Any, List, Set, TYPE_CHECKING, Optional, Dict
-from typing_extensions import Literal, Self
-from pydantic import Field
+from typing_extensions import Self
 
 ASCENDING_ANY_OF_SCHEMAS = ["TunableBool", "bool"]
 
+
 class Ascending(BaseModel):
     """
     Ascending
@@ -39,7 +39,7 @@ class Ascending(BaseModel):
         actual_instance: Optional[Union[TunableBool, bool]] = None
     else:
         actual_instance: Any = None
-    any_of_schemas: Set[str] = { "TunableBool", "bool" }
+    any_of_schemas: Set[str] = {"TunableBool", "bool"}
 
     model_config = {
         "validate_assignment": True,
@@ -49,14 +49,18 @@ class Ascending(BaseModel):
     def __init__(self, *args, **kwargs) -> None:
         if args:
             if len(args) > 1:
-                raise ValueError("If a position argument is used, only 1 is allowed to set `actual_instance`")
+                raise ValueError(
+                    "If a position argument is used, only 1 is allowed to set `actual_instance`"
+                )
             if kwargs:
-                raise ValueError("If a position argument is used, keyword arguments cannot be used.")
+                raise ValueError(
+                    "If a position argument is used, keyword arguments cannot be used."
+                )
             super().__init__(actual_instance=args[0])
         else:
             super().__init__(**kwargs)
 
-    @field_validator('actual_instance')
+    @field_validator("actual_instance")
     def actual_instance_must_validate_anyof(cls, v):
         instance = Ascending.model_construct()
         error_messages = []
@@ -74,7 +78,10 @@ class Ascending(BaseModel):
 
         if error_messages:
             # no match
-            raise ValueError("No match found when setting the actual_instance in Ascending with anyOf schemas: TunableBool, bool. Details: " + ", ".join(error_messages))
+            raise ValueError(
+                "No match found when setting the actual_instance in Ascending with anyOf schemas: TunableBool, bool. Details: "
+                + ", ".join(error_messages)
+            )
         else:
             return v
 
@@ -101,11 +108,14 @@ class Ascending(BaseModel):
             instance.actual_instance = TunableBool.from_json(json_str)
             return instance
         except (ValidationError, ValueError) as e:
-             error_messages.append(str(e))
+            error_messages.append(str(e))
 
         if error_messages:
             # no match
-            raise ValueError("No match found when deserializing the JSON string into Ascending with anyOf schemas: TunableBool, bool. Details: " + ", ".join(error_messages))
+            raise ValueError(
+                "No match found when deserializing the JSON string into Ascending with anyOf schemas: TunableBool, bool. Details: "
+                + ", ".join(error_messages)
+            )
         else:
             return instance
 
@@ -114,7 +124,9 @@ class Ascending(BaseModel):
         if self.actual_instance is None:
             return "null"
 
-        if hasattr(self.actual_instance, "to_json") and callable(self.actual_instance.to_json):
+        if hasattr(self.actual_instance, "to_json") and callable(
+            self.actual_instance.to_json
+        ):
             return self.actual_instance.to_json()
         else:
             return json.dumps(self.actual_instance)
@@ -124,7 +136,9 @@ class Ascending(BaseModel):
         if self.actual_instance is None:
             return None
 
-        if hasattr(self.actual_instance, "to_dict") and callable(self.actual_instance.to_dict):
+        if hasattr(self.actual_instance, "to_dict") and callable(
+            self.actual_instance.to_dict
+        ):
             return self.actual_instance.to_dict()
         else:
             return self.actual_instance
@@ -132,5 +146,3 @@ class Ascending(BaseModel):
     def to_str(self) -> str:
         """Returns the string representation of the actual instance"""
         return pprint.pformat(self.model_dump())
-
-

shaped/autogen/models/attn_dropout_prob.py

@@ -5,7 +5,7 @@
 
     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: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.

shaped/autogen/models/attribute_journey.py

@@ -5,7 +5,7 @@
 
     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: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.

shaped/autogen/models/attribute_value.py

@@ -5,7 +5,7 @@
 
     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: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.

shaped/autogen/models/autoscaling_config.py

@@ -5,7 +5,7 @@
 
     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: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.

shaped/autogen/models/aws_pinpoint_table_config.py

@@ -5,7 +5,7 @@
 
     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: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.

shaped/autogen/models/batch_size.py

@@ -5,7 +5,7 @@
 
     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: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.

shaped/autogen/models/batch_size1.py

@@ -5,7 +5,7 @@
 
     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: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.

shaped/autogen/models/batch_size2.py

@@ -5,7 +5,7 @@
 
     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: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.

shaped/autogen/models/big_query_table_config.py

@@ -5,7 +5,7 @@
 
     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: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.

shaped/autogen/models/bm25.py

@@ -5,7 +5,7 @@
 
     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: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.

shaped/autogen/models/boosted_reorder_step.py

@@ -5,7 +5,7 @@
 
     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: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.

shaped/autogen/models/canary_rollout.py

@@ -5,7 +5,7 @@
 
     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: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.

shaped/autogen/models/candidate_attributes_retrieve_step.py

@@ -5,7 +5,7 @@
 
     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: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.
@@ -30,7 +30,6 @@ class CandidateAttributesRetrieveStep(BaseModel):
     limit: Optional[StrictInt] = None
     name: Optional[StrictStr] = None
     type: Optional[StrictStr] = Field(default='candidate_attributes', description="Retrieve step type discriminator.")
-    additional_properties: Dict[str, Any] = {}
     __properties: ClassVar[List[str]] = ["item_attributes", "limit", "name", "type"]
 
     @field_validator('type')
@@ -73,10 +72,8 @@ class CandidateAttributesRetrieveStep(BaseModel):
         * `None` is only added to the output dict for nullable fields that
           were set at model initialization. Other fields with value `None`
           are ignored.
-        * Fields in `self.additional_properties` are added to the output dict.
         """
         excluded_fields: Set[str] = set([
-            "additional_properties",
         ])
 
         _dict = self.model_dump(
@@ -84,11 +81,6 @@ class CandidateAttributesRetrieveStep(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
-        # puts key-value pairs in additional_properties in the top level
-        if self.additional_properties is not None:
-            for _key, _value in self.additional_properties.items():
-                _dict[_key] = _value
-
         # set to None if limit (nullable) is None
         # and model_fields_set contains the field
         if self.limit is None and "limit" in self.model_fields_set:
@@ -116,11 +108,6 @@ class CandidateAttributesRetrieveStep(BaseModel):
             "name": obj.get("name"),
             "type": obj.get("type") if obj.get("type") is not None else 'candidate_attributes'
         })
-        # store additional fields in additional_properties
-        for _key in obj.keys():
-            if _key not in cls.__properties:
-                _obj.additional_properties[_key] = obj.get(_key)
-
         return _obj
 
 

shaped/autogen/models/candidate_ids_retrieve_step.py

@@ -5,7 +5,7 @@
 
     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: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.

shaped/autogen/models/candidate_retrieval_strategy.py

@@ -5,7 +5,7 @@
 
     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: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.

shaped/autogen/models/clickhouse_table_config.py

@@ -5,7 +5,7 @@
 
     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: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.

shaped/autogen/models/column_order_retrieve_step.py

@@ -5,7 +5,7 @@
 
     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: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.

shaped/autogen/models/column_ordering.py

@@ -5,7 +5,7 @@
 
     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: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.

shaped/autogen/models/create_table_response.py

@@ -5,7 +5,7 @@
 
     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: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.

shaped/autogen/models/create_view_response.py

@@ -5,7 +5,7 @@
 
     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: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.