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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (341) hide show
  1. shaped/__init__.py +59 -4
  2. shaped/autogen/__init__.py +527 -66
  3. shaped/autogen/api/__init__.py +4 -3
  4. shaped/autogen/api/engine_api.py +1467 -0
  5. shaped/autogen/api/{dataset_api.py → query_api.py} +159 -194
  6. shaped/autogen/api/table_api.py +1494 -0
  7. shaped/autogen/api/{model_management_api.py → view_api.py} +179 -181
  8. shaped/autogen/api_client.py +18 -11
  9. shaped/autogen/configuration.py +22 -9
  10. shaped/autogen/exceptions.py +25 -5
  11. shaped/autogen/models/__init__.py +247 -52
  12. shaped/autogen/models/ai_enrichment_view_config.py +123 -0
  13. shaped/autogen/models/{path.py → algorithm.py} +19 -19
  14. shaped/autogen/models/amplitude_table_config.py +106 -0
  15. shaped/autogen/models/ascending.py +148 -0
  16. shaped/autogen/models/attn_dropout_prob.py +136 -0
  17. shaped/autogen/models/attribute_journey.py +124 -0
  18. shaped/autogen/models/attribute_value.py +178 -0
  19. shaped/autogen/models/autoscaling_config.py +95 -0
  20. shaped/autogen/models/aws_pinpoint_table_config.py +108 -0
  21. shaped/autogen/models/batch_size.py +136 -0
  22. shaped/autogen/models/batch_size1.py +136 -0
  23. shaped/autogen/models/batch_size2.py +136 -0
  24. shaped/autogen/models/big_query_table_config.py +147 -0
  25. shaped/autogen/models/bm25.py +136 -0
  26. shaped/autogen/models/boosted_reorder_step.py +125 -0
  27. shaped/autogen/models/canary_rollout.py +99 -0
  28. shaped/autogen/models/candidate_attributes_retrieve_step.py +113 -0
  29. shaped/autogen/models/candidate_ids_retrieve_step.py +113 -0
  30. shaped/autogen/models/candidate_retrieval_strategy.py +41 -0
  31. shaped/autogen/models/clickhouse_table_config.py +146 -0
  32. shaped/autogen/models/column_order_retrieve_step.py +123 -0
  33. shaped/autogen/models/column_ordering.py +91 -0
  34. shaped/autogen/models/create_table_response.py +87 -0
  35. shaped/autogen/models/create_view_response.py +87 -0
  36. shaped/autogen/models/custom_table_config.py +135 -0
  37. shaped/autogen/models/data_compute_config.py +89 -0
  38. shaped/autogen/models/data_config.py +145 -0
  39. shaped/autogen/models/data_config_interaction_table.py +146 -0
  40. shaped/autogen/models/data_split_config.py +88 -0
  41. shaped/autogen/models/data_split_strategy.py +37 -0
  42. shaped/autogen/models/data_tier.py +37 -0
  43. shaped/autogen/models/default.py +246 -0
  44. shaped/autogen/models/delete_engine_response.py +87 -0
  45. shaped/autogen/models/delete_table_response.py +87 -0
  46. shaped/autogen/models/delete_view_response.py +87 -0
  47. shaped/autogen/models/deployment_config.py +117 -0
  48. shaped/autogen/models/distance_function.py +38 -0
  49. shaped/autogen/models/diversity_reorder_step.py +137 -0
  50. shaped/autogen/models/dropout_rate.py +136 -0
  51. shaped/autogen/models/dynamo_db_table_config.py +160 -0
  52. shaped/autogen/models/dynamo_db_table_config_scan_kwargs_value.py +138 -0
  53. shaped/autogen/models/embedder_batch_size.py +136 -0
  54. shaped/autogen/models/embedding_config.py +93 -0
  55. shaped/autogen/models/embedding_dim.py +136 -0
  56. shaped/autogen/models/embedding_dims.py +136 -0
  57. shaped/autogen/models/embedding_size.py +136 -0
  58. shaped/autogen/models/encoder.py +140 -0
  59. shaped/autogen/models/encoding_pooling_strategy.py +38 -0
  60. shaped/autogen/models/engine.py +109 -0
  61. shaped/autogen/models/engine_config_v2.py +152 -0
  62. shaped/autogen/models/engine_details_response.py +120 -0
  63. shaped/autogen/models/engine_schema.py +113 -0
  64. shaped/autogen/models/engine_schema_user_inner.py +146 -0
  65. shaped/autogen/models/entity_config.py +109 -0
  66. shaped/autogen/models/entity_journey.py +161 -0
  67. shaped/autogen/models/entity_type.py +38 -0
  68. shaped/autogen/models/evaluation_config.py +92 -0
  69. shaped/autogen/models/exploration_reorder_step.py +125 -0
  70. shaped/autogen/models/expression_filter_step.py +106 -0
  71. shaped/autogen/models/factors.py +136 -0
  72. shaped/autogen/models/factors1.py +136 -0
  73. shaped/autogen/models/feature.py +90 -0
  74. shaped/autogen/models/feature_type.py +60 -0
  75. shaped/autogen/models/file_table_config.py +112 -0
  76. shaped/autogen/models/filter_config.py +99 -0
  77. shaped/autogen/models/filter_dataset.py +164 -0
  78. shaped/autogen/models/filter_index_type.py +36 -0
  79. shaped/autogen/models/filter_retrieve_step.py +113 -0
  80. shaped/autogen/models/filter_step_explanation.py +165 -0
  81. shaped/autogen/models/filter_table.py +140 -0
  82. shaped/autogen/models/filter_type.py +134 -0
  83. shaped/autogen/models/global_filter.py +102 -0
  84. shaped/autogen/models/hidden_dropout_prob.py +136 -0
  85. shaped/autogen/models/hidden_size.py +136 -0
  86. shaped/autogen/models/hidden_size1.py +136 -0
  87. shaped/autogen/models/http_problem_response.py +115 -0
  88. shaped/autogen/models/http_validation_error.py +2 -2
  89. shaped/autogen/models/hugging_face_encoder.py +115 -0
  90. shaped/autogen/models/iceberg_table_config.py +154 -0
  91. shaped/autogen/models/index_config.py +101 -0
  92. shaped/autogen/models/inner_entity_id.py +144 -0
  93. shaped/autogen/models/inner_size.py +136 -0
  94. shaped/autogen/models/inner_size1.py +136 -0
  95. shaped/autogen/models/inner_uid.py +144 -0
  96. shaped/autogen/models/interaction_config.py +122 -0
  97. shaped/autogen/models/interaction_pooling_encoder.py +104 -0
  98. shaped/autogen/models/interaction_round_robin_encoder.py +104 -0
  99. shaped/autogen/models/item_attribute_pooling_encoder.py +111 -0
  100. shaped/autogen/models/journey.py +140 -0
  101. shaped/autogen/models/kafka_table_config.py +129 -0
  102. shaped/autogen/models/kinesis_table_config.py +140 -0
  103. shaped/autogen/models/kinesis_table_config_column_schema_value.py +136 -0
  104. shaped/autogen/models/label.py +90 -0
  105. shaped/autogen/models/label_type.py +36 -0
  106. shaped/autogen/models/laplace_smoothing.py +136 -0
  107. shaped/autogen/models/latency_scaling_policy.py +112 -0
  108. shaped/autogen/models/learning_rate.py +136 -0
  109. shaped/autogen/models/learning_rate1.py +136 -0
  110. shaped/autogen/models/learning_rate2.py +136 -0
  111. shaped/autogen/models/learning_rate3.py +136 -0
  112. shaped/autogen/models/lexical_search_mode.py +99 -0
  113. shaped/autogen/models/list_engines_response.py +95 -0
  114. shaped/autogen/models/list_tables_response.py +95 -0
  115. shaped/autogen/models/list_views_response.py +95 -0
  116. shaped/autogen/models/location_inner.py +138 -0
  117. shaped/autogen/models/loss_types.py +37 -0
  118. shaped/autogen/models/lr.py +136 -0
  119. shaped/autogen/models/lr1.py +136 -0
  120. shaped/autogen/models/lr2.py +136 -0
  121. shaped/autogen/models/max_depth.py +136 -0
  122. shaped/autogen/models/max_leaves.py +136 -0
  123. shaped/autogen/models/max_seq_length.py +136 -0
  124. shaped/autogen/models/max_seq_length1.py +136 -0
  125. shaped/autogen/models/max_seq_length2.py +136 -0
  126. shaped/autogen/models/mode.py +134 -0
  127. shaped/autogen/models/mode1.py +136 -0
  128. shaped/autogen/models/mode2.py +150 -0
  129. shaped/autogen/models/models_inner.py +308 -0
  130. shaped/autogen/models/mongo_db_table_config.py +147 -0
  131. shaped/autogen/models/mssql_table_config.py +155 -0
  132. shaped/autogen/models/{my_sql_dataset_config.py → my_sql_table_config.py} +45 -28
  133. shaped/autogen/models/n_epochs.py +136 -0
  134. shaped/autogen/models/n_epochs1.py +136 -0
  135. shaped/autogen/models/n_epochs2.py +136 -0
  136. shaped/autogen/models/n_estimators.py +136 -0
  137. shaped/autogen/models/n_heads.py +136 -0
  138. shaped/autogen/models/n_layers.py +136 -0
  139. shaped/autogen/models/neg_per_positive.py +136 -0
  140. shaped/autogen/models/negative_samples_count.py +136 -0
  141. shaped/autogen/models/ngram_tokenizer.py +103 -0
  142. shaped/autogen/models/no_op_config.py +117 -0
  143. shaped/autogen/models/num_blocks.py +136 -0
  144. shaped/autogen/models/num_heads.py +136 -0
  145. shaped/autogen/models/num_leaves.py +136 -0
  146. shaped/autogen/models/objective.py +40 -0
  147. shaped/autogen/models/objective1.py +134 -0
  148. shaped/autogen/models/online_store_config.py +89 -0
  149. shaped/autogen/models/pagination_config.py +87 -0
  150. shaped/autogen/models/parameter_definition.py +96 -0
  151. shaped/autogen/models/parameters_value.py +240 -0
  152. shaped/autogen/models/passthrough_score.py +104 -0
  153. shaped/autogen/models/personal_filter.py +104 -0
  154. shaped/autogen/models/pipeline_stage_explanation.py +118 -0
  155. shaped/autogen/models/policy.py +134 -0
  156. shaped/autogen/models/pool_fn.py +152 -0
  157. shaped/autogen/models/pooling_function.py +37 -0
  158. shaped/autogen/models/{postgres_dataset_config.py → postgres_table_config.py} +66 -28
  159. shaped/autogen/models/posthog_table_config.py +133 -0
  160. shaped/autogen/models/prebuilt_filter_step.py +113 -0
  161. shaped/autogen/models/precomputed_item_embedding.py +99 -0
  162. shaped/autogen/models/precomputed_user_embedding.py +99 -0
  163. shaped/autogen/models/query.py +136 -0
  164. shaped/autogen/models/query1.py +136 -0
  165. shaped/autogen/models/query_any_of.py +172 -0
  166. shaped/autogen/models/query_config.py +140 -0
  167. shaped/autogen/models/query_definition.py +106 -0
  168. shaped/autogen/models/query_encoder.py +194 -0
  169. shaped/autogen/models/query_explanation.py +201 -0
  170. shaped/autogen/models/query_request.py +121 -0
  171. shaped/autogen/models/query_result.py +113 -0
  172. shaped/autogen/models/query_table_config.py +99 -0
  173. shaped/autogen/models/rank_item_attribute_values_query_config.py +122 -0
  174. shaped/autogen/models/rank_query_config.py +167 -0
  175. shaped/autogen/models/rank_query_config_filter_inner.py +149 -0
  176. shaped/autogen/models/rank_query_config_reorder_inner.py +186 -0
  177. shaped/autogen/models/rank_query_config_retrieve_inner.py +265 -0
  178. shaped/autogen/models/recreate_rollout.py +97 -0
  179. shaped/autogen/models/{redshift_dataset_config.py → redshift_table_config.py} +48 -25
  180. shaped/autogen/models/reference_table_config.py +113 -0
  181. shaped/autogen/models/regularization.py +136 -0
  182. shaped/autogen/models/reorder_inner.py +149 -0
  183. shaped/autogen/models/reorder_step_explanation.py +207 -0
  184. shaped/autogen/models/request.py +378 -0
  185. shaped/autogen/models/request1.py +140 -0
  186. shaped/autogen/models/requests_per_second_scaling_policy.py +112 -0
  187. shaped/autogen/models/resource_config.py +100 -0
  188. shaped/autogen/models/response_get_view_details_views_view_name_get.py +134 -0
  189. shaped/autogen/models/result.py +132 -0
  190. shaped/autogen/models/result_embeddings_value.py +127 -0
  191. shaped/autogen/models/retrieval_scores_value.py +127 -0
  192. shaped/autogen/models/retrieve_inner.py +196 -0
  193. shaped/autogen/models/retrieve_step_explanation.py +172 -0
  194. shaped/autogen/models/retriever.py +196 -0
  195. shaped/autogen/models/retriever1.py +196 -0
  196. shaped/autogen/models/rollout_config.py +91 -0
  197. shaped/autogen/models/rudderstack_table_config.py +106 -0
  198. shaped/autogen/models/sampling_strategy.py +36 -0
  199. shaped/autogen/models/saved_query_info_response.py +103 -0
  200. shaped/autogen/models/saved_query_list_response.py +87 -0
  201. shaped/autogen/models/saved_query_request.py +115 -0
  202. shaped/autogen/models/schema_config.py +117 -0
  203. shaped/autogen/models/score.py +134 -0
  204. shaped/autogen/models/score_ensemble.py +127 -0
  205. shaped/autogen/models/score_ensemble_policy_config.py +165 -0
  206. shaped/autogen/models/score_ensemble_policy_config_policies_inner.py +755 -0
  207. shaped/autogen/models/score_step_explanation.py +224 -0
  208. shaped/autogen/models/search_config.py +105 -0
  209. shaped/autogen/models/segment_table_config.py +106 -0
  210. shaped/autogen/models/sequence_length.py +136 -0
  211. shaped/autogen/models/server_config.py +100 -0
  212. shaped/autogen/models/setup_engine_response.py +87 -0
  213. shaped/autogen/models/shaped_internal_recsys_policies_als_model_policy_als_model_policy_config.py +154 -0
  214. shaped/autogen/models/shaped_internal_recsys_policies_beeformer_model_policy_beeformer_model_policy_beeformer_model_policy_config.py +160 -0
  215. shaped/autogen/models/shaped_internal_recsys_policies_bert_model_policy_bert_model_policy_bert_model_policy_config.py +215 -0
  216. shaped/autogen/models/shaped_internal_recsys_policies_chronological_model_policy_chronological_model_policy_config.py +156 -0
  217. shaped/autogen/models/shaped_internal_recsys_policies_elsa_model_policy_elsa_model_policy_elsa_model_policy_config.py +141 -0
  218. shaped/autogen/models/shaped_internal_recsys_policies_gsasrec_model_policy_gsasrec_model_policy_gsas_rec_model_policy_config.py +211 -0
  219. shaped/autogen/models/shaped_internal_recsys_policies_item2vec_model_policy_item2_vec_model_policy_config.py +140 -0
  220. shaped/autogen/models/shaped_internal_recsys_policies_item_content_similarity_model_policy_item_content_similarity_model_policy_config.py +195 -0
  221. shaped/autogen/models/shaped_internal_recsys_policies_lightgbm_model_policy_lightgbm_model_policy_light_gbm_model_policy_config.py +245 -0
  222. shaped/autogen/models/shaped_internal_recsys_policies_ngram_model_policy_ngram_model_policy_config.py +119 -0
  223. shaped/autogen/models/shaped_internal_recsys_policies_popular_model_policy_popular_model_policy_config.py +161 -0
  224. shaped/autogen/models/shaped_internal_recsys_policies_random_model_policy_random_model_policy_config.py +108 -0
  225. shaped/autogen/models/shaped_internal_recsys_policies_recently_popular_policy_recently_popular_policy_config.py +143 -0
  226. shaped/autogen/models/shaped_internal_recsys_policies_rising_popular_policy_rising_popular_policy_config.py +123 -0
  227. shaped/autogen/models/shaped_internal_recsys_policies_sasrec_model_policy_sasrec_model_policy_sas_rec_model_policy_config.py +230 -0
  228. shaped/autogen/models/shaped_internal_recsys_policies_svd_model_policy_svd_model_policy_config.py +119 -0
  229. shaped/autogen/models/shaped_internal_recsys_policies_two_tower_model_policy_two_tower_model_policy_two_tower_model_policy_config.py +165 -0
  230. shaped/autogen/models/shaped_internal_recsys_policies_user_item_content_similarity_model_policy_user_item_content_similarity_model_policy_config.py +153 -0
  231. shaped/autogen/models/shaped_internal_recsys_policies_widedeep_model_policy_wide_deep_model_policy_config.py +137 -0
  232. shaped/autogen/models/shaped_internal_recsys_policies_xgboost_model_policy_xg_boost_model_policy_config.py +155 -0
  233. shaped/autogen/models/shopify_table_config.py +156 -0
  234. shaped/autogen/models/similarity_retrieve_step.py +123 -0
  235. shaped/autogen/models/{snowflake_dataset_config.py → snowflake_table_config.py} +47 -18
  236. shaped/autogen/models/sql_transform_type.py +37 -0
  237. shaped/autogen/models/sql_view_config.py +111 -0
  238. shaped/autogen/models/stemmer_tokenizer.py +105 -0
  239. shaped/autogen/models/step_explanation.py +137 -0
  240. shaped/autogen/models/steps_inner.py +179 -0
  241. shaped/autogen/models/strategy.py +134 -0
  242. shaped/autogen/models/table.py +102 -0
  243. shaped/autogen/models/table_deployment_type.py +38 -0
  244. shaped/autogen/models/table_insert_arguments.py +87 -0
  245. shaped/autogen/models/table_insert_response.py +87 -0
  246. shaped/autogen/models/text_encoding.py +148 -0
  247. shaped/autogen/models/text_search_retrieve_step.py +121 -0
  248. shaped/autogen/models/time_frequency.py +136 -0
  249. shaped/autogen/models/time_window.py +136 -0
  250. shaped/autogen/models/time_window_in_days.py +154 -0
  251. shaped/autogen/models/tokenizer.py +149 -0
  252. shaped/autogen/models/trained_model_encoder.py +99 -0
  253. shaped/autogen/models/training_compute_config.py +99 -0
  254. shaped/autogen/models/training_config.py +121 -0
  255. shaped/autogen/models/training_config_models_inner.py +540 -0
  256. shaped/autogen/models/training_strategy.py +37 -0
  257. shaped/autogen/models/transform_status.py +41 -0
  258. shaped/autogen/models/trending_mode.py +37 -0
  259. shaped/autogen/models/truncate_filter_step.py +106 -0
  260. shaped/autogen/models/tunable_bool.py +97 -0
  261. shaped/autogen/models/tunable_float.py +118 -0
  262. shaped/autogen/models/tunable_int.py +118 -0
  263. shaped/autogen/models/tunable_int_categorical.py +99 -0
  264. shaped/autogen/models/tunable_string.py +99 -0
  265. shaped/autogen/models/tuning_config.py +89 -0
  266. shaped/autogen/models/type.py +150 -0
  267. shaped/autogen/models/update_table_response.py +87 -0
  268. shaped/autogen/models/update_view_response.py +87 -0
  269. shaped/autogen/models/user_attribute_pooling_encoder.py +124 -0
  270. shaped/autogen/models/user_inner.py +134 -0
  271. shaped/autogen/models/val_split.py +136 -0
  272. shaped/autogen/models/validation_error.py +13 -3
  273. shaped/autogen/models/validation_error_loc_inner.py +150 -0
  274. shaped/autogen/models/value_type.py +7 -5
  275. shaped/autogen/models/vector_search_mode.py +101 -0
  276. shaped/autogen/models/view.py +104 -0
  277. shaped/autogen/models/view_details_ai.py +140 -0
  278. shaped/autogen/models/view_details_ai_schema_value.py +153 -0
  279. shaped/autogen/models/view_details_sql.py +140 -0
  280. shaped/autogen/models/view_status.py +41 -0
  281. shaped/autogen/models/weight_decay.py +136 -0
  282. shaped/autogen/models/whitespace_tokenizer.py +97 -0
  283. shaped/autogen/models/window_size.py +136 -0
  284. shaped/autogen/rest.py +10 -4
  285. shaped/cli/shaped_cli.py +163 -35
  286. shaped/client.py +591 -171
  287. shaped/config_builders.py +705 -0
  288. shaped/query_builder.py +781 -0
  289. {shaped-2.0.1.dist-info → shaped-2.0.4.dist-info}/METADATA +141 -6
  290. shaped-2.0.4.dist-info/RECORD +296 -0
  291. {shaped-2.0.1.dist-info → shaped-2.0.4.dist-info}/WHEEL +1 -1
  292. shaped-2.0.4.dist-info/entry_points.txt +2 -0
  293. shaped/autogen/api/model_inference_api.py +0 -2825
  294. shaped/autogen/models/amplitude_dataset_config.py +0 -96
  295. shaped/autogen/models/aws_pinpoint_dataset_config.py +0 -96
  296. shaped/autogen/models/big_query_dataset_config.py +0 -114
  297. shaped/autogen/models/complement_items_request.py +0 -99
  298. shaped/autogen/models/complement_items_response.py +0 -89
  299. shaped/autogen/models/connectors_inner.py +0 -134
  300. shaped/autogen/models/create_dataset_arguments.py +0 -263
  301. shaped/autogen/models/create_embedding_response.py +0 -87
  302. shaped/autogen/models/create_item_embedding_request.py +0 -89
  303. shaped/autogen/models/create_model_arguments.py +0 -107
  304. shaped/autogen/models/create_model_response.py +0 -87
  305. shaped/autogen/models/create_user_embedding_request.py +0 -89
  306. shaped/autogen/models/custom_dataset_config.py +0 -115
  307. shaped/autogen/models/dataset_config.py +0 -101
  308. shaped/autogen/models/dataset_schema_type.py +0 -47
  309. shaped/autogen/models/datasets_inner.py +0 -91
  310. shaped/autogen/models/delete_model_response.py +0 -87
  311. shaped/autogen/models/fetch_config.py +0 -95
  312. shaped/autogen/models/file_config.py +0 -105
  313. shaped/autogen/models/file_source_config.py +0 -89
  314. shaped/autogen/models/inference_config.py +0 -101
  315. shaped/autogen/models/insert_model_response.py +0 -87
  316. shaped/autogen/models/interaction.py +0 -87
  317. shaped/autogen/models/list_datasets_response.py +0 -95
  318. shaped/autogen/models/list_models_response.py +0 -95
  319. shaped/autogen/models/model_config.py +0 -99
  320. shaped/autogen/models/model_response.py +0 -95
  321. shaped/autogen/models/mongo_db_dataset_config.py +0 -119
  322. shaped/autogen/models/post_rank_request.py +0 -117
  323. shaped/autogen/models/rank_attribute_response.py +0 -89
  324. shaped/autogen/models/rank_grid_attribute_request.py +0 -91
  325. shaped/autogen/models/rank_grid_attribute_request1.py +0 -93
  326. shaped/autogen/models/rank_grid_attribute_response.py +0 -91
  327. shaped/autogen/models/rank_response.py +0 -91
  328. shaped/autogen/models/retrieve_request.py +0 -101
  329. shaped/autogen/models/retrieve_response.py +0 -91
  330. shaped/autogen/models/retriever_top_k_override.py +0 -97
  331. shaped/autogen/models/rudder_stack_dataset_config.py +0 -96
  332. shaped/autogen/models/segment_dataset_config.py +0 -96
  333. shaped/autogen/models/similar_item_request.py +0 -101
  334. shaped/autogen/models/similar_response.py +0 -89
  335. shaped/autogen/models/similar_users_request.py +0 -99
  336. shaped/autogen/models/successful_response.py +0 -87
  337. shaped/autogen/models/view_model_response.py +0 -99
  338. shaped-2.0.1.dist-info/RECORD +0 -73
  339. shaped-2.0.1.dist-info/entry_points.txt +0 -2
  340. {shaped-2.0.1.dist-info → shaped-2.0.4.dist-info}/top_level.txt +0 -0
  341. {shaped-2.0.1.dist-info → shaped-2.0.4.dist-info}/zip-safe +0 -0
@@ -5,9 +5,9 @@
5
5
  """
6
6
  Shaped API
7
7
 
8
- 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.
8
+ 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`
9
9
 
10
- The version of the OpenAPI document: 1.0.1
10
+ The version of the OpenAPI document: 2.0.4
11
11
  Generated by OpenAPI Generator (https://openapi-generator.tech)
12
12
 
13
13
  Do not edit the class manually.
@@ -16,72 +16,533 @@
16
16
 
17
17
  __version__ = "1.0.0"
18
18
 
19
+ # Define package exports
20
+ __all__ = [
21
+ "EngineApi",
22
+ "QueryApi",
23
+ "TableApi",
24
+ "ViewApi",
25
+ "ApiResponse",
26
+ "ApiClient",
27
+ "Configuration",
28
+ "OpenApiException",
29
+ "ApiTypeError",
30
+ "ApiValueError",
31
+ "ApiKeyError",
32
+ "ApiAttributeError",
33
+ "ApiException",
34
+ "AIEnrichmentViewConfig",
35
+ "AWSPinpointTableConfig",
36
+ "Algorithm",
37
+ "AmplitudeTableConfig",
38
+ "AttnDropoutProb",
39
+ "AttributeJourney",
40
+ "AttributeValue",
41
+ "AutoscalingConfig",
42
+ "BatchSize",
43
+ "BatchSize1",
44
+ "BatchSize2",
45
+ "BigQueryTableConfig",
46
+ "Bm25",
47
+ "BoostedReorderStep",
48
+ "CanaryRollout",
49
+ "CandidateAttributesRetrieveStep",
50
+ "CandidateIdsRetrieveStep",
51
+ "CandidateRetrievalStrategy",
52
+ "ClickhouseTableConfig",
53
+ "ColumnOrderRetrieveStep",
54
+ "ColumnOrdering",
55
+ "CreateTableResponse",
56
+ "CreateViewResponse",
57
+ "CustomTableConfig",
58
+ "DataComputeConfig",
59
+ "DataConfig",
60
+ "DataConfigInteractionTable",
61
+ "DataSplitConfig",
62
+ "DataSplitStrategy",
63
+ "DataTier",
64
+ "Default",
65
+ "DeleteEngineResponse",
66
+ "DeleteTableResponse",
67
+ "DeleteViewResponse",
68
+ "DeploymentConfig",
69
+ "DiversityReorderStep",
70
+ "DropoutRate",
71
+ "DynamoDBTableConfig",
72
+ "DynamoDBTableConfigScanKwargsValue",
73
+ "EmbedderBatchSize",
74
+ "EmbeddingConfig",
75
+ "EmbeddingDim",
76
+ "EmbeddingDims",
77
+ "EmbeddingSize",
78
+ "Encoder",
79
+ "Engine",
80
+ "EngineConfigV2",
81
+ "EngineDetailsResponse",
82
+ "EngineSchema",
83
+ "EntityConfig",
84
+ "EntityJourney",
85
+ "EntityType",
86
+ "EvaluationConfig",
87
+ "ExplorationReorderStep",
88
+ "ExpressionFilterStep",
89
+ "Factors",
90
+ "Factors1",
91
+ "Feature",
92
+ "FeatureType",
93
+ "FileTableConfig",
94
+ "FilterConfig",
95
+ "FilterIndexType",
96
+ "FilterRetrieveStep",
97
+ "FilterStepExplanation",
98
+ "FilterTable",
99
+ "FilterType",
100
+ "GlobalFilter",
101
+ "HTTPValidationError",
102
+ "HiddenDropoutProb",
103
+ "HiddenSize",
104
+ "HiddenSize1",
105
+ "HttpProblemResponse",
106
+ "HuggingFaceEncoder",
107
+ "IcebergTableConfig",
108
+ "IndexConfig",
109
+ "InnerEntityId",
110
+ "InnerSize",
111
+ "InnerSize1",
112
+ "InnerUid",
113
+ "InteractionConfig",
114
+ "InteractionPoolingEncoder",
115
+ "InteractionRoundRobinEncoder",
116
+ "ItemAttributePoolingEncoder",
117
+ "Journey",
118
+ "KafkaTableConfig",
119
+ "KinesisTableConfig",
120
+ "KinesisTableConfigColumnSchemaValue",
121
+ "Label",
122
+ "LabelType",
123
+ "LaplaceSmoothing",
124
+ "LatencyScalingPolicy",
125
+ "LearningRate",
126
+ "LearningRate1",
127
+ "LearningRate2",
128
+ "LearningRate3",
129
+ "LexicalSearchMode",
130
+ "ListEnginesResponse",
131
+ "ListTablesResponse",
132
+ "ListViewsResponse",
133
+ "LocationInner",
134
+ "LossTypes",
135
+ "Lr",
136
+ "Lr1",
137
+ "Lr2",
138
+ "MSSQLTableConfig",
139
+ "MaxDepth",
140
+ "MaxLeaves",
141
+ "MaxSeqLength",
142
+ "MaxSeqLength1",
143
+ "MaxSeqLength2",
144
+ "Mode",
145
+ "Mode1",
146
+ "ModelsInner",
147
+ "MongoDBTableConfig",
148
+ "MySQLTableConfig",
149
+ "NEpochs",
150
+ "NEpochs1",
151
+ "NEpochs2",
152
+ "NEstimators",
153
+ "NHeads",
154
+ "NLayers",
155
+ "NegPerPositive",
156
+ "NegativeSamplesCount",
157
+ "NgramTokenizer",
158
+ "NoOpConfig",
159
+ "NumBlocks",
160
+ "NumHeads",
161
+ "NumLeaves",
162
+ "Objective",
163
+ "Objective1",
164
+ "OnlineStoreConfig",
165
+ "PaginationConfig",
166
+ "ParameterDefinition",
167
+ "ParametersValue",
168
+ "PassthroughScore",
169
+ "PersonalFilter",
170
+ "PipelineStageExplanation",
171
+ "Policy",
172
+ "PoolingFunction",
173
+ "PostgresTableConfig",
174
+ "PosthogTableConfig",
175
+ "PrebuiltFilterStep",
176
+ "PrecomputedItemEmbedding",
177
+ "PrecomputedUserEmbedding",
178
+ "Query",
179
+ "Query1",
180
+ "QueryConfig",
181
+ "QueryDefinition",
182
+ "QueryEncoder",
183
+ "QueryExplanation",
184
+ "QueryRequest",
185
+ "QueryResult",
186
+ "QueryTableConfig",
187
+ "RankItemAttributeValuesQueryConfig",
188
+ "RankQueryConfig",
189
+ "RankQueryConfigFilterInner",
190
+ "RecreateRollout",
191
+ "RedshiftTableConfig",
192
+ "ReferenceTableConfig",
193
+ "Regularization",
194
+ "ReorderInner",
195
+ "ReorderStepExplanation",
196
+ "Request",
197
+ "Request1",
198
+ "RequestsPerSecondScalingPolicy",
199
+ "ResourceConfig",
200
+ "ResponseGetViewDetailsViewsViewNameGet",
201
+ "Result",
202
+ "ResultEmbeddingsValue",
203
+ "RetrievalScoresValue",
204
+ "RetrieveInner",
205
+ "RetrieveStepExplanation",
206
+ "Retriever",
207
+ "Retriever1",
208
+ "RolloutConfig",
209
+ "RudderstackTableConfig",
210
+ "SQLTransformType",
211
+ "SQLViewConfig",
212
+ "SamplingStrategy",
213
+ "SavedQueryInfoResponse",
214
+ "SavedQueryListResponse",
215
+ "SavedQueryRequest",
216
+ "SchemaConfig",
217
+ "Score",
218
+ "ScoreEnsemble",
219
+ "ScoreStepExplanation",
220
+ "SearchConfig",
221
+ "SegmentTableConfig",
222
+ "SequenceLength",
223
+ "ServerConfig",
224
+ "SetupEngineResponse",
225
+ "ShapedInternalRecsysPoliciesAlsModelPolicyALSModelPolicyConfig",
226
+ "ShapedInternalRecsysPoliciesBeeformerModelPolicyBeeformerModelPolicyBeeformerModelPolicyConfig",
227
+ "ShapedInternalRecsysPoliciesBertModelPolicyBertModelPolicyBERTModelPolicyConfig",
228
+ "ShapedInternalRecsysPoliciesElsaModelPolicyElsaModelPolicyELSAModelPolicyConfig",
229
+ "ShapedInternalRecsysPoliciesGsasrecModelPolicyGsasrecModelPolicyGSASRecModelPolicyConfig",
230
+ "ShapedInternalRecsysPoliciesItem2vecModelPolicyItem2VecModelPolicyConfig",
231
+ "ShapedInternalRecsysPoliciesLightgbmModelPolicyLightgbmModelPolicyLightGBMModelPolicyConfig",
232
+ "ShapedInternalRecsysPoliciesNgramModelPolicyNgramModelPolicyConfig",
233
+ "ShapedInternalRecsysPoliciesRisingPopularPolicyRisingPopularPolicyConfig",
234
+ "ShapedInternalRecsysPoliciesSasrecModelPolicySasrecModelPolicySASRecModelPolicyConfig",
235
+ "ShapedInternalRecsysPoliciesSvdModelPolicySVDModelPolicyConfig",
236
+ "ShapedInternalRecsysPoliciesTwoTowerModelPolicyTwoTowerModelPolicyTwoTowerModelPolicyConfig",
237
+ "ShapedInternalRecsysPoliciesWidedeepModelPolicyWideDeepModelPolicyConfig",
238
+ "ShapedInternalRecsysPoliciesXgboostModelPolicyXGBoostModelPolicyConfig",
239
+ "ShopifyTableConfig",
240
+ "SimilarityRetrieveStep",
241
+ "SnowflakeTableConfig",
242
+ "StemmerTokenizer",
243
+ "StepExplanation",
244
+ "StepsInner",
245
+ "Strategy",
246
+ "Table",
247
+ "TableDeploymentType",
248
+ "TableInsertArguments",
249
+ "TableInsertResponse",
250
+ "TextSearchRetrieveStep",
251
+ "TimeFrequency",
252
+ "TimeWindow",
253
+ "Tokenizer",
254
+ "TrainedModelEncoder",
255
+ "TrainingComputeConfig",
256
+ "TrainingConfig",
257
+ "TrainingStrategy",
258
+ "TransformStatus",
259
+ "TruncateFilterStep",
260
+ "TunableBool",
261
+ "TunableFloat",
262
+ "TunableInt",
263
+ "TunableIntCategorical",
264
+ "TunableString",
265
+ "TuningConfig",
266
+ "UpdateTableResponse",
267
+ "UpdateViewResponse",
268
+ "UserAttributePoolingEncoder",
269
+ "UserInner",
270
+ "ValSplit",
271
+ "ValidationError",
272
+ "ValueType",
273
+ "VectorSearchMode",
274
+ "View",
275
+ "ViewDetailsAI",
276
+ "ViewDetailsAISchemaValue",
277
+ "ViewDetailsSQL",
278
+ "WeightDecay",
279
+ "WhitespaceTokenizer",
280
+ "WindowSize",
281
+ ]
282
+
19
283
  # import apis into sdk package
20
- from shaped.autogen.api.dataset_api import DatasetApi
21
- from shaped.autogen.api.model_inference_api import ModelInferenceApi
22
- from shaped.autogen.api.model_management_api import ModelManagementApi
284
+ from shaped.autogen.api.engine_api import EngineApi as EngineApi
285
+ from shaped.autogen.api.query_api import QueryApi as QueryApi
286
+ from shaped.autogen.api.table_api import TableApi as TableApi
287
+ from shaped.autogen.api.view_api import ViewApi as ViewApi
23
288
 
24
289
  # import ApiClient
25
- from shaped.autogen.api_response import ApiResponse
26
- from shaped.autogen.api_client import ApiClient
27
- from shaped.autogen.configuration import Configuration
28
- from shaped.autogen.exceptions import OpenApiException
29
- from shaped.autogen.exceptions import ApiTypeError
30
- from shaped.autogen.exceptions import ApiValueError
31
- from shaped.autogen.exceptions import ApiKeyError
32
- from shaped.autogen.exceptions import ApiAttributeError
33
- from shaped.autogen.exceptions import ApiException
290
+ from shaped.autogen.api_response import ApiResponse as ApiResponse
291
+ from shaped.autogen.api_client import ApiClient as ApiClient
292
+ from shaped.autogen.configuration import Configuration as Configuration
293
+ from shaped.autogen.exceptions import OpenApiException as OpenApiException
294
+ from shaped.autogen.exceptions import ApiTypeError as ApiTypeError
295
+ from shaped.autogen.exceptions import ApiValueError as ApiValueError
296
+ from shaped.autogen.exceptions import ApiKeyError as ApiKeyError
297
+ from shaped.autogen.exceptions import ApiAttributeError as ApiAttributeError
298
+ from shaped.autogen.exceptions import ApiException as ApiException
34
299
 
35
300
  # import models into sdk package
36
- from shaped.autogen.models.aws_pinpoint_dataset_config import AWSPinpointDatasetConfig
37
- from shaped.autogen.models.amplitude_dataset_config import AmplitudeDatasetConfig
38
- from shaped.autogen.models.big_query_dataset_config import BigQueryDatasetConfig
39
- from shaped.autogen.models.complement_items_request import ComplementItemsRequest
40
- from shaped.autogen.models.complement_items_response import ComplementItemsResponse
41
- from shaped.autogen.models.connectors_inner import ConnectorsInner
42
- from shaped.autogen.models.create_dataset_arguments import CreateDatasetArguments
43
- from shaped.autogen.models.create_embedding_response import CreateEmbeddingResponse
44
- from shaped.autogen.models.create_item_embedding_request import CreateItemEmbeddingRequest
45
- from shaped.autogen.models.create_model_arguments import CreateModelArguments
46
- from shaped.autogen.models.create_model_response import CreateModelResponse
47
- from shaped.autogen.models.create_user_embedding_request import CreateUserEmbeddingRequest
48
- from shaped.autogen.models.custom_dataset_config import CustomDatasetConfig
49
- from shaped.autogen.models.dataset_config import DatasetConfig
50
- from shaped.autogen.models.dataset_schema_type import DatasetSchemaType
51
- from shaped.autogen.models.datasets_inner import DatasetsInner
52
- from shaped.autogen.models.delete_model_response import DeleteModelResponse
53
- from shaped.autogen.models.fetch_config import FetchConfig
54
- from shaped.autogen.models.file_config import FileConfig
55
- from shaped.autogen.models.file_source_config import FileSourceConfig
56
- from shaped.autogen.models.http_validation_error import HTTPValidationError
57
- from shaped.autogen.models.inference_config import InferenceConfig
58
- from shaped.autogen.models.insert_model_response import InsertModelResponse
59
- from shaped.autogen.models.interaction import Interaction
60
- from shaped.autogen.models.list_datasets_response import ListDatasetsResponse
61
- from shaped.autogen.models.list_models_response import ListModelsResponse
62
- from shaped.autogen.models.model_config import ModelConfig
63
- from shaped.autogen.models.model_response import ModelResponse
64
- from shaped.autogen.models.mongo_db_dataset_config import MongoDBDatasetConfig
65
- from shaped.autogen.models.my_sql_dataset_config import MySQLDatasetConfig
66
- from shaped.autogen.models.path import Path
67
- from shaped.autogen.models.post_rank_request import PostRankRequest
68
- from shaped.autogen.models.postgres_dataset_config import PostgresDatasetConfig
69
- from shaped.autogen.models.rank_attribute_response import RankAttributeResponse
70
- from shaped.autogen.models.rank_grid_attribute_request import RankGridAttributeRequest
71
- from shaped.autogen.models.rank_grid_attribute_request1 import RankGridAttributeRequest1
72
- from shaped.autogen.models.rank_grid_attribute_response import RankGridAttributeResponse
73
- from shaped.autogen.models.rank_response import RankResponse
74
- from shaped.autogen.models.redshift_dataset_config import RedshiftDatasetConfig
75
- from shaped.autogen.models.retrieve_request import RetrieveRequest
76
- from shaped.autogen.models.retrieve_response import RetrieveResponse
77
- from shaped.autogen.models.retriever_top_k_override import RetrieverTopKOverride
78
- from shaped.autogen.models.rudder_stack_dataset_config import RudderStackDatasetConfig
79
- from shaped.autogen.models.segment_dataset_config import SegmentDatasetConfig
80
- from shaped.autogen.models.similar_item_request import SimilarItemRequest
81
- from shaped.autogen.models.similar_response import SimilarResponse
82
- from shaped.autogen.models.similar_users_request import SimilarUsersRequest
83
- from shaped.autogen.models.snowflake_dataset_config import SnowflakeDatasetConfig
84
- from shaped.autogen.models.successful_response import SuccessfulResponse
85
- from shaped.autogen.models.validation_error import ValidationError
86
- from shaped.autogen.models.value_type import ValueType
87
- from shaped.autogen.models.view_model_response import ViewModelResponse
301
+ from shaped.autogen.models.ai_enrichment_view_config import AIEnrichmentViewConfig as AIEnrichmentViewConfig
302
+ from shaped.autogen.models.aws_pinpoint_table_config import AWSPinpointTableConfig as AWSPinpointTableConfig
303
+ from shaped.autogen.models.algorithm import Algorithm as Algorithm
304
+ from shaped.autogen.models.amplitude_table_config import AmplitudeTableConfig as AmplitudeTableConfig
305
+ from shaped.autogen.models.attn_dropout_prob import AttnDropoutProb as AttnDropoutProb
306
+ from shaped.autogen.models.attribute_journey import AttributeJourney as AttributeJourney
307
+ from shaped.autogen.models.attribute_value import AttributeValue as AttributeValue
308
+ from shaped.autogen.models.autoscaling_config import AutoscalingConfig as AutoscalingConfig
309
+ from shaped.autogen.models.batch_size import BatchSize as BatchSize
310
+ from shaped.autogen.models.batch_size1 import BatchSize1 as BatchSize1
311
+ from shaped.autogen.models.batch_size2 import BatchSize2 as BatchSize2
312
+ from shaped.autogen.models.big_query_table_config import BigQueryTableConfig as BigQueryTableConfig
313
+ from shaped.autogen.models.bm25 import Bm25 as Bm25
314
+ from shaped.autogen.models.boosted_reorder_step import BoostedReorderStep as BoostedReorderStep
315
+ from shaped.autogen.models.canary_rollout import CanaryRollout as CanaryRollout
316
+ from shaped.autogen.models.candidate_attributes_retrieve_step import CandidateAttributesRetrieveStep as CandidateAttributesRetrieveStep
317
+ from shaped.autogen.models.candidate_ids_retrieve_step import CandidateIdsRetrieveStep as CandidateIdsRetrieveStep
318
+ from shaped.autogen.models.candidate_retrieval_strategy import CandidateRetrievalStrategy as CandidateRetrievalStrategy
319
+ from shaped.autogen.models.clickhouse_table_config import ClickhouseTableConfig as ClickhouseTableConfig
320
+ from shaped.autogen.models.column_order_retrieve_step import ColumnOrderRetrieveStep as ColumnOrderRetrieveStep
321
+ from shaped.autogen.models.column_ordering import ColumnOrdering as ColumnOrdering
322
+ from shaped.autogen.models.create_table_response import CreateTableResponse as CreateTableResponse
323
+ from shaped.autogen.models.create_view_response import CreateViewResponse as CreateViewResponse
324
+ from shaped.autogen.models.custom_table_config import CustomTableConfig as CustomTableConfig
325
+ from shaped.autogen.models.data_compute_config import DataComputeConfig as DataComputeConfig
326
+ from shaped.autogen.models.data_config import DataConfig as DataConfig
327
+ from shaped.autogen.models.data_config_interaction_table import DataConfigInteractionTable as DataConfigInteractionTable
328
+ from shaped.autogen.models.data_split_config import DataSplitConfig as DataSplitConfig
329
+ from shaped.autogen.models.data_split_strategy import DataSplitStrategy as DataSplitStrategy
330
+ from shaped.autogen.models.data_tier import DataTier as DataTier
331
+ from shaped.autogen.models.default import Default as Default
332
+ from shaped.autogen.models.delete_engine_response import DeleteEngineResponse as DeleteEngineResponse
333
+ from shaped.autogen.models.delete_table_response import DeleteTableResponse as DeleteTableResponse
334
+ from shaped.autogen.models.delete_view_response import DeleteViewResponse as DeleteViewResponse
335
+ from shaped.autogen.models.deployment_config import DeploymentConfig as DeploymentConfig
336
+ from shaped.autogen.models.diversity_reorder_step import DiversityReorderStep as DiversityReorderStep
337
+ from shaped.autogen.models.dropout_rate import DropoutRate as DropoutRate
338
+ from shaped.autogen.models.dynamo_db_table_config import DynamoDBTableConfig as DynamoDBTableConfig
339
+ from shaped.autogen.models.dynamo_db_table_config_scan_kwargs_value import DynamoDBTableConfigScanKwargsValue as DynamoDBTableConfigScanKwargsValue
340
+ from shaped.autogen.models.embedder_batch_size import EmbedderBatchSize as EmbedderBatchSize
341
+ from shaped.autogen.models.embedding_config import EmbeddingConfig as EmbeddingConfig
342
+ from shaped.autogen.models.embedding_dim import EmbeddingDim as EmbeddingDim
343
+ from shaped.autogen.models.embedding_dims import EmbeddingDims as EmbeddingDims
344
+ from shaped.autogen.models.embedding_size import EmbeddingSize as EmbeddingSize
345
+ from shaped.autogen.models.encoder import Encoder as Encoder
346
+ from shaped.autogen.models.engine import Engine as Engine
347
+ from shaped.autogen.models.engine_config_v2 import EngineConfigV2 as EngineConfigV2
348
+ from shaped.autogen.models.engine_details_response import EngineDetailsResponse as EngineDetailsResponse
349
+ from shaped.autogen.models.engine_schema import EngineSchema as EngineSchema
350
+ from shaped.autogen.models.entity_config import EntityConfig as EntityConfig
351
+ from shaped.autogen.models.entity_journey import EntityJourney as EntityJourney
352
+ from shaped.autogen.models.entity_type import EntityType as EntityType
353
+ from shaped.autogen.models.evaluation_config import EvaluationConfig as EvaluationConfig
354
+ from shaped.autogen.models.exploration_reorder_step import ExplorationReorderStep as ExplorationReorderStep
355
+ from shaped.autogen.models.expression_filter_step import ExpressionFilterStep as ExpressionFilterStep
356
+ from shaped.autogen.models.factors import Factors as Factors
357
+ from shaped.autogen.models.factors1 import Factors1 as Factors1
358
+ from shaped.autogen.models.feature import Feature as Feature
359
+ from shaped.autogen.models.feature_type import FeatureType as FeatureType
360
+ from shaped.autogen.models.file_table_config import FileTableConfig as FileTableConfig
361
+ from shaped.autogen.models.filter_config import FilterConfig as FilterConfig
362
+ from shaped.autogen.models.filter_index_type import FilterIndexType as FilterIndexType
363
+ from shaped.autogen.models.filter_retrieve_step import FilterRetrieveStep as FilterRetrieveStep
364
+ from shaped.autogen.models.filter_step_explanation import FilterStepExplanation as FilterStepExplanation
365
+ from shaped.autogen.models.filter_table import FilterTable as FilterTable
366
+ from shaped.autogen.models.filter_type import FilterType as FilterType
367
+ from shaped.autogen.models.global_filter import GlobalFilter as GlobalFilter
368
+ from shaped.autogen.models.http_validation_error import HTTPValidationError as HTTPValidationError
369
+ from shaped.autogen.models.hidden_dropout_prob import HiddenDropoutProb as HiddenDropoutProb
370
+ from shaped.autogen.models.hidden_size import HiddenSize as HiddenSize
371
+ from shaped.autogen.models.hidden_size1 import HiddenSize1 as HiddenSize1
372
+ from shaped.autogen.models.http_problem_response import HttpProblemResponse as HttpProblemResponse
373
+ from shaped.autogen.models.hugging_face_encoder import HuggingFaceEncoder as HuggingFaceEncoder
374
+ from shaped.autogen.models.iceberg_table_config import IcebergTableConfig as IcebergTableConfig
375
+ from shaped.autogen.models.index_config import IndexConfig as IndexConfig
376
+ from shaped.autogen.models.inner_entity_id import InnerEntityId as InnerEntityId
377
+ from shaped.autogen.models.inner_size import InnerSize as InnerSize
378
+ from shaped.autogen.models.inner_size1 import InnerSize1 as InnerSize1
379
+ from shaped.autogen.models.inner_uid import InnerUid as InnerUid
380
+ from shaped.autogen.models.interaction_config import InteractionConfig as InteractionConfig
381
+ from shaped.autogen.models.interaction_pooling_encoder import InteractionPoolingEncoder as InteractionPoolingEncoder
382
+ from shaped.autogen.models.interaction_round_robin_encoder import InteractionRoundRobinEncoder as InteractionRoundRobinEncoder
383
+ from shaped.autogen.models.item_attribute_pooling_encoder import ItemAttributePoolingEncoder as ItemAttributePoolingEncoder
384
+ from shaped.autogen.models.journey import Journey as Journey
385
+ from shaped.autogen.models.kafka_table_config import KafkaTableConfig as KafkaTableConfig
386
+ from shaped.autogen.models.kinesis_table_config import KinesisTableConfig as KinesisTableConfig
387
+ from shaped.autogen.models.kinesis_table_config_column_schema_value import KinesisTableConfigColumnSchemaValue as KinesisTableConfigColumnSchemaValue
388
+ from shaped.autogen.models.label import Label as Label
389
+ from shaped.autogen.models.label_type import LabelType as LabelType
390
+ from shaped.autogen.models.laplace_smoothing import LaplaceSmoothing as LaplaceSmoothing
391
+ from shaped.autogen.models.latency_scaling_policy import LatencyScalingPolicy as LatencyScalingPolicy
392
+ from shaped.autogen.models.learning_rate import LearningRate as LearningRate
393
+ from shaped.autogen.models.learning_rate1 import LearningRate1 as LearningRate1
394
+ from shaped.autogen.models.learning_rate2 import LearningRate2 as LearningRate2
395
+ from shaped.autogen.models.learning_rate3 import LearningRate3 as LearningRate3
396
+ from shaped.autogen.models.lexical_search_mode import LexicalSearchMode as LexicalSearchMode
397
+ from shaped.autogen.models.list_engines_response import ListEnginesResponse as ListEnginesResponse
398
+ from shaped.autogen.models.list_tables_response import ListTablesResponse as ListTablesResponse
399
+ from shaped.autogen.models.list_views_response import ListViewsResponse as ListViewsResponse
400
+ from shaped.autogen.models.location_inner import LocationInner as LocationInner
401
+ from shaped.autogen.models.loss_types import LossTypes as LossTypes
402
+ from shaped.autogen.models.lr import Lr as Lr
403
+ from shaped.autogen.models.lr1 import Lr1 as Lr1
404
+ from shaped.autogen.models.lr2 import Lr2 as Lr2
405
+ from shaped.autogen.models.mssql_table_config import MSSQLTableConfig as MSSQLTableConfig
406
+ from shaped.autogen.models.max_depth import MaxDepth as MaxDepth
407
+ from shaped.autogen.models.max_leaves import MaxLeaves as MaxLeaves
408
+ from shaped.autogen.models.max_seq_length import MaxSeqLength as MaxSeqLength
409
+ from shaped.autogen.models.max_seq_length1 import MaxSeqLength1 as MaxSeqLength1
410
+ from shaped.autogen.models.max_seq_length2 import MaxSeqLength2 as MaxSeqLength2
411
+ from shaped.autogen.models.mode import Mode as Mode
412
+ from shaped.autogen.models.mode1 import Mode1 as Mode1
413
+ from shaped.autogen.models.models_inner import ModelsInner as ModelsInner
414
+ from shaped.autogen.models.mongo_db_table_config import MongoDBTableConfig as MongoDBTableConfig
415
+ from shaped.autogen.models.my_sql_table_config import MySQLTableConfig as MySQLTableConfig
416
+ from shaped.autogen.models.n_epochs import NEpochs as NEpochs
417
+ from shaped.autogen.models.n_epochs1 import NEpochs1 as NEpochs1
418
+ from shaped.autogen.models.n_epochs2 import NEpochs2 as NEpochs2
419
+ from shaped.autogen.models.n_estimators import NEstimators as NEstimators
420
+ from shaped.autogen.models.n_heads import NHeads as NHeads
421
+ from shaped.autogen.models.n_layers import NLayers as NLayers
422
+ from shaped.autogen.models.neg_per_positive import NegPerPositive as NegPerPositive
423
+ from shaped.autogen.models.negative_samples_count import NegativeSamplesCount as NegativeSamplesCount
424
+ from shaped.autogen.models.ngram_tokenizer import NgramTokenizer as NgramTokenizer
425
+ from shaped.autogen.models.no_op_config import NoOpConfig as NoOpConfig
426
+ from shaped.autogen.models.num_blocks import NumBlocks as NumBlocks
427
+ from shaped.autogen.models.num_heads import NumHeads as NumHeads
428
+ from shaped.autogen.models.num_leaves import NumLeaves as NumLeaves
429
+ from shaped.autogen.models.objective import Objective as Objective
430
+ from shaped.autogen.models.objective1 import Objective1 as Objective1
431
+ from shaped.autogen.models.online_store_config import OnlineStoreConfig as OnlineStoreConfig
432
+ from shaped.autogen.models.pagination_config import PaginationConfig as PaginationConfig
433
+ from shaped.autogen.models.parameter_definition import ParameterDefinition as ParameterDefinition
434
+ from shaped.autogen.models.parameters_value import ParametersValue as ParametersValue
435
+ from shaped.autogen.models.passthrough_score import PassthroughScore as PassthroughScore
436
+ from shaped.autogen.models.personal_filter import PersonalFilter as PersonalFilter
437
+ from shaped.autogen.models.pipeline_stage_explanation import PipelineStageExplanation as PipelineStageExplanation
438
+ from shaped.autogen.models.policy import Policy as Policy
439
+ from shaped.autogen.models.pooling_function import PoolingFunction as PoolingFunction
440
+ from shaped.autogen.models.postgres_table_config import PostgresTableConfig as PostgresTableConfig
441
+ from shaped.autogen.models.posthog_table_config import PosthogTableConfig as PosthogTableConfig
442
+ from shaped.autogen.models.prebuilt_filter_step import PrebuiltFilterStep as PrebuiltFilterStep
443
+ from shaped.autogen.models.precomputed_item_embedding import PrecomputedItemEmbedding as PrecomputedItemEmbedding
444
+ from shaped.autogen.models.precomputed_user_embedding import PrecomputedUserEmbedding as PrecomputedUserEmbedding
445
+ from shaped.autogen.models.query import Query as Query
446
+ from shaped.autogen.models.query1 import Query1 as Query1
447
+ from shaped.autogen.models.query_config import QueryConfig as QueryConfig
448
+ from shaped.autogen.models.query_definition import QueryDefinition as QueryDefinition
449
+ from shaped.autogen.models.query_encoder import QueryEncoder as QueryEncoder
450
+ from shaped.autogen.models.query_explanation import QueryExplanation as QueryExplanation
451
+ from shaped.autogen.models.query_request import QueryRequest as QueryRequest
452
+ from shaped.autogen.models.query_result import QueryResult as QueryResult
453
+ from shaped.autogen.models.query_table_config import QueryTableConfig as QueryTableConfig
454
+ from shaped.autogen.models.rank_item_attribute_values_query_config import RankItemAttributeValuesQueryConfig as RankItemAttributeValuesQueryConfig
455
+ from shaped.autogen.models.rank_query_config import RankQueryConfig as RankQueryConfig
456
+ from shaped.autogen.models.rank_query_config_filter_inner import RankQueryConfigFilterInner as RankQueryConfigFilterInner
457
+ from shaped.autogen.models.recreate_rollout import RecreateRollout as RecreateRollout
458
+ from shaped.autogen.models.redshift_table_config import RedshiftTableConfig as RedshiftTableConfig
459
+ from shaped.autogen.models.reference_table_config import ReferenceTableConfig as ReferenceTableConfig
460
+ from shaped.autogen.models.regularization import Regularization as Regularization
461
+ from shaped.autogen.models.reorder_inner import ReorderInner as ReorderInner
462
+ from shaped.autogen.models.reorder_step_explanation import ReorderStepExplanation as ReorderStepExplanation
463
+ from shaped.autogen.models.request import Request as Request
464
+ from shaped.autogen.models.request1 import Request1 as Request1
465
+ from shaped.autogen.models.requests_per_second_scaling_policy import RequestsPerSecondScalingPolicy as RequestsPerSecondScalingPolicy
466
+ from shaped.autogen.models.resource_config import ResourceConfig as ResourceConfig
467
+ from shaped.autogen.models.response_get_view_details_views_view_name_get import ResponseGetViewDetailsViewsViewNameGet as ResponseGetViewDetailsViewsViewNameGet
468
+ from shaped.autogen.models.result import Result as Result
469
+ from shaped.autogen.models.result_embeddings_value import ResultEmbeddingsValue as ResultEmbeddingsValue
470
+ from shaped.autogen.models.retrieval_scores_value import RetrievalScoresValue as RetrievalScoresValue
471
+ from shaped.autogen.models.retrieve_inner import RetrieveInner as RetrieveInner
472
+ from shaped.autogen.models.retrieve_step_explanation import RetrieveStepExplanation as RetrieveStepExplanation
473
+ from shaped.autogen.models.retriever import Retriever as Retriever
474
+ from shaped.autogen.models.retriever1 import Retriever1 as Retriever1
475
+ from shaped.autogen.models.rollout_config import RolloutConfig as RolloutConfig
476
+ from shaped.autogen.models.rudderstack_table_config import RudderstackTableConfig as RudderstackTableConfig
477
+ from shaped.autogen.models.sql_transform_type import SQLTransformType as SQLTransformType
478
+ from shaped.autogen.models.sql_view_config import SQLViewConfig as SQLViewConfig
479
+ from shaped.autogen.models.sampling_strategy import SamplingStrategy as SamplingStrategy
480
+ from shaped.autogen.models.saved_query_info_response import SavedQueryInfoResponse as SavedQueryInfoResponse
481
+ from shaped.autogen.models.saved_query_list_response import SavedQueryListResponse as SavedQueryListResponse
482
+ from shaped.autogen.models.saved_query_request import SavedQueryRequest as SavedQueryRequest
483
+ from shaped.autogen.models.schema_config import SchemaConfig as SchemaConfig
484
+ from shaped.autogen.models.score import Score as Score
485
+ from shaped.autogen.models.score_ensemble import ScoreEnsemble as ScoreEnsemble
486
+ from shaped.autogen.models.score_step_explanation import ScoreStepExplanation as ScoreStepExplanation
487
+ from shaped.autogen.models.search_config import SearchConfig as SearchConfig
488
+ from shaped.autogen.models.segment_table_config import SegmentTableConfig as SegmentTableConfig
489
+ from shaped.autogen.models.sequence_length import SequenceLength as SequenceLength
490
+ from shaped.autogen.models.server_config import ServerConfig as ServerConfig
491
+ from shaped.autogen.models.setup_engine_response import SetupEngineResponse as SetupEngineResponse
492
+ from shaped.autogen.models.shaped_internal_recsys_policies_als_model_policy_als_model_policy_config import ShapedInternalRecsysPoliciesAlsModelPolicyALSModelPolicyConfig as ShapedInternalRecsysPoliciesAlsModelPolicyALSModelPolicyConfig
493
+ from shaped.autogen.models.shaped_internal_recsys_policies_beeformer_model_policy_beeformer_model_policy_beeformer_model_policy_config import ShapedInternalRecsysPoliciesBeeformerModelPolicyBeeformerModelPolicyBeeformerModelPolicyConfig as ShapedInternalRecsysPoliciesBeeformerModelPolicyBeeformerModelPolicyBeeformerModelPolicyConfig
494
+ from shaped.autogen.models.shaped_internal_recsys_policies_bert_model_policy_bert_model_policy_bert_model_policy_config import ShapedInternalRecsysPoliciesBertModelPolicyBertModelPolicyBERTModelPolicyConfig as ShapedInternalRecsysPoliciesBertModelPolicyBertModelPolicyBERTModelPolicyConfig
495
+ from shaped.autogen.models.shaped_internal_recsys_policies_elsa_model_policy_elsa_model_policy_elsa_model_policy_config import ShapedInternalRecsysPoliciesElsaModelPolicyElsaModelPolicyELSAModelPolicyConfig as ShapedInternalRecsysPoliciesElsaModelPolicyElsaModelPolicyELSAModelPolicyConfig
496
+ from shaped.autogen.models.shaped_internal_recsys_policies_gsasrec_model_policy_gsasrec_model_policy_gsas_rec_model_policy_config import ShapedInternalRecsysPoliciesGsasrecModelPolicyGsasrecModelPolicyGSASRecModelPolicyConfig as ShapedInternalRecsysPoliciesGsasrecModelPolicyGsasrecModelPolicyGSASRecModelPolicyConfig
497
+ from shaped.autogen.models.shaped_internal_recsys_policies_item2vec_model_policy_item2_vec_model_policy_config import ShapedInternalRecsysPoliciesItem2vecModelPolicyItem2VecModelPolicyConfig as ShapedInternalRecsysPoliciesItem2vecModelPolicyItem2VecModelPolicyConfig
498
+ from shaped.autogen.models.shaped_internal_recsys_policies_lightgbm_model_policy_lightgbm_model_policy_light_gbm_model_policy_config import ShapedInternalRecsysPoliciesLightgbmModelPolicyLightgbmModelPolicyLightGBMModelPolicyConfig as ShapedInternalRecsysPoliciesLightgbmModelPolicyLightgbmModelPolicyLightGBMModelPolicyConfig
499
+ from shaped.autogen.models.shaped_internal_recsys_policies_ngram_model_policy_ngram_model_policy_config import ShapedInternalRecsysPoliciesNgramModelPolicyNgramModelPolicyConfig as ShapedInternalRecsysPoliciesNgramModelPolicyNgramModelPolicyConfig
500
+ from shaped.autogen.models.shaped_internal_recsys_policies_rising_popular_policy_rising_popular_policy_config import ShapedInternalRecsysPoliciesRisingPopularPolicyRisingPopularPolicyConfig as ShapedInternalRecsysPoliciesRisingPopularPolicyRisingPopularPolicyConfig
501
+ from shaped.autogen.models.shaped_internal_recsys_policies_sasrec_model_policy_sasrec_model_policy_sas_rec_model_policy_config import ShapedInternalRecsysPoliciesSasrecModelPolicySasrecModelPolicySASRecModelPolicyConfig as ShapedInternalRecsysPoliciesSasrecModelPolicySasrecModelPolicySASRecModelPolicyConfig
502
+ from shaped.autogen.models.shaped_internal_recsys_policies_svd_model_policy_svd_model_policy_config import ShapedInternalRecsysPoliciesSvdModelPolicySVDModelPolicyConfig as ShapedInternalRecsysPoliciesSvdModelPolicySVDModelPolicyConfig
503
+ from shaped.autogen.models.shaped_internal_recsys_policies_two_tower_model_policy_two_tower_model_policy_two_tower_model_policy_config import ShapedInternalRecsysPoliciesTwoTowerModelPolicyTwoTowerModelPolicyTwoTowerModelPolicyConfig as ShapedInternalRecsysPoliciesTwoTowerModelPolicyTwoTowerModelPolicyTwoTowerModelPolicyConfig
504
+ from shaped.autogen.models.shaped_internal_recsys_policies_widedeep_model_policy_wide_deep_model_policy_config import ShapedInternalRecsysPoliciesWidedeepModelPolicyWideDeepModelPolicyConfig as ShapedInternalRecsysPoliciesWidedeepModelPolicyWideDeepModelPolicyConfig
505
+ from shaped.autogen.models.shaped_internal_recsys_policies_xgboost_model_policy_xg_boost_model_policy_config import ShapedInternalRecsysPoliciesXgboostModelPolicyXGBoostModelPolicyConfig as ShapedInternalRecsysPoliciesXgboostModelPolicyXGBoostModelPolicyConfig
506
+ from shaped.autogen.models.shopify_table_config import ShopifyTableConfig as ShopifyTableConfig
507
+ from shaped.autogen.models.similarity_retrieve_step import SimilarityRetrieveStep as SimilarityRetrieveStep
508
+ from shaped.autogen.models.snowflake_table_config import SnowflakeTableConfig as SnowflakeTableConfig
509
+ from shaped.autogen.models.stemmer_tokenizer import StemmerTokenizer as StemmerTokenizer
510
+ from shaped.autogen.models.step_explanation import StepExplanation as StepExplanation
511
+ from shaped.autogen.models.steps_inner import StepsInner as StepsInner
512
+ from shaped.autogen.models.strategy import Strategy as Strategy
513
+ from shaped.autogen.models.table import Table as Table
514
+ from shaped.autogen.models.table_deployment_type import TableDeploymentType as TableDeploymentType
515
+ from shaped.autogen.models.table_insert_arguments import TableInsertArguments as TableInsertArguments
516
+ from shaped.autogen.models.table_insert_response import TableInsertResponse as TableInsertResponse
517
+ from shaped.autogen.models.text_search_retrieve_step import TextSearchRetrieveStep as TextSearchRetrieveStep
518
+ from shaped.autogen.models.time_frequency import TimeFrequency as TimeFrequency
519
+ from shaped.autogen.models.time_window import TimeWindow as TimeWindow
520
+ from shaped.autogen.models.tokenizer import Tokenizer as Tokenizer
521
+ from shaped.autogen.models.trained_model_encoder import TrainedModelEncoder as TrainedModelEncoder
522
+ from shaped.autogen.models.training_compute_config import TrainingComputeConfig as TrainingComputeConfig
523
+ from shaped.autogen.models.training_config import TrainingConfig as TrainingConfig
524
+ from shaped.autogen.models.training_strategy import TrainingStrategy as TrainingStrategy
525
+ from shaped.autogen.models.transform_status import TransformStatus as TransformStatus
526
+ from shaped.autogen.models.truncate_filter_step import TruncateFilterStep as TruncateFilterStep
527
+ from shaped.autogen.models.tunable_bool import TunableBool as TunableBool
528
+ from shaped.autogen.models.tunable_float import TunableFloat as TunableFloat
529
+ from shaped.autogen.models.tunable_int import TunableInt as TunableInt
530
+ from shaped.autogen.models.tunable_int_categorical import TunableIntCategorical as TunableIntCategorical
531
+ from shaped.autogen.models.tunable_string import TunableString as TunableString
532
+ from shaped.autogen.models.tuning_config import TuningConfig as TuningConfig
533
+ from shaped.autogen.models.update_table_response import UpdateTableResponse as UpdateTableResponse
534
+ from shaped.autogen.models.update_view_response import UpdateViewResponse as UpdateViewResponse
535
+ from shaped.autogen.models.user_attribute_pooling_encoder import UserAttributePoolingEncoder as UserAttributePoolingEncoder
536
+ from shaped.autogen.models.user_inner import UserInner as UserInner
537
+ from shaped.autogen.models.val_split import ValSplit as ValSplit
538
+ from shaped.autogen.models.validation_error import ValidationError as ValidationError
539
+ from shaped.autogen.models.value_type import ValueType as ValueType
540
+ from shaped.autogen.models.vector_search_mode import VectorSearchMode as VectorSearchMode
541
+ from shaped.autogen.models.view import View as View
542
+ from shaped.autogen.models.view_details_ai import ViewDetailsAI as ViewDetailsAI
543
+ from shaped.autogen.models.view_details_ai_schema_value import ViewDetailsAISchemaValue as ViewDetailsAISchemaValue
544
+ from shaped.autogen.models.view_details_sql import ViewDetailsSQL as ViewDetailsSQL
545
+ from shaped.autogen.models.weight_decay import WeightDecay as WeightDecay
546
+ from shaped.autogen.models.whitespace_tokenizer import WhitespaceTokenizer as WhitespaceTokenizer
547
+ from shaped.autogen.models.window_size import WindowSize as WindowSize
548
+
@@ -1,7 +1,8 @@
1
1
  # flake8: noqa
2
2
 
3
3
  # import apis into api package
4
- from shaped.autogen.api.dataset_api import DatasetApi
5
- from shaped.autogen.api.model_inference_api import ModelInferenceApi
6
- from shaped.autogen.api.model_management_api import ModelManagementApi
4
+ from shaped.autogen.api.engine_api import EngineApi
5
+ from shaped.autogen.api.query_api import QueryApi
6
+ from shaped.autogen.api.table_api import TableApi
7
+ from shaped.autogen.api.view_api import ViewApi
7
8