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,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