orca-sdk 0.0.97__py3-none-any.whl → 0.0.98__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 (298) hide show
  1. orca_sdk/__init__.py +1 -0
  2. orca_sdk/_shared/__init__.py +1 -0
  3. orca_sdk/_utils/analysis_ui.py +5 -5
  4. orca_sdk/_utils/auth.py +23 -33
  5. orca_sdk/_utils/pagination.py +126 -0
  6. orca_sdk/_utils/pagination_test.py +132 -0
  7. orca_sdk/classification_model.py +188 -126
  8. orca_sdk/classification_model_test.py +57 -8
  9. orca_sdk/client.py +3515 -0
  10. orca_sdk/conftest.py +10 -0
  11. orca_sdk/credentials.py +59 -21
  12. orca_sdk/credentials_test.py +20 -0
  13. orca_sdk/datasource.py +42 -76
  14. orca_sdk/embedding_model.py +225 -71
  15. orca_sdk/embedding_model_test.py +27 -36
  16. orca_sdk/job.py +49 -45
  17. orca_sdk/job_test.py +16 -0
  18. orca_sdk/memoryset.py +340 -353
  19. orca_sdk/memoryset_test.py +7 -11
  20. orca_sdk/regression_model.py +120 -111
  21. orca_sdk/regression_model_test.py +15 -0
  22. orca_sdk/telemetry.py +162 -139
  23. {orca_sdk-0.0.97.dist-info → orca_sdk-0.0.98.dist-info}/METADATA +2 -5
  24. orca_sdk-0.0.98.dist-info/RECORD +40 -0
  25. orca_sdk/_generated_api_client/__init__.py +0 -3
  26. orca_sdk/_generated_api_client/api/__init__.py +0 -307
  27. orca_sdk/_generated_api_client/api/auth/__init__.py +0 -0
  28. orca_sdk/_generated_api_client/api/auth/check_authentication_auth_get.py +0 -128
  29. orca_sdk/_generated_api_client/api/auth/create_api_key_auth_api_key_post.py +0 -170
  30. orca_sdk/_generated_api_client/api/auth/create_org_plan_auth_org_plan_post.py +0 -168
  31. orca_sdk/_generated_api_client/api/auth/delete_api_key_auth_api_key_name_or_id_delete.py +0 -156
  32. orca_sdk/_generated_api_client/api/auth/delete_org_auth_org_delete.py +0 -130
  33. orca_sdk/_generated_api_client/api/auth/get_org_plan_auth_org_plan_get.py +0 -122
  34. orca_sdk/_generated_api_client/api/auth/list_api_keys_auth_api_key_get.py +0 -127
  35. orca_sdk/_generated_api_client/api/auth/update_org_plan_auth_org_plan_put.py +0 -168
  36. orca_sdk/_generated_api_client/api/classification_model/__init__.py +0 -0
  37. orca_sdk/_generated_api_client/api/classification_model/create_classification_model_classification_model_post.py +0 -170
  38. orca_sdk/_generated_api_client/api/classification_model/delete_classification_model_classification_model_name_or_id_delete.py +0 -156
  39. orca_sdk/_generated_api_client/api/classification_model/delete_classification_model_evaluation_classification_model_model_name_or_id_evaluation_task_id_delete.py +0 -168
  40. orca_sdk/_generated_api_client/api/classification_model/evaluate_classification_model_classification_model_model_name_or_id_evaluation_post.py +0 -183
  41. orca_sdk/_generated_api_client/api/classification_model/get_classification_model_classification_model_name_or_id_get.py +0 -156
  42. orca_sdk/_generated_api_client/api/classification_model/get_classification_model_evaluation_classification_model_model_name_or_id_evaluation_task_id_get.py +0 -170
  43. orca_sdk/_generated_api_client/api/classification_model/list_classification_model_evaluations_classification_model_model_name_or_id_evaluation_get.py +0 -161
  44. orca_sdk/_generated_api_client/api/classification_model/list_classification_models_classification_model_get.py +0 -127
  45. orca_sdk/_generated_api_client/api/classification_model/predict_label_gpu_classification_model_name_or_id_prediction_post.py +0 -190
  46. orca_sdk/_generated_api_client/api/classification_model/update_classification_model_classification_model_name_or_id_patch.py +0 -183
  47. orca_sdk/_generated_api_client/api/datasource/__init__.py +0 -0
  48. orca_sdk/_generated_api_client/api/datasource/create_datasource_from_content_datasource_post.py +0 -224
  49. orca_sdk/_generated_api_client/api/datasource/create_datasource_from_files_datasource_upload_post.py +0 -229
  50. orca_sdk/_generated_api_client/api/datasource/create_embedding_evaluation_datasource_name_or_id_embedding_evaluation_post.py +0 -183
  51. orca_sdk/_generated_api_client/api/datasource/delete_datasource_datasource_name_or_id_delete.py +0 -156
  52. orca_sdk/_generated_api_client/api/datasource/download_datasource_datasource_name_or_id_download_get.py +0 -172
  53. orca_sdk/_generated_api_client/api/datasource/get_datasource_datasource_name_or_id_get.py +0 -156
  54. orca_sdk/_generated_api_client/api/datasource/get_embedding_evaluation_datasource_name_or_id_embedding_evaluation_task_id_get.py +0 -169
  55. orca_sdk/_generated_api_client/api/datasource/list_datasources_datasource_get.py +0 -127
  56. orca_sdk/_generated_api_client/api/datasource/list_embedding_evaluations_datasource_name_or_id_embedding_evaluation_get.py +0 -235
  57. orca_sdk/_generated_api_client/api/default/__init__.py +0 -0
  58. orca_sdk/_generated_api_client/api/default/healthcheck_get.py +0 -118
  59. orca_sdk/_generated_api_client/api/default/healthcheck_gpu_get.py +0 -118
  60. orca_sdk/_generated_api_client/api/finetuned_embedding_model/__init__.py +0 -0
  61. orca_sdk/_generated_api_client/api/finetuned_embedding_model/create_finetuned_embedding_model_finetuned_embedding_model_post.py +0 -168
  62. orca_sdk/_generated_api_client/api/finetuned_embedding_model/delete_finetuned_embedding_model_finetuned_embedding_model_name_or_id_delete.py +0 -156
  63. orca_sdk/_generated_api_client/api/finetuned_embedding_model/embed_with_finetuned_model_gpu_finetuned_embedding_model_name_or_id_embedding_post.py +0 -189
  64. orca_sdk/_generated_api_client/api/finetuned_embedding_model/get_finetuned_embedding_model_finetuned_embedding_model_name_or_id_get.py +0 -156
  65. orca_sdk/_generated_api_client/api/finetuned_embedding_model/list_finetuned_embedding_models_finetuned_embedding_model_get.py +0 -127
  66. orca_sdk/_generated_api_client/api/memoryset/__init__.py +0 -0
  67. orca_sdk/_generated_api_client/api/memoryset/analyze_memoryset_memoryset_name_or_id_analysis_post.py +0 -183
  68. orca_sdk/_generated_api_client/api/memoryset/batch_delete_memoryset_batch_delete_memoryset_post.py +0 -168
  69. orca_sdk/_generated_api_client/api/memoryset/clone_memoryset_memoryset_name_or_id_clone_post.py +0 -181
  70. orca_sdk/_generated_api_client/api/memoryset/create_memoryset_memoryset_post.py +0 -168
  71. orca_sdk/_generated_api_client/api/memoryset/delete_memories_memoryset_name_or_id_memories_delete_post.py +0 -181
  72. orca_sdk/_generated_api_client/api/memoryset/delete_memory_memoryset_name_or_id_memory_memory_id_delete.py +0 -167
  73. orca_sdk/_generated_api_client/api/memoryset/delete_memoryset_memoryset_name_or_id_delete.py +0 -156
  74. orca_sdk/_generated_api_client/api/memoryset/get_analysis_memoryset_name_or_id_analysis_analysis_task_id_get.py +0 -169
  75. orca_sdk/_generated_api_client/api/memoryset/get_memories_memoryset_name_or_id_memories_get_post.py +0 -210
  76. orca_sdk/_generated_api_client/api/memoryset/get_memory_memoryset_name_or_id_memory_memory_id_get.py +0 -186
  77. orca_sdk/_generated_api_client/api/memoryset/get_memoryset_memoryset_name_or_id_get.py +0 -156
  78. orca_sdk/_generated_api_client/api/memoryset/insert_memories_gpu_memoryset_name_or_id_memory_post.py +0 -188
  79. orca_sdk/_generated_api_client/api/memoryset/list_analyses_memoryset_name_or_id_analysis_get.py +0 -235
  80. orca_sdk/_generated_api_client/api/memoryset/list_memorysets_memoryset_get.py +0 -180
  81. orca_sdk/_generated_api_client/api/memoryset/memoryset_lookup_gpu_memoryset_name_or_id_lookup_post.py +0 -212
  82. orca_sdk/_generated_api_client/api/memoryset/potential_duplicate_groups_memoryset_name_or_id_potential_duplicate_groups_get.py +0 -195
  83. orca_sdk/_generated_api_client/api/memoryset/query_memoryset_memoryset_name_or_id_memories_post.py +0 -210
  84. orca_sdk/_generated_api_client/api/memoryset/suggest_cascading_edits_memoryset_name_or_id_memory_memory_id_cascading_edits_post.py +0 -233
  85. orca_sdk/_generated_api_client/api/memoryset/update_memories_gpu_memoryset_name_or_id_memories_patch.py +0 -216
  86. orca_sdk/_generated_api_client/api/memoryset/update_memory_gpu_memoryset_name_or_id_memory_patch.py +0 -205
  87. orca_sdk/_generated_api_client/api/memoryset/update_memoryset_memoryset_name_or_id_patch.py +0 -183
  88. orca_sdk/_generated_api_client/api/predictive_model/__init__.py +0 -0
  89. orca_sdk/_generated_api_client/api/predictive_model/list_predictive_models_predictive_model_get.py +0 -150
  90. orca_sdk/_generated_api_client/api/pretrained_embedding_model/__init__.py +0 -0
  91. orca_sdk/_generated_api_client/api/pretrained_embedding_model/embed_with_pretrained_model_gpu_pretrained_embedding_model_model_name_embedding_post.py +0 -192
  92. orca_sdk/_generated_api_client/api/pretrained_embedding_model/get_pretrained_embedding_model_pretrained_embedding_model_model_name_get.py +0 -161
  93. orca_sdk/_generated_api_client/api/pretrained_embedding_model/list_pretrained_embedding_models_pretrained_embedding_model_get.py +0 -127
  94. orca_sdk/_generated_api_client/api/regression_model/__init__.py +0 -0
  95. orca_sdk/_generated_api_client/api/regression_model/create_regression_model_regression_model_post.py +0 -170
  96. orca_sdk/_generated_api_client/api/regression_model/delete_regression_model_evaluation_regression_model_model_name_or_id_evaluation_task_id_delete.py +0 -168
  97. orca_sdk/_generated_api_client/api/regression_model/delete_regression_model_regression_model_name_or_id_delete.py +0 -154
  98. orca_sdk/_generated_api_client/api/regression_model/evaluate_regression_model_regression_model_model_name_or_id_evaluation_post.py +0 -183
  99. orca_sdk/_generated_api_client/api/regression_model/get_regression_model_evaluation_regression_model_model_name_or_id_evaluation_task_id_get.py +0 -170
  100. orca_sdk/_generated_api_client/api/regression_model/get_regression_model_regression_model_name_or_id_get.py +0 -156
  101. orca_sdk/_generated_api_client/api/regression_model/list_regression_model_evaluations_regression_model_model_name_or_id_evaluation_get.py +0 -161
  102. orca_sdk/_generated_api_client/api/regression_model/list_regression_models_regression_model_get.py +0 -127
  103. orca_sdk/_generated_api_client/api/regression_model/predict_score_gpu_regression_model_name_or_id_prediction_post.py +0 -190
  104. orca_sdk/_generated_api_client/api/regression_model/update_regression_model_regression_model_name_or_id_patch.py +0 -183
  105. orca_sdk/_generated_api_client/api/task/__init__.py +0 -0
  106. orca_sdk/_generated_api_client/api/task/abort_task_task_task_id_abort_delete.py +0 -154
  107. orca_sdk/_generated_api_client/api/task/get_task_status_task_task_id_status_get.py +0 -156
  108. orca_sdk/_generated_api_client/api/task/get_task_task_task_id_get.py +0 -156
  109. orca_sdk/_generated_api_client/api/task/list_tasks_task_get.py +0 -288
  110. orca_sdk/_generated_api_client/api/telemetry/__init__.py +0 -0
  111. orca_sdk/_generated_api_client/api/telemetry/count_predictions_telemetry_prediction_count_post.py +0 -168
  112. orca_sdk/_generated_api_client/api/telemetry/drop_feedback_category_with_data_telemetry_feedback_category_name_or_id_delete.py +0 -162
  113. orca_sdk/_generated_api_client/api/telemetry/explain_prediction_telemetry_prediction_prediction_id_explanation_get.py +0 -182
  114. orca_sdk/_generated_api_client/api/telemetry/generate_memory_suggestions_telemetry_prediction_prediction_id_memory_suggestions_post.py +0 -239
  115. orca_sdk/_generated_api_client/api/telemetry/get_action_recommendation_telemetry_prediction_prediction_id_action_get.py +0 -192
  116. orca_sdk/_generated_api_client/api/telemetry/get_feedback_category_telemetry_feedback_category_name_or_id_get.py +0 -156
  117. orca_sdk/_generated_api_client/api/telemetry/get_prediction_telemetry_prediction_prediction_id_get.py +0 -180
  118. orca_sdk/_generated_api_client/api/telemetry/list_feedback_categories_telemetry_feedback_category_get.py +0 -127
  119. orca_sdk/_generated_api_client/api/telemetry/list_memories_with_feedback_telemetry_memories_post.py +0 -198
  120. orca_sdk/_generated_api_client/api/telemetry/list_predictions_telemetry_prediction_post.py +0 -198
  121. orca_sdk/_generated_api_client/api/telemetry/record_prediction_feedback_telemetry_prediction_feedback_put.py +0 -171
  122. orca_sdk/_generated_api_client/api/telemetry/update_prediction_telemetry_prediction_prediction_id_patch.py +0 -181
  123. orca_sdk/_generated_api_client/client.py +0 -216
  124. orca_sdk/_generated_api_client/errors.py +0 -38
  125. orca_sdk/_generated_api_client/models/__init__.py +0 -345
  126. orca_sdk/_generated_api_client/models/action_recommendation.py +0 -82
  127. orca_sdk/_generated_api_client/models/action_recommendation_action.py +0 -11
  128. orca_sdk/_generated_api_client/models/add_memory_recommendations.py +0 -85
  129. orca_sdk/_generated_api_client/models/add_memory_suggestion.py +0 -79
  130. orca_sdk/_generated_api_client/models/analyze_neighbor_labels_result.py +0 -116
  131. orca_sdk/_generated_api_client/models/api_key_metadata.py +0 -137
  132. orca_sdk/_generated_api_client/models/api_key_metadata_scope_item.py +0 -9
  133. orca_sdk/_generated_api_client/models/base_label_prediction_result.py +0 -130
  134. orca_sdk/_generated_api_client/models/base_model.py +0 -55
  135. orca_sdk/_generated_api_client/models/base_score_prediction_result.py +0 -108
  136. orca_sdk/_generated_api_client/models/body_create_datasource_from_files_datasource_upload_post.py +0 -145
  137. orca_sdk/_generated_api_client/models/cascade_edit_suggestions_request.py +0 -154
  138. orca_sdk/_generated_api_client/models/cascading_edit_suggestion.py +0 -92
  139. orca_sdk/_generated_api_client/models/class_representatives.py +0 -92
  140. orca_sdk/_generated_api_client/models/classification_evaluation_request.py +0 -148
  141. orca_sdk/_generated_api_client/models/classification_metrics.py +0 -259
  142. orca_sdk/_generated_api_client/models/classification_model_metadata.py +0 -227
  143. orca_sdk/_generated_api_client/models/classification_prediction_request.py +0 -220
  144. orca_sdk/_generated_api_client/models/clone_memoryset_request.py +0 -210
  145. orca_sdk/_generated_api_client/models/cluster_metrics.py +0 -78
  146. orca_sdk/_generated_api_client/models/column_info.py +0 -145
  147. orca_sdk/_generated_api_client/models/column_type.py +0 -14
  148. orca_sdk/_generated_api_client/models/constraint_violation_error_response.py +0 -81
  149. orca_sdk/_generated_api_client/models/constraint_violation_error_response_status_code.py +0 -8
  150. orca_sdk/_generated_api_client/models/count_predictions_request.py +0 -195
  151. orca_sdk/_generated_api_client/models/create_api_key_request.py +0 -120
  152. orca_sdk/_generated_api_client/models/create_api_key_request_scope_item.py +0 -9
  153. orca_sdk/_generated_api_client/models/create_api_key_response.py +0 -145
  154. orca_sdk/_generated_api_client/models/create_api_key_response_scope_item.py +0 -9
  155. orca_sdk/_generated_api_client/models/create_classification_model_request.py +0 -237
  156. orca_sdk/_generated_api_client/models/create_datasource_from_content_request.py +0 -101
  157. orca_sdk/_generated_api_client/models/create_memoryset_request.py +0 -365
  158. orca_sdk/_generated_api_client/models/create_memoryset_request_index_params.py +0 -66
  159. orca_sdk/_generated_api_client/models/create_memoryset_request_index_type.py +0 -13
  160. orca_sdk/_generated_api_client/models/create_org_plan_request.py +0 -73
  161. orca_sdk/_generated_api_client/models/create_org_plan_request_tier.py +0 -11
  162. orca_sdk/_generated_api_client/models/create_regression_model_request.py +0 -157
  163. orca_sdk/_generated_api_client/models/datasource_metadata.py +0 -156
  164. orca_sdk/_generated_api_client/models/delete_memories_request.py +0 -70
  165. orca_sdk/_generated_api_client/models/delete_memorysets_request.py +0 -70
  166. orca_sdk/_generated_api_client/models/embed_request.py +0 -155
  167. orca_sdk/_generated_api_client/models/embedding_evaluation_payload.py +0 -205
  168. orca_sdk/_generated_api_client/models/embedding_evaluation_request.py +0 -197
  169. orca_sdk/_generated_api_client/models/embedding_evaluation_response.py +0 -158
  170. orca_sdk/_generated_api_client/models/embedding_evaluation_result.py +0 -86
  171. orca_sdk/_generated_api_client/models/embedding_finetuning_method.py +0 -9
  172. orca_sdk/_generated_api_client/models/embedding_model_result.py +0 -123
  173. orca_sdk/_generated_api_client/models/evaluation_response.py +0 -153
  174. orca_sdk/_generated_api_client/models/evaluation_response_classification_metrics.py +0 -140
  175. orca_sdk/_generated_api_client/models/evaluation_response_regression_metrics.py +0 -140
  176. orca_sdk/_generated_api_client/models/feedback_metrics.py +0 -85
  177. orca_sdk/_generated_api_client/models/feedback_type.py +0 -9
  178. orca_sdk/_generated_api_client/models/filter_item.py +0 -239
  179. orca_sdk/_generated_api_client/models/filter_item_field_type_0_item.py +0 -17
  180. orca_sdk/_generated_api_client/models/filter_item_field_type_1_item_type_0.py +0 -8
  181. orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_0.py +0 -8
  182. orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_1.py +0 -22
  183. orca_sdk/_generated_api_client/models/filter_item_op.py +0 -16
  184. orca_sdk/_generated_api_client/models/finetune_embedding_model_request.py +0 -259
  185. orca_sdk/_generated_api_client/models/finetune_embedding_model_request_training_args.py +0 -66
  186. orca_sdk/_generated_api_client/models/finetuned_embedding_model_metadata.py +0 -166
  187. orca_sdk/_generated_api_client/models/get_memories_request.py +0 -70
  188. orca_sdk/_generated_api_client/models/http_validation_error.py +0 -86
  189. orca_sdk/_generated_api_client/models/internal_server_error_response.py +0 -81
  190. orca_sdk/_generated_api_client/models/internal_server_error_response_status_code.py +0 -8
  191. orca_sdk/_generated_api_client/models/label_class_metrics.py +0 -108
  192. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup.py +0 -210
  193. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup_metadata.py +0 -68
  194. orca_sdk/_generated_api_client/models/label_prediction_with_memories_and_feedback.py +0 -288
  195. orca_sdk/_generated_api_client/models/labeled_memory.py +0 -186
  196. orca_sdk/_generated_api_client/models/labeled_memory_insert.py +0 -128
  197. orca_sdk/_generated_api_client/models/labeled_memory_insert_metadata.py +0 -68
  198. orca_sdk/_generated_api_client/models/labeled_memory_lookup.py +0 -194
  199. orca_sdk/_generated_api_client/models/labeled_memory_lookup_metadata.py +0 -68
  200. orca_sdk/_generated_api_client/models/labeled_memory_metadata.py +0 -68
  201. orca_sdk/_generated_api_client/models/labeled_memory_update.py +0 -171
  202. orca_sdk/_generated_api_client/models/labeled_memory_update_metadata_type_0.py +0 -68
  203. orca_sdk/_generated_api_client/models/labeled_memory_with_feedback_metrics.py +0 -207
  204. orca_sdk/_generated_api_client/models/labeled_memory_with_feedback_metrics_feedback_metrics.py +0 -68
  205. orca_sdk/_generated_api_client/models/labeled_memory_with_feedback_metrics_metadata.py +0 -68
  206. orca_sdk/_generated_api_client/models/list_memories_request.py +0 -104
  207. orca_sdk/_generated_api_client/models/list_predictions_request.py +0 -319
  208. orca_sdk/_generated_api_client/models/lookup_request.py +0 -101
  209. orca_sdk/_generated_api_client/models/lookup_score_metrics.py +0 -94
  210. orca_sdk/_generated_api_client/models/memory_metrics.py +0 -263
  211. orca_sdk/_generated_api_client/models/memory_type.py +0 -9
  212. orca_sdk/_generated_api_client/models/memoryset_analysis_configs.py +0 -245
  213. orca_sdk/_generated_api_client/models/memoryset_analysis_request.py +0 -105
  214. orca_sdk/_generated_api_client/models/memoryset_analysis_response.py +0 -182
  215. orca_sdk/_generated_api_client/models/memoryset_class_patterns_analysis_config.py +0 -79
  216. orca_sdk/_generated_api_client/models/memoryset_class_patterns_metrics.py +0 -138
  217. orca_sdk/_generated_api_client/models/memoryset_cluster_analysis_config.py +0 -202
  218. orca_sdk/_generated_api_client/models/memoryset_cluster_analysis_config_clustering_method.py +0 -9
  219. orca_sdk/_generated_api_client/models/memoryset_cluster_analysis_config_partitioning_method.py +0 -10
  220. orca_sdk/_generated_api_client/models/memoryset_cluster_metrics.py +0 -100
  221. orca_sdk/_generated_api_client/models/memoryset_duplicate_analysis_config.py +0 -70
  222. orca_sdk/_generated_api_client/models/memoryset_duplicate_metrics.py +0 -70
  223. orca_sdk/_generated_api_client/models/memoryset_label_analysis_config.py +0 -70
  224. orca_sdk/_generated_api_client/models/memoryset_label_metrics.py +0 -116
  225. orca_sdk/_generated_api_client/models/memoryset_metadata.py +0 -333
  226. orca_sdk/_generated_api_client/models/memoryset_metadata_index_params.py +0 -55
  227. orca_sdk/_generated_api_client/models/memoryset_metadata_index_type.py +0 -13
  228. orca_sdk/_generated_api_client/models/memoryset_metrics.py +0 -265
  229. orca_sdk/_generated_api_client/models/memoryset_neighbor_analysis_config.py +0 -83
  230. orca_sdk/_generated_api_client/models/memoryset_neighbor_metrics.py +0 -76
  231. orca_sdk/_generated_api_client/models/memoryset_neighbor_metrics_lookup_score_metrics.py +0 -68
  232. orca_sdk/_generated_api_client/models/memoryset_projection_analysis_config.py +0 -79
  233. orca_sdk/_generated_api_client/models/memoryset_projection_metrics.py +0 -55
  234. orca_sdk/_generated_api_client/models/memoryset_update.py +0 -121
  235. orca_sdk/_generated_api_client/models/not_found_error_response.py +0 -99
  236. orca_sdk/_generated_api_client/models/not_found_error_response_resource_type_0.py +0 -23
  237. orca_sdk/_generated_api_client/models/not_found_error_response_status_code.py +0 -8
  238. orca_sdk/_generated_api_client/models/org_plan.py +0 -99
  239. orca_sdk/_generated_api_client/models/org_plan_tier.py +0 -11
  240. orca_sdk/_generated_api_client/models/paginated_task.py +0 -108
  241. orca_sdk/_generated_api_client/models/paginated_union_labeled_memory_with_feedback_metrics_scored_memory_with_feedback_metrics.py +0 -135
  242. orca_sdk/_generated_api_client/models/pr_curve.py +0 -86
  243. orca_sdk/_generated_api_client/models/prediction_feedback.py +0 -157
  244. orca_sdk/_generated_api_client/models/prediction_feedback_category.py +0 -115
  245. orca_sdk/_generated_api_client/models/prediction_feedback_request.py +0 -122
  246. orca_sdk/_generated_api_client/models/prediction_feedback_result.py +0 -102
  247. orca_sdk/_generated_api_client/models/prediction_sort_item_item_type_0.py +0 -10
  248. orca_sdk/_generated_api_client/models/prediction_sort_item_item_type_1.py +0 -9
  249. orca_sdk/_generated_api_client/models/predictive_model_update.py +0 -111
  250. orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py +0 -115
  251. orca_sdk/_generated_api_client/models/pretrained_embedding_model_name.py +0 -17
  252. orca_sdk/_generated_api_client/models/rac_head_type.py +0 -11
  253. orca_sdk/_generated_api_client/models/rar_head_type.py +0 -8
  254. orca_sdk/_generated_api_client/models/regression_evaluation_request.py +0 -148
  255. orca_sdk/_generated_api_client/models/regression_metrics.py +0 -172
  256. orca_sdk/_generated_api_client/models/regression_model_metadata.py +0 -191
  257. orca_sdk/_generated_api_client/models/regression_prediction_request.py +0 -195
  258. orca_sdk/_generated_api_client/models/roc_curve.py +0 -86
  259. orca_sdk/_generated_api_client/models/score_prediction_memory_lookup.py +0 -196
  260. orca_sdk/_generated_api_client/models/score_prediction_memory_lookup_metadata.py +0 -68
  261. orca_sdk/_generated_api_client/models/score_prediction_with_memories_and_feedback.py +0 -252
  262. orca_sdk/_generated_api_client/models/scored_memory.py +0 -172
  263. orca_sdk/_generated_api_client/models/scored_memory_insert.py +0 -128
  264. orca_sdk/_generated_api_client/models/scored_memory_insert_metadata.py +0 -68
  265. orca_sdk/_generated_api_client/models/scored_memory_lookup.py +0 -180
  266. orca_sdk/_generated_api_client/models/scored_memory_lookup_metadata.py +0 -68
  267. orca_sdk/_generated_api_client/models/scored_memory_metadata.py +0 -68
  268. orca_sdk/_generated_api_client/models/scored_memory_update.py +0 -171
  269. orca_sdk/_generated_api_client/models/scored_memory_update_metadata_type_0.py +0 -68
  270. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics.py +0 -193
  271. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics_feedback_metrics.py +0 -68
  272. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics_metadata.py +0 -68
  273. orca_sdk/_generated_api_client/models/service_unavailable_error_response.py +0 -81
  274. orca_sdk/_generated_api_client/models/service_unavailable_error_response_status_code.py +0 -8
  275. orca_sdk/_generated_api_client/models/task.py +0 -198
  276. orca_sdk/_generated_api_client/models/task_status.py +0 -14
  277. orca_sdk/_generated_api_client/models/task_status_info.py +0 -133
  278. orca_sdk/_generated_api_client/models/telemetry_field_type_0_item_type_0.py +0 -8
  279. orca_sdk/_generated_api_client/models/telemetry_field_type_0_item_type_2.py +0 -9
  280. orca_sdk/_generated_api_client/models/telemetry_field_type_1_item_type_0.py +0 -8
  281. orca_sdk/_generated_api_client/models/telemetry_field_type_1_item_type_1.py +0 -8
  282. orca_sdk/_generated_api_client/models/telemetry_filter_item.py +0 -217
  283. orca_sdk/_generated_api_client/models/telemetry_filter_item_op.py +0 -15
  284. orca_sdk/_generated_api_client/models/telemetry_memories_request.py +0 -181
  285. orca_sdk/_generated_api_client/models/telemetry_sort_options.py +0 -185
  286. orca_sdk/_generated_api_client/models/telemetry_sort_options_direction.py +0 -9
  287. orca_sdk/_generated_api_client/models/unauthenticated_error_response.py +0 -73
  288. orca_sdk/_generated_api_client/models/unauthenticated_error_response_status_code.py +0 -8
  289. orca_sdk/_generated_api_client/models/unauthorized_error_response.py +0 -81
  290. orca_sdk/_generated_api_client/models/unauthorized_error_response_status_code.py +0 -8
  291. orca_sdk/_generated_api_client/models/update_org_plan_request.py +0 -73
  292. orca_sdk/_generated_api_client/models/update_org_plan_request_tier.py +0 -11
  293. orca_sdk/_generated_api_client/models/update_prediction_request.py +0 -133
  294. orca_sdk/_generated_api_client/models/validation_error.py +0 -99
  295. orca_sdk/_generated_api_client/py.typed +0 -1
  296. orca_sdk/_generated_api_client/types.py +0 -56
  297. orca_sdk-0.0.97.dist-info/RECORD +0 -309
  298. {orca_sdk-0.0.97.dist-info → orca_sdk-0.0.98.dist-info}/WHEEL +0 -0
@@ -1,148 +0,0 @@
1
- """
2
- This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
-
4
- It is a customized template from the openapi-python-client tool's default template:
5
- https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
6
-
7
- The main change is:
8
- - Fix typing issues
9
- """
10
-
11
- # flake8: noqa: C901
12
-
13
- from typing import Any, List, Type, TypeVar, Union, cast
14
-
15
- from attrs import define as _attrs_define
16
- from attrs import field as _attrs_field
17
-
18
- from ..types import UNSET, Unset
19
-
20
- T = TypeVar("T", bound="RegressionEvaluationRequest")
21
-
22
-
23
- @_attrs_define
24
- class RegressionEvaluationRequest:
25
- """
26
- Attributes:
27
- datasource_id (str):
28
- datasource_score_column (str):
29
- datasource_value_column (str):
30
- memoryset_override_id (Union[None, Unset, str]):
31
- record_telemetry (Union[Unset, bool]): Default: False.
32
- telemetry_tags (Union[List[str], None, Unset]):
33
- """
34
-
35
- datasource_id: str
36
- datasource_score_column: str
37
- datasource_value_column: str
38
- memoryset_override_id: Union[None, Unset, str] = UNSET
39
- record_telemetry: Union[Unset, bool] = False
40
- telemetry_tags: Union[List[str], None, Unset] = UNSET
41
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
42
-
43
- def to_dict(self) -> dict[str, Any]:
44
- datasource_id = self.datasource_id
45
-
46
- datasource_score_column = self.datasource_score_column
47
-
48
- datasource_value_column = self.datasource_value_column
49
-
50
- memoryset_override_id: Union[None, Unset, str]
51
- if isinstance(self.memoryset_override_id, Unset):
52
- memoryset_override_id = UNSET
53
- else:
54
- memoryset_override_id = self.memoryset_override_id
55
-
56
- record_telemetry = self.record_telemetry
57
-
58
- telemetry_tags: Union[List[str], None, Unset]
59
- if isinstance(self.telemetry_tags, Unset):
60
- telemetry_tags = UNSET
61
- elif isinstance(self.telemetry_tags, list):
62
- telemetry_tags = self.telemetry_tags
63
-
64
- else:
65
- telemetry_tags = self.telemetry_tags
66
-
67
- field_dict: dict[str, Any] = {}
68
- field_dict.update(self.additional_properties)
69
- field_dict.update(
70
- {
71
- "datasource_id": datasource_id,
72
- "datasource_score_column": datasource_score_column,
73
- "datasource_value_column": datasource_value_column,
74
- }
75
- )
76
- if memoryset_override_id is not UNSET:
77
- field_dict["memoryset_override_id"] = memoryset_override_id
78
- if record_telemetry is not UNSET:
79
- field_dict["record_telemetry"] = record_telemetry
80
- if telemetry_tags is not UNSET:
81
- field_dict["telemetry_tags"] = telemetry_tags
82
-
83
- return field_dict
84
-
85
- @classmethod
86
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
87
- d = src_dict.copy()
88
- datasource_id = d.pop("datasource_id")
89
-
90
- datasource_score_column = d.pop("datasource_score_column")
91
-
92
- datasource_value_column = d.pop("datasource_value_column")
93
-
94
- def _parse_memoryset_override_id(data: object) -> Union[None, Unset, str]:
95
- if data is None:
96
- return data
97
- if isinstance(data, Unset):
98
- return data
99
- return cast(Union[None, Unset, str], data)
100
-
101
- memoryset_override_id = _parse_memoryset_override_id(d.pop("memoryset_override_id", UNSET))
102
-
103
- record_telemetry = d.pop("record_telemetry", UNSET)
104
-
105
- def _parse_telemetry_tags(data: object) -> Union[List[str], None, Unset]:
106
- if data is None:
107
- return data
108
- if isinstance(data, Unset):
109
- return data
110
- try:
111
- if not isinstance(data, list):
112
- raise TypeError()
113
- telemetry_tags_type_0 = cast(List[str], data)
114
-
115
- return telemetry_tags_type_0
116
- except: # noqa: E722
117
- pass
118
- return cast(Union[List[str], None, Unset], data)
119
-
120
- telemetry_tags = _parse_telemetry_tags(d.pop("telemetry_tags", UNSET))
121
-
122
- regression_evaluation_request = cls(
123
- datasource_id=datasource_id,
124
- datasource_score_column=datasource_score_column,
125
- datasource_value_column=datasource_value_column,
126
- memoryset_override_id=memoryset_override_id,
127
- record_telemetry=record_telemetry,
128
- telemetry_tags=telemetry_tags,
129
- )
130
-
131
- regression_evaluation_request.additional_properties = d
132
- return regression_evaluation_request
133
-
134
- @property
135
- def additional_keys(self) -> list[str]:
136
- return list(self.additional_properties.keys())
137
-
138
- def __getitem__(self, key: str) -> Any:
139
- return self.additional_properties[key]
140
-
141
- def __setitem__(self, key: str, value: Any) -> None:
142
- self.additional_properties[key] = value
143
-
144
- def __delitem__(self, key: str) -> None:
145
- del self.additional_properties[key]
146
-
147
- def __contains__(self, key: str) -> bool:
148
- return key in self.additional_properties
@@ -1,172 +0,0 @@
1
- """
2
- This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
-
4
- It is a customized template from the openapi-python-client tool's default template:
5
- https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
6
-
7
- The main change is:
8
- - Fix typing issues
9
- """
10
-
11
- # flake8: noqa: C901
12
-
13
- from typing import Any, Type, TypeVar, Union, cast
14
-
15
- from attrs import define as _attrs_define
16
- from attrs import field as _attrs_field
17
-
18
- from ..types import UNSET, Unset
19
-
20
- T = TypeVar("T", bound="RegressionMetrics")
21
-
22
-
23
- @_attrs_define
24
- class RegressionMetrics:
25
- """
26
- Attributes:
27
- mse (float):
28
- rmse (float):
29
- mae (float):
30
- r2 (float):
31
- explained_variance (float):
32
- loss (float):
33
- anomaly_score_mean (Union[None, Unset, float]):
34
- anomaly_score_median (Union[None, Unset, float]):
35
- anomaly_score_variance (Union[None, Unset, float]):
36
- """
37
-
38
- mse: float
39
- rmse: float
40
- mae: float
41
- r2: float
42
- explained_variance: float
43
- loss: float
44
- anomaly_score_mean: Union[None, Unset, float] = UNSET
45
- anomaly_score_median: Union[None, Unset, float] = UNSET
46
- anomaly_score_variance: Union[None, Unset, float] = UNSET
47
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
48
-
49
- def to_dict(self) -> dict[str, Any]:
50
- mse = self.mse
51
-
52
- rmse = self.rmse
53
-
54
- mae = self.mae
55
-
56
- r2 = self.r2
57
-
58
- explained_variance = self.explained_variance
59
-
60
- loss = self.loss
61
-
62
- anomaly_score_mean: Union[None, Unset, float]
63
- if isinstance(self.anomaly_score_mean, Unset):
64
- anomaly_score_mean = UNSET
65
- else:
66
- anomaly_score_mean = self.anomaly_score_mean
67
-
68
- anomaly_score_median: Union[None, Unset, float]
69
- if isinstance(self.anomaly_score_median, Unset):
70
- anomaly_score_median = UNSET
71
- else:
72
- anomaly_score_median = self.anomaly_score_median
73
-
74
- anomaly_score_variance: Union[None, Unset, float]
75
- if isinstance(self.anomaly_score_variance, Unset):
76
- anomaly_score_variance = UNSET
77
- else:
78
- anomaly_score_variance = self.anomaly_score_variance
79
-
80
- field_dict: dict[str, Any] = {}
81
- field_dict.update(self.additional_properties)
82
- field_dict.update(
83
- {
84
- "mse": mse,
85
- "rmse": rmse,
86
- "mae": mae,
87
- "r2": r2,
88
- "explained_variance": explained_variance,
89
- "loss": loss,
90
- }
91
- )
92
- if anomaly_score_mean is not UNSET:
93
- field_dict["anomaly_score_mean"] = anomaly_score_mean
94
- if anomaly_score_median is not UNSET:
95
- field_dict["anomaly_score_median"] = anomaly_score_median
96
- if anomaly_score_variance is not UNSET:
97
- field_dict["anomaly_score_variance"] = anomaly_score_variance
98
-
99
- return field_dict
100
-
101
- @classmethod
102
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
103
- d = src_dict.copy()
104
- mse = d.pop("mse")
105
-
106
- rmse = d.pop("rmse")
107
-
108
- mae = d.pop("mae")
109
-
110
- r2 = d.pop("r2")
111
-
112
- explained_variance = d.pop("explained_variance")
113
-
114
- loss = d.pop("loss")
115
-
116
- def _parse_anomaly_score_mean(data: object) -> Union[None, Unset, float]:
117
- if data is None:
118
- return data
119
- if isinstance(data, Unset):
120
- return data
121
- return cast(Union[None, Unset, float], data)
122
-
123
- anomaly_score_mean = _parse_anomaly_score_mean(d.pop("anomaly_score_mean", UNSET))
124
-
125
- def _parse_anomaly_score_median(data: object) -> Union[None, Unset, float]:
126
- if data is None:
127
- return data
128
- if isinstance(data, Unset):
129
- return data
130
- return cast(Union[None, Unset, float], data)
131
-
132
- anomaly_score_median = _parse_anomaly_score_median(d.pop("anomaly_score_median", UNSET))
133
-
134
- def _parse_anomaly_score_variance(data: object) -> Union[None, Unset, float]:
135
- if data is None:
136
- return data
137
- if isinstance(data, Unset):
138
- return data
139
- return cast(Union[None, Unset, float], data)
140
-
141
- anomaly_score_variance = _parse_anomaly_score_variance(d.pop("anomaly_score_variance", UNSET))
142
-
143
- regression_metrics = cls(
144
- mse=mse,
145
- rmse=rmse,
146
- mae=mae,
147
- r2=r2,
148
- explained_variance=explained_variance,
149
- loss=loss,
150
- anomaly_score_mean=anomaly_score_mean,
151
- anomaly_score_median=anomaly_score_median,
152
- anomaly_score_variance=anomaly_score_variance,
153
- )
154
-
155
- regression_metrics.additional_properties = d
156
- return regression_metrics
157
-
158
- @property
159
- def additional_keys(self) -> list[str]:
160
- return list(self.additional_properties.keys())
161
-
162
- def __getitem__(self, key: str) -> Any:
163
- return self.additional_properties[key]
164
-
165
- def __setitem__(self, key: str, value: Any) -> None:
166
- self.additional_properties[key] = value
167
-
168
- def __delitem__(self, key: str) -> None:
169
- del self.additional_properties[key]
170
-
171
- def __contains__(self, key: str) -> bool:
172
- return key in self.additional_properties
@@ -1,191 +0,0 @@
1
- """
2
- This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
-
4
- It is a customized template from the openapi-python-client tool's default template:
5
- https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
6
-
7
- The main change is:
8
- - Fix typing issues
9
- """
10
-
11
- # flake8: noqa: C901
12
-
13
- import datetime
14
- from enum import Enum
15
- from typing import Any, Type, TypeVar, Union, cast
16
-
17
- from attrs import define as _attrs_define
18
- from attrs import field as _attrs_field
19
- from dateutil.parser import isoparse
20
-
21
- from ..models.rar_head_type import RARHeadType
22
-
23
- T = TypeVar("T", bound="RegressionModelMetadata")
24
-
25
-
26
- @_attrs_define
27
- class RegressionModelMetadata:
28
- """
29
- Attributes:
30
- id (str):
31
- org_id (str):
32
- name (str):
33
- description (Union[None, str]):
34
- notes (Union[None, str]):
35
- version (int):
36
- memoryset_id (str):
37
- memory_lookup_count (int):
38
- storage_path (str):
39
- memoryset_collection_name (str):
40
- created_at (datetime.datetime):
41
- updated_at (datetime.datetime):
42
- locked (bool):
43
- head_type (RARHeadType):
44
- """
45
-
46
- id: str
47
- org_id: str
48
- name: str
49
- description: Union[None, str]
50
- notes: Union[None, str]
51
- version: int
52
- memoryset_id: str
53
- memory_lookup_count: int
54
- storage_path: str
55
- memoryset_collection_name: str
56
- created_at: datetime.datetime
57
- updated_at: datetime.datetime
58
- locked: bool
59
- head_type: RARHeadType
60
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
61
-
62
- def to_dict(self) -> dict[str, Any]:
63
- id = self.id
64
-
65
- org_id = self.org_id
66
-
67
- name = self.name
68
-
69
- description: Union[None, str]
70
- description = self.description
71
-
72
- notes: Union[None, str]
73
- notes = self.notes
74
-
75
- version = self.version
76
-
77
- memoryset_id = self.memoryset_id
78
-
79
- memory_lookup_count = self.memory_lookup_count
80
-
81
- storage_path = self.storage_path
82
-
83
- memoryset_collection_name = self.memoryset_collection_name
84
-
85
- created_at = self.created_at.isoformat()
86
-
87
- updated_at = self.updated_at.isoformat()
88
-
89
- locked = self.locked
90
-
91
- head_type = self.head_type.value if isinstance(self.head_type, Enum) else self.head_type
92
-
93
- field_dict: dict[str, Any] = {}
94
- field_dict.update(self.additional_properties)
95
- field_dict.update(
96
- {
97
- "id": id,
98
- "org_id": org_id,
99
- "name": name,
100
- "description": description,
101
- "notes": notes,
102
- "version": version,
103
- "memoryset_id": memoryset_id,
104
- "memory_lookup_count": memory_lookup_count,
105
- "storage_path": storage_path,
106
- "memoryset_collection_name": memoryset_collection_name,
107
- "created_at": created_at,
108
- "updated_at": updated_at,
109
- "locked": locked,
110
- "head_type": head_type,
111
- }
112
- )
113
-
114
- return field_dict
115
-
116
- @classmethod
117
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
118
- d = src_dict.copy()
119
- id = d.pop("id")
120
-
121
- org_id = d.pop("org_id")
122
-
123
- name = d.pop("name")
124
-
125
- def _parse_description(data: object) -> Union[None, str]:
126
- if data is None:
127
- return data
128
- return cast(Union[None, str], data)
129
-
130
- description = _parse_description(d.pop("description"))
131
-
132
- def _parse_notes(data: object) -> Union[None, str]:
133
- if data is None:
134
- return data
135
- return cast(Union[None, str], data)
136
-
137
- notes = _parse_notes(d.pop("notes"))
138
-
139
- version = d.pop("version")
140
-
141
- memoryset_id = d.pop("memoryset_id")
142
-
143
- memory_lookup_count = d.pop("memory_lookup_count")
144
-
145
- storage_path = d.pop("storage_path")
146
-
147
- memoryset_collection_name = d.pop("memoryset_collection_name")
148
-
149
- created_at = isoparse(d.pop("created_at"))
150
-
151
- updated_at = isoparse(d.pop("updated_at"))
152
-
153
- locked = d.pop("locked")
154
-
155
- head_type = RARHeadType(d.pop("head_type"))
156
-
157
- regression_model_metadata = cls(
158
- id=id,
159
- org_id=org_id,
160
- name=name,
161
- description=description,
162
- notes=notes,
163
- version=version,
164
- memoryset_id=memoryset_id,
165
- memory_lookup_count=memory_lookup_count,
166
- storage_path=storage_path,
167
- memoryset_collection_name=memoryset_collection_name,
168
- created_at=created_at,
169
- updated_at=updated_at,
170
- locked=locked,
171
- head_type=head_type,
172
- )
173
-
174
- regression_model_metadata.additional_properties = d
175
- return regression_model_metadata
176
-
177
- @property
178
- def additional_keys(self) -> list[str]:
179
- return list(self.additional_properties.keys())
180
-
181
- def __getitem__(self, key: str) -> Any:
182
- return self.additional_properties[key]
183
-
184
- def __setitem__(self, key: str, value: Any) -> None:
185
- self.additional_properties[key] = value
186
-
187
- def __delitem__(self, key: str) -> None:
188
- del self.additional_properties[key]
189
-
190
- def __contains__(self, key: str) -> bool:
191
- return key in self.additional_properties
@@ -1,195 +0,0 @@
1
- """
2
- This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
-
4
- It is a customized template from the openapi-python-client tool's default template:
5
- https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
6
-
7
- The main change is:
8
- - Fix typing issues
9
- """
10
-
11
- # flake8: noqa: C901
12
-
13
- from typing import Any, List, Type, TypeVar, Union, cast
14
-
15
- from attrs import define as _attrs_define
16
- from attrs import field as _attrs_field
17
-
18
- from ..types import UNSET, Unset
19
-
20
- T = TypeVar("T", bound="RegressionPredictionRequest")
21
-
22
-
23
- @_attrs_define
24
- class RegressionPredictionRequest:
25
- """
26
- Attributes:
27
- input_values (List[str]):
28
- expected_scores (Union[List[float], None, Unset]):
29
- tags (Union[Unset, List[str]]):
30
- memoryset_override_id (Union[None, Unset, str]):
31
- save_telemetry (Union[Unset, bool]): Default: True.
32
- save_telemetry_synchronously (Union[Unset, bool]): Default: False.
33
- """
34
-
35
- input_values: List[str]
36
- expected_scores: Union[List[float], None, Unset] = UNSET
37
- tags: Union[Unset, List[str]] = UNSET
38
- memoryset_override_id: Union[None, Unset, str] = UNSET
39
- save_telemetry: Union[Unset, bool] = True
40
- save_telemetry_synchronously: Union[Unset, bool] = False
41
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
42
-
43
- def to_dict(self) -> dict[str, Any]:
44
- input_values: List[str]
45
- if isinstance(self.input_values, list):
46
- input_values = self.input_values
47
-
48
- expected_scores: Union[List[float], None, Unset]
49
- if isinstance(self.expected_scores, Unset):
50
- expected_scores = UNSET
51
- elif isinstance(self.expected_scores, list):
52
- expected_scores = self.expected_scores
53
-
54
- else:
55
- expected_scores = self.expected_scores
56
-
57
- tags: Union[Unset, List[str]] = UNSET
58
- if not isinstance(self.tags, Unset):
59
- tags = self.tags
60
-
61
- memoryset_override_id: Union[None, Unset, str]
62
- if isinstance(self.memoryset_override_id, Unset):
63
- memoryset_override_id = UNSET
64
- else:
65
- memoryset_override_id = self.memoryset_override_id
66
-
67
- save_telemetry = self.save_telemetry
68
-
69
- save_telemetry_synchronously = self.save_telemetry_synchronously
70
-
71
- field_dict: dict[str, Any] = {}
72
- field_dict.update(self.additional_properties)
73
- field_dict.update(
74
- {
75
- "input_values": input_values,
76
- }
77
- )
78
- if expected_scores is not UNSET:
79
- field_dict["expected_scores"] = expected_scores
80
- if tags is not UNSET:
81
- field_dict["tags"] = tags
82
- if memoryset_override_id is not UNSET:
83
- field_dict["memoryset_override_id"] = memoryset_override_id
84
- if save_telemetry is not UNSET:
85
- field_dict["save_telemetry"] = save_telemetry
86
- if save_telemetry_synchronously is not UNSET:
87
- field_dict["save_telemetry_synchronously"] = save_telemetry_synchronously
88
-
89
- return field_dict
90
-
91
- @classmethod
92
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
93
- d = src_dict.copy()
94
-
95
- def _parse_input_values(data: object) -> List[str]:
96
- try:
97
- if not isinstance(data, list):
98
- raise TypeError()
99
- input_values_type_0 = cast(List[str], data)
100
-
101
- return input_values_type_0
102
- except: # noqa: E722
103
- pass
104
- try:
105
- if not isinstance(data, list):
106
- raise TypeError()
107
- input_values_type_1 = cast(List[str], data)
108
-
109
- return input_values_type_1
110
- except: # noqa: E722
111
- pass
112
- try:
113
- if not isinstance(data, list):
114
- raise TypeError()
115
- input_values_type_2 = cast(List[str], data)
116
-
117
- return input_values_type_2
118
- except: # noqa: E722
119
- pass
120
- if not isinstance(data, list):
121
- raise TypeError()
122
- input_values_type_3 = []
123
- _input_values_type_3 = data
124
- for input_values_type_3_item_data in _input_values_type_3:
125
-
126
- def _parse_input_values_type_3_item(data: object) -> str:
127
- return cast(str, data)
128
-
129
- input_values_type_3_item = _parse_input_values_type_3_item(input_values_type_3_item_data)
130
-
131
- input_values_type_3.append(input_values_type_3_item)
132
-
133
- return input_values_type_3
134
-
135
- input_values = _parse_input_values(d.pop("input_values"))
136
-
137
- def _parse_expected_scores(data: object) -> Union[List[float], None, Unset]:
138
- if data is None:
139
- return data
140
- if isinstance(data, Unset):
141
- return data
142
- try:
143
- if not isinstance(data, list):
144
- raise TypeError()
145
- expected_scores_type_0 = cast(List[float], data)
146
-
147
- return expected_scores_type_0
148
- except: # noqa: E722
149
- pass
150
- return cast(Union[List[float], None, Unset], data)
151
-
152
- expected_scores = _parse_expected_scores(d.pop("expected_scores", UNSET))
153
-
154
- tags = cast(List[str], d.pop("tags", UNSET))
155
-
156
- def _parse_memoryset_override_id(data: object) -> Union[None, Unset, str]:
157
- if data is None:
158
- return data
159
- if isinstance(data, Unset):
160
- return data
161
- return cast(Union[None, Unset, str], data)
162
-
163
- memoryset_override_id = _parse_memoryset_override_id(d.pop("memoryset_override_id", UNSET))
164
-
165
- save_telemetry = d.pop("save_telemetry", UNSET)
166
-
167
- save_telemetry_synchronously = d.pop("save_telemetry_synchronously", UNSET)
168
-
169
- regression_prediction_request = cls(
170
- input_values=input_values,
171
- expected_scores=expected_scores,
172
- tags=tags,
173
- memoryset_override_id=memoryset_override_id,
174
- save_telemetry=save_telemetry,
175
- save_telemetry_synchronously=save_telemetry_synchronously,
176
- )
177
-
178
- regression_prediction_request.additional_properties = d
179
- return regression_prediction_request
180
-
181
- @property
182
- def additional_keys(self) -> list[str]:
183
- return list(self.additional_properties.keys())
184
-
185
- def __getitem__(self, key: str) -> Any:
186
- return self.additional_properties[key]
187
-
188
- def __setitem__(self, key: str, value: Any) -> None:
189
- self.additional_properties[key] = value
190
-
191
- def __delitem__(self, key: str) -> None:
192
- del self.additional_properties[key]
193
-
194
- def __contains__(self, key: str) -> bool:
195
- return key in self.additional_properties