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,182 +0,0 @@
1
- """
2
- This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
-
4
- It is a customized template from the openapi-python-client tool's default template:
5
- https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
6
-
7
- The main change is:
8
- - Fix typing issues
9
- """
10
-
11
- # flake8: noqa: C901
12
-
13
- import datetime
14
- from enum import Enum
15
- from typing import TYPE_CHECKING, Any, Dict, Type, TypeVar, Union, cast
16
-
17
- from attrs import define as _attrs_define
18
- from attrs import field as _attrs_field
19
- from dateutil.parser import isoparse
20
-
21
- from ..models.task_status import TaskStatus
22
-
23
- if TYPE_CHECKING:
24
- from ..models.memoryset_analysis_configs import MemorysetAnalysisConfigs
25
- from ..models.memoryset_metrics import MemorysetMetrics
26
-
27
-
28
- T = TypeVar("T", bound="MemorysetAnalysisResponse")
29
-
30
-
31
- @_attrs_define
32
- class MemorysetAnalysisResponse:
33
- """
34
- Attributes:
35
- task_id (str):
36
- org_id (str):
37
- memoryset_id (str):
38
- status (TaskStatus): Status of task in the task queue
39
- lookup_count (int):
40
- batch_size (int):
41
- clear_metrics (bool):
42
- configs (MemorysetAnalysisConfigs):
43
- results (Union['MemorysetMetrics', None]):
44
- created_at (datetime.datetime):
45
- updated_at (datetime.datetime):
46
- """
47
-
48
- task_id: str
49
- org_id: str
50
- memoryset_id: str
51
- status: TaskStatus
52
- lookup_count: int
53
- batch_size: int
54
- clear_metrics: bool
55
- configs: "MemorysetAnalysisConfigs"
56
- results: Union["MemorysetMetrics", None]
57
- created_at: datetime.datetime
58
- updated_at: datetime.datetime
59
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
60
-
61
- def to_dict(self) -> dict[str, Any]:
62
- from ..models.memoryset_metrics import MemorysetMetrics
63
-
64
- task_id = self.task_id
65
-
66
- org_id = self.org_id
67
-
68
- memoryset_id = self.memoryset_id
69
-
70
- status = self.status.value if isinstance(self.status, Enum) else self.status
71
-
72
- lookup_count = self.lookup_count
73
-
74
- batch_size = self.batch_size
75
-
76
- clear_metrics = self.clear_metrics
77
-
78
- configs = self.configs.to_dict()
79
-
80
- results: Union[Dict[str, Any], None]
81
- if isinstance(self.results, MemorysetMetrics):
82
- results = self.results.to_dict()
83
- else:
84
- results = self.results
85
-
86
- created_at = self.created_at.isoformat()
87
-
88
- updated_at = self.updated_at.isoformat()
89
-
90
- field_dict: dict[str, Any] = {}
91
- field_dict.update(self.additional_properties)
92
- field_dict.update(
93
- {
94
- "task_id": task_id,
95
- "org_id": org_id,
96
- "memoryset_id": memoryset_id,
97
- "status": status,
98
- "lookup_count": lookup_count,
99
- "batch_size": batch_size,
100
- "clear_metrics": clear_metrics,
101
- "configs": configs,
102
- "results": results,
103
- "created_at": created_at,
104
- "updated_at": updated_at,
105
- }
106
- )
107
-
108
- return field_dict
109
-
110
- @classmethod
111
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
112
- from ..models.memoryset_analysis_configs import MemorysetAnalysisConfigs
113
- from ..models.memoryset_metrics import MemorysetMetrics
114
-
115
- d = src_dict.copy()
116
- task_id = d.pop("task_id")
117
-
118
- org_id = d.pop("org_id")
119
-
120
- memoryset_id = d.pop("memoryset_id")
121
-
122
- status = TaskStatus(d.pop("status"))
123
-
124
- lookup_count = d.pop("lookup_count")
125
-
126
- batch_size = d.pop("batch_size")
127
-
128
- clear_metrics = d.pop("clear_metrics")
129
-
130
- configs = MemorysetAnalysisConfigs.from_dict(d.pop("configs"))
131
-
132
- def _parse_results(data: object) -> Union["MemorysetMetrics", None]:
133
- if data is None:
134
- return data
135
- try:
136
- if not isinstance(data, dict):
137
- raise TypeError()
138
- results_type_0 = MemorysetMetrics.from_dict(data)
139
-
140
- return results_type_0
141
- except: # noqa: E722
142
- pass
143
- return cast(Union["MemorysetMetrics", None], data)
144
-
145
- results = _parse_results(d.pop("results"))
146
-
147
- created_at = isoparse(d.pop("created_at"))
148
-
149
- updated_at = isoparse(d.pop("updated_at"))
150
-
151
- memoryset_analysis_response = cls(
152
- task_id=task_id,
153
- org_id=org_id,
154
- memoryset_id=memoryset_id,
155
- status=status,
156
- lookup_count=lookup_count,
157
- batch_size=batch_size,
158
- clear_metrics=clear_metrics,
159
- configs=configs,
160
- results=results,
161
- created_at=created_at,
162
- updated_at=updated_at,
163
- )
164
-
165
- memoryset_analysis_response.additional_properties = d
166
- return memoryset_analysis_response
167
-
168
- @property
169
- def additional_keys(self) -> list[str]:
170
- return list(self.additional_properties.keys())
171
-
172
- def __getitem__(self, key: str) -> Any:
173
- return self.additional_properties[key]
174
-
175
- def __setitem__(self, key: str, value: Any) -> None:
176
- self.additional_properties[key] = value
177
-
178
- def __delitem__(self, key: str) -> None:
179
- del self.additional_properties[key]
180
-
181
- def __contains__(self, key: str) -> bool:
182
- return key in self.additional_properties
@@ -1,79 +0,0 @@
1
- """
2
- This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
-
4
- It is a customized template from the openapi-python-client tool's default template:
5
- https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
6
-
7
- The main change is:
8
- - Fix typing issues
9
- """
10
-
11
- # flake8: noqa: C901
12
-
13
- from typing import Any, Type, TypeVar, Union
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="MemorysetClassPatternsAnalysisConfig")
21
-
22
-
23
- @_attrs_define
24
- class MemorysetClassPatternsAnalysisConfig:
25
- """
26
- Attributes:
27
- representatives_per_class (Union[Unset, int]): Default: 5.
28
- enable_patterns_description (Union[Unset, bool]): Default: True.
29
- """
30
-
31
- representatives_per_class: Union[Unset, int] = 5
32
- enable_patterns_description: Union[Unset, bool] = True
33
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
34
-
35
- def to_dict(self) -> dict[str, Any]:
36
- representatives_per_class = self.representatives_per_class
37
-
38
- enable_patterns_description = self.enable_patterns_description
39
-
40
- field_dict: dict[str, Any] = {}
41
- field_dict.update(self.additional_properties)
42
- field_dict.update({})
43
- if representatives_per_class is not UNSET:
44
- field_dict["representatives_per_class"] = representatives_per_class
45
- if enable_patterns_description is not UNSET:
46
- field_dict["enable_patterns_description"] = enable_patterns_description
47
-
48
- return field_dict
49
-
50
- @classmethod
51
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
52
- d = src_dict.copy()
53
- representatives_per_class = d.pop("representatives_per_class", UNSET)
54
-
55
- enable_patterns_description = d.pop("enable_patterns_description", UNSET)
56
-
57
- memoryset_class_patterns_analysis_config = cls(
58
- representatives_per_class=representatives_per_class,
59
- enable_patterns_description=enable_patterns_description,
60
- )
61
-
62
- memoryset_class_patterns_analysis_config.additional_properties = d
63
- return memoryset_class_patterns_analysis_config
64
-
65
- @property
66
- def additional_keys(self) -> list[str]:
67
- return list(self.additional_properties.keys())
68
-
69
- def __getitem__(self, key: str) -> Any:
70
- return self.additional_properties[key]
71
-
72
- def __setitem__(self, key: str, value: Any) -> None:
73
- self.additional_properties[key] = value
74
-
75
- def __delitem__(self, key: str) -> None:
76
- del self.additional_properties[key]
77
-
78
- def __contains__(self, key: str) -> bool:
79
- return key in self.additional_properties
@@ -1,138 +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, 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
- if TYPE_CHECKING:
21
- from ..models.class_representatives import ClassRepresentatives
22
-
23
-
24
- T = TypeVar("T", bound="MemorysetClassPatternsMetrics")
25
-
26
-
27
- @_attrs_define
28
- class MemorysetClassPatternsMetrics:
29
- """
30
- Attributes:
31
- class_representatives (List['ClassRepresentatives']):
32
- mean_spread (float):
33
- variance_spread (float):
34
- mean_uniformity (float):
35
- variance_uniformity (float):
36
- patterns_description (Union[None, Unset, str]):
37
- """
38
-
39
- class_representatives: List["ClassRepresentatives"]
40
- mean_spread: float
41
- variance_spread: float
42
- mean_uniformity: float
43
- variance_uniformity: float
44
- patterns_description: Union[None, Unset, str] = UNSET
45
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
46
-
47
- def to_dict(self) -> dict[str, Any]:
48
- class_representatives = []
49
- for class_representatives_item_data in self.class_representatives:
50
- class_representatives_item = class_representatives_item_data.to_dict()
51
- class_representatives.append(class_representatives_item)
52
-
53
- mean_spread = self.mean_spread
54
-
55
- variance_spread = self.variance_spread
56
-
57
- mean_uniformity = self.mean_uniformity
58
-
59
- variance_uniformity = self.variance_uniformity
60
-
61
- patterns_description: Union[None, Unset, str]
62
- if isinstance(self.patterns_description, Unset):
63
- patterns_description = UNSET
64
- else:
65
- patterns_description = self.patterns_description
66
-
67
- field_dict: dict[str, Any] = {}
68
- field_dict.update(self.additional_properties)
69
- field_dict.update(
70
- {
71
- "class_representatives": class_representatives,
72
- "mean_spread": mean_spread,
73
- "variance_spread": variance_spread,
74
- "mean_uniformity": mean_uniformity,
75
- "variance_uniformity": variance_uniformity,
76
- }
77
- )
78
- if patterns_description is not UNSET:
79
- field_dict["patterns_description"] = patterns_description
80
-
81
- return field_dict
82
-
83
- @classmethod
84
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
85
- from ..models.class_representatives import ClassRepresentatives
86
-
87
- d = src_dict.copy()
88
- class_representatives = []
89
- _class_representatives = d.pop("class_representatives")
90
- for class_representatives_item_data in _class_representatives:
91
- class_representatives_item = ClassRepresentatives.from_dict(class_representatives_item_data)
92
-
93
- class_representatives.append(class_representatives_item)
94
-
95
- mean_spread = d.pop("mean_spread")
96
-
97
- variance_spread = d.pop("variance_spread")
98
-
99
- mean_uniformity = d.pop("mean_uniformity")
100
-
101
- variance_uniformity = d.pop("variance_uniformity")
102
-
103
- def _parse_patterns_description(data: object) -> Union[None, Unset, str]:
104
- if data is None:
105
- return data
106
- if isinstance(data, Unset):
107
- return data
108
- return cast(Union[None, Unset, str], data)
109
-
110
- patterns_description = _parse_patterns_description(d.pop("patterns_description", UNSET))
111
-
112
- memoryset_class_patterns_metrics = cls(
113
- class_representatives=class_representatives,
114
- mean_spread=mean_spread,
115
- variance_spread=variance_spread,
116
- mean_uniformity=mean_uniformity,
117
- variance_uniformity=variance_uniformity,
118
- patterns_description=patterns_description,
119
- )
120
-
121
- memoryset_class_patterns_metrics.additional_properties = d
122
- return memoryset_class_patterns_metrics
123
-
124
- @property
125
- def additional_keys(self) -> list[str]:
126
- return list(self.additional_properties.keys())
127
-
128
- def __getitem__(self, key: str) -> Any:
129
- return self.additional_properties[key]
130
-
131
- def __setitem__(self, key: str, value: Any) -> None:
132
- self.additional_properties[key] = value
133
-
134
- def __delitem__(self, key: str) -> None:
135
- del self.additional_properties[key]
136
-
137
- def __contains__(self, key: str) -> bool:
138
- return key in self.additional_properties
@@ -1,202 +0,0 @@
1
- """
2
- This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
-
4
- It is a customized template from the openapi-python-client tool's default template:
5
- https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
6
-
7
- The main change is:
8
- - Fix typing issues
9
- """
10
-
11
- # flake8: noqa: C901
12
-
13
- from typing import Any, Type, TypeVar, Union, cast
14
-
15
- from attrs import define as _attrs_define
16
- from attrs import field as _attrs_field
17
-
18
- from ..models.memoryset_cluster_analysis_config_clustering_method import MemorysetClusterAnalysisConfigClusteringMethod
19
- from ..models.memoryset_cluster_analysis_config_partitioning_method import (
20
- MemorysetClusterAnalysisConfigPartitioningMethod,
21
- )
22
- from ..types import UNSET, Unset
23
-
24
- T = TypeVar("T", bound="MemorysetClusterAnalysisConfig")
25
-
26
-
27
- @_attrs_define
28
- class MemorysetClusterAnalysisConfig:
29
- """
30
- Attributes:
31
- min_cluster_size (Union[None, Unset, int]):
32
- max_cluster_size (Union[None, Unset, int]):
33
- clustering_method (Union[Unset, MemorysetClusterAnalysisConfigClusteringMethod]): Default:
34
- MemorysetClusterAnalysisConfigClusteringMethod.GRAPH.
35
- min_cluster_distance (Union[Unset, float]): Default: 0.0.
36
- partitioning_method (Union[Unset, MemorysetClusterAnalysisConfigPartitioningMethod]): Default:
37
- MemorysetClusterAnalysisConfigPartitioningMethod.NG.
38
- resolution (Union[None, Unset, float]):
39
- num_iterations (Union[Unset, int]): Default: 2.
40
- random_state (Union[None, Unset, int]): Default: 42.
41
- """
42
-
43
- min_cluster_size: Union[None, Unset, int] = UNSET
44
- max_cluster_size: Union[None, Unset, int] = UNSET
45
- clustering_method: Union[Unset, MemorysetClusterAnalysisConfigClusteringMethod] = (
46
- MemorysetClusterAnalysisConfigClusteringMethod.GRAPH
47
- )
48
- min_cluster_distance: Union[Unset, float] = 0.0
49
- partitioning_method: Union[Unset, MemorysetClusterAnalysisConfigPartitioningMethod] = (
50
- MemorysetClusterAnalysisConfigPartitioningMethod.NG
51
- )
52
- resolution: Union[None, Unset, float] = UNSET
53
- num_iterations: Union[Unset, int] = 2
54
- random_state: Union[None, Unset, int] = 42
55
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
56
-
57
- def to_dict(self) -> dict[str, Any]:
58
- min_cluster_size: Union[None, Unset, int]
59
- if isinstance(self.min_cluster_size, Unset):
60
- min_cluster_size = UNSET
61
- else:
62
- min_cluster_size = self.min_cluster_size
63
-
64
- max_cluster_size: Union[None, Unset, int]
65
- if isinstance(self.max_cluster_size, Unset):
66
- max_cluster_size = UNSET
67
- else:
68
- max_cluster_size = self.max_cluster_size
69
-
70
- clustering_method: Union[Unset, str] = UNSET
71
- if not isinstance(self.clustering_method, Unset):
72
- clustering_method = self.clustering_method.value
73
-
74
- min_cluster_distance = self.min_cluster_distance
75
-
76
- partitioning_method: Union[Unset, str] = UNSET
77
- if not isinstance(self.partitioning_method, Unset):
78
- partitioning_method = self.partitioning_method.value
79
-
80
- resolution: Union[None, Unset, float]
81
- if isinstance(self.resolution, Unset):
82
- resolution = UNSET
83
- else:
84
- resolution = self.resolution
85
-
86
- num_iterations = self.num_iterations
87
-
88
- random_state: Union[None, Unset, int]
89
- if isinstance(self.random_state, Unset):
90
- random_state = UNSET
91
- else:
92
- random_state = self.random_state
93
-
94
- field_dict: dict[str, Any] = {}
95
- field_dict.update(self.additional_properties)
96
- field_dict.update({})
97
- if min_cluster_size is not UNSET:
98
- field_dict["min_cluster_size"] = min_cluster_size
99
- if max_cluster_size is not UNSET:
100
- field_dict["max_cluster_size"] = max_cluster_size
101
- if clustering_method is not UNSET:
102
- field_dict["clustering_method"] = clustering_method
103
- if min_cluster_distance is not UNSET:
104
- field_dict["min_cluster_distance"] = min_cluster_distance
105
- if partitioning_method is not UNSET:
106
- field_dict["partitioning_method"] = partitioning_method
107
- if resolution is not UNSET:
108
- field_dict["resolution"] = resolution
109
- if num_iterations is not UNSET:
110
- field_dict["num_iterations"] = num_iterations
111
- if random_state is not UNSET:
112
- field_dict["random_state"] = random_state
113
-
114
- return field_dict
115
-
116
- @classmethod
117
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
118
- d = src_dict.copy()
119
-
120
- def _parse_min_cluster_size(data: object) -> Union[None, Unset, int]:
121
- if data is None:
122
- return data
123
- if isinstance(data, Unset):
124
- return data
125
- return cast(Union[None, Unset, int], data)
126
-
127
- min_cluster_size = _parse_min_cluster_size(d.pop("min_cluster_size", UNSET))
128
-
129
- def _parse_max_cluster_size(data: object) -> Union[None, Unset, int]:
130
- if data is None:
131
- return data
132
- if isinstance(data, Unset):
133
- return data
134
- return cast(Union[None, Unset, int], data)
135
-
136
- max_cluster_size = _parse_max_cluster_size(d.pop("max_cluster_size", UNSET))
137
-
138
- _clustering_method = d.pop("clustering_method", UNSET)
139
- clustering_method: Union[Unset, MemorysetClusterAnalysisConfigClusteringMethod]
140
- if isinstance(_clustering_method, Unset):
141
- clustering_method = UNSET
142
- else:
143
- clustering_method = MemorysetClusterAnalysisConfigClusteringMethod(_clustering_method)
144
-
145
- min_cluster_distance = d.pop("min_cluster_distance", UNSET)
146
-
147
- _partitioning_method = d.pop("partitioning_method", UNSET)
148
- partitioning_method: Union[Unset, MemorysetClusterAnalysisConfigPartitioningMethod]
149
- if isinstance(_partitioning_method, Unset):
150
- partitioning_method = UNSET
151
- else:
152
- partitioning_method = MemorysetClusterAnalysisConfigPartitioningMethod(_partitioning_method)
153
-
154
- def _parse_resolution(data: object) -> Union[None, Unset, float]:
155
- if data is None:
156
- return data
157
- if isinstance(data, Unset):
158
- return data
159
- return cast(Union[None, Unset, float], data)
160
-
161
- resolution = _parse_resolution(d.pop("resolution", UNSET))
162
-
163
- num_iterations = d.pop("num_iterations", UNSET)
164
-
165
- def _parse_random_state(data: object) -> Union[None, Unset, int]:
166
- if data is None:
167
- return data
168
- if isinstance(data, Unset):
169
- return data
170
- return cast(Union[None, Unset, int], data)
171
-
172
- random_state = _parse_random_state(d.pop("random_state", UNSET))
173
-
174
- memoryset_cluster_analysis_config = cls(
175
- min_cluster_size=min_cluster_size,
176
- max_cluster_size=max_cluster_size,
177
- clustering_method=clustering_method,
178
- min_cluster_distance=min_cluster_distance,
179
- partitioning_method=partitioning_method,
180
- resolution=resolution,
181
- num_iterations=num_iterations,
182
- random_state=random_state,
183
- )
184
-
185
- memoryset_cluster_analysis_config.additional_properties = d
186
- return memoryset_cluster_analysis_config
187
-
188
- @property
189
- def additional_keys(self) -> list[str]:
190
- return list(self.additional_properties.keys())
191
-
192
- def __getitem__(self, key: str) -> Any:
193
- return self.additional_properties[key]
194
-
195
- def __setitem__(self, key: str, value: Any) -> None:
196
- self.additional_properties[key] = value
197
-
198
- def __delitem__(self, key: str) -> None:
199
- del self.additional_properties[key]
200
-
201
- def __contains__(self, key: str) -> bool:
202
- return key in self.additional_properties
@@ -1,9 +0,0 @@
1
- from enum import Enum
2
-
3
-
4
- class MemorysetClusterAnalysisConfigClusteringMethod(str, Enum):
5
- DENSITY = "density"
6
- GRAPH = "graph"
7
-
8
- def __str__(self) -> str:
9
- return str(self.value)
@@ -1,10 +0,0 @@
1
- from enum import Enum
2
-
3
-
4
- class MemorysetClusterAnalysisConfigPartitioningMethod(str, Enum):
5
- CPM = "cpm"
6
- NG = "ng"
7
- RB = "rb"
8
-
9
- def __str__(self) -> str:
10
- return str(self.value)