orca-sdk 0.0.97__py3-none-any.whl → 0.0.100__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 (299) 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 +3563 -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 +80 -93
  14. orca_sdk/datasource_test.py +41 -7
  15. orca_sdk/embedding_model.py +225 -71
  16. orca_sdk/embedding_model_test.py +27 -36
  17. orca_sdk/job.py +49 -45
  18. orca_sdk/job_test.py +16 -0
  19. orca_sdk/memoryset.py +340 -353
  20. orca_sdk/memoryset_test.py +7 -11
  21. orca_sdk/regression_model.py +120 -111
  22. orca_sdk/regression_model_test.py +15 -0
  23. orca_sdk/telemetry.py +162 -139
  24. {orca_sdk-0.0.97.dist-info → orca_sdk-0.0.100.dist-info}/METADATA +2 -5
  25. orca_sdk-0.0.100.dist-info/RECORD +40 -0
  26. orca_sdk/_generated_api_client/__init__.py +0 -3
  27. orca_sdk/_generated_api_client/api/__init__.py +0 -307
  28. orca_sdk/_generated_api_client/api/auth/__init__.py +0 -0
  29. orca_sdk/_generated_api_client/api/auth/check_authentication_auth_get.py +0 -128
  30. orca_sdk/_generated_api_client/api/auth/create_api_key_auth_api_key_post.py +0 -170
  31. orca_sdk/_generated_api_client/api/auth/create_org_plan_auth_org_plan_post.py +0 -168
  32. orca_sdk/_generated_api_client/api/auth/delete_api_key_auth_api_key_name_or_id_delete.py +0 -156
  33. orca_sdk/_generated_api_client/api/auth/delete_org_auth_org_delete.py +0 -130
  34. orca_sdk/_generated_api_client/api/auth/get_org_plan_auth_org_plan_get.py +0 -122
  35. orca_sdk/_generated_api_client/api/auth/list_api_keys_auth_api_key_get.py +0 -127
  36. orca_sdk/_generated_api_client/api/auth/update_org_plan_auth_org_plan_put.py +0 -168
  37. orca_sdk/_generated_api_client/api/classification_model/__init__.py +0 -0
  38. orca_sdk/_generated_api_client/api/classification_model/create_classification_model_classification_model_post.py +0 -170
  39. orca_sdk/_generated_api_client/api/classification_model/delete_classification_model_classification_model_name_or_id_delete.py +0 -156
  40. 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
  41. orca_sdk/_generated_api_client/api/classification_model/evaluate_classification_model_classification_model_model_name_or_id_evaluation_post.py +0 -183
  42. orca_sdk/_generated_api_client/api/classification_model/get_classification_model_classification_model_name_or_id_get.py +0 -156
  43. 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
  44. orca_sdk/_generated_api_client/api/classification_model/list_classification_model_evaluations_classification_model_model_name_or_id_evaluation_get.py +0 -161
  45. orca_sdk/_generated_api_client/api/classification_model/list_classification_models_classification_model_get.py +0 -127
  46. orca_sdk/_generated_api_client/api/classification_model/predict_label_gpu_classification_model_name_or_id_prediction_post.py +0 -190
  47. orca_sdk/_generated_api_client/api/classification_model/update_classification_model_classification_model_name_or_id_patch.py +0 -183
  48. orca_sdk/_generated_api_client/api/datasource/__init__.py +0 -0
  49. orca_sdk/_generated_api_client/api/datasource/create_datasource_from_content_datasource_post.py +0 -224
  50. orca_sdk/_generated_api_client/api/datasource/create_datasource_from_files_datasource_upload_post.py +0 -229
  51. orca_sdk/_generated_api_client/api/datasource/create_embedding_evaluation_datasource_name_or_id_embedding_evaluation_post.py +0 -183
  52. orca_sdk/_generated_api_client/api/datasource/delete_datasource_datasource_name_or_id_delete.py +0 -156
  53. orca_sdk/_generated_api_client/api/datasource/download_datasource_datasource_name_or_id_download_get.py +0 -172
  54. orca_sdk/_generated_api_client/api/datasource/get_datasource_datasource_name_or_id_get.py +0 -156
  55. orca_sdk/_generated_api_client/api/datasource/get_embedding_evaluation_datasource_name_or_id_embedding_evaluation_task_id_get.py +0 -169
  56. orca_sdk/_generated_api_client/api/datasource/list_datasources_datasource_get.py +0 -127
  57. orca_sdk/_generated_api_client/api/datasource/list_embedding_evaluations_datasource_name_or_id_embedding_evaluation_get.py +0 -235
  58. orca_sdk/_generated_api_client/api/default/__init__.py +0 -0
  59. orca_sdk/_generated_api_client/api/default/healthcheck_get.py +0 -118
  60. orca_sdk/_generated_api_client/api/default/healthcheck_gpu_get.py +0 -118
  61. orca_sdk/_generated_api_client/api/finetuned_embedding_model/__init__.py +0 -0
  62. orca_sdk/_generated_api_client/api/finetuned_embedding_model/create_finetuned_embedding_model_finetuned_embedding_model_post.py +0 -168
  63. orca_sdk/_generated_api_client/api/finetuned_embedding_model/delete_finetuned_embedding_model_finetuned_embedding_model_name_or_id_delete.py +0 -156
  64. 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
  65. orca_sdk/_generated_api_client/api/finetuned_embedding_model/get_finetuned_embedding_model_finetuned_embedding_model_name_or_id_get.py +0 -156
  66. orca_sdk/_generated_api_client/api/finetuned_embedding_model/list_finetuned_embedding_models_finetuned_embedding_model_get.py +0 -127
  67. orca_sdk/_generated_api_client/api/memoryset/__init__.py +0 -0
  68. orca_sdk/_generated_api_client/api/memoryset/analyze_memoryset_memoryset_name_or_id_analysis_post.py +0 -183
  69. orca_sdk/_generated_api_client/api/memoryset/batch_delete_memoryset_batch_delete_memoryset_post.py +0 -168
  70. orca_sdk/_generated_api_client/api/memoryset/clone_memoryset_memoryset_name_or_id_clone_post.py +0 -181
  71. orca_sdk/_generated_api_client/api/memoryset/create_memoryset_memoryset_post.py +0 -168
  72. orca_sdk/_generated_api_client/api/memoryset/delete_memories_memoryset_name_or_id_memories_delete_post.py +0 -181
  73. orca_sdk/_generated_api_client/api/memoryset/delete_memory_memoryset_name_or_id_memory_memory_id_delete.py +0 -167
  74. orca_sdk/_generated_api_client/api/memoryset/delete_memoryset_memoryset_name_or_id_delete.py +0 -156
  75. orca_sdk/_generated_api_client/api/memoryset/get_analysis_memoryset_name_or_id_analysis_analysis_task_id_get.py +0 -169
  76. orca_sdk/_generated_api_client/api/memoryset/get_memories_memoryset_name_or_id_memories_get_post.py +0 -210
  77. orca_sdk/_generated_api_client/api/memoryset/get_memory_memoryset_name_or_id_memory_memory_id_get.py +0 -186
  78. orca_sdk/_generated_api_client/api/memoryset/get_memoryset_memoryset_name_or_id_get.py +0 -156
  79. orca_sdk/_generated_api_client/api/memoryset/insert_memories_gpu_memoryset_name_or_id_memory_post.py +0 -188
  80. orca_sdk/_generated_api_client/api/memoryset/list_analyses_memoryset_name_or_id_analysis_get.py +0 -235
  81. orca_sdk/_generated_api_client/api/memoryset/list_memorysets_memoryset_get.py +0 -180
  82. orca_sdk/_generated_api_client/api/memoryset/memoryset_lookup_gpu_memoryset_name_or_id_lookup_post.py +0 -212
  83. orca_sdk/_generated_api_client/api/memoryset/potential_duplicate_groups_memoryset_name_or_id_potential_duplicate_groups_get.py +0 -195
  84. orca_sdk/_generated_api_client/api/memoryset/query_memoryset_memoryset_name_or_id_memories_post.py +0 -210
  85. orca_sdk/_generated_api_client/api/memoryset/suggest_cascading_edits_memoryset_name_or_id_memory_memory_id_cascading_edits_post.py +0 -233
  86. orca_sdk/_generated_api_client/api/memoryset/update_memories_gpu_memoryset_name_or_id_memories_patch.py +0 -216
  87. orca_sdk/_generated_api_client/api/memoryset/update_memory_gpu_memoryset_name_or_id_memory_patch.py +0 -205
  88. orca_sdk/_generated_api_client/api/memoryset/update_memoryset_memoryset_name_or_id_patch.py +0 -183
  89. orca_sdk/_generated_api_client/api/predictive_model/__init__.py +0 -0
  90. orca_sdk/_generated_api_client/api/predictive_model/list_predictive_models_predictive_model_get.py +0 -150
  91. orca_sdk/_generated_api_client/api/pretrained_embedding_model/__init__.py +0 -0
  92. orca_sdk/_generated_api_client/api/pretrained_embedding_model/embed_with_pretrained_model_gpu_pretrained_embedding_model_model_name_embedding_post.py +0 -192
  93. orca_sdk/_generated_api_client/api/pretrained_embedding_model/get_pretrained_embedding_model_pretrained_embedding_model_model_name_get.py +0 -161
  94. orca_sdk/_generated_api_client/api/pretrained_embedding_model/list_pretrained_embedding_models_pretrained_embedding_model_get.py +0 -127
  95. orca_sdk/_generated_api_client/api/regression_model/__init__.py +0 -0
  96. orca_sdk/_generated_api_client/api/regression_model/create_regression_model_regression_model_post.py +0 -170
  97. 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
  98. orca_sdk/_generated_api_client/api/regression_model/delete_regression_model_regression_model_name_or_id_delete.py +0 -154
  99. orca_sdk/_generated_api_client/api/regression_model/evaluate_regression_model_regression_model_model_name_or_id_evaluation_post.py +0 -183
  100. 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
  101. orca_sdk/_generated_api_client/api/regression_model/get_regression_model_regression_model_name_or_id_get.py +0 -156
  102. orca_sdk/_generated_api_client/api/regression_model/list_regression_model_evaluations_regression_model_model_name_or_id_evaluation_get.py +0 -161
  103. orca_sdk/_generated_api_client/api/regression_model/list_regression_models_regression_model_get.py +0 -127
  104. orca_sdk/_generated_api_client/api/regression_model/predict_score_gpu_regression_model_name_or_id_prediction_post.py +0 -190
  105. orca_sdk/_generated_api_client/api/regression_model/update_regression_model_regression_model_name_or_id_patch.py +0 -183
  106. orca_sdk/_generated_api_client/api/task/__init__.py +0 -0
  107. orca_sdk/_generated_api_client/api/task/abort_task_task_task_id_abort_delete.py +0 -154
  108. orca_sdk/_generated_api_client/api/task/get_task_status_task_task_id_status_get.py +0 -156
  109. orca_sdk/_generated_api_client/api/task/get_task_task_task_id_get.py +0 -156
  110. orca_sdk/_generated_api_client/api/task/list_tasks_task_get.py +0 -288
  111. orca_sdk/_generated_api_client/api/telemetry/__init__.py +0 -0
  112. orca_sdk/_generated_api_client/api/telemetry/count_predictions_telemetry_prediction_count_post.py +0 -168
  113. orca_sdk/_generated_api_client/api/telemetry/drop_feedback_category_with_data_telemetry_feedback_category_name_or_id_delete.py +0 -162
  114. orca_sdk/_generated_api_client/api/telemetry/explain_prediction_telemetry_prediction_prediction_id_explanation_get.py +0 -182
  115. orca_sdk/_generated_api_client/api/telemetry/generate_memory_suggestions_telemetry_prediction_prediction_id_memory_suggestions_post.py +0 -239
  116. orca_sdk/_generated_api_client/api/telemetry/get_action_recommendation_telemetry_prediction_prediction_id_action_get.py +0 -192
  117. orca_sdk/_generated_api_client/api/telemetry/get_feedback_category_telemetry_feedback_category_name_or_id_get.py +0 -156
  118. orca_sdk/_generated_api_client/api/telemetry/get_prediction_telemetry_prediction_prediction_id_get.py +0 -180
  119. orca_sdk/_generated_api_client/api/telemetry/list_feedback_categories_telemetry_feedback_category_get.py +0 -127
  120. orca_sdk/_generated_api_client/api/telemetry/list_memories_with_feedback_telemetry_memories_post.py +0 -198
  121. orca_sdk/_generated_api_client/api/telemetry/list_predictions_telemetry_prediction_post.py +0 -198
  122. orca_sdk/_generated_api_client/api/telemetry/record_prediction_feedback_telemetry_prediction_feedback_put.py +0 -171
  123. orca_sdk/_generated_api_client/api/telemetry/update_prediction_telemetry_prediction_prediction_id_patch.py +0 -181
  124. orca_sdk/_generated_api_client/client.py +0 -216
  125. orca_sdk/_generated_api_client/errors.py +0 -38
  126. orca_sdk/_generated_api_client/models/__init__.py +0 -345
  127. orca_sdk/_generated_api_client/models/action_recommendation.py +0 -82
  128. orca_sdk/_generated_api_client/models/action_recommendation_action.py +0 -11
  129. orca_sdk/_generated_api_client/models/add_memory_recommendations.py +0 -85
  130. orca_sdk/_generated_api_client/models/add_memory_suggestion.py +0 -79
  131. orca_sdk/_generated_api_client/models/analyze_neighbor_labels_result.py +0 -116
  132. orca_sdk/_generated_api_client/models/api_key_metadata.py +0 -137
  133. orca_sdk/_generated_api_client/models/api_key_metadata_scope_item.py +0 -9
  134. orca_sdk/_generated_api_client/models/base_label_prediction_result.py +0 -130
  135. orca_sdk/_generated_api_client/models/base_model.py +0 -55
  136. orca_sdk/_generated_api_client/models/base_score_prediction_result.py +0 -108
  137. orca_sdk/_generated_api_client/models/body_create_datasource_from_files_datasource_upload_post.py +0 -145
  138. orca_sdk/_generated_api_client/models/cascade_edit_suggestions_request.py +0 -154
  139. orca_sdk/_generated_api_client/models/cascading_edit_suggestion.py +0 -92
  140. orca_sdk/_generated_api_client/models/class_representatives.py +0 -92
  141. orca_sdk/_generated_api_client/models/classification_evaluation_request.py +0 -148
  142. orca_sdk/_generated_api_client/models/classification_metrics.py +0 -259
  143. orca_sdk/_generated_api_client/models/classification_model_metadata.py +0 -227
  144. orca_sdk/_generated_api_client/models/classification_prediction_request.py +0 -220
  145. orca_sdk/_generated_api_client/models/clone_memoryset_request.py +0 -210
  146. orca_sdk/_generated_api_client/models/cluster_metrics.py +0 -78
  147. orca_sdk/_generated_api_client/models/column_info.py +0 -145
  148. orca_sdk/_generated_api_client/models/column_type.py +0 -14
  149. orca_sdk/_generated_api_client/models/constraint_violation_error_response.py +0 -81
  150. orca_sdk/_generated_api_client/models/constraint_violation_error_response_status_code.py +0 -8
  151. orca_sdk/_generated_api_client/models/count_predictions_request.py +0 -195
  152. orca_sdk/_generated_api_client/models/create_api_key_request.py +0 -120
  153. orca_sdk/_generated_api_client/models/create_api_key_request_scope_item.py +0 -9
  154. orca_sdk/_generated_api_client/models/create_api_key_response.py +0 -145
  155. orca_sdk/_generated_api_client/models/create_api_key_response_scope_item.py +0 -9
  156. orca_sdk/_generated_api_client/models/create_classification_model_request.py +0 -237
  157. orca_sdk/_generated_api_client/models/create_datasource_from_content_request.py +0 -101
  158. orca_sdk/_generated_api_client/models/create_memoryset_request.py +0 -365
  159. orca_sdk/_generated_api_client/models/create_memoryset_request_index_params.py +0 -66
  160. orca_sdk/_generated_api_client/models/create_memoryset_request_index_type.py +0 -13
  161. orca_sdk/_generated_api_client/models/create_org_plan_request.py +0 -73
  162. orca_sdk/_generated_api_client/models/create_org_plan_request_tier.py +0 -11
  163. orca_sdk/_generated_api_client/models/create_regression_model_request.py +0 -157
  164. orca_sdk/_generated_api_client/models/datasource_metadata.py +0 -156
  165. orca_sdk/_generated_api_client/models/delete_memories_request.py +0 -70
  166. orca_sdk/_generated_api_client/models/delete_memorysets_request.py +0 -70
  167. orca_sdk/_generated_api_client/models/embed_request.py +0 -155
  168. orca_sdk/_generated_api_client/models/embedding_evaluation_payload.py +0 -205
  169. orca_sdk/_generated_api_client/models/embedding_evaluation_request.py +0 -197
  170. orca_sdk/_generated_api_client/models/embedding_evaluation_response.py +0 -158
  171. orca_sdk/_generated_api_client/models/embedding_evaluation_result.py +0 -86
  172. orca_sdk/_generated_api_client/models/embedding_finetuning_method.py +0 -9
  173. orca_sdk/_generated_api_client/models/embedding_model_result.py +0 -123
  174. orca_sdk/_generated_api_client/models/evaluation_response.py +0 -153
  175. orca_sdk/_generated_api_client/models/evaluation_response_classification_metrics.py +0 -140
  176. orca_sdk/_generated_api_client/models/evaluation_response_regression_metrics.py +0 -140
  177. orca_sdk/_generated_api_client/models/feedback_metrics.py +0 -85
  178. orca_sdk/_generated_api_client/models/feedback_type.py +0 -9
  179. orca_sdk/_generated_api_client/models/filter_item.py +0 -239
  180. orca_sdk/_generated_api_client/models/filter_item_field_type_0_item.py +0 -17
  181. orca_sdk/_generated_api_client/models/filter_item_field_type_1_item_type_0.py +0 -8
  182. orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_0.py +0 -8
  183. orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_1.py +0 -22
  184. orca_sdk/_generated_api_client/models/filter_item_op.py +0 -16
  185. orca_sdk/_generated_api_client/models/finetune_embedding_model_request.py +0 -259
  186. orca_sdk/_generated_api_client/models/finetune_embedding_model_request_training_args.py +0 -66
  187. orca_sdk/_generated_api_client/models/finetuned_embedding_model_metadata.py +0 -166
  188. orca_sdk/_generated_api_client/models/get_memories_request.py +0 -70
  189. orca_sdk/_generated_api_client/models/http_validation_error.py +0 -86
  190. orca_sdk/_generated_api_client/models/internal_server_error_response.py +0 -81
  191. orca_sdk/_generated_api_client/models/internal_server_error_response_status_code.py +0 -8
  192. orca_sdk/_generated_api_client/models/label_class_metrics.py +0 -108
  193. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup.py +0 -210
  194. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup_metadata.py +0 -68
  195. orca_sdk/_generated_api_client/models/label_prediction_with_memories_and_feedback.py +0 -288
  196. orca_sdk/_generated_api_client/models/labeled_memory.py +0 -186
  197. orca_sdk/_generated_api_client/models/labeled_memory_insert.py +0 -128
  198. orca_sdk/_generated_api_client/models/labeled_memory_insert_metadata.py +0 -68
  199. orca_sdk/_generated_api_client/models/labeled_memory_lookup.py +0 -194
  200. orca_sdk/_generated_api_client/models/labeled_memory_lookup_metadata.py +0 -68
  201. orca_sdk/_generated_api_client/models/labeled_memory_metadata.py +0 -68
  202. orca_sdk/_generated_api_client/models/labeled_memory_update.py +0 -171
  203. orca_sdk/_generated_api_client/models/labeled_memory_update_metadata_type_0.py +0 -68
  204. orca_sdk/_generated_api_client/models/labeled_memory_with_feedback_metrics.py +0 -207
  205. orca_sdk/_generated_api_client/models/labeled_memory_with_feedback_metrics_feedback_metrics.py +0 -68
  206. orca_sdk/_generated_api_client/models/labeled_memory_with_feedback_metrics_metadata.py +0 -68
  207. orca_sdk/_generated_api_client/models/list_memories_request.py +0 -104
  208. orca_sdk/_generated_api_client/models/list_predictions_request.py +0 -319
  209. orca_sdk/_generated_api_client/models/lookup_request.py +0 -101
  210. orca_sdk/_generated_api_client/models/lookup_score_metrics.py +0 -94
  211. orca_sdk/_generated_api_client/models/memory_metrics.py +0 -263
  212. orca_sdk/_generated_api_client/models/memory_type.py +0 -9
  213. orca_sdk/_generated_api_client/models/memoryset_analysis_configs.py +0 -245
  214. orca_sdk/_generated_api_client/models/memoryset_analysis_request.py +0 -105
  215. orca_sdk/_generated_api_client/models/memoryset_analysis_response.py +0 -182
  216. orca_sdk/_generated_api_client/models/memoryset_class_patterns_analysis_config.py +0 -79
  217. orca_sdk/_generated_api_client/models/memoryset_class_patterns_metrics.py +0 -138
  218. orca_sdk/_generated_api_client/models/memoryset_cluster_analysis_config.py +0 -202
  219. orca_sdk/_generated_api_client/models/memoryset_cluster_analysis_config_clustering_method.py +0 -9
  220. orca_sdk/_generated_api_client/models/memoryset_cluster_analysis_config_partitioning_method.py +0 -10
  221. orca_sdk/_generated_api_client/models/memoryset_cluster_metrics.py +0 -100
  222. orca_sdk/_generated_api_client/models/memoryset_duplicate_analysis_config.py +0 -70
  223. orca_sdk/_generated_api_client/models/memoryset_duplicate_metrics.py +0 -70
  224. orca_sdk/_generated_api_client/models/memoryset_label_analysis_config.py +0 -70
  225. orca_sdk/_generated_api_client/models/memoryset_label_metrics.py +0 -116
  226. orca_sdk/_generated_api_client/models/memoryset_metadata.py +0 -333
  227. orca_sdk/_generated_api_client/models/memoryset_metadata_index_params.py +0 -55
  228. orca_sdk/_generated_api_client/models/memoryset_metadata_index_type.py +0 -13
  229. orca_sdk/_generated_api_client/models/memoryset_metrics.py +0 -265
  230. orca_sdk/_generated_api_client/models/memoryset_neighbor_analysis_config.py +0 -83
  231. orca_sdk/_generated_api_client/models/memoryset_neighbor_metrics.py +0 -76
  232. orca_sdk/_generated_api_client/models/memoryset_neighbor_metrics_lookup_score_metrics.py +0 -68
  233. orca_sdk/_generated_api_client/models/memoryset_projection_analysis_config.py +0 -79
  234. orca_sdk/_generated_api_client/models/memoryset_projection_metrics.py +0 -55
  235. orca_sdk/_generated_api_client/models/memoryset_update.py +0 -121
  236. orca_sdk/_generated_api_client/models/not_found_error_response.py +0 -99
  237. orca_sdk/_generated_api_client/models/not_found_error_response_resource_type_0.py +0 -23
  238. orca_sdk/_generated_api_client/models/not_found_error_response_status_code.py +0 -8
  239. orca_sdk/_generated_api_client/models/org_plan.py +0 -99
  240. orca_sdk/_generated_api_client/models/org_plan_tier.py +0 -11
  241. orca_sdk/_generated_api_client/models/paginated_task.py +0 -108
  242. orca_sdk/_generated_api_client/models/paginated_union_labeled_memory_with_feedback_metrics_scored_memory_with_feedback_metrics.py +0 -135
  243. orca_sdk/_generated_api_client/models/pr_curve.py +0 -86
  244. orca_sdk/_generated_api_client/models/prediction_feedback.py +0 -157
  245. orca_sdk/_generated_api_client/models/prediction_feedback_category.py +0 -115
  246. orca_sdk/_generated_api_client/models/prediction_feedback_request.py +0 -122
  247. orca_sdk/_generated_api_client/models/prediction_feedback_result.py +0 -102
  248. orca_sdk/_generated_api_client/models/prediction_sort_item_item_type_0.py +0 -10
  249. orca_sdk/_generated_api_client/models/prediction_sort_item_item_type_1.py +0 -9
  250. orca_sdk/_generated_api_client/models/predictive_model_update.py +0 -111
  251. orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py +0 -115
  252. orca_sdk/_generated_api_client/models/pretrained_embedding_model_name.py +0 -17
  253. orca_sdk/_generated_api_client/models/rac_head_type.py +0 -11
  254. orca_sdk/_generated_api_client/models/rar_head_type.py +0 -8
  255. orca_sdk/_generated_api_client/models/regression_evaluation_request.py +0 -148
  256. orca_sdk/_generated_api_client/models/regression_metrics.py +0 -172
  257. orca_sdk/_generated_api_client/models/regression_model_metadata.py +0 -191
  258. orca_sdk/_generated_api_client/models/regression_prediction_request.py +0 -195
  259. orca_sdk/_generated_api_client/models/roc_curve.py +0 -86
  260. orca_sdk/_generated_api_client/models/score_prediction_memory_lookup.py +0 -196
  261. orca_sdk/_generated_api_client/models/score_prediction_memory_lookup_metadata.py +0 -68
  262. orca_sdk/_generated_api_client/models/score_prediction_with_memories_and_feedback.py +0 -252
  263. orca_sdk/_generated_api_client/models/scored_memory.py +0 -172
  264. orca_sdk/_generated_api_client/models/scored_memory_insert.py +0 -128
  265. orca_sdk/_generated_api_client/models/scored_memory_insert_metadata.py +0 -68
  266. orca_sdk/_generated_api_client/models/scored_memory_lookup.py +0 -180
  267. orca_sdk/_generated_api_client/models/scored_memory_lookup_metadata.py +0 -68
  268. orca_sdk/_generated_api_client/models/scored_memory_metadata.py +0 -68
  269. orca_sdk/_generated_api_client/models/scored_memory_update.py +0 -171
  270. orca_sdk/_generated_api_client/models/scored_memory_update_metadata_type_0.py +0 -68
  271. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics.py +0 -193
  272. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics_feedback_metrics.py +0 -68
  273. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics_metadata.py +0 -68
  274. orca_sdk/_generated_api_client/models/service_unavailable_error_response.py +0 -81
  275. orca_sdk/_generated_api_client/models/service_unavailable_error_response_status_code.py +0 -8
  276. orca_sdk/_generated_api_client/models/task.py +0 -198
  277. orca_sdk/_generated_api_client/models/task_status.py +0 -14
  278. orca_sdk/_generated_api_client/models/task_status_info.py +0 -133
  279. orca_sdk/_generated_api_client/models/telemetry_field_type_0_item_type_0.py +0 -8
  280. orca_sdk/_generated_api_client/models/telemetry_field_type_0_item_type_2.py +0 -9
  281. orca_sdk/_generated_api_client/models/telemetry_field_type_1_item_type_0.py +0 -8
  282. orca_sdk/_generated_api_client/models/telemetry_field_type_1_item_type_1.py +0 -8
  283. orca_sdk/_generated_api_client/models/telemetry_filter_item.py +0 -217
  284. orca_sdk/_generated_api_client/models/telemetry_filter_item_op.py +0 -15
  285. orca_sdk/_generated_api_client/models/telemetry_memories_request.py +0 -181
  286. orca_sdk/_generated_api_client/models/telemetry_sort_options.py +0 -185
  287. orca_sdk/_generated_api_client/models/telemetry_sort_options_direction.py +0 -9
  288. orca_sdk/_generated_api_client/models/unauthenticated_error_response.py +0 -73
  289. orca_sdk/_generated_api_client/models/unauthenticated_error_response_status_code.py +0 -8
  290. orca_sdk/_generated_api_client/models/unauthorized_error_response.py +0 -81
  291. orca_sdk/_generated_api_client/models/unauthorized_error_response_status_code.py +0 -8
  292. orca_sdk/_generated_api_client/models/update_org_plan_request.py +0 -73
  293. orca_sdk/_generated_api_client/models/update_org_plan_request_tier.py +0 -11
  294. orca_sdk/_generated_api_client/models/update_prediction_request.py +0 -133
  295. orca_sdk/_generated_api_client/models/validation_error.py +0 -99
  296. orca_sdk/_generated_api_client/py.typed +0 -1
  297. orca_sdk/_generated_api_client/types.py +0 -56
  298. orca_sdk-0.0.97.dist-info/RECORD +0 -309
  299. {orca_sdk-0.0.97.dist-info → orca_sdk-0.0.100.dist-info}/WHEEL +0 -0
@@ -1,263 +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="MemoryMetrics")
21
-
22
-
23
- @_attrs_define
24
- class MemoryMetrics:
25
- """
26
- Attributes:
27
- is_duplicate (Union[Unset, bool]):
28
- duplicate_memory_ids (Union[Unset, List[str]]):
29
- has_potential_duplicates (Union[Unset, bool]):
30
- potential_duplicate_memory_ids (Union[List[str], None, Unset]):
31
- cluster (Union[Unset, int]):
32
- embedding_2d (Union[Unset, List[float]]):
33
- anomaly_score (Union[Unset, float]):
34
- neighbor_label_logits (Union[Unset, List[float]]):
35
- neighbor_predicted_label (Union[Unset, int]):
36
- neighbor_predicted_label_ambiguity (Union[Unset, float]):
37
- neighbor_predicted_label_confidence (Union[Unset, float]):
38
- current_label_neighbor_confidence (Union[Unset, float]):
39
- normalized_neighbor_label_entropy (Union[Unset, float]):
40
- neighbor_predicted_label_matches_current_label (Union[None, Unset, bool]):
41
- spread (Union[Unset, float]):
42
- uniformity (Union[Unset, float]):
43
- """
44
-
45
- is_duplicate: Union[Unset, bool] = UNSET
46
- duplicate_memory_ids: Union[Unset, List[str]] = UNSET
47
- has_potential_duplicates: Union[Unset, bool] = UNSET
48
- potential_duplicate_memory_ids: Union[List[str], None, Unset] = UNSET
49
- cluster: Union[Unset, int] = UNSET
50
- embedding_2d: Union[Unset, List[float]] = UNSET
51
- anomaly_score: Union[Unset, float] = UNSET
52
- neighbor_label_logits: Union[Unset, List[float]] = UNSET
53
- neighbor_predicted_label: Union[Unset, int] = UNSET
54
- neighbor_predicted_label_ambiguity: Union[Unset, float] = UNSET
55
- neighbor_predicted_label_confidence: Union[Unset, float] = UNSET
56
- current_label_neighbor_confidence: Union[Unset, float] = UNSET
57
- normalized_neighbor_label_entropy: Union[Unset, float] = UNSET
58
- neighbor_predicted_label_matches_current_label: Union[None, Unset, bool] = UNSET
59
- spread: Union[Unset, float] = UNSET
60
- uniformity: Union[Unset, float] = UNSET
61
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
62
-
63
- def to_dict(self) -> dict[str, Any]:
64
- is_duplicate = self.is_duplicate
65
-
66
- duplicate_memory_ids: Union[Unset, List[str]] = UNSET
67
- if not isinstance(self.duplicate_memory_ids, Unset):
68
- duplicate_memory_ids = self.duplicate_memory_ids
69
-
70
- has_potential_duplicates = self.has_potential_duplicates
71
-
72
- potential_duplicate_memory_ids: Union[List[str], None, Unset]
73
- if isinstance(self.potential_duplicate_memory_ids, Unset):
74
- potential_duplicate_memory_ids = UNSET
75
- elif isinstance(self.potential_duplicate_memory_ids, list):
76
- potential_duplicate_memory_ids = self.potential_duplicate_memory_ids
77
-
78
- else:
79
- potential_duplicate_memory_ids = self.potential_duplicate_memory_ids
80
-
81
- cluster = self.cluster
82
-
83
- embedding_2d: Union[Unset, List[float]] = UNSET
84
- if not isinstance(self.embedding_2d, Unset):
85
- embedding_2d = []
86
- for embedding_2d_item_data in self.embedding_2d:
87
- embedding_2d_item: float
88
- embedding_2d_item = embedding_2d_item_data
89
- embedding_2d.append(embedding_2d_item)
90
-
91
- anomaly_score = self.anomaly_score
92
-
93
- neighbor_label_logits: Union[Unset, List[float]] = UNSET
94
- if not isinstance(self.neighbor_label_logits, Unset):
95
- neighbor_label_logits = self.neighbor_label_logits
96
-
97
- neighbor_predicted_label = self.neighbor_predicted_label
98
-
99
- neighbor_predicted_label_ambiguity = self.neighbor_predicted_label_ambiguity
100
-
101
- neighbor_predicted_label_confidence = self.neighbor_predicted_label_confidence
102
-
103
- current_label_neighbor_confidence = self.current_label_neighbor_confidence
104
-
105
- normalized_neighbor_label_entropy = self.normalized_neighbor_label_entropy
106
-
107
- neighbor_predicted_label_matches_current_label: Union[None, Unset, bool]
108
- if isinstance(self.neighbor_predicted_label_matches_current_label, Unset):
109
- neighbor_predicted_label_matches_current_label = UNSET
110
- else:
111
- neighbor_predicted_label_matches_current_label = self.neighbor_predicted_label_matches_current_label
112
-
113
- spread = self.spread
114
-
115
- uniformity = self.uniformity
116
-
117
- field_dict: dict[str, Any] = {}
118
- field_dict.update(self.additional_properties)
119
- field_dict.update({})
120
- if is_duplicate is not UNSET:
121
- field_dict["is_duplicate"] = is_duplicate
122
- if duplicate_memory_ids is not UNSET:
123
- field_dict["duplicate_memory_ids"] = duplicate_memory_ids
124
- if has_potential_duplicates is not UNSET:
125
- field_dict["has_potential_duplicates"] = has_potential_duplicates
126
- if potential_duplicate_memory_ids is not UNSET:
127
- field_dict["potential_duplicate_memory_ids"] = potential_duplicate_memory_ids
128
- if cluster is not UNSET:
129
- field_dict["cluster"] = cluster
130
- if embedding_2d is not UNSET:
131
- field_dict["embedding_2d"] = embedding_2d
132
- if anomaly_score is not UNSET:
133
- field_dict["anomaly_score"] = anomaly_score
134
- if neighbor_label_logits is not UNSET:
135
- field_dict["neighbor_label_logits"] = neighbor_label_logits
136
- if neighbor_predicted_label is not UNSET:
137
- field_dict["neighbor_predicted_label"] = neighbor_predicted_label
138
- if neighbor_predicted_label_ambiguity is not UNSET:
139
- field_dict["neighbor_predicted_label_ambiguity"] = neighbor_predicted_label_ambiguity
140
- if neighbor_predicted_label_confidence is not UNSET:
141
- field_dict["neighbor_predicted_label_confidence"] = neighbor_predicted_label_confidence
142
- if current_label_neighbor_confidence is not UNSET:
143
- field_dict["current_label_neighbor_confidence"] = current_label_neighbor_confidence
144
- if normalized_neighbor_label_entropy is not UNSET:
145
- field_dict["normalized_neighbor_label_entropy"] = normalized_neighbor_label_entropy
146
- if neighbor_predicted_label_matches_current_label is not UNSET:
147
- field_dict["neighbor_predicted_label_matches_current_label"] = (
148
- neighbor_predicted_label_matches_current_label
149
- )
150
- if spread is not UNSET:
151
- field_dict["spread"] = spread
152
- if uniformity is not UNSET:
153
- field_dict["uniformity"] = uniformity
154
-
155
- return field_dict
156
-
157
- @classmethod
158
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
159
- d = src_dict.copy()
160
- is_duplicate = d.pop("is_duplicate", UNSET)
161
-
162
- duplicate_memory_ids = cast(List[str], d.pop("duplicate_memory_ids", UNSET))
163
-
164
- has_potential_duplicates = d.pop("has_potential_duplicates", UNSET)
165
-
166
- def _parse_potential_duplicate_memory_ids(data: object) -> Union[List[str], None, Unset]:
167
- if data is None:
168
- return data
169
- if isinstance(data, Unset):
170
- return data
171
- try:
172
- if not isinstance(data, list):
173
- raise TypeError()
174
- potential_duplicate_memory_ids_type_0 = cast(List[str], data)
175
-
176
- return potential_duplicate_memory_ids_type_0
177
- except: # noqa: E722
178
- pass
179
- return cast(Union[List[str], None, Unset], data)
180
-
181
- potential_duplicate_memory_ids = _parse_potential_duplicate_memory_ids(
182
- d.pop("potential_duplicate_memory_ids", UNSET)
183
- )
184
-
185
- cluster = d.pop("cluster", UNSET)
186
-
187
- embedding_2d = []
188
- _embedding_2d = d.pop("embedding_2d", UNSET)
189
- for embedding_2d_item_data in _embedding_2d or []:
190
-
191
- def _parse_embedding_2d_item(data: object) -> float:
192
- return cast(float, data)
193
-
194
- embedding_2d_item = _parse_embedding_2d_item(embedding_2d_item_data)
195
-
196
- embedding_2d.append(embedding_2d_item)
197
-
198
- anomaly_score = d.pop("anomaly_score", UNSET)
199
-
200
- neighbor_label_logits = cast(List[float], d.pop("neighbor_label_logits", UNSET))
201
-
202
- neighbor_predicted_label = d.pop("neighbor_predicted_label", UNSET)
203
-
204
- neighbor_predicted_label_ambiguity = d.pop("neighbor_predicted_label_ambiguity", UNSET)
205
-
206
- neighbor_predicted_label_confidence = d.pop("neighbor_predicted_label_confidence", UNSET)
207
-
208
- current_label_neighbor_confidence = d.pop("current_label_neighbor_confidence", UNSET)
209
-
210
- normalized_neighbor_label_entropy = d.pop("normalized_neighbor_label_entropy", UNSET)
211
-
212
- def _parse_neighbor_predicted_label_matches_current_label(data: object) -> Union[None, Unset, bool]:
213
- if data is None:
214
- return data
215
- if isinstance(data, Unset):
216
- return data
217
- return cast(Union[None, Unset, bool], data)
218
-
219
- neighbor_predicted_label_matches_current_label = _parse_neighbor_predicted_label_matches_current_label(
220
- d.pop("neighbor_predicted_label_matches_current_label", UNSET)
221
- )
222
-
223
- spread = d.pop("spread", UNSET)
224
-
225
- uniformity = d.pop("uniformity", UNSET)
226
-
227
- memory_metrics = cls(
228
- is_duplicate=is_duplicate,
229
- duplicate_memory_ids=duplicate_memory_ids,
230
- has_potential_duplicates=has_potential_duplicates,
231
- potential_duplicate_memory_ids=potential_duplicate_memory_ids,
232
- cluster=cluster,
233
- embedding_2d=embedding_2d,
234
- anomaly_score=anomaly_score,
235
- neighbor_label_logits=neighbor_label_logits,
236
- neighbor_predicted_label=neighbor_predicted_label,
237
- neighbor_predicted_label_ambiguity=neighbor_predicted_label_ambiguity,
238
- neighbor_predicted_label_confidence=neighbor_predicted_label_confidence,
239
- current_label_neighbor_confidence=current_label_neighbor_confidence,
240
- normalized_neighbor_label_entropy=normalized_neighbor_label_entropy,
241
- neighbor_predicted_label_matches_current_label=neighbor_predicted_label_matches_current_label,
242
- spread=spread,
243
- uniformity=uniformity,
244
- )
245
-
246
- memory_metrics.additional_properties = d
247
- return memory_metrics
248
-
249
- @property
250
- def additional_keys(self) -> list[str]:
251
- return list(self.additional_properties.keys())
252
-
253
- def __getitem__(self, key: str) -> Any:
254
- return self.additional_properties[key]
255
-
256
- def __setitem__(self, key: str, value: Any) -> None:
257
- self.additional_properties[key] = value
258
-
259
- def __delitem__(self, key: str) -> None:
260
- del self.additional_properties[key]
261
-
262
- def __contains__(self, key: str) -> bool:
263
- return key in self.additional_properties
@@ -1,9 +0,0 @@
1
- from enum import Enum
2
-
3
-
4
- class MemoryType(str, Enum):
5
- LABELED = "LABELED"
6
- SCORED = "SCORED"
7
-
8
- def __str__(self) -> str:
9
- return str(self.value)
@@ -1,245 +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 TYPE_CHECKING, Any, Dict, Type, TypeVar, Union, cast
14
-
15
- from attrs import define as _attrs_define
16
-
17
- from ..types import UNSET, Unset
18
-
19
- if TYPE_CHECKING:
20
- from ..models.memoryset_class_patterns_analysis_config import MemorysetClassPatternsAnalysisConfig
21
- from ..models.memoryset_cluster_analysis_config import MemorysetClusterAnalysisConfig
22
- from ..models.memoryset_duplicate_analysis_config import MemorysetDuplicateAnalysisConfig
23
- from ..models.memoryset_label_analysis_config import MemorysetLabelAnalysisConfig
24
- from ..models.memoryset_neighbor_analysis_config import MemorysetNeighborAnalysisConfig
25
- from ..models.memoryset_projection_analysis_config import MemorysetProjectionAnalysisConfig
26
-
27
-
28
- T = TypeVar("T", bound="MemorysetAnalysisConfigs")
29
-
30
-
31
- @_attrs_define
32
- class MemorysetAnalysisConfigs:
33
- """
34
- Attributes:
35
- neighbor (Union['MemorysetNeighborAnalysisConfig', None, Unset]):
36
- label (Union['MemorysetLabelAnalysisConfig', None, Unset]):
37
- duplicate (Union['MemorysetDuplicateAnalysisConfig', None, Unset]):
38
- projection (Union['MemorysetProjectionAnalysisConfig', None, Unset]):
39
- cluster (Union['MemorysetClusterAnalysisConfig', None, Unset]):
40
- class_patterns (Union['MemorysetClassPatternsAnalysisConfig', None, Unset]):
41
- """
42
-
43
- neighbor: Union["MemorysetNeighborAnalysisConfig", None, Unset] = UNSET
44
- label: Union["MemorysetLabelAnalysisConfig", None, Unset] = UNSET
45
- duplicate: Union["MemorysetDuplicateAnalysisConfig", None, Unset] = UNSET
46
- projection: Union["MemorysetProjectionAnalysisConfig", None, Unset] = UNSET
47
- cluster: Union["MemorysetClusterAnalysisConfig", None, Unset] = UNSET
48
- class_patterns: Union["MemorysetClassPatternsAnalysisConfig", None, Unset] = UNSET
49
-
50
- def to_dict(self) -> dict[str, Any]:
51
- from ..models.memoryset_class_patterns_analysis_config import MemorysetClassPatternsAnalysisConfig
52
- from ..models.memoryset_cluster_analysis_config import MemorysetClusterAnalysisConfig
53
- from ..models.memoryset_duplicate_analysis_config import MemorysetDuplicateAnalysisConfig
54
- from ..models.memoryset_label_analysis_config import MemorysetLabelAnalysisConfig
55
- from ..models.memoryset_neighbor_analysis_config import MemorysetNeighborAnalysisConfig
56
- from ..models.memoryset_projection_analysis_config import MemorysetProjectionAnalysisConfig
57
-
58
- neighbor: Union[Dict[str, Any], None, Unset]
59
- if isinstance(self.neighbor, Unset):
60
- neighbor = UNSET
61
- elif isinstance(self.neighbor, MemorysetNeighborAnalysisConfig):
62
- neighbor = self.neighbor.to_dict()
63
- else:
64
- neighbor = self.neighbor
65
-
66
- label: Union[Dict[str, Any], None, Unset]
67
- if isinstance(self.label, Unset):
68
- label = UNSET
69
- elif isinstance(self.label, MemorysetLabelAnalysisConfig):
70
- label = self.label.to_dict()
71
- else:
72
- label = self.label
73
-
74
- duplicate: Union[Dict[str, Any], None, Unset]
75
- if isinstance(self.duplicate, Unset):
76
- duplicate = UNSET
77
- elif isinstance(self.duplicate, MemorysetDuplicateAnalysisConfig):
78
- duplicate = self.duplicate.to_dict()
79
- else:
80
- duplicate = self.duplicate
81
-
82
- projection: Union[Dict[str, Any], None, Unset]
83
- if isinstance(self.projection, Unset):
84
- projection = UNSET
85
- elif isinstance(self.projection, MemorysetProjectionAnalysisConfig):
86
- projection = self.projection.to_dict()
87
- else:
88
- projection = self.projection
89
-
90
- cluster: Union[Dict[str, Any], None, Unset]
91
- if isinstance(self.cluster, Unset):
92
- cluster = UNSET
93
- elif isinstance(self.cluster, MemorysetClusterAnalysisConfig):
94
- cluster = self.cluster.to_dict()
95
- else:
96
- cluster = self.cluster
97
-
98
- class_patterns: Union[Dict[str, Any], None, Unset]
99
- if isinstance(self.class_patterns, Unset):
100
- class_patterns = UNSET
101
- elif isinstance(self.class_patterns, MemorysetClassPatternsAnalysisConfig):
102
- class_patterns = self.class_patterns.to_dict()
103
- else:
104
- class_patterns = self.class_patterns
105
-
106
- field_dict: dict[str, Any] = {}
107
- field_dict.update({})
108
- if neighbor is not UNSET:
109
- field_dict["neighbor"] = neighbor
110
- if label is not UNSET:
111
- field_dict["label"] = label
112
- if duplicate is not UNSET:
113
- field_dict["duplicate"] = duplicate
114
- if projection is not UNSET:
115
- field_dict["projection"] = projection
116
- if cluster is not UNSET:
117
- field_dict["cluster"] = cluster
118
- if class_patterns is not UNSET:
119
- field_dict["class_patterns"] = class_patterns
120
-
121
- return field_dict
122
-
123
- @classmethod
124
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
125
- from ..models.memoryset_class_patterns_analysis_config import MemorysetClassPatternsAnalysisConfig
126
- from ..models.memoryset_cluster_analysis_config import MemorysetClusterAnalysisConfig
127
- from ..models.memoryset_duplicate_analysis_config import MemorysetDuplicateAnalysisConfig
128
- from ..models.memoryset_label_analysis_config import MemorysetLabelAnalysisConfig
129
- from ..models.memoryset_neighbor_analysis_config import MemorysetNeighborAnalysisConfig
130
- from ..models.memoryset_projection_analysis_config import MemorysetProjectionAnalysisConfig
131
-
132
- d = src_dict.copy()
133
-
134
- def _parse_neighbor(data: object) -> Union["MemorysetNeighborAnalysisConfig", None, Unset]:
135
- if data is None:
136
- return data
137
- if isinstance(data, Unset):
138
- return data
139
- try:
140
- if not isinstance(data, dict):
141
- raise TypeError()
142
- neighbor_type_0 = MemorysetNeighborAnalysisConfig.from_dict(data)
143
-
144
- return neighbor_type_0
145
- except: # noqa: E722
146
- pass
147
- return cast(Union["MemorysetNeighborAnalysisConfig", None, Unset], data)
148
-
149
- neighbor = _parse_neighbor(d.pop("neighbor", UNSET))
150
-
151
- def _parse_label(data: object) -> Union["MemorysetLabelAnalysisConfig", None, Unset]:
152
- if data is None:
153
- return data
154
- if isinstance(data, Unset):
155
- return data
156
- try:
157
- if not isinstance(data, dict):
158
- raise TypeError()
159
- label_type_0 = MemorysetLabelAnalysisConfig.from_dict(data)
160
-
161
- return label_type_0
162
- except: # noqa: E722
163
- pass
164
- return cast(Union["MemorysetLabelAnalysisConfig", None, Unset], data)
165
-
166
- label = _parse_label(d.pop("label", UNSET))
167
-
168
- def _parse_duplicate(data: object) -> Union["MemorysetDuplicateAnalysisConfig", None, Unset]:
169
- if data is None:
170
- return data
171
- if isinstance(data, Unset):
172
- return data
173
- try:
174
- if not isinstance(data, dict):
175
- raise TypeError()
176
- duplicate_type_0 = MemorysetDuplicateAnalysisConfig.from_dict(data)
177
-
178
- return duplicate_type_0
179
- except: # noqa: E722
180
- pass
181
- return cast(Union["MemorysetDuplicateAnalysisConfig", None, Unset], data)
182
-
183
- duplicate = _parse_duplicate(d.pop("duplicate", UNSET))
184
-
185
- def _parse_projection(data: object) -> Union["MemorysetProjectionAnalysisConfig", None, Unset]:
186
- if data is None:
187
- return data
188
- if isinstance(data, Unset):
189
- return data
190
- try:
191
- if not isinstance(data, dict):
192
- raise TypeError()
193
- projection_type_0 = MemorysetProjectionAnalysisConfig.from_dict(data)
194
-
195
- return projection_type_0
196
- except: # noqa: E722
197
- pass
198
- return cast(Union["MemorysetProjectionAnalysisConfig", None, Unset], data)
199
-
200
- projection = _parse_projection(d.pop("projection", UNSET))
201
-
202
- def _parse_cluster(data: object) -> Union["MemorysetClusterAnalysisConfig", None, Unset]:
203
- if data is None:
204
- return data
205
- if isinstance(data, Unset):
206
- return data
207
- try:
208
- if not isinstance(data, dict):
209
- raise TypeError()
210
- cluster_type_0 = MemorysetClusterAnalysisConfig.from_dict(data)
211
-
212
- return cluster_type_0
213
- except: # noqa: E722
214
- pass
215
- return cast(Union["MemorysetClusterAnalysisConfig", None, Unset], data)
216
-
217
- cluster = _parse_cluster(d.pop("cluster", UNSET))
218
-
219
- def _parse_class_patterns(data: object) -> Union["MemorysetClassPatternsAnalysisConfig", None, Unset]:
220
- if data is None:
221
- return data
222
- if isinstance(data, Unset):
223
- return data
224
- try:
225
- if not isinstance(data, dict):
226
- raise TypeError()
227
- class_patterns_type_0 = MemorysetClassPatternsAnalysisConfig.from_dict(data)
228
-
229
- return class_patterns_type_0
230
- except: # noqa: E722
231
- pass
232
- return cast(Union["MemorysetClassPatternsAnalysisConfig", None, Unset], data)
233
-
234
- class_patterns = _parse_class_patterns(d.pop("class_patterns", UNSET))
235
-
236
- memoryset_analysis_configs = cls(
237
- neighbor=neighbor,
238
- label=label,
239
- duplicate=duplicate,
240
- projection=projection,
241
- cluster=cluster,
242
- class_patterns=class_patterns,
243
- )
244
-
245
- return memoryset_analysis_configs
@@ -1,105 +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 TYPE_CHECKING, Any, Type, TypeVar, Union
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
- if TYPE_CHECKING:
21
- from ..models.memoryset_analysis_configs import MemorysetAnalysisConfigs
22
-
23
-
24
- T = TypeVar("T", bound="MemorysetAnalysisRequest")
25
-
26
-
27
- @_attrs_define
28
- class MemorysetAnalysisRequest:
29
- """
30
- Attributes:
31
- configs (MemorysetAnalysisConfigs):
32
- lookup_count (Union[Unset, int]): Default: 15.
33
- batch_size (Union[Unset, int]): Default: 32.
34
- clear_metrics (Union[Unset, bool]): Default: False.
35
- """
36
-
37
- configs: "MemorysetAnalysisConfigs"
38
- lookup_count: Union[Unset, int] = 15
39
- batch_size: Union[Unset, int] = 32
40
- clear_metrics: 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
- configs = self.configs.to_dict()
45
-
46
- lookup_count = self.lookup_count
47
-
48
- batch_size = self.batch_size
49
-
50
- clear_metrics = self.clear_metrics
51
-
52
- field_dict: dict[str, Any] = {}
53
- field_dict.update(self.additional_properties)
54
- field_dict.update(
55
- {
56
- "configs": configs,
57
- }
58
- )
59
- if lookup_count is not UNSET:
60
- field_dict["lookup_count"] = lookup_count
61
- if batch_size is not UNSET:
62
- field_dict["batch_size"] = batch_size
63
- if clear_metrics is not UNSET:
64
- field_dict["clear_metrics"] = clear_metrics
65
-
66
- return field_dict
67
-
68
- @classmethod
69
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
70
- from ..models.memoryset_analysis_configs import MemorysetAnalysisConfigs
71
-
72
- d = src_dict.copy()
73
- configs = MemorysetAnalysisConfigs.from_dict(d.pop("configs"))
74
-
75
- lookup_count = d.pop("lookup_count", UNSET)
76
-
77
- batch_size = d.pop("batch_size", UNSET)
78
-
79
- clear_metrics = d.pop("clear_metrics", UNSET)
80
-
81
- memoryset_analysis_request = cls(
82
- configs=configs,
83
- lookup_count=lookup_count,
84
- batch_size=batch_size,
85
- clear_metrics=clear_metrics,
86
- )
87
-
88
- memoryset_analysis_request.additional_properties = d
89
- return memoryset_analysis_request
90
-
91
- @property
92
- def additional_keys(self) -> list[str]:
93
- return list(self.additional_properties.keys())
94
-
95
- def __getitem__(self, key: str) -> Any:
96
- return self.additional_properties[key]
97
-
98
- def __setitem__(self, key: str, value: Any) -> None:
99
- self.additional_properties[key] = value
100
-
101
- def __delitem__(self, key: str) -> None:
102
- del self.additional_properties[key]
103
-
104
- def __contains__(self, key: str) -> bool:
105
- return key in self.additional_properties