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

shaped/autogen/models/rank_query_config_reorder_inner.py

@@ -13,20 +13,24 @@
 
 
 from __future__ import annotations
-from inspect import getfullargspec
+
 import json
 import pprint
 import re  # noqa: F401
-from pydantic import BaseModel, ConfigDict, Field, StrictStr, ValidationError, field_validator
-from typing import Optional
+from typing import TYPE_CHECKING, Any, Dict, Optional, Set, Union
+
+from pydantic import BaseModel, ValidationError, field_validator
 from shaped.autogen.models.boosted_reorder_step import BoostedReorderStep
 from shaped.autogen.models.diversity_reorder_step import DiversityReorderStep
 from shaped.autogen.models.exploration_reorder_step import ExplorationReorderStep
-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
+
+RANKQUERYCONFIGREORDERINNER_ANY_OF_SCHEMAS = [
+    "BoostedReorderStep",
+    "DiversityReorderStep",
+    "ExplorationReorderStep",
+]
 
-RANKQUERYCONFIGREORDERINNER_ANY_OF_SCHEMAS = ["BoostedReorderStep", "DiversityReorderStep", "ExplorationReorderStep"]
 
 class RankQueryConfigReorderInner(BaseModel):
     """
@@ -40,10 +44,16 @@ class RankQueryConfigReorderInner(BaseModel):
     # data type: DiversityReorderStep
     anyof_schema_3_validator: Optional[DiversityReorderStep] = None
     if TYPE_CHECKING:
-        actual_instance: Optional[Union[BoostedReorderStep, DiversityReorderStep, ExplorationReorderStep]] = None
+        actual_instance: Optional[
+            Union[BoostedReorderStep, DiversityReorderStep, ExplorationReorderStep]
+        ] = None
     else:
         actual_instance: Any = None
-    any_of_schemas: Set[str] = { "BoostedReorderStep", "DiversityReorderStep", "ExplorationReorderStep" }
+    any_of_schemas: Set[str] = {
+        "BoostedReorderStep",
+        "DiversityReorderStep",
+        "ExplorationReorderStep",
+    }
 
     model_config = {
         "validate_assignment": True,
@@ -53,38 +63,51 @@ class RankQueryConfigReorderInner(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 = RankQueryConfigReorderInner.model_construct()
         error_messages = []
         # validate data type: ExplorationReorderStep
         if not isinstance(v, ExplorationReorderStep):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `ExplorationReorderStep`")
+            error_messages.append(
+                f"Error! Input type `{type(v)}` is not `ExplorationReorderStep`"
+            )
         else:
             return v
 
         # validate data type: BoostedReorderStep
         if not isinstance(v, BoostedReorderStep):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `BoostedReorderStep`")
+            error_messages.append(
+                f"Error! Input type `{type(v)}` is not `BoostedReorderStep`"
+            )
         else:
             return v
 
         # validate data type: DiversityReorderStep
         if not isinstance(v, DiversityReorderStep):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `DiversityReorderStep`")
+            error_messages.append(
+                f"Error! Input type `{type(v)}` is not `DiversityReorderStep`"
+            )
         else:
             return v
 
         if error_messages:
             # no match
-            raise ValueError("No match found when setting the actual_instance in RankQueryConfigReorderInner with anyOf schemas: BoostedReorderStep, DiversityReorderStep, ExplorationReorderStep. Details: " + ", ".join(error_messages))
+            raise ValueError(
+                "No match found when setting the actual_instance in RankQueryConfigReorderInner with anyOf schemas: BoostedReorderStep, DiversityReorderStep, ExplorationReorderStep. Details: "
+                + ", ".join(error_messages)
+            )
         else:
             return v
 
@@ -102,23 +125,26 @@ class RankQueryConfigReorderInner(BaseModel):
             instance.actual_instance = ExplorationReorderStep.from_json(json_str)
             return instance
         except (ValidationError, ValueError) as e:
-             error_messages.append(str(e))
+            error_messages.append(str(e))
         # anyof_schema_2_validator: Optional[BoostedReorderStep] = None
         try:
             instance.actual_instance = BoostedReorderStep.from_json(json_str)
             return instance
         except (ValidationError, ValueError) as e:
-             error_messages.append(str(e))
+            error_messages.append(str(e))
         # anyof_schema_3_validator: Optional[DiversityReorderStep] = None
         try:
             instance.actual_instance = DiversityReorderStep.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 RankQueryConfigReorderInner with anyOf schemas: BoostedReorderStep, DiversityReorderStep, ExplorationReorderStep. Details: " + ", ".join(error_messages))
+            raise ValueError(
+                "No match found when deserializing the JSON string into RankQueryConfigReorderInner with anyOf schemas: BoostedReorderStep, DiversityReorderStep, ExplorationReorderStep. Details: "
+                + ", ".join(error_messages)
+            )
         else:
             return instance
 
@@ -127,17 +153,30 @@ class RankQueryConfigReorderInner(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)
 
-    def to_dict(self) -> Optional[Union[Dict[str, Any], BoostedReorderStep, DiversityReorderStep, ExplorationReorderStep]]:
+    def to_dict(
+        self,
+    ) -> Optional[
+        Union[
+            Dict[str, Any],
+            BoostedReorderStep,
+            DiversityReorderStep,
+            ExplorationReorderStep,
+        ]
+    ]:
         """Returns the dict representation of the actual instance"""
         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
@@ -145,5 +184,3 @@ class RankQueryConfigReorderInner(BaseModel):
     def to_str(self) -> str:
         """Returns the string representation of the actual instance"""
         return pprint.pformat(self.model_dump())
-
-

shaped/autogen/models/rank_query_config_retrieve_inner.py

@@ -13,26 +13,37 @@
 
 
 from __future__ import annotations
+
 import json
 import pprint
-from pydantic import BaseModel, ConfigDict, Field, StrictStr, ValidationError, field_validator
-from typing import Any, List, Optional
-from shaped.autogen.models.candidate_attributes_retrieve_step import CandidateAttributesRetrieveStep
+from typing import Any, Dict, Optional, Set, Union
+
+from pydantic import BaseModel, ConfigDict, ValidationError, field_validator
+from shaped.autogen.models.candidate_attributes_retrieve_step import (
+    CandidateAttributesRetrieveStep,
+)
 from shaped.autogen.models.candidate_ids_retrieve_step import CandidateIdsRetrieveStep
 from shaped.autogen.models.column_order_retrieve_step import ColumnOrderRetrieveStep
 from shaped.autogen.models.filter_retrieve_step import FilterRetrieveStep
 from shaped.autogen.models.similarity_retrieve_step import SimilarityRetrieveStep
 from shaped.autogen.models.text_search_retrieve_step import TextSearchRetrieveStep
-from pydantic import StrictStr, Field
-from typing import Union, List, Set, Optional, Dict
-from typing_extensions import Literal, Self
+from typing_extensions import Self
+
+RANKQUERYCONFIGRETRIEVEINNER_ONE_OF_SCHEMAS = [
+    "CandidateAttributesRetrieveStep",
+    "CandidateIdsRetrieveStep",
+    "ColumnOrderRetrieveStep",
+    "FilterRetrieveStep",
+    "SimilarityRetrieveStep",
+    "TextSearchRetrieveStep",
+]
 
-RANKQUERYCONFIGRETRIEVEINNER_ONE_OF_SCHEMAS = ["CandidateAttributesRetrieveStep", "CandidateIdsRetrieveStep", "ColumnOrderRetrieveStep", "FilterRetrieveStep", "SimilarityRetrieveStep", "TextSearchRetrieveStep"]
 
 class RankQueryConfigRetrieveInner(BaseModel):
     """
     RankQueryConfigRetrieveInner
     """
+
     # data type: ColumnOrderRetrieveStep
     oneof_schema_1_validator: Optional[ColumnOrderRetrieveStep] = None
     # data type: TextSearchRetrieveStep
@@ -45,69 +56,105 @@ class RankQueryConfigRetrieveInner(BaseModel):
     oneof_schema_5_validator: Optional[CandidateIdsRetrieveStep] = None
     # data type: CandidateAttributesRetrieveStep
     oneof_schema_6_validator: Optional[CandidateAttributesRetrieveStep] = None
-    actual_instance: Optional[Union[CandidateAttributesRetrieveStep, CandidateIdsRetrieveStep, ColumnOrderRetrieveStep, FilterRetrieveStep, SimilarityRetrieveStep, TextSearchRetrieveStep]] = None
-    one_of_schemas: Set[str] = { "CandidateAttributesRetrieveStep", "CandidateIdsRetrieveStep", "ColumnOrderRetrieveStep", "FilterRetrieveStep", "SimilarityRetrieveStep", "TextSearchRetrieveStep" }
+    actual_instance: Optional[
+        Union[
+            CandidateAttributesRetrieveStep,
+            CandidateIdsRetrieveStep,
+            ColumnOrderRetrieveStep,
+            FilterRetrieveStep,
+            SimilarityRetrieveStep,
+            TextSearchRetrieveStep,
+        ]
+    ] = None
+    one_of_schemas: Set[str] = {
+        "CandidateAttributesRetrieveStep",
+        "CandidateIdsRetrieveStep",
+        "ColumnOrderRetrieveStep",
+        "FilterRetrieveStep",
+        "SimilarityRetrieveStep",
+        "TextSearchRetrieveStep",
+    }
 
     model_config = ConfigDict(
         validate_assignment=True,
         protected_namespaces=(),
     )
 
-
-    discriminator_value_class_map: Dict[str, str] = {
-    }
+    discriminator_value_class_map: Dict[str, str] = {}
 
     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_oneof(cls, v):
         instance = RankQueryConfigRetrieveInner.model_construct()
         error_messages = []
         match = 0
         # validate data type: ColumnOrderRetrieveStep
         if not isinstance(v, ColumnOrderRetrieveStep):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `ColumnOrderRetrieveStep`")
+            error_messages.append(
+                f"Error! Input type `{type(v)}` is not `ColumnOrderRetrieveStep`"
+            )
         else:
             match += 1
         # validate data type: TextSearchRetrieveStep
         if not isinstance(v, TextSearchRetrieveStep):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `TextSearchRetrieveStep`")
+            error_messages.append(
+                f"Error! Input type `{type(v)}` is not `TextSearchRetrieveStep`"
+            )
         else:
             match += 1
         # validate data type: SimilarityRetrieveStep
         if not isinstance(v, SimilarityRetrieveStep):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `SimilarityRetrieveStep`")
+            error_messages.append(
+                f"Error! Input type `{type(v)}` is not `SimilarityRetrieveStep`"
+            )
         else:
             match += 1
         # validate data type: FilterRetrieveStep
         if not isinstance(v, FilterRetrieveStep):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `FilterRetrieveStep`")
+            error_messages.append(
+                f"Error! Input type `{type(v)}` is not `FilterRetrieveStep`"
+            )
         else:
             match += 1
         # validate data type: CandidateIdsRetrieveStep
         if not isinstance(v, CandidateIdsRetrieveStep):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `CandidateIdsRetrieveStep`")
+            error_messages.append(
+                f"Error! Input type `{type(v)}` is not `CandidateIdsRetrieveStep`"
+            )
         else:
             match += 1
         # validate data type: CandidateAttributesRetrieveStep
         if not isinstance(v, CandidateAttributesRetrieveStep):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `CandidateAttributesRetrieveStep`")
+            error_messages.append(
+                f"Error! Input type `{type(v)}` is not `CandidateAttributesRetrieveStep`"
+            )
         else:
             match += 1
         if match > 1:
             # more than 1 match
-            raise ValueError("Multiple matches found when setting `actual_instance` in RankQueryConfigRetrieveInner with oneOf schemas: CandidateAttributesRetrieveStep, CandidateIdsRetrieveStep, ColumnOrderRetrieveStep, FilterRetrieveStep, SimilarityRetrieveStep, TextSearchRetrieveStep. Details: " + ", ".join(error_messages))
+            raise ValueError(
+                "Multiple matches found when setting `actual_instance` in RankQueryConfigRetrieveInner with oneOf schemas: CandidateAttributesRetrieveStep, CandidateIdsRetrieveStep, ColumnOrderRetrieveStep, FilterRetrieveStep, SimilarityRetrieveStep, TextSearchRetrieveStep. Details: "
+                + ", ".join(error_messages)
+            )
         elif match == 0:
             # no match
-            raise ValueError("No match found when setting `actual_instance` in RankQueryConfigRetrieveInner with oneOf schemas: CandidateAttributesRetrieveStep, CandidateIdsRetrieveStep, ColumnOrderRetrieveStep, FilterRetrieveStep, SimilarityRetrieveStep, TextSearchRetrieveStep. Details: " + ", ".join(error_messages))
+            raise ValueError(
+                "No match found when setting `actual_instance` in RankQueryConfigRetrieveInner with oneOf schemas: CandidateAttributesRetrieveStep, CandidateIdsRetrieveStep, ColumnOrderRetrieveStep, FilterRetrieveStep, SimilarityRetrieveStep, TextSearchRetrieveStep. Details: "
+                + ", ".join(error_messages)
+            )
         else:
             return v
 
@@ -154,17 +201,25 @@ class RankQueryConfigRetrieveInner(BaseModel):
             error_messages.append(str(e))
         # deserialize data into CandidateAttributesRetrieveStep
         try:
-            instance.actual_instance = CandidateAttributesRetrieveStep.from_json(json_str)
+            instance.actual_instance = CandidateAttributesRetrieveStep.from_json(
+                json_str
+            )
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
 
         if match > 1:
             # more than 1 match
-            raise ValueError("Multiple matches found when deserializing the JSON string into RankQueryConfigRetrieveInner with oneOf schemas: CandidateAttributesRetrieveStep, CandidateIdsRetrieveStep, ColumnOrderRetrieveStep, FilterRetrieveStep, SimilarityRetrieveStep, TextSearchRetrieveStep. Details: " + ", ".join(error_messages))
+            raise ValueError(
+                "Multiple matches found when deserializing the JSON string into RankQueryConfigRetrieveInner with oneOf schemas: CandidateAttributesRetrieveStep, CandidateIdsRetrieveStep, ColumnOrderRetrieveStep, FilterRetrieveStep, SimilarityRetrieveStep, TextSearchRetrieveStep. Details: "
+                + ", ".join(error_messages)
+            )
         elif match == 0:
             # no match
-            raise ValueError("No match found when deserializing the JSON string into RankQueryConfigRetrieveInner with oneOf schemas: CandidateAttributesRetrieveStep, CandidateIdsRetrieveStep, ColumnOrderRetrieveStep, FilterRetrieveStep, SimilarityRetrieveStep, TextSearchRetrieveStep. Details: " + ", ".join(error_messages))
+            raise ValueError(
+                "No match found when deserializing the JSON string into RankQueryConfigRetrieveInner with oneOf schemas: CandidateAttributesRetrieveStep, CandidateIdsRetrieveStep, ColumnOrderRetrieveStep, FilterRetrieveStep, SimilarityRetrieveStep, TextSearchRetrieveStep. Details: "
+                + ", ".join(error_messages)
+            )
         else:
             return instance
 
@@ -173,17 +228,33 @@ class RankQueryConfigRetrieveInner(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)
 
-    def to_dict(self) -> Optional[Union[Dict[str, Any], CandidateAttributesRetrieveStep, CandidateIdsRetrieveStep, ColumnOrderRetrieveStep, FilterRetrieveStep, SimilarityRetrieveStep, TextSearchRetrieveStep]]:
+    def to_dict(
+        self,
+    ) -> Optional[
+        Union[
+            Dict[str, Any],
+            CandidateAttributesRetrieveStep,
+            CandidateIdsRetrieveStep,
+            ColumnOrderRetrieveStep,
+            FilterRetrieveStep,
+            SimilarityRetrieveStep,
+            TextSearchRetrieveStep,
+        ]
+    ]:
         """Returns the dict representation of the actual instance"""
         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:
             # primitive type
@@ -192,5 +263,3 @@ class RankQueryConfigRetrieveInner(BaseModel):
     def to_str(self) -> str:
         """Returns the string representation of the actual instance"""
         return pprint.pformat(self.model_dump())
-
-

shaped/autogen/models/recreate_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/redshift_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/reference_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/regularization.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/reorder_inner.py

@@ -0,0 +1,149 @@
+# coding: utf-8
+
+"""
+    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.  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.4
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+from inspect import getfullargspec
+import json
+import pprint
+import re  # noqa: F401
+from pydantic import BaseModel, ConfigDict, Field, StrictStr, ValidationError, field_validator
+from typing import Optional
+from shaped.autogen.models.boosted_reorder_step import BoostedReorderStep
+from shaped.autogen.models.diversity_reorder_step import DiversityReorderStep
+from shaped.autogen.models.exploration_reorder_step import ExplorationReorderStep
+from typing import Union, Any, List, Set, TYPE_CHECKING, Optional, Dict
+from typing_extensions import Literal, Self
+from pydantic import Field
+
+REORDERINNER_ANY_OF_SCHEMAS = ["BoostedReorderStep", "DiversityReorderStep", "ExplorationReorderStep"]
+
+class ReorderInner(BaseModel):
+    """
+    ReorderInner
+    """
+
+    # data type: ExplorationReorderStep
+    anyof_schema_1_validator: Optional[ExplorationReorderStep] = None
+    # data type: BoostedReorderStep
+    anyof_schema_2_validator: Optional[BoostedReorderStep] = None
+    # data type: DiversityReorderStep
+    anyof_schema_3_validator: Optional[DiversityReorderStep] = None
+    if TYPE_CHECKING:
+        actual_instance: Optional[Union[BoostedReorderStep, DiversityReorderStep, ExplorationReorderStep]] = None
+    else:
+        actual_instance: Any = None
+    any_of_schemas: Set[str] = { "BoostedReorderStep", "DiversityReorderStep", "ExplorationReorderStep" }
+
+    model_config = {
+        "validate_assignment": True,
+        "protected_namespaces": (),
+    }
+
+    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`")
+            if kwargs:
+                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')
+    def actual_instance_must_validate_anyof(cls, v):
+        instance = ReorderInner.model_construct()
+        error_messages = []
+        # validate data type: ExplorationReorderStep
+        if not isinstance(v, ExplorationReorderStep):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `ExplorationReorderStep`")
+        else:
+            return v
+
+        # validate data type: BoostedReorderStep
+        if not isinstance(v, BoostedReorderStep):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `BoostedReorderStep`")
+        else:
+            return v
+
+        # validate data type: DiversityReorderStep
+        if not isinstance(v, DiversityReorderStep):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `DiversityReorderStep`")
+        else:
+            return v
+
+        if error_messages:
+            # no match
+            raise ValueError("No match found when setting the actual_instance in ReorderInner with anyOf schemas: BoostedReorderStep, DiversityReorderStep, ExplorationReorderStep. Details: " + ", ".join(error_messages))
+        else:
+            return v
+
+    @classmethod
+    def from_dict(cls, obj: Dict[str, Any]) -> Self:
+        return cls.from_json(json.dumps(obj))
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Returns the object represented by the json string"""
+        instance = cls.model_construct()
+        error_messages = []
+        # anyof_schema_1_validator: Optional[ExplorationReorderStep] = None
+        try:
+            instance.actual_instance = ExplorationReorderStep.from_json(json_str)
+            return instance
+        except (ValidationError, ValueError) as e:
+             error_messages.append(str(e))
+        # anyof_schema_2_validator: Optional[BoostedReorderStep] = None
+        try:
+            instance.actual_instance = BoostedReorderStep.from_json(json_str)
+            return instance
+        except (ValidationError, ValueError) as e:
+             error_messages.append(str(e))
+        # anyof_schema_3_validator: Optional[DiversityReorderStep] = None
+        try:
+            instance.actual_instance = DiversityReorderStep.from_json(json_str)
+            return instance
+        except (ValidationError, ValueError) as e:
+             error_messages.append(str(e))
+
+        if error_messages:
+            # no match
+            raise ValueError("No match found when deserializing the JSON string into ReorderInner with anyOf schemas: BoostedReorderStep, DiversityReorderStep, ExplorationReorderStep. Details: " + ", ".join(error_messages))
+        else:
+            return instance
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the actual instance"""
+        if self.actual_instance is None:
+            return "null"
+
+        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)
+
+    def to_dict(self) -> Optional[Union[Dict[str, Any], BoostedReorderStep, DiversityReorderStep, ExplorationReorderStep]]:
+        """Returns the dict representation of the actual instance"""
+        if self.actual_instance is None:
+            return None
+
+        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
+
+    def to_str(self) -> str:
+        """Returns the string representation of the actual instance"""
+        return pprint.pformat(self.model_dump())
+
+