shaped 0.14.1__py3-none-any.whl → 1.0.2__py3-none-any.whl

shaped/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "1.0.0"
+__version__ = "1.0.1"
 from .client import Client
 from .autogen.models.inference_config import InferenceConfig
 from .autogen.models.retriever_top_k_override import RetrieverTopKOverride

shaped/autogen/__init__.py

@@ -73,6 +73,7 @@ from shaped.autogen.models.rank_grid_attribute_response import RankGridAttribute
 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

shaped/autogen/api/dataset_api.py

@@ -272,7 +272,9 @@ class DatasetApi:
         _query_params: List[Tuple[str, str]] = []
         _header_params: Dict[str, Optional[str]] = _headers or {}
         _form_params: List[Tuple[str, str]] = []
-        _files: Dict[str, Union[str, bytes]] = {}
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
         _body_params: Optional[bytes] = None
 
         # process the path parameters
@@ -561,7 +563,9 @@ class DatasetApi:
         _query_params: List[Tuple[str, str]] = []
         _header_params: Dict[str, Optional[str]] = _headers or {}
         _form_params: List[Tuple[str, str]] = []
-        _files: Dict[str, Union[str, bytes]] = {}
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
         _body_params: Optional[bytes] = None
 
         # process the path parameters
@@ -837,7 +841,9 @@ class DatasetApi:
         _query_params: List[Tuple[str, str]] = []
         _header_params: Dict[str, Optional[str]] = _headers or {}
         _form_params: List[Tuple[str, str]] = []
-        _files: Dict[str, Union[str, bytes]] = {}
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
         _body_params: Optional[bytes] = None
 
         # process the path parameters
@@ -1100,7 +1106,9 @@ class DatasetApi:
         _query_params: List[Tuple[str, str]] = []
         _header_params: Dict[str, Optional[str]] = _headers or {}
         _form_params: List[Tuple[str, str]] = []
-        _files: Dict[str, Union[str, bytes]] = {}
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
         _body_params: Optional[bytes] = None
 
         # process the path parameters

shaped/autogen/api/model_inference_api.py

@@ -31,6 +31,7 @@ from shaped.autogen.models.rank_grid_attribute_request1 import RankGridAttribute
 from shaped.autogen.models.rank_grid_attribute_response import RankGridAttributeResponse
 from shaped.autogen.models.rank_response import RankResponse
 from shaped.autogen.models.retrieve_request import RetrieveRequest
+from shaped.autogen.models.retrieve_response import RetrieveResponse
 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
@@ -297,7 +298,9 @@ class ModelInferenceApi:
         _query_params: List[Tuple[str, str]] = []
         _header_params: Dict[str, Optional[str]] = _headers or {}
         _form_params: List[Tuple[str, str]] = []
-        _files: Dict[str, Union[str, bytes]] = {}
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
         _body_params: Optional[bytes] = None
 
         # process the path parameters
@@ -601,7 +604,9 @@ class ModelInferenceApi:
         _query_params: List[Tuple[str, str]] = []
         _header_params: Dict[str, Optional[str]] = _headers or {}
         _form_params: List[Tuple[str, str]] = []
-        _files: Dict[str, Union[str, bytes]] = {}
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
         _body_params: Optional[bytes] = None
 
         # process the path parameters
@@ -905,7 +910,9 @@ class ModelInferenceApi:
         _query_params: List[Tuple[str, str]] = []
         _header_params: Dict[str, Optional[str]] = _headers or {}
         _form_params: List[Tuple[str, str]] = []
-        _files: Dict[str, Union[str, bytes]] = {}
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
         _body_params: Optional[bytes] = None
 
         # process the path parameters
@@ -1209,7 +1216,9 @@ class ModelInferenceApi:
         _query_params: List[Tuple[str, str]] = []
         _header_params: Dict[str, Optional[str]] = _headers or {}
         _form_params: List[Tuple[str, str]] = []
-        _files: Dict[str, Union[str, bytes]] = {}
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
         _body_params: Optional[bytes] = None
 
         # process the path parameters
@@ -1513,7 +1522,9 @@ class ModelInferenceApi:
         _query_params: List[Tuple[str, str]] = []
         _header_params: Dict[str, Optional[str]] = _headers or {}
         _form_params: List[Tuple[str, str]] = []
-        _files: Dict[str, Union[str, bytes]] = {}
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
         _body_params: Optional[bytes] = None
 
         # process the path parameters
@@ -1817,7 +1828,9 @@ class ModelInferenceApi:
         _query_params: List[Tuple[str, str]] = []
         _header_params: Dict[str, Optional[str]] = _headers or {}
         _form_params: List[Tuple[str, str]] = []
-        _files: Dict[str, Union[str, bytes]] = {}
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
         _body_params: Optional[bytes] = None
 
         # process the path parameters
@@ -1895,7 +1908,7 @@ class ModelInferenceApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> RankResponse:
+    ) -> RetrieveResponse:
         """Retrieve
 
         Retrieve returns relevant item_ids for the given input text or user query. It can be used instead of rank if the filtering, scoring and ordering stages aren't needed for the final ranking. Typically people use this endpoint over rank if they want to reduce latency and complexity of the ranking pipeline and only need a subset of the functionality, e.g. just search but without personalization. 
@@ -1939,7 +1952,7 @@ class ModelInferenceApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "RankResponse",
+            '200': "RetrieveResponse",
             '422': "HTTPValidationError",
         }
         response_data = self.api_client.call_api(
@@ -1971,7 +1984,7 @@ class ModelInferenceApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> ApiResponse[RankResponse]:
+    ) -> ApiResponse[RetrieveResponse]:
         """Retrieve
 
         Retrieve returns relevant item_ids for the given input text or user query. It can be used instead of rank if the filtering, scoring and ordering stages aren't needed for the final ranking. Typically people use this endpoint over rank if they want to reduce latency and complexity of the ranking pipeline and only need a subset of the functionality, e.g. just search but without personalization. 
@@ -2015,7 +2028,7 @@ class ModelInferenceApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "RankResponse",
+            '200': "RetrieveResponse",
             '422': "HTTPValidationError",
         }
         response_data = self.api_client.call_api(
@@ -2091,7 +2104,7 @@ class ModelInferenceApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "RankResponse",
+            '200': "RetrieveResponse",
             '422': "HTTPValidationError",
         }
         response_data = self.api_client.call_api(
@@ -2121,7 +2134,9 @@ class ModelInferenceApi:
         _query_params: List[Tuple[str, str]] = []
         _header_params: Dict[str, Optional[str]] = _headers or {}
         _form_params: List[Tuple[str, str]] = []
-        _files: Dict[str, Union[str, bytes]] = {}
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
         _body_params: Optional[bytes] = None
 
         # process the path parameters
@@ -2425,7 +2440,9 @@ class ModelInferenceApi:
         _query_params: List[Tuple[str, str]] = []
         _header_params: Dict[str, Optional[str]] = _headers or {}
         _form_params: List[Tuple[str, str]] = []
-        _files: Dict[str, Union[str, bytes]] = {}
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
         _body_params: Optional[bytes] = None
 
         # process the path parameters
@@ -2742,7 +2759,9 @@ class ModelInferenceApi:
         _query_params: List[Tuple[str, str]] = []
         _header_params: Dict[str, Optional[str]] = _headers or {}
         _form_params: List[Tuple[str, str]] = []
-        _files: Dict[str, Union[str, bytes]] = {}
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
         _body_params: Optional[bytes] = None
 
         # process the path parameters

shaped/autogen/api/model_management_api.py

@@ -273,7 +273,9 @@ class ModelManagementApi:
         _query_params: List[Tuple[str, str]] = []
         _header_params: Dict[str, Optional[str]] = _headers or {}
         _form_params: List[Tuple[str, str]] = []
-        _files: Dict[str, Union[str, bytes]] = {}
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
         _body_params: Optional[bytes] = None
 
         # process the path parameters
@@ -549,7 +551,9 @@ class ModelManagementApi:
         _query_params: List[Tuple[str, str]] = []
         _header_params: Dict[str, Optional[str]] = _headers or {}
         _form_params: List[Tuple[str, str]] = []
-        _files: Dict[str, Union[str, bytes]] = {}
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
         _body_params: Optional[bytes] = None
 
         # process the path parameters
@@ -812,7 +816,9 @@ class ModelManagementApi:
         _query_params: List[Tuple[str, str]] = []
         _header_params: Dict[str, Optional[str]] = _headers or {}
         _form_params: List[Tuple[str, str]] = []
-        _files: Dict[str, Union[str, bytes]] = {}
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
         _body_params: Optional[bytes] = None
 
         # process the path parameters
@@ -1086,7 +1092,9 @@ class ModelManagementApi:
         _query_params: List[Tuple[str, str]] = []
         _header_params: Dict[str, Optional[str]] = _headers or {}
         _form_params: List[Tuple[str, str]] = []
-        _files: Dict[str, Union[str, bytes]] = {}
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
         _body_params: Optional[bytes] = None
 
         # process the path parameters
@@ -1375,7 +1383,9 @@ class ModelManagementApi:
         _query_params: List[Tuple[str, str]] = []
         _header_params: Dict[str, Optional[str]] = _headers or {}
         _form_params: List[Tuple[str, str]] = []
-        _files: Dict[str, Union[str, bytes]] = {}
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
         _body_params: Optional[bytes] = None
 
         # process the path parameters

shaped/autogen/api_client.py

@@ -15,6 +15,7 @@
 import datetime
 from dateutil.parser import parse
 from enum import Enum
+import decimal
 import json
 import mimetypes
 import os
@@ -66,6 +67,7 @@ class ApiClient:
         'bool': bool,
         'date': datetime.date,
         'datetime': datetime.datetime,
+        'decimal': decimal.Decimal,
         'object': object,
     }
     _pool = None
@@ -338,6 +340,7 @@ class ApiClient:
         If obj is str, int, long, float, bool, return directly.
         If obj is datetime.datetime, datetime.date
             convert to string in iso8601 format.
+        If obj is decimal.Decimal return string representation.
         If obj is list, sanitize each element in the list.
         If obj is dict, return the dict.
         If obj is OpenAPI model, return the properties dict.
@@ -363,6 +366,8 @@ class ApiClient:
             )
         elif isinstance(obj, (datetime.datetime, datetime.date)):
             return obj.isoformat()
+        elif isinstance(obj, decimal.Decimal):
+            return str(obj)
 
         elif isinstance(obj, dict):
             obj_dict = obj
@@ -399,12 +404,12 @@ class ApiClient:
                 data = json.loads(response_text)
             except ValueError:
                 data = response_text
-        elif content_type.startswith("application/json"):
+        elif re.match(r'^application/(json|[\w!#$&.+-^_]+\+json)\s*(;|$)', content_type, re.IGNORECASE):
             if response_text == "":
                 data = ""
             else:
                 data = json.loads(response_text)
-        elif content_type.startswith("text/plain"):
+        elif re.match(r'^text\/[a-z.+-]+\s*(;|$)', content_type, re.IGNORECASE):
             data = response_text
         else:
             raise ApiException(
@@ -454,6 +459,8 @@ class ApiClient:
             return self.__deserialize_date(data)
         elif klass == datetime.datetime:
             return self.__deserialize_datetime(data)
+        elif klass == decimal.Decimal:
+            return decimal.Decimal(data)
         elif issubclass(klass, Enum):
             return self.__deserialize_enum(data, klass)
         else:
@@ -528,7 +535,10 @@ class ApiClient:
 
         return "&".join(["=".join(map(str, item)) for item in new_params])
 
-    def files_parameters(self, files: Dict[str, Union[str, bytes]]):
+    def files_parameters(
+        self,
+        files: Dict[str, Union[str, bytes, List[str], List[bytes], Tuple[str, bytes]]],
+    ):
         """Builds form parameters.
 
         :param files: File parameters.
@@ -543,6 +553,12 @@ class ApiClient:
             elif isinstance(v, bytes):
                 filename = k
                 filedata = v
+            elif isinstance(v, tuple):
+                filename, filedata = v
+            elif isinstance(v, list):
+                for file_param in v:
+                    params.extend(self.files_parameters({k: file_param}))
+                continue
             else:
                 raise ValueError("Unsupported file value")
             mimetype = (

shaped/autogen/configuration.py

@@ -13,14 +13,16 @@
 
 
 import copy
+import http.client as httplib
 import logging
 from logging import FileHandler
 import multiprocessing
 import sys
-from typing import Optional
+from typing import Any, ClassVar, Dict, List, Literal, Optional, TypedDict
+from typing_extensions import NotRequired, Self
+
 import urllib3
 
-import http.client as httplib
 
 JSON_SCHEMA_VALIDATION_KEYWORDS = {
     'multipleOf', 'maximum', 'exclusiveMaximum',
@@ -28,6 +30,106 @@ JSON_SCHEMA_VALIDATION_KEYWORDS = {
     'minLength', 'pattern', 'maxItems', 'minItems'
 }
 
+ServerVariablesT = Dict[str, str]
+
+GenericAuthSetting = TypedDict(
+    "GenericAuthSetting",
+    {
+        "type": str,
+        "in": str,
+        "key": str,
+        "value": str,
+    },
+)
+
+
+OAuth2AuthSetting = TypedDict(
+    "OAuth2AuthSetting",
+    {
+        "type": Literal["oauth2"],
+        "in": Literal["header"],
+        "key": Literal["Authorization"],
+        "value": str,
+    },
+)
+
+
+APIKeyAuthSetting = TypedDict(
+    "APIKeyAuthSetting",
+    {
+        "type": Literal["api_key"],
+        "in": str,
+        "key": str,
+        "value": Optional[str],
+    },
+)
+
+
+BasicAuthSetting = TypedDict(
+    "BasicAuthSetting",
+    {
+        "type": Literal["basic"],
+        "in": Literal["header"],
+        "key": Literal["Authorization"],
+        "value": Optional[str],
+    },
+)
+
+
+BearerFormatAuthSetting = TypedDict(
+    "BearerFormatAuthSetting",
+    {
+        "type": Literal["bearer"],
+        "in": Literal["header"],
+        "format": Literal["JWT"],
+        "key": Literal["Authorization"],
+        "value": str,
+    },
+)
+
+
+BearerAuthSetting = TypedDict(
+    "BearerAuthSetting",
+    {
+        "type": Literal["bearer"],
+        "in": Literal["header"],
+        "key": Literal["Authorization"],
+        "value": str,
+    },
+)
+
+
+HTTPSignatureAuthSetting = TypedDict(
+    "HTTPSignatureAuthSetting",
+    {
+        "type": Literal["http-signature"],
+        "in": Literal["header"],
+        "key": Literal["Authorization"],
+        "value": None,
+    },
+)
+
+
+AuthSettings = TypedDict(
+    "AuthSettings",
+    {
+    },
+    total=False,
+)
+
+
+class HostSettingVariable(TypedDict):
+    description: str
+    default_value: str
+    enum_values: List[str]
+
+
+class HostSetting(TypedDict):
+    url: str
+    description: str
+    variables: NotRequired[Dict[str, HostSettingVariable]]
+
+
 class Configuration:
     """This class contains various settings of the API client.
 
@@ -61,20 +163,26 @@ class Configuration:
 
     """
 
-    _default = None
-
-    def __init__(self, host=None,
-                 api_key=None, api_key_prefix=None,
-                 username=None, password=None,
-                 access_token=None,
-                 server_index=None, server_variables=None,
-                 server_operation_index=None, server_operation_variables=None,
-                 ignore_operation_servers=False,
-                 ssl_ca_cert=None,
-                 retries=None,
-                 *,
-                 debug: Optional[bool] = None
-                 ) -> None:
+    _default: ClassVar[Optional[Self]] = None
+
+    def __init__(
+        self,
+        host: Optional[str]=None,
+        api_key: Optional[Dict[str, str]]=None,
+        api_key_prefix: Optional[Dict[str, str]]=None,
+        username: Optional[str]=None,
+        password: Optional[str]=None,
+        access_token: Optional[str]=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,
+        *,
+        debug: Optional[bool] = None,
+    ) -> None:
         """Constructor
         """
         self._base_path = "https://api.shaped.ai/v1" if host is None else host
@@ -198,7 +306,7 @@ class Configuration:
         """date format
         """
 
-    def __deepcopy__(self, memo):
+    def __deepcopy__(self, memo:  Dict[int, Any]) -> Self:
         cls = self.__class__
         result = cls.__new__(cls)
         memo[id(self)] = result
@@ -212,11 +320,11 @@ class Configuration:
         result.debug = self.debug
         return result
 
-    def __setattr__(self, name, value):
+    def __setattr__(self, name: str, value: Any) -> None:
         object.__setattr__(self, name, value)
 
     @classmethod
-    def set_default(cls, default):
+    def set_default(cls, default: Optional[Self]) -> None:
         """Set default instance of configuration.
 
         It stores default configuration, which can be
@@ -227,7 +335,7 @@ class Configuration:
         cls._default = default
 
     @classmethod
-    def get_default_copy(cls):
+    def get_default_copy(cls) -> Self:
         """Deprecated. Please use `get_default` instead.
 
         Deprecated. Please use `get_default` instead.
@@ -237,7 +345,7 @@ class Configuration:
         return cls.get_default()
 
     @classmethod
-    def get_default(cls):
+    def get_default(cls) -> Self:
         """Return the default configuration.
 
         This method returns newly created, based on default constructor,
@@ -247,11 +355,11 @@ class Configuration:
         :return: The configuration object.
         """
         if cls._default is None:
-            cls._default = Configuration()
+            cls._default = cls()
         return cls._default
 
     @property
-    def logger_file(self):
+    def logger_file(self) -> Optional[str]:
         """The logger file.
 
         If the logger_file is None, then add stream handler and remove file
@@ -263,7 +371,7 @@ class Configuration:
         return self.__logger_file
 
     @logger_file.setter
-    def logger_file(self, value):
+    def logger_file(self, value: Optional[str]) -> None:
         """The logger file.
 
         If the logger_file is None, then add stream handler and remove file
@@ -282,7 +390,7 @@ class Configuration:
                 logger.addHandler(self.logger_file_handler)
 
     @property
-    def debug(self):
+    def debug(self) -> bool:
         """Debug status
 
         :param value: The debug status, True or False.
@@ -291,7 +399,7 @@ class Configuration:
         return self.__debug
 
     @debug.setter
-    def debug(self, value):
+    def debug(self, value: bool) -> None:
         """Debug status
 
         :param value: The debug status, True or False.
@@ -313,7 +421,7 @@ class Configuration:
             httplib.HTTPConnection.debuglevel = 0
 
     @property
-    def logger_format(self):
+    def logger_format(self) -> str:
         """The logger format.
 
         The logger_formatter will be updated when sets logger_format.
@@ -324,7 +432,7 @@ class Configuration:
         return self.__logger_format
 
     @logger_format.setter
-    def logger_format(self, value):
+    def logger_format(self, value: str) -> None:
         """The logger format.
 
         The logger_formatter will be updated when sets logger_format.
@@ -335,7 +443,7 @@ class Configuration:
         self.__logger_format = value
         self.logger_formatter = logging.Formatter(self.__logger_format)
 
-    def get_api_key_with_prefix(self, identifier, alias=None):
+    def get_api_key_with_prefix(self, identifier: str, alias: Optional[str]=None) -> Optional[str]:
         """Gets API key (with prefix if set).
 
         :param identifier: The identifier of apiKey.
@@ -352,7 +460,9 @@ class Configuration:
             else:
                 return key
 
-    def get_basic_auth_token(self):
+        return None
+
+    def get_basic_auth_token(self) -> Optional[str]:
         """Gets HTTP basic authentication header (string).
 
         :return: The token for basic HTTP authentication.
@@ -367,15 +477,15 @@ class Configuration:
             basic_auth=username + ':' + password
         ).get('authorization')
 
-    def auth_settings(self):
+    def auth_settings(self)-> AuthSettings:
         """Gets Auth Settings dict for api client.
 
         :return: The Auth Settings information dict.
         """
-        auth = {}
+        auth: AuthSettings = {}
         return auth
 
-    def to_debug_report(self):
+    def to_debug_report(self) -> str:
         """Gets the essential information for debugging.
 
         :return: The report for debugging.
@@ -387,7 +497,7 @@ class Configuration:
                "SDK Package Version: 1.0.0".\
                format(env=sys.platform, pyversion=sys.version)
 
-    def get_host_settings(self):
+    def get_host_settings(self) -> List[HostSetting]:
         """Gets an array of host settings
 
         :return: An array of host settings
@@ -399,7 +509,12 @@ class Configuration:
             }
         ]
 
-    def get_host_from_settings(self, index, variables=None, servers=None):
+    def get_host_from_settings(
+        self,
+        index: Optional[int],
+        variables: Optional[ServerVariablesT]=None,
+        servers: Optional[List[HostSetting]]=None,
+    ) -> str:
         """Gets host URL based on the index and variables
         :param index: array index of the host settings
         :param variables: hash of variable and the corresponding value
@@ -439,12 +554,12 @@ class Configuration:
         return url
 
     @property
-    def host(self):
+    def host(self) -> str:
         """Return generated host."""
         return self.get_host_from_settings(self.server_index, variables=self.server_variables)
 
     @host.setter
-    def host(self, value):
+    def host(self, value: str) -> None:
         """Fix base path."""
         self._base_path = value
         self.server_index = None

shaped/autogen/models/__init__.py

@@ -54,6 +54,7 @@ from shaped.autogen.models.rank_grid_attribute_response import RankGridAttribute
 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

shaped/autogen/models/complement_items_request.py

@@ -17,7 +17,7 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, StrictBool, StrictStr
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
 from shaped.autogen.models.inference_config import InferenceConfig
 from typing import Optional, Set
@@ -30,8 +30,9 @@ class ComplementItemsRequest(BaseModel):
     item_ids: List[StrictStr]
     user_id: Optional[StrictStr] = None
     return_metadata: Optional[StrictBool] = None
+    filter_predicate: Optional[StrictStr] = Field(default=None, description="A SQL where query that can be used to filter candidate items.")
     config: Optional[InferenceConfig] = None
-    __properties: ClassVar[List[str]] = ["item_ids", "user_id", "return_metadata", "config"]
+    __properties: ClassVar[List[str]] = ["item_ids", "user_id", "return_metadata", "filter_predicate", "config"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -90,6 +91,7 @@ class ComplementItemsRequest(BaseModel):
             "item_ids": obj.get("item_ids"),
             "user_id": obj.get("user_id"),
             "return_metadata": obj.get("return_metadata"),
+            "filter_predicate": obj.get("filter_predicate"),
             "config": InferenceConfig.from_dict(obj["config"]) if obj.get("config") is not None else None
         })
         return _obj

shaped/autogen/models/create_model_arguments.py

@@ -79,9 +79,9 @@ class CreateModelArguments(BaseModel):
         # override the default output from pydantic by calling `to_dict()` of each item in connectors (list)
         _items = []
         if self.connectors:
-            for _item in self.connectors:
-                if _item:
-                    _items.append(_item.to_dict())
+            for _item_connectors in self.connectors:
+                if _item_connectors:
+                    _items.append(_item_connectors.to_dict())
             _dict['connectors'] = _items
         # override the default output from pydantic by calling `to_dict()` of fetch
         if self.fetch:

shaped/autogen/models/http_validation_error.py

@@ -72,9 +72,9 @@ class HTTPValidationError(BaseModel):
         # override the default output from pydantic by calling `to_dict()` of each item in detail (list)
         _items = []
         if self.detail:
-            for _item in self.detail:
-                if _item:
-                    _items.append(_item.to_dict())
+            for _item_detail in self.detail:
+                if _item_detail:
+                    _items.append(_item_detail.to_dict())
             _dict['detail'] = _items
         return _dict
 

shaped/autogen/models/list_datasets_response.py

@@ -72,9 +72,9 @@ class ListDatasetsResponse(BaseModel):
         # override the default output from pydantic by calling `to_dict()` of each item in datasets (list)
         _items = []
         if self.datasets:
-            for _item in self.datasets:
-                if _item:
-                    _items.append(_item.to_dict())
+            for _item_datasets in self.datasets:
+                if _item_datasets:
+                    _items.append(_item_datasets.to_dict())
             _dict['datasets'] = _items
         return _dict
 

shaped/autogen/models/list_models_response.py

@@ -72,9 +72,9 @@ class ListModelsResponse(BaseModel):
         # override the default output from pydantic by calling `to_dict()` of each item in models (list)
         _items = []
         if self.models:
-            for _item in self.models:
-                if _item:
-                    _items.append(_item.to_dict())
+            for _item_models in self.models:
+                if _item_models:
+                    _items.append(_item_models.to_dict())
             _dict['models'] = _items
         return _dict
 

shaped/autogen/models/post_rank_request.py

@@ -82,9 +82,9 @@ class PostRankRequest(BaseModel):
         # override the default output from pydantic by calling `to_dict()` of each item in interactions (list)
         _items = []
         if self.interactions:
-            for _item in self.interactions:
-                if _item:
-                    _items.append(_item.to_dict())
+            for _item_interactions in self.interactions:
+                if _item_interactions:
+                    _items.append(_item_interactions.to_dict())
             _dict['interactions'] = _items
         # override the default output from pydantic by calling `to_dict()` of config
         if self.config:

shaped/autogen/models/retrieve_response.py

@@ -0,0 +1,91 @@
+# 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.   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. 
+
+    The version of the OpenAPI document: 1.0.1
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, StrictFloat, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional, Union
+from typing import Optional, Set
+from typing_extensions import Self
+
+class RetrieveResponse(BaseModel):
+    """
+    RetrieveResponse
+    """ # noqa: E501
+    ids: List[StrictStr]
+    scores: Optional[List[Union[StrictFloat, StrictInt]]] = None
+    metadata: Optional[List[Dict[str, Any]]] = None
+    __properties: ClassVar[List[str]] = ["ids", "scores", "metadata"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of RetrieveResponse from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of RetrieveResponse from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "ids": obj.get("ids"),
+            "scores": obj.get("scores"),
+            "metadata": obj.get("metadata")
+        })
+        return _obj
+
+

shaped/autogen/models/retriever_top_k_override.py

@@ -28,11 +28,11 @@ class RetrieverTopKOverride(BaseModel):
     """ # noqa: E501
     knn: Optional[Union[StrictFloat, StrictInt]] = Field(default=None, description="The number of items to retrieve from the knn retriever. This retriever makes a vector store lookup to find the most relevant items for the given query. If not set, 300 will be retrieved. Note, when a filter predicate or text query all other retrievers are disabled. ")
     chronological: Optional[Union[StrictFloat, StrictInt]] = Field(default=None, description="The number of items to retrieve from the chronological retriever. If not set, 300 will be retrieved. ")
-    toplist: Optional[Union[StrictFloat, StrictInt]] = Field(default=None, description="The number of items to retrieve from the toplist retriever. Toplist refers to the most popular items historically. If not set, 300 will be retrieved. ")
+    popular: Optional[Union[StrictFloat, StrictInt]] = Field(default=None, description="The number of items to retrieve from the Popular retriever. Popular refers to the most popular items historically. If not set, 300 will be retrieved. ")
     trending: Optional[Union[StrictFloat, StrictInt]] = Field(default=None, description="The number of items to retrieve from the trending retriever. Trending refers to recently popular items. If not set, 300 will be retrieved ")
     random: Optional[Union[StrictFloat, StrictInt]] = Field(default=None, description="The number of items to retrieve from the random retriever. If not set, 0 will be retrieved. ")
     cold_start: Optional[Union[StrictFloat, StrictInt]] = Field(default=None, description="The number of items to retrieve from the cold-start retriever. This retriever pulls from the set of items with the lowest number of interactions. If not set, 50 will be retrieved. ")
-    __properties: ClassVar[List[str]] = ["knn", "chronological", "toplist", "trending", "random", "cold_start"]
+    __properties: ClassVar[List[str]] = ["knn", "chronological", "popular", "trending", "random", "cold_start"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -87,7 +87,7 @@ class RetrieverTopKOverride(BaseModel):
         _obj = cls.model_validate({
             "knn": obj.get("knn"),
             "chronological": obj.get("chronological"),
-            "toplist": obj.get("toplist"),
+            "popular": obj.get("popular"),
             "trending": obj.get("trending"),
             "random": obj.get("random"),
             "cold_start": obj.get("cold_start")

shaped/autogen/models/similar_item_request.py

@@ -31,8 +31,9 @@ class SimilarItemRequest(BaseModel):
     user_id: Optional[StrictStr] = Field(default=None, description="An optional user to personalize results for.")
     return_metadata: Optional[StrictBool] = Field(default=False, description="If true, return the corresponding metadata for the ranked ids.")
     flush_paginations: Optional[StrictBool] = Field(default=False, description="Clears the pagination store for the given input user. This is useful if you want to implement paginations on client side or if you want to start the rankings again, e.g. on a page refresh. ")
+    filter_predicate: Optional[StrictStr] = Field(default=None, description="A SQL where query that can be used to filter candidate items.")
     config: Optional[InferenceConfig] = None
-    __properties: ClassVar[List[str]] = ["item_id", "user_id", "return_metadata", "flush_paginations", "config"]
+    __properties: ClassVar[List[str]] = ["item_id", "user_id", "return_metadata", "flush_paginations", "filter_predicate", "config"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -92,6 +93,7 @@ class SimilarItemRequest(BaseModel):
             "user_id": obj.get("user_id"),
             "return_metadata": obj.get("return_metadata") if obj.get("return_metadata") is not None else False,
             "flush_paginations": obj.get("flush_paginations") if obj.get("flush_paginations") is not None else False,
+            "filter_predicate": obj.get("filter_predicate"),
             "config": InferenceConfig.from_dict(obj["config"]) if obj.get("config") is not None else None
         })
         return _obj

shaped/autogen/models/similar_users_request.py

@@ -30,8 +30,9 @@ class SimilarUsersRequest(BaseModel):
     user_id: StrictStr = Field(description="The query user to find similar results for.")
     return_metadata: Optional[StrictBool] = Field(default=False, description="If true, return the corresponding metadata for the ranked ids.")
     flush_paginations: Optional[StrictBool] = Field(default=False, description="Clears the pagination store for the given input user. This is useful if you want to implement paginations on client side or if you want to start the rankings again, e.g. on a page refresh. ")
+    filter_predicate: Optional[StrictStr] = Field(default=None, description="A SQL where query that can be used to filter candidate users.")
     config: Optional[InferenceConfig] = None
-    __properties: ClassVar[List[str]] = ["user_id", "return_metadata", "flush_paginations", "config"]
+    __properties: ClassVar[List[str]] = ["user_id", "return_metadata", "flush_paginations", "filter_predicate", "config"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -90,6 +91,7 @@ class SimilarUsersRequest(BaseModel):
             "user_id": obj.get("user_id"),
             "return_metadata": obj.get("return_metadata") if obj.get("return_metadata") is not None else False,
             "flush_paginations": obj.get("flush_paginations") if obj.get("flush_paginations") is not None else False,
+            "filter_predicate": obj.get("filter_predicate"),
             "config": InferenceConfig.from_dict(obj["config"]) if obj.get("config") is not None else None
         })
         return _obj

shaped/autogen/rest.py

@@ -225,7 +225,7 @@ class RESTClientObject:
                         headers=headers,
                         preload_content=False
                     )
-                elif headers['Content-Type'] == 'text/plain' and isinstance(body, bool):
+                elif headers['Content-Type'].startswith('text/') and isinstance(body, bool):
                     request_body = "true" if body else "false"
                     r = self.pool_manager.request(
                         method,

shaped/cli/shaped_cli.py

@@ -260,9 +260,9 @@ def create_dataset_from_uri(
 
     chunk_column_definitions = {col: "String" for col in chunk_columns}
     dataset_definition = {
-        "dataset_name": name,
+        "name": name,
         "schema_type": "CUSTOM",
-        "schema": chunk_column_definitions,
+        "column_schema": chunk_column_definitions,
     }
     dataset_created = _create_dataset_from_payload(json.dumps(dataset_definition))
     if dataset_created:

shaped/client.py

@@ -122,6 +122,7 @@ class Client:
         user_id: Optional[StrictStr] = None,
         return_metadata: Optional[StrictBool] = None,
         flush_paginations: Optional[StrictBool] = None,
+        filter_predicate: Optional[StrictStr] = None,
         config: Optional[InferenceConfig] = None,
     ):
         """
@@ -135,6 +136,7 @@ class Client:
             user_id=user_id,
             return_metadata=return_metadata,
             flush_paginations=flush_paginations,
+            filter_predicate=filter_predicate,
             config=config,
         )
         return (
@@ -152,6 +154,7 @@ class Client:
         user_id: StrictStr = None,
         return_metadata: Optional[StrictBool] = None,
         flush_paginations: Optional[StrictBool] = None,
+        filter_predicate: Optional[StrictStr] = None,
         config: Optional[InferenceConfig] = None,
     ):
         """
@@ -162,6 +165,7 @@ class Client:
             user_id=user_id,
             return_metadata=return_metadata,
             flush_paginations=flush_paginations,
+            filter_predicate=filter_predicate,
             config=config,
         )
         return (
@@ -180,6 +184,7 @@ class Client:
         item_ids: List[StrictStr],
         user_id: Optional[StrictStr] = None,
         return_metadata: Optional[StrictBool] = None,
+        filter_predicate: Optional[StrictStr] = None,
         config: Optional[InferenceConfig] = None,
     ):
         """
@@ -193,6 +198,7 @@ class Client:
             item_ids=item_ids,
             user_id=user_id,
             return_metadata=return_metadata,
+            filter_predicate=filter_predicate,
             config=config,
         )
         return (

{shaped-0.14.1.dist-info → shaped-1.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: shaped
-Version: 0.14.1
+Version: 1.0.2
 Summary: CLI and SDK tools for interacting with the Shaped API.
 Home-page: https://github.com/shaped-ai/shaped-cli
 Author: Shaped Team
@@ -13,21 +13,31 @@ Classifier: Development Status :: 5 - Production/Stable
 Classifier: Intended Audience :: Developers
 Requires-Python: >=3.8, <3.12
 Description-Content-Type: text/markdown
-Requires-Dist: typer >=0.7.0
-Requires-Dist: requests >=2.28.1
-Requires-Dist: pydantic >=2.8.2
-Requires-Dist: pyyaml >=6.0
-Requires-Dist: pyarrow ==11.0.0
-Requires-Dist: pandas ==1.5.3
-Requires-Dist: tqdm ==4.65.0
-Requires-Dist: s3fs ==0.4.2
-Requires-Dist: fsspec ==2023.5.0
-Requires-Dist: numpy ==1.26.4
-Requires-Dist: urllib3 <2.1.0,>=1.25.3
+Requires-Dist: typer>=0.7.0
+Requires-Dist: requests>=2.28.1
+Requires-Dist: pydantic>=2.8.2
+Requires-Dist: pyyaml>=6.0
+Requires-Dist: pyarrow==11.0.0
+Requires-Dist: pandas==1.5.3
+Requires-Dist: tqdm==4.65.0
+Requires-Dist: s3fs==0.4.2
+Requires-Dist: fsspec==2023.5.0
+Requires-Dist: numpy==1.26.4
+Requires-Dist: urllib3<2.1.0,>=1.25.3
 Requires-Dist: python-dateutil
-Requires-Dist: typing-extensions >=4.7.1
-Requires-Dist: pytest >=6.2.5
-Requires-Dist: pytest-mock ==3.14.0
+Requires-Dist: typing-extensions>=4.7.1
+Requires-Dist: pytest>=6.2.5
+Requires-Dist: pytest-mock==3.14.0
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
 
 # Python SDK
 

{shaped-0.14.1.dist-info → shaped-1.0.2.dist-info}/RECORD

@@ -1,26 +1,26 @@
-shaped/__init__.py,sha256=oPg0RwEQOD2WTwzuujrETTWpcCL9r8q76EE2tFAELVk,184
-shaped/client.py,sha256=v_u9LyP_1BeoFd0YOSiuNQey7Uou0XBGo6aCoabua0o,8069
-shaped/autogen/__init__.py,sha256=JbYo6slNApI-rKmogQZ0KrDbsWI8m0dIYdJ_PRCdzNY,6790
-shaped/autogen/api_client.py,sha256=HX_8aj2K5PVYdKE6gtiU5NiviukXoVVN5ju4Y7bXdDM,28704
+shaped/__init__.py,sha256=ikq1AG3uapaJAzN7dx3V4WLl6Hx95vQTr_d8fO8ebJ4,184
+shaped/client.py,sha256=4poRVpvupQEm3RjdQJ2zuwIkFwTZziXY0sln-v9mQck,8372
+shaped/autogen/__init__.py,sha256=wKYfoKohE39ud3mAORGhwGT9Ouq9A-VoN2N2lVvM1TA,6859
+shaped/autogen/api_client.py,sha256=JC6ZcpvzPYK1krMeIdH-sXBkR2XsFM0HCO5YbmFylB4,29368
 shaped/autogen/api_response.py,sha256=eMxw1mpmJcoGZ3gs9z6jM4oYoZ10Gjk333s9sKxGv7s,652
-shaped/autogen/configuration.py,sha256=YMXxgCxr4yAEKO54qZNDNSPst9X48NDvnHoeFd4sLP4,16976
+shaped/autogen/configuration.py,sha256=rTPXrCmxc1gn375s6gxaeR5-DzKUZ31bJnplg77PmSE,19513
 shaped/autogen/exceptions.py,sha256=5muki093ghdifnr96i9wLiRBACuzCRIcken82pKvbXM,7873
-shaped/autogen/rest.py,sha256=7iqvWI7oXJ3TCbQrX4JPjmMAGNqBQ6Wm_-M20zvBrhM,11309
+shaped/autogen/rest.py,sha256=ycxoXj-E1oIl1x4xuogvQJViYYrDaUgcxSNXZ6UlTik,11313
 shaped/autogen/api/__init__.py,sha256=ljAq90mlHSU5qdMLEZpf4uLd5SyLR8DcllMgDsZLFo0,242
-shaped/autogen/api/dataset_api.py,sha256=7Ek2Ta3z0GDQir-Sw7LPamVDsf72p7efF8_bVsWN8KU,48859
-shaped/autogen/api/model_inference_api.py,sha256=3ZleRwYBcEJpxIjkRH-vaVvrjNeGTCgBQK83K0f_wGU,130193
-shaped/autogen/api/model_management_api.py,sha256=ksFmi9JvuH6KBdGaCQz4p-iRn2mWJgR8zYcBhnHJ4b8,60494
-shaped/autogen/models/__init__.py,sha256=4ppE18cZEL4IeP3UrM873TALzRlCoOXw2WbxAwmHeDg,6052
+shaped/autogen/api/dataset_api.py,sha256=RkOwlddbLtop1Zhdm4agP4Eic5nRVes-INylNRftQ3Q,49143
+shaped/autogen/api/model_inference_api.py,sha256=JBIFALi-eR1_U4Bt_jjJf_4Mu3Mx6_CKYs3uYi1naRU,130921
+shaped/autogen/api/model_management_api.py,sha256=1qRUUFCbcle1KtGH6OzhmsbK6Fk36zzAZ_CiTPHn0l4,60849
+shaped/autogen/models/__init__.py,sha256=wEJt1anUDMx3Q3NnW_f-eCXee_nPrWWw8h8y1ab-IFw,6121
 shaped/autogen/models/amplitude_dataset_config.py,sha256=wsH4E2dATObPZHPnbLK9QQT5pyqmAp0cjqDCHHKhios,4770
 shaped/autogen/models/aws_pinpoint_dataset_config.py,sha256=bhCN4SezUSQ66ofuib_k7o51QNLwpTb-45HK7ubAm10,4852
 shaped/autogen/models/big_query_dataset_config.py,sha256=xNfaQOvWjDQe-tlH7bBSNvsKlYiuo0589noZ-4dPGU8,5711
-shaped/autogen/models/complement_items_request.py,sha256=k4HtzOtUq4Pl1_EX3bfsI6uGu4ZtnVkHuVwp-RWJiSg,5060
+shaped/autogen/models/complement_items_request.py,sha256=KJLw_ztHn8smrgrgB7iinHAULG7KtjULPk6bvUVv9gY,5289
 shaped/autogen/models/complement_items_response.py,sha256=eg3f9wtW94ErrPWt3L_yQkCQi4qDkZXWDl4IHnaHTuI,4530
 shaped/autogen/models/connectors_inner.py,sha256=tsiFiE21KUz3mUqASR1OQyIwyGQC8G5KsxA6pBK9J6w,6750
 shaped/autogen/models/create_dataset_arguments.py,sha256=-yKKVi_vUMJJcilbU4BxDFN5Z0pWnJDch87b6FEvyyQ,15473
 shaped/autogen/models/create_embedding_response.py,sha256=2SiEooeVQq342thw-iap1dlW0wtUdi3OcxSGHDllvLk,4485
 shaped/autogen/models/create_item_embedding_request.py,sha256=if10EuiWUoE0RORd6ni4ou3lnOV1XAlBgqx5Xxrmbic,4566
-shaped/autogen/models/create_model_arguments.py,sha256=EJAf_8bFd1uTjndcZiZYoByy3hz5ukUbzwEgx07EqpU,5593
+shaped/autogen/models/create_model_arguments.py,sha256=9IvLu5K75w6DO9lizNoeJVZi44ajP2BVeo0lHx5QOY0,5626
 shaped/autogen/models/create_model_response.py,sha256=vi8bUhKlbGGXvv-JxTZ9xMzxO26_acRw9lLkHDZNEEQ,4413
 shaped/autogen/models/create_user_embedding_request.py,sha256=jmv43RlAgY_4V2mvWUF74PolAo5FmTslH-jKpjlBMQE,4566
 shaped/autogen/models/custom_dataset_config.py,sha256=st8Fs4sio4lPV35ot_gW99mdarQdnJSjKSxy8t1J3kQ,6190
@@ -31,18 +31,18 @@ shaped/autogen/models/delete_model_response.py,sha256=X23iIe_LSq2kKkHuLcFgmBhizB
 shaped/autogen/models/fetch_config.py,sha256=p5tLHOmxAgY7iQ0O3Uwsd-ABmJZ7fS60oz-gAGUBINw,6032
 shaped/autogen/models/file_config.py,sha256=Xd3dVYLUjVxqnEugQsnrPEzhLO-ygbZBIVMr1idkjHg,5163
 shaped/autogen/models/file_source_config.py,sha256=pbKudXQqyx8DG_GTFGHnCTOJ3EedZLCQq7gzhxIr9cQ,4559
-shaped/autogen/models/http_validation_error.py,sha256=lJ6SHMqKo2lckZ_Jr__o-XmAdIWFCvLFEbhppyWDWmw,4884
+shaped/autogen/models/http_validation_error.py,sha256=gQLYz56RE9XsXHl25Hus142IENnZn5kpDMp8DRXyxyA,4905
 shaped/autogen/models/inference_config.py,sha256=TL0RUsyV112PCZYA71b-FjMFyS54xwce2LrhA0KwuYg,7148
 shaped/autogen/models/insert_model_response.py,sha256=5_Y_ZeXmlwVPUvwVuZfbGhZqS24Aw63dvhT519i_nGI,4432
 shaped/autogen/models/interaction.py,sha256=YuHXFfejjkAnYogyjJZHWZ0nrmg3yHsbN7EC43U4T6E,4373
-shaped/autogen/models/list_datasets_response.py,sha256=8iat9Cp0sMgGDQbQnZoamxCRBdmzYmqHEoiRzmdpX0o,4871
-shaped/autogen/models/list_models_response.py,sha256=5AHiQyVJqD1HrObUMW2-UmC-HiRRclPzO_s17K1jPKQ,4845
+shaped/autogen/models/list_datasets_response.py,sha256=ogXMA72jgaHLSBgQuuWTsbFcbGeS2v5GzhdqnclTBT4,4898
+shaped/autogen/models/list_models_response.py,sha256=lLDGfP2T7Vh9sqeq6bNGc1kB--B1vNQ62sXSuqAVfBI,4866
 shaped/autogen/models/model_config.py,sha256=2iDd3ZOEVoK5vcCnEwNJoruEDN-5ojBCtNWTuNLimz0,5915
 shaped/autogen/models/model_response.py,sha256=J9iKVIUlYIalXPa7Hti76NRbMlTfecOJCchT1dxSKH0,5428
 shaped/autogen/models/mongo_db_dataset_config.py,sha256=yNjuT5CBZAtc-Gc0RtCvcUtEcWihd9h-7qzrZiniWOg,6012
 shaped/autogen/models/my_sql_dataset_config.py,sha256=63ODzapQguEyVmE6LPIGljpYfRnm9aOZMSxJmVdb8K8,8797
 shaped/autogen/models/path.py,sha256=dyMZsvg5oekbRNWTLHWzCGo4pE2yNwaCH-zS05_tExk,6741
-shaped/autogen/models/post_rank_request.py,sha256=5IBH4R16YF8GBMxl8UNvWih1eQjrhkEzqQvM-Q6xdj4,7975
+shaped/autogen/models/post_rank_request.py,sha256=oaJh8y-nZpNGbGkA7lvZtjBzwbnp04zqzxAlTS_fBQc,8014
 shaped/autogen/models/postgres_dataset_config.py,sha256=loTx56lxMCuirIgIQF41EnGOB70Jw14CnfIeZTj7nw4,8815
 shaped/autogen/models/rank_attribute_response.py,sha256=ZRhyj3bwo16Ul78LSGPtKuJ-2xPyw8JsauzbrkFpHzk,4561
 shaped/autogen/models/rank_grid_attribute_request.py,sha256=84vVMGO8Td2j8qRfDz1l8gb3QM1a-1yY9UNmjA6pZ9A,4732
@@ -51,22 +51,23 @@ shaped/autogen/models/rank_grid_attribute_response.py,sha256=1GsgjCBZuBLn6H3P64H
 shaped/autogen/models/rank_response.py,sha256=ttP-e5744WVpA30FYHosoyNF5ytqOUoYvi4coc9tsLk,4616
 shaped/autogen/models/redshift_dataset_config.py,sha256=PB8w0ecT2vbfEvWGLOjX7lQexJGocf8d_4apPUa86Yo,6743
 shaped/autogen/models/retrieve_request.py,sha256=Llrk-9gAZS6u1q8NRv8MiRiOsAJi9QhT46Rj8hzomqQ,6066
-shaped/autogen/models/retriever_top_k_override.py,sha256=cP-VpLDAT_jS4ArrBt5Le2OXvA8WLIDg0zSK5a5wry0,6190
+shaped/autogen/models/retrieve_response.py,sha256=AsnkhIZp8rfPoH0u7imbOZ7FfK-yMGxBS368s8TJhM4,4649
+shaped/autogen/models/retriever_top_k_override.py,sha256=PQ3dLXp2Pq_l3NooKWMqA3IPpduvyyp5W_YVom3FTm8,6190
 shaped/autogen/models/rudder_stack_dataset_config.py,sha256=z3fUCHf7sPAh5yUCz8_GDNVN1LHazu3VsLvDt5EFHwM,4782
 shaped/autogen/models/segment_dataset_config.py,sha256=1OPaweyV4Pd2KGpWByAAl_yaPR-PN8R_cg_teft1XLQ,4758
-shaped/autogen/models/similar_item_request.py,sha256=1GKr8wMmoj7D0Zc6WkmohZRvyvOHXKGjmpY8DAx5OPE,5741
+shaped/autogen/models/similar_item_request.py,sha256=pPnq9EqIpQ56KiKqdqOHCFvLwyDrtboH0r75KNbvfog,5963
 shaped/autogen/models/similar_response.py,sha256=kk8tY1oJgnuehLD4orddtAnnb5zewtVXst9NPEoFjiQ,4498
-shaped/autogen/models/similar_users_request.py,sha256=U9UGO5SKOQjNnMTgharXHE0L3iXRpM7i5Iy4pXcosJg,5576
+shaped/autogen/models/similar_users_request.py,sha256=xbtRy90CcS2U8_rBOE8GMfOFDfUCgc5VDMIJmn5WXOg,5798
 shaped/autogen/models/snowflake_dataset_config.py,sha256=Io0IKwcE69pzJrbONXPWdMYkfjw6J8dBT1h9iKN_NRk,6974
 shaped/autogen/models/successful_response.py,sha256=sQ3oVhrvJUQ9ApgDd4jsr7eZfZvu-goghD8xZCeDFzk,4428
 shaped/autogen/models/validation_error.py,sha256=kVstUJ1nwQ6Y9kssgDBpeoKGFScWegporYC6klUoMe0,4438
 shaped/autogen/models/value_type.py,sha256=uxlGNPto3CK17jHKDRvAl3BqNCPkHlxfRBPdofi_qdM,2917
 shaped/autogen/models/view_model_response.py,sha256=4xzesZSHMcPCZHsevo5Bt_3P9jBNiEbd_PFTWQ5Q9KU,5757
 shaped/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-shaped/cli/shaped_cli.py,sha256=kOxCiPivUEjNPs4mNETGpi43_PzTXp61dN-dr9fp7C0,12822
-shaped-0.14.1.dist-info/METADATA,sha256=thrHS_hfUqWd74uh_yUpaFgZhGnjlJV3wZtjLBW7_YU,5448
-shaped-0.14.1.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-shaped-0.14.1.dist-info/entry_points.txt,sha256=4xoVmnNTKtmzjGNX5ezp9MCWLgi7880TtfqFByN1u5U,53
-shaped-0.14.1.dist-info/top_level.txt,sha256=w-lDaoadQVYpze9N9gZyK9qngb7fZCJ-KCdHLGvt0SU,7
-shaped-0.14.1.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-shaped-0.14.1.dist-info/RECORD,,
+shaped/cli/shaped_cli.py,sha256=k2GSO1zNKvf_o_bRcJoJUZxGQxYdyedbL7XJvPJguMU,12821
+shaped-1.0.2.dist-info/METADATA,sha256=IyMKtYZlOL-NnTz3-W_9csMg_JjdmRx-poVGDrH9GJQ,5648
+shaped-1.0.2.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+shaped-1.0.2.dist-info/entry_points.txt,sha256=4xoVmnNTKtmzjGNX5ezp9MCWLgi7880TtfqFByN1u5U,53
+shaped-1.0.2.dist-info/top_level.txt,sha256=w-lDaoadQVYpze9N9gZyK9qngb7fZCJ-KCdHLGvt0SU,7
+shaped-1.0.2.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+shaped-1.0.2.dist-info/RECORD,,

{shaped-0.14.1.dist-info → shaped-1.0.2.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.1.0)
+Generator: setuptools (75.8.0)
 Root-Is-Purelib: true
 Tag: py3-none-any