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

shaped/autogen/api/query_api.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.
@@ -44,7 +44,6 @@ class QueryApi:
     @validate_call
     def execute_ad_hoc_query_query_post(
         self,
-        engine_name: StrictStr,
         query_request: QueryRequest,
         _request_timeout: Union[
             None,
@@ -63,8 +62,6 @@ class QueryApi:
 
         Execute a query with the provided QueryConfig. Use this endpoint to execute on-demand queries for development or experimentation.
 
-        :param engine_name: (required)
-        :type engine_name: str
         :param query_request: (required)
         :type query_request: QueryRequest
         :param _request_timeout: timeout setting for this request. If one
@@ -90,7 +87,6 @@ class QueryApi:
         """ # noqa: E501
 
         _param = self._execute_ad_hoc_query_query_post_serialize(
-            engine_name=engine_name,
             query_request=query_request,
             _request_auth=_request_auth,
             _content_type=_content_type,
@@ -116,7 +112,6 @@ class QueryApi:
     @validate_call
     def execute_ad_hoc_query_query_post_with_http_info(
         self,
-        engine_name: StrictStr,
         query_request: QueryRequest,
         _request_timeout: Union[
             None,
@@ -135,8 +130,6 @@ class QueryApi:
 
         Execute a query with the provided QueryConfig. Use this endpoint to execute on-demand queries for development or experimentation.
 
-        :param engine_name: (required)
-        :type engine_name: str
         :param query_request: (required)
         :type query_request: QueryRequest
         :param _request_timeout: timeout setting for this request. If one
@@ -162,7 +155,6 @@ class QueryApi:
         """ # noqa: E501
 
         _param = self._execute_ad_hoc_query_query_post_serialize(
-            engine_name=engine_name,
             query_request=query_request,
             _request_auth=_request_auth,
             _content_type=_content_type,
@@ -188,7 +180,6 @@ class QueryApi:
     @validate_call
     def execute_ad_hoc_query_query_post_without_preload_content(
         self,
-        engine_name: StrictStr,
         query_request: QueryRequest,
         _request_timeout: Union[
             None,
@@ -207,8 +198,6 @@ class QueryApi:
 
         Execute a query with the provided QueryConfig. Use this endpoint to execute on-demand queries for development or experimentation.
 
-        :param engine_name: (required)
-        :type engine_name: str
         :param query_request: (required)
         :type query_request: QueryRequest
         :param _request_timeout: timeout setting for this request. If one
@@ -234,7 +223,6 @@ class QueryApi:
         """ # noqa: E501
 
         _param = self._execute_ad_hoc_query_query_post_serialize(
-            engine_name=engine_name,
             query_request=query_request,
             _request_auth=_request_auth,
             _content_type=_content_type,
@@ -255,7 +243,6 @@ class QueryApi:
 
     def _execute_ad_hoc_query_query_post_serialize(
         self,
-        engine_name,
         query_request,
         _request_auth,
         _content_type,
@@ -278,8 +265,6 @@ class QueryApi:
         _body_params: Optional[bytes] = None
 
         # process the path parameters
-        if engine_name is not None:
-            _path_params['engine_name'] = engine_name
         # process the query parameters
         # process the header parameters
         # process the form parameters
@@ -316,7 +301,7 @@ class QueryApi:
 
         return self.api_client.param_serialize(
             method='POST',
-            resource_path='/engines/{engine_name}/query',
+            resource_path='/query',
             path_params=_path_params,
             query_params=_query_params,
             header_params=_header_params,
@@ -335,7 +320,6 @@ class QueryApi:
     @validate_call
     def execute_saved_query_queries_query_name_post(
         self,
-        engine_name: StrictStr,
         query_name: StrictStr,
         saved_query_request: SavedQueryRequest,
         _request_timeout: Union[
@@ -355,8 +339,6 @@ class QueryApi:
 
         Execute a saved query by name. The `parameters` field defines inputs for the saved query, such as a search string or user ID.
 
-        :param engine_name: (required)
-        :type engine_name: str
         :param query_name: (required)
         :type query_name: str
         :param saved_query_request: (required)
@@ -384,7 +366,6 @@ class QueryApi:
         """ # noqa: E501
 
         _param = self._execute_saved_query_queries_query_name_post_serialize(
-            engine_name=engine_name,
             query_name=query_name,
             saved_query_request=saved_query_request,
             _request_auth=_request_auth,
@@ -411,7 +392,6 @@ class QueryApi:
     @validate_call
     def execute_saved_query_queries_query_name_post_with_http_info(
         self,
-        engine_name: StrictStr,
         query_name: StrictStr,
         saved_query_request: SavedQueryRequest,
         _request_timeout: Union[
@@ -431,8 +411,6 @@ class QueryApi:
 
         Execute a saved query by name. The `parameters` field defines inputs for the saved query, such as a search string or user ID.
 
-        :param engine_name: (required)
-        :type engine_name: str
         :param query_name: (required)
         :type query_name: str
         :param saved_query_request: (required)
@@ -460,7 +438,6 @@ class QueryApi:
         """ # noqa: E501
 
         _param = self._execute_saved_query_queries_query_name_post_serialize(
-            engine_name=engine_name,
             query_name=query_name,
             saved_query_request=saved_query_request,
             _request_auth=_request_auth,
@@ -487,7 +464,6 @@ class QueryApi:
     @validate_call
     def execute_saved_query_queries_query_name_post_without_preload_content(
         self,
-        engine_name: StrictStr,
         query_name: StrictStr,
         saved_query_request: SavedQueryRequest,
         _request_timeout: Union[
@@ -507,8 +483,6 @@ class QueryApi:
 
         Execute a saved query by name. The `parameters` field defines inputs for the saved query, such as a search string or user ID.
 
-        :param engine_name: (required)
-        :type engine_name: str
         :param query_name: (required)
         :type query_name: str
         :param saved_query_request: (required)
@@ -536,7 +510,6 @@ class QueryApi:
         """ # noqa: E501
 
         _param = self._execute_saved_query_queries_query_name_post_serialize(
-            engine_name=engine_name,
             query_name=query_name,
             saved_query_request=saved_query_request,
             _request_auth=_request_auth,
@@ -558,7 +531,6 @@ class QueryApi:
 
     def _execute_saved_query_queries_query_name_post_serialize(
         self,
-        engine_name,
         query_name,
         saved_query_request,
         _request_auth,
@@ -582,8 +554,6 @@ class QueryApi:
         _body_params: Optional[bytes] = None
 
         # process the path parameters
-        if engine_name is not None:
-            _path_params['engine_name'] = engine_name
         if query_name is not None:
             _path_params['query_name'] = query_name
         # process the query parameters
@@ -622,7 +592,7 @@ class QueryApi:
 
         return self.api_client.param_serialize(
             method='POST',
-            resource_path='/engines/{engine_name}/queries/{query_name}',
+            resource_path='/queries/{query_name}',
             path_params=_path_params,
             query_params=_query_params,
             header_params=_header_params,
@@ -641,7 +611,6 @@ class QueryApi:
     @validate_call
     def get_saved_query_info_queries_query_name_get(
         self,
-        engine_name: StrictStr,
         query_name: StrictStr,
         _request_timeout: Union[
             None,
@@ -660,8 +629,6 @@ class QueryApi:
 
         Get information about a specific saved query.
 
-        :param engine_name: (required)
-        :type engine_name: str
         :param query_name: (required)
         :type query_name: str
         :param _request_timeout: timeout setting for this request. If one
@@ -687,7 +654,6 @@ class QueryApi:
         """ # noqa: E501
 
         _param = self._get_saved_query_info_queries_query_name_get_serialize(
-            engine_name=engine_name,
             query_name=query_name,
             _request_auth=_request_auth,
             _content_type=_content_type,
@@ -713,7 +679,6 @@ class QueryApi:
     @validate_call
     def get_saved_query_info_queries_query_name_get_with_http_info(
         self,
-        engine_name: StrictStr,
         query_name: StrictStr,
         _request_timeout: Union[
             None,
@@ -732,8 +697,6 @@ class QueryApi:
 
         Get information about a specific saved query.
 
-        :param engine_name: (required)
-        :type engine_name: str
         :param query_name: (required)
         :type query_name: str
         :param _request_timeout: timeout setting for this request. If one
@@ -759,7 +722,6 @@ class QueryApi:
         """ # noqa: E501
 
         _param = self._get_saved_query_info_queries_query_name_get_serialize(
-            engine_name=engine_name,
             query_name=query_name,
             _request_auth=_request_auth,
             _content_type=_content_type,
@@ -785,7 +747,6 @@ class QueryApi:
     @validate_call
     def get_saved_query_info_queries_query_name_get_without_preload_content(
         self,
-        engine_name: StrictStr,
         query_name: StrictStr,
         _request_timeout: Union[
             None,
@@ -804,8 +765,6 @@ class QueryApi:
 
         Get information about a specific saved query.
 
-        :param engine_name: (required)
-        :type engine_name: str
         :param query_name: (required)
         :type query_name: str
         :param _request_timeout: timeout setting for this request. If one
@@ -831,7 +790,6 @@ class QueryApi:
         """ # noqa: E501
 
         _param = self._get_saved_query_info_queries_query_name_get_serialize(
-            engine_name=engine_name,
             query_name=query_name,
             _request_auth=_request_auth,
             _content_type=_content_type,
@@ -852,7 +810,6 @@ class QueryApi:
 
     def _get_saved_query_info_queries_query_name_get_serialize(
         self,
-        engine_name,
         query_name,
         _request_auth,
         _content_type,
@@ -875,8 +832,6 @@ class QueryApi:
         _body_params: Optional[bytes] = None
 
         # process the path parameters
-        if engine_name is not None:
-            _path_params['engine_name'] = engine_name
         if query_name is not None:
             _path_params['query_name'] = query_name
         # process the query parameters
@@ -900,7 +855,7 @@ class QueryApi:
 
         return self.api_client.param_serialize(
             method='GET',
-            resource_path='/engines/{engine_name}/queries/{query_name}',
+            resource_path='/queries/{query_name}',
             path_params=_path_params,
             query_params=_query_params,
             header_params=_header_params,
@@ -919,7 +874,6 @@ class QueryApi:
     @validate_call
     def list_saved_queries_queries_get(
         self,
-        engine_name: StrictStr,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -935,10 +889,8 @@ class QueryApi:
     ) -> SavedQueryListResponse:
         """List Saved Queries
 
-        List all available saved queries for an engine.
+        List all available saved queries.
 
-        :param engine_name: (required)
-        :type engine_name: str
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -962,7 +914,6 @@ class QueryApi:
         """ # noqa: E501
 
         _param = self._list_saved_queries_queries_get_serialize(
-            engine_name=engine_name,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -986,7 +937,6 @@ class QueryApi:
     @validate_call
     def list_saved_queries_queries_get_with_http_info(
         self,
-        engine_name: StrictStr,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -1002,10 +952,8 @@ class QueryApi:
     ) -> ApiResponse[SavedQueryListResponse]:
         """List Saved Queries
 
-        List all available saved queries for an engine.
+        List all available saved queries.
 
-        :param engine_name: (required)
-        :type engine_name: str
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -1029,7 +977,6 @@ class QueryApi:
         """ # noqa: E501
 
         _param = self._list_saved_queries_queries_get_serialize(
-            engine_name=engine_name,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -1053,7 +1000,6 @@ class QueryApi:
     @validate_call
     def list_saved_queries_queries_get_without_preload_content(
         self,
-        engine_name: StrictStr,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -1069,10 +1015,8 @@ class QueryApi:
     ) -> RESTResponseType:
         """List Saved Queries
 
-        List all available saved queries for an engine.
+        List all available saved queries.
 
-        :param engine_name: (required)
-        :type engine_name: str
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -1096,7 +1040,6 @@ class QueryApi:
         """ # noqa: E501
 
         _param = self._list_saved_queries_queries_get_serialize(
-            engine_name=engine_name,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -1115,7 +1058,6 @@ class QueryApi:
 
     def _list_saved_queries_queries_get_serialize(
         self,
-        engine_name,
         _request_auth,
         _content_type,
         _headers,
@@ -1137,8 +1079,6 @@ class QueryApi:
         _body_params: Optional[bytes] = None
 
         # process the path parameters
-        if engine_name is not None:
-            _path_params['engine_name'] = engine_name
         # process the query parameters
         # process the header parameters
         # process the form parameters
@@ -1160,7 +1100,7 @@ class QueryApi:
 
         return self.api_client.param_serialize(
             method='GET',
-            resource_path='/engines/{engine_name}/queries',
+            resource_path='/queries',
             path_params=_path_params,
             query_params=_query_params,
             header_params=_header_params,

shaped/autogen/api/table_api.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/api/view_api.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/api_client.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.
@@ -312,7 +312,7 @@ class ApiClient:
                 return_data = self.__deserialize_file(response_data)
             elif response_type is not None:
                 match = None
-                content_type = response_data.getheader('content-type')
+                content_type = response_data.headers.get('content-type')
                 if content_type is not None:
                     match = re.search(r"charset=([a-zA-Z\-\d]+)[\s;]?", content_type)
                 encoding = match.group(1) if match else "utf-8"
@@ -329,7 +329,7 @@ class ApiClient:
         return ApiResponse(
             status_code = response_data.status,
             data = return_data,
-            headers = response_data.getheaders(),
+            headers = response_data.headers,
             raw_data = response_data.data
         )
 
@@ -701,7 +701,7 @@ class ApiClient:
         os.close(fd)
         os.remove(path)
 
-        content_disposition = response.getheader("Content-Disposition")
+        content_disposition = response.headers.get("Content-Disposition")
         if content_disposition:
             m = re.search(
                 r'filename=[\'"]?([^\'"\s]+)[\'"]?',

shaped/autogen/configuration.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.
@@ -163,7 +163,7 @@ class Configuration:
     :param ca_cert_data: verify the peer using concatenated CA certificate data
       in PEM (str) or DER (bytes) format.
     :param cert_file: the path to a client certificate file, for mTLS.
-    :param key_file: the path to a client key file, for mTLS. 
+    :param key_file: the path to a client key file, for mTLS.
 
     """
 
@@ -484,6 +484,7 @@ class Configuration:
         password = ""
         if self.password is not None:
             password = self.password
+
         return urllib3.util.make_headers(
             basic_auth=username + ':' + password
         ).get('authorization')
@@ -504,7 +505,7 @@ class Configuration:
         return "Python SDK Debug Report:\n"\
                "OS: {env}\n"\
                "Python Version: {pyversion}\n"\
-               "Version of the API: 2.0.0\n"\
+               "Version of the API: 2.0.4\n"\
                "SDK Package Version: 1.0.0".\
                format(env=sys.platform, pyversion=sys.version)
 
@@ -553,6 +554,7 @@ class Configuration:
                 variable_name, variable['default_value'])
 
             if 'enum_values' in variable \
+                    and variable['enum_values'] \
                     and used_value not in variable['enum_values']:
                 raise ValueError(
                     "The variable `{0}` in the host URL has invalid value "

shaped/autogen/exceptions.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.
@@ -128,7 +128,7 @@ class ApiException(OpenApiException):
                     self.body = http_resp.data.decode('utf-8')
                 except Exception:
                     pass
-            self.headers = http_resp.getheaders()
+            self.headers = http_resp.headers
 
     @classmethod
     def from_response(
@@ -169,8 +169,11 @@ class ApiException(OpenApiException):
             error_message += "HTTP response headers: {0}\n".format(
                 self.headers)
 
-        if self.data or self.body:
-            error_message += "HTTP response body: {0}\n".format(self.data or self.body)
+        if self.body:
+            error_message += "HTTP response body: {0}\n".format(self.body)
+
+        if self.data:
+            error_message += "HTTP response data: {0}\n".format(self.data)
 
         return error_message