orca-sdk 0.0.96__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 (269) hide show
  1. orca_sdk/__init__.py +2 -5
  2. orca_sdk/_shared/__init__.py +1 -0
  3. orca_sdk/_shared/metrics.py +1 -1
  4. orca_sdk/_utils/analysis_ui.py +5 -5
  5. orca_sdk/_utils/auth.py +23 -33
  6. orca_sdk/_utils/pagination.py +126 -0
  7. orca_sdk/_utils/pagination_test.py +132 -0
  8. orca_sdk/classification_model.py +188 -126
  9. orca_sdk/classification_model_test.py +102 -0
  10. orca_sdk/client.py +3515 -0
  11. orca_sdk/conftest.py +10 -0
  12. orca_sdk/credentials.py +73 -21
  13. orca_sdk/credentials_test.py +20 -0
  14. orca_sdk/datasource.py +186 -81
  15. orca_sdk/datasource_test.py +194 -0
  16. orca_sdk/embedding_model.py +267 -75
  17. orca_sdk/embedding_model_test.py +32 -14
  18. orca_sdk/job.py +59 -54
  19. orca_sdk/job_test.py +50 -0
  20. orca_sdk/memoryset.py +372 -345
  21. orca_sdk/memoryset_test.py +7 -11
  22. orca_sdk/regression_model.py +120 -111
  23. orca_sdk/regression_model_test.py +15 -0
  24. orca_sdk/telemetry.py +229 -115
  25. {orca_sdk-0.0.96.dist-info → orca_sdk-0.0.98.dist-info}/METADATA +19 -5
  26. orca_sdk-0.0.98.dist-info/RECORD +40 -0
  27. orca_sdk/_generated_api_client/__init__.py +0 -3
  28. orca_sdk/_generated_api_client/api/__init__.py +0 -287
  29. orca_sdk/_generated_api_client/api/auth/__init__.py +0 -0
  30. orca_sdk/_generated_api_client/api/auth/check_authentication_auth_get.py +0 -128
  31. orca_sdk/_generated_api_client/api/auth/create_api_key_auth_api_key_post.py +0 -170
  32. orca_sdk/_generated_api_client/api/auth/delete_api_key_auth_api_key_name_or_id_delete.py +0 -156
  33. orca_sdk/_generated_api_client/api/auth/delete_org_auth_org_delete.py +0 -130
  34. orca_sdk/_generated_api_client/api/auth/list_api_keys_auth_api_key_get.py +0 -127
  35. orca_sdk/_generated_api_client/api/classification_model/__init__.py +0 -0
  36. orca_sdk/_generated_api_client/api/classification_model/create_classification_model_gpu_classification_model_post.py +0 -170
  37. orca_sdk/_generated_api_client/api/classification_model/delete_classification_model_classification_model_name_or_id_delete.py +0 -156
  38. 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
  39. orca_sdk/_generated_api_client/api/classification_model/evaluate_classification_model_classification_model_model_name_or_id_evaluation_post.py +0 -183
  40. orca_sdk/_generated_api_client/api/classification_model/get_classification_model_classification_model_name_or_id_get.py +0 -156
  41. 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
  42. orca_sdk/_generated_api_client/api/classification_model/list_classification_model_evaluations_classification_model_model_name_or_id_evaluation_get.py +0 -161
  43. orca_sdk/_generated_api_client/api/classification_model/list_classification_models_classification_model_get.py +0 -127
  44. orca_sdk/_generated_api_client/api/classification_model/predict_label_gpu_classification_model_name_or_id_prediction_post.py +0 -190
  45. orca_sdk/_generated_api_client/api/classification_model/update_classification_model_classification_model_name_or_id_patch.py +0 -183
  46. orca_sdk/_generated_api_client/api/datasource/__init__.py +0 -0
  47. orca_sdk/_generated_api_client/api/datasource/create_datasource_datasource_post.py +0 -167
  48. orca_sdk/_generated_api_client/api/datasource/create_embedding_evaluation_datasource_name_or_id_embedding_evaluation_post.py +0 -183
  49. orca_sdk/_generated_api_client/api/datasource/delete_datasource_datasource_name_or_id_delete.py +0 -156
  50. orca_sdk/_generated_api_client/api/datasource/download_datasource_datasource_name_or_id_download_get.py +0 -172
  51. orca_sdk/_generated_api_client/api/datasource/get_datasource_datasource_name_or_id_get.py +0 -156
  52. orca_sdk/_generated_api_client/api/datasource/get_embedding_evaluation_datasource_name_or_id_embedding_evaluation_task_id_get.py +0 -169
  53. orca_sdk/_generated_api_client/api/datasource/list_datasources_datasource_get.py +0 -127
  54. orca_sdk/_generated_api_client/api/datasource/list_embedding_evaluations_datasource_name_or_id_embedding_evaluation_get.py +0 -235
  55. orca_sdk/_generated_api_client/api/default/__init__.py +0 -0
  56. orca_sdk/_generated_api_client/api/default/healthcheck_get.py +0 -118
  57. orca_sdk/_generated_api_client/api/default/healthcheck_gpu_get.py +0 -118
  58. orca_sdk/_generated_api_client/api/finetuned_embedding_model/__init__.py +0 -0
  59. orca_sdk/_generated_api_client/api/finetuned_embedding_model/create_finetuned_embedding_model_finetuned_embedding_model_post.py +0 -168
  60. orca_sdk/_generated_api_client/api/finetuned_embedding_model/delete_finetuned_embedding_model_finetuned_embedding_model_name_or_id_delete.py +0 -156
  61. 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
  62. orca_sdk/_generated_api_client/api/finetuned_embedding_model/get_finetuned_embedding_model_finetuned_embedding_model_name_or_id_get.py +0 -156
  63. orca_sdk/_generated_api_client/api/finetuned_embedding_model/list_finetuned_embedding_models_finetuned_embedding_model_get.py +0 -127
  64. orca_sdk/_generated_api_client/api/memoryset/__init__.py +0 -0
  65. orca_sdk/_generated_api_client/api/memoryset/analyze_memoryset_memoryset_name_or_id_analysis_post.py +0 -183
  66. orca_sdk/_generated_api_client/api/memoryset/batch_delete_memoryset_batch_delete_memoryset_post.py +0 -168
  67. orca_sdk/_generated_api_client/api/memoryset/clone_memoryset_memoryset_name_or_id_clone_post.py +0 -181
  68. orca_sdk/_generated_api_client/api/memoryset/create_memoryset_memoryset_post.py +0 -168
  69. orca_sdk/_generated_api_client/api/memoryset/delete_memories_memoryset_name_or_id_memories_delete_post.py +0 -181
  70. orca_sdk/_generated_api_client/api/memoryset/delete_memory_memoryset_name_or_id_memory_memory_id_delete.py +0 -167
  71. orca_sdk/_generated_api_client/api/memoryset/delete_memoryset_memoryset_name_or_id_delete.py +0 -156
  72. orca_sdk/_generated_api_client/api/memoryset/get_analysis_memoryset_name_or_id_analysis_analysis_task_id_get.py +0 -169
  73. orca_sdk/_generated_api_client/api/memoryset/get_memories_memoryset_name_or_id_memories_get_post.py +0 -210
  74. orca_sdk/_generated_api_client/api/memoryset/get_memory_memoryset_name_or_id_memory_memory_id_get.py +0 -186
  75. orca_sdk/_generated_api_client/api/memoryset/get_memoryset_memoryset_name_or_id_get.py +0 -156
  76. orca_sdk/_generated_api_client/api/memoryset/insert_memories_gpu_memoryset_name_or_id_memory_post.py +0 -188
  77. orca_sdk/_generated_api_client/api/memoryset/list_analyses_memoryset_name_or_id_analysis_get.py +0 -235
  78. orca_sdk/_generated_api_client/api/memoryset/list_memorysets_memoryset_get.py +0 -180
  79. orca_sdk/_generated_api_client/api/memoryset/memoryset_lookup_gpu_memoryset_name_or_id_lookup_post.py +0 -212
  80. orca_sdk/_generated_api_client/api/memoryset/potential_duplicate_groups_memoryset_name_or_id_potential_duplicate_groups_get.py +0 -195
  81. orca_sdk/_generated_api_client/api/memoryset/query_memoryset_memoryset_name_or_id_memories_post.py +0 -210
  82. orca_sdk/_generated_api_client/api/memoryset/suggest_cascading_edits_memoryset_name_or_id_memory_memory_id_cascading_edits_post.py +0 -233
  83. orca_sdk/_generated_api_client/api/memoryset/update_memories_gpu_memoryset_name_or_id_memories_patch.py +0 -216
  84. orca_sdk/_generated_api_client/api/memoryset/update_memory_gpu_memoryset_name_or_id_memory_patch.py +0 -205
  85. orca_sdk/_generated_api_client/api/memoryset/update_memoryset_memoryset_name_or_id_patch.py +0 -183
  86. orca_sdk/_generated_api_client/api/predictive_model/__init__.py +0 -0
  87. orca_sdk/_generated_api_client/api/predictive_model/list_predictive_models_predictive_model_get.py +0 -150
  88. orca_sdk/_generated_api_client/api/pretrained_embedding_model/__init__.py +0 -0
  89. orca_sdk/_generated_api_client/api/pretrained_embedding_model/embed_with_pretrained_model_gpu_pretrained_embedding_model_model_name_embedding_post.py +0 -192
  90. orca_sdk/_generated_api_client/api/pretrained_embedding_model/get_pretrained_embedding_model_pretrained_embedding_model_model_name_get.py +0 -161
  91. orca_sdk/_generated_api_client/api/pretrained_embedding_model/list_pretrained_embedding_models_pretrained_embedding_model_get.py +0 -127
  92. orca_sdk/_generated_api_client/api/regression_model/__init__.py +0 -0
  93. orca_sdk/_generated_api_client/api/regression_model/create_regression_model_gpu_regression_model_post.py +0 -170
  94. 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
  95. orca_sdk/_generated_api_client/api/regression_model/delete_regression_model_regression_model_name_or_id_delete.py +0 -154
  96. orca_sdk/_generated_api_client/api/regression_model/evaluate_regression_model_regression_model_model_name_or_id_evaluation_post.py +0 -183
  97. 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
  98. orca_sdk/_generated_api_client/api/regression_model/get_regression_model_regression_model_name_or_id_get.py +0 -156
  99. orca_sdk/_generated_api_client/api/regression_model/list_regression_model_evaluations_regression_model_model_name_or_id_evaluation_get.py +0 -161
  100. orca_sdk/_generated_api_client/api/regression_model/list_regression_models_regression_model_get.py +0 -127
  101. orca_sdk/_generated_api_client/api/regression_model/predict_score_gpu_regression_model_name_or_id_prediction_post.py +0 -190
  102. orca_sdk/_generated_api_client/api/regression_model/update_regression_model_regression_model_name_or_id_patch.py +0 -183
  103. orca_sdk/_generated_api_client/api/task/__init__.py +0 -0
  104. orca_sdk/_generated_api_client/api/task/abort_task_task_task_id_abort_delete.py +0 -154
  105. orca_sdk/_generated_api_client/api/task/get_task_status_task_task_id_status_get.py +0 -156
  106. orca_sdk/_generated_api_client/api/task/get_task_task_task_id_get.py +0 -156
  107. orca_sdk/_generated_api_client/api/task/list_tasks_task_get.py +0 -293
  108. orca_sdk/_generated_api_client/api/telemetry/__init__.py +0 -0
  109. orca_sdk/_generated_api_client/api/telemetry/count_predictions_telemetry_prediction_count_post.py +0 -168
  110. orca_sdk/_generated_api_client/api/telemetry/drop_feedback_category_with_data_telemetry_feedback_category_name_or_id_delete.py +0 -162
  111. orca_sdk/_generated_api_client/api/telemetry/explain_prediction_telemetry_prediction_prediction_id_explanation_get.py +0 -182
  112. orca_sdk/_generated_api_client/api/telemetry/get_feedback_category_telemetry_feedback_category_name_or_id_get.py +0 -156
  113. orca_sdk/_generated_api_client/api/telemetry/get_prediction_telemetry_prediction_prediction_id_get.py +0 -180
  114. orca_sdk/_generated_api_client/api/telemetry/list_feedback_categories_telemetry_feedback_category_get.py +0 -127
  115. orca_sdk/_generated_api_client/api/telemetry/list_memories_with_feedback_telemetry_memories_post.py +0 -198
  116. orca_sdk/_generated_api_client/api/telemetry/list_predictions_telemetry_prediction_post.py +0 -198
  117. orca_sdk/_generated_api_client/api/telemetry/record_prediction_feedback_telemetry_prediction_feedback_put.py +0 -171
  118. orca_sdk/_generated_api_client/api/telemetry/update_prediction_telemetry_prediction_prediction_id_patch.py +0 -181
  119. orca_sdk/_generated_api_client/client.py +0 -216
  120. orca_sdk/_generated_api_client/errors.py +0 -38
  121. orca_sdk/_generated_api_client/models/__init__.py +0 -295
  122. orca_sdk/_generated_api_client/models/analyze_neighbor_labels_result.py +0 -116
  123. orca_sdk/_generated_api_client/models/api_key_metadata.py +0 -137
  124. orca_sdk/_generated_api_client/models/api_key_metadata_scope_item.py +0 -9
  125. orca_sdk/_generated_api_client/models/base_label_prediction_result.py +0 -130
  126. orca_sdk/_generated_api_client/models/base_model.py +0 -55
  127. orca_sdk/_generated_api_client/models/base_score_prediction_result.py +0 -108
  128. orca_sdk/_generated_api_client/models/body_create_datasource_datasource_post.py +0 -207
  129. orca_sdk/_generated_api_client/models/cascade_edit_suggestions_request.py +0 -154
  130. orca_sdk/_generated_api_client/models/cascading_edit_suggestion.py +0 -92
  131. orca_sdk/_generated_api_client/models/classification_evaluation_request.py +0 -148
  132. orca_sdk/_generated_api_client/models/classification_metrics.py +0 -259
  133. orca_sdk/_generated_api_client/models/classification_model_metadata.py +0 -213
  134. orca_sdk/_generated_api_client/models/classification_prediction_request.py +0 -220
  135. orca_sdk/_generated_api_client/models/clone_memoryset_request.py +0 -170
  136. orca_sdk/_generated_api_client/models/cluster_metrics.py +0 -78
  137. orca_sdk/_generated_api_client/models/column_info.py +0 -145
  138. orca_sdk/_generated_api_client/models/column_type.py +0 -14
  139. orca_sdk/_generated_api_client/models/constraint_violation_error_response.py +0 -80
  140. orca_sdk/_generated_api_client/models/count_predictions_request.py +0 -195
  141. orca_sdk/_generated_api_client/models/create_api_key_request.py +0 -120
  142. orca_sdk/_generated_api_client/models/create_api_key_request_scope_item.py +0 -9
  143. orca_sdk/_generated_api_client/models/create_api_key_response.py +0 -145
  144. orca_sdk/_generated_api_client/models/create_api_key_response_scope_item.py +0 -9
  145. orca_sdk/_generated_api_client/models/create_classification_model_request.py +0 -197
  146. orca_sdk/_generated_api_client/models/create_memoryset_request.py +0 -325
  147. orca_sdk/_generated_api_client/models/create_memoryset_request_index_params.py +0 -66
  148. orca_sdk/_generated_api_client/models/create_memoryset_request_index_type.py +0 -13
  149. orca_sdk/_generated_api_client/models/create_regression_model_request.py +0 -137
  150. orca_sdk/_generated_api_client/models/datasource_metadata.py +0 -156
  151. orca_sdk/_generated_api_client/models/delete_memories_request.py +0 -70
  152. orca_sdk/_generated_api_client/models/delete_memorysets_request.py +0 -70
  153. orca_sdk/_generated_api_client/models/embed_request.py +0 -135
  154. orca_sdk/_generated_api_client/models/embedding_evaluation_payload.py +0 -187
  155. orca_sdk/_generated_api_client/models/embedding_evaluation_request.py +0 -179
  156. orca_sdk/_generated_api_client/models/embedding_evaluation_response.py +0 -158
  157. orca_sdk/_generated_api_client/models/embedding_evaluation_result.py +0 -86
  158. orca_sdk/_generated_api_client/models/embedding_finetuning_method.py +0 -9
  159. orca_sdk/_generated_api_client/models/embedding_model_result.py +0 -114
  160. orca_sdk/_generated_api_client/models/evaluation_response.py +0 -153
  161. orca_sdk/_generated_api_client/models/evaluation_response_classification_metrics.py +0 -140
  162. orca_sdk/_generated_api_client/models/evaluation_response_regression_metrics.py +0 -140
  163. orca_sdk/_generated_api_client/models/feedback_metrics.py +0 -85
  164. orca_sdk/_generated_api_client/models/feedback_type.py +0 -9
  165. orca_sdk/_generated_api_client/models/filter_item.py +0 -231
  166. orca_sdk/_generated_api_client/models/filter_item_field_type_0_item.py +0 -17
  167. orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_1.py +0 -20
  168. orca_sdk/_generated_api_client/models/filter_item_op.py +0 -16
  169. orca_sdk/_generated_api_client/models/finetune_embedding_model_request.py +0 -259
  170. orca_sdk/_generated_api_client/models/finetune_embedding_model_request_training_args.py +0 -66
  171. orca_sdk/_generated_api_client/models/finetuned_embedding_model_metadata.py +0 -166
  172. orca_sdk/_generated_api_client/models/get_memories_request.py +0 -70
  173. orca_sdk/_generated_api_client/models/http_validation_error.py +0 -86
  174. orca_sdk/_generated_api_client/models/internal_server_error_response.py +0 -80
  175. orca_sdk/_generated_api_client/models/label_class_metrics.py +0 -108
  176. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup.py +0 -210
  177. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup_metadata.py +0 -68
  178. orca_sdk/_generated_api_client/models/label_prediction_with_memories_and_feedback.py +0 -288
  179. orca_sdk/_generated_api_client/models/labeled_memory.py +0 -186
  180. orca_sdk/_generated_api_client/models/labeled_memory_insert.py +0 -128
  181. orca_sdk/_generated_api_client/models/labeled_memory_insert_metadata.py +0 -68
  182. orca_sdk/_generated_api_client/models/labeled_memory_lookup.py +0 -194
  183. orca_sdk/_generated_api_client/models/labeled_memory_lookup_metadata.py +0 -68
  184. orca_sdk/_generated_api_client/models/labeled_memory_metadata.py +0 -68
  185. orca_sdk/_generated_api_client/models/labeled_memory_metrics.py +0 -246
  186. orca_sdk/_generated_api_client/models/labeled_memory_update.py +0 -171
  187. orca_sdk/_generated_api_client/models/labeled_memory_update_metadata_type_0.py +0 -68
  188. orca_sdk/_generated_api_client/models/labeled_memory_with_feedback_metrics.py +0 -207
  189. orca_sdk/_generated_api_client/models/labeled_memory_with_feedback_metrics_feedback_metrics.py +0 -68
  190. orca_sdk/_generated_api_client/models/labeled_memory_with_feedback_metrics_metadata.py +0 -68
  191. orca_sdk/_generated_api_client/models/list_memories_request.py +0 -104
  192. orca_sdk/_generated_api_client/models/list_predictions_request.py +0 -319
  193. orca_sdk/_generated_api_client/models/lookup_request.py +0 -81
  194. orca_sdk/_generated_api_client/models/lookup_score_metrics.py +0 -94
  195. orca_sdk/_generated_api_client/models/memory_metrics.py +0 -165
  196. orca_sdk/_generated_api_client/models/memory_type.py +0 -9
  197. orca_sdk/_generated_api_client/models/memoryset_analysis_configs.py +0 -212
  198. orca_sdk/_generated_api_client/models/memoryset_analysis_request.py +0 -105
  199. orca_sdk/_generated_api_client/models/memoryset_analysis_response.py +0 -182
  200. orca_sdk/_generated_api_client/models/memoryset_cluster_analysis_config.py +0 -202
  201. orca_sdk/_generated_api_client/models/memoryset_cluster_analysis_config_clustering_method.py +0 -9
  202. orca_sdk/_generated_api_client/models/memoryset_cluster_analysis_config_partitioning_method.py +0 -10
  203. orca_sdk/_generated_api_client/models/memoryset_cluster_metrics.py +0 -100
  204. orca_sdk/_generated_api_client/models/memoryset_duplicate_analysis_config.py +0 -70
  205. orca_sdk/_generated_api_client/models/memoryset_duplicate_metrics.py +0 -70
  206. orca_sdk/_generated_api_client/models/memoryset_label_analysis_config.py +0 -70
  207. orca_sdk/_generated_api_client/models/memoryset_label_metrics.py +0 -116
  208. orca_sdk/_generated_api_client/models/memoryset_metadata.py +0 -291
  209. orca_sdk/_generated_api_client/models/memoryset_metadata_index_params.py +0 -55
  210. orca_sdk/_generated_api_client/models/memoryset_metadata_index_type.py +0 -13
  211. orca_sdk/_generated_api_client/models/memoryset_metrics.py +0 -232
  212. orca_sdk/_generated_api_client/models/memoryset_neighbor_analysis_config.py +0 -83
  213. orca_sdk/_generated_api_client/models/memoryset_neighbor_metrics.py +0 -76
  214. orca_sdk/_generated_api_client/models/memoryset_neighbor_metrics_lookup_score_metrics.py +0 -68
  215. orca_sdk/_generated_api_client/models/memoryset_projection_analysis_config.py +0 -79
  216. orca_sdk/_generated_api_client/models/memoryset_projection_metrics.py +0 -55
  217. orca_sdk/_generated_api_client/models/memoryset_update.py +0 -101
  218. orca_sdk/_generated_api_client/models/not_found_error_response.py +0 -100
  219. orca_sdk/_generated_api_client/models/not_found_error_response_resource_type_0.py +0 -22
  220. orca_sdk/_generated_api_client/models/paginated_union_labeled_memory_with_feedback_metrics_scored_memory_with_feedback_metrics.py +0 -135
  221. orca_sdk/_generated_api_client/models/pr_curve.py +0 -86
  222. orca_sdk/_generated_api_client/models/prediction_feedback.py +0 -157
  223. orca_sdk/_generated_api_client/models/prediction_feedback_category.py +0 -115
  224. orca_sdk/_generated_api_client/models/prediction_feedback_request.py +0 -122
  225. orca_sdk/_generated_api_client/models/prediction_feedback_result.py +0 -102
  226. orca_sdk/_generated_api_client/models/prediction_sort_item_item_type_0.py +0 -10
  227. orca_sdk/_generated_api_client/models/prediction_sort_item_item_type_1.py +0 -9
  228. orca_sdk/_generated_api_client/models/predictive_model_update.py +0 -91
  229. orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py +0 -107
  230. orca_sdk/_generated_api_client/models/pretrained_embedding_model_name.py +0 -17
  231. orca_sdk/_generated_api_client/models/rac_head_type.py +0 -11
  232. orca_sdk/_generated_api_client/models/rar_head_type.py +0 -8
  233. orca_sdk/_generated_api_client/models/regression_evaluation_request.py +0 -148
  234. orca_sdk/_generated_api_client/models/regression_metrics.py +0 -172
  235. orca_sdk/_generated_api_client/models/regression_model_metadata.py +0 -177
  236. orca_sdk/_generated_api_client/models/regression_prediction_request.py +0 -195
  237. orca_sdk/_generated_api_client/models/roc_curve.py +0 -86
  238. orca_sdk/_generated_api_client/models/score_prediction_memory_lookup.py +0 -196
  239. orca_sdk/_generated_api_client/models/score_prediction_memory_lookup_metadata.py +0 -68
  240. orca_sdk/_generated_api_client/models/score_prediction_with_memories_and_feedback.py +0 -252
  241. orca_sdk/_generated_api_client/models/scored_memory.py +0 -172
  242. orca_sdk/_generated_api_client/models/scored_memory_insert.py +0 -128
  243. orca_sdk/_generated_api_client/models/scored_memory_insert_metadata.py +0 -68
  244. orca_sdk/_generated_api_client/models/scored_memory_lookup.py +0 -180
  245. orca_sdk/_generated_api_client/models/scored_memory_lookup_metadata.py +0 -68
  246. orca_sdk/_generated_api_client/models/scored_memory_metadata.py +0 -68
  247. orca_sdk/_generated_api_client/models/scored_memory_update.py +0 -171
  248. orca_sdk/_generated_api_client/models/scored_memory_update_metadata_type_0.py +0 -68
  249. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics.py +0 -193
  250. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics_feedback_metrics.py +0 -68
  251. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics_metadata.py +0 -68
  252. orca_sdk/_generated_api_client/models/service_unavailable_error_response.py +0 -80
  253. orca_sdk/_generated_api_client/models/task.py +0 -198
  254. orca_sdk/_generated_api_client/models/task_status.py +0 -14
  255. orca_sdk/_generated_api_client/models/task_status_info.py +0 -133
  256. orca_sdk/_generated_api_client/models/telemetry_field_type_0_item_type_2.py +0 -9
  257. orca_sdk/_generated_api_client/models/telemetry_filter_item.py +0 -205
  258. orca_sdk/_generated_api_client/models/telemetry_filter_item_op.py +0 -15
  259. orca_sdk/_generated_api_client/models/telemetry_memories_request.py +0 -181
  260. orca_sdk/_generated_api_client/models/telemetry_sort_options.py +0 -173
  261. orca_sdk/_generated_api_client/models/telemetry_sort_options_direction.py +0 -9
  262. orca_sdk/_generated_api_client/models/unauthenticated_error_response.py +0 -72
  263. orca_sdk/_generated_api_client/models/unauthorized_error_response.py +0 -80
  264. orca_sdk/_generated_api_client/models/update_prediction_request.py +0 -133
  265. orca_sdk/_generated_api_client/models/validation_error.py +0 -99
  266. orca_sdk/_generated_api_client/py.typed +0 -1
  267. orca_sdk/_generated_api_client/types.py +0 -56
  268. orca_sdk-0.0.96.dist-info/RECORD +0 -278
  269. {orca_sdk-0.0.96.dist-info → orca_sdk-0.0.98.dist-info}/WHEEL +0 -0
@@ -1,212 +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_cluster_analysis_config import MemorysetClusterAnalysisConfig
21
- from ..models.memoryset_duplicate_analysis_config import MemorysetDuplicateAnalysisConfig
22
- from ..models.memoryset_label_analysis_config import MemorysetLabelAnalysisConfig
23
- from ..models.memoryset_neighbor_analysis_config import MemorysetNeighborAnalysisConfig
24
- from ..models.memoryset_projection_analysis_config import MemorysetProjectionAnalysisConfig
25
-
26
-
27
- T = TypeVar("T", bound="MemorysetAnalysisConfigs")
28
-
29
-
30
- @_attrs_define
31
- class MemorysetAnalysisConfigs:
32
- """
33
- Attributes:
34
- neighbor (Union['MemorysetNeighborAnalysisConfig', None, Unset]):
35
- label (Union['MemorysetLabelAnalysisConfig', None, Unset]):
36
- duplicate (Union['MemorysetDuplicateAnalysisConfig', None, Unset]):
37
- projection (Union['MemorysetProjectionAnalysisConfig', None, Unset]):
38
- cluster (Union['MemorysetClusterAnalysisConfig', None, Unset]):
39
- """
40
-
41
- neighbor: Union["MemorysetNeighborAnalysisConfig", None, Unset] = UNSET
42
- label: Union["MemorysetLabelAnalysisConfig", None, Unset] = UNSET
43
- duplicate: Union["MemorysetDuplicateAnalysisConfig", None, Unset] = UNSET
44
- projection: Union["MemorysetProjectionAnalysisConfig", None, Unset] = UNSET
45
- cluster: Union["MemorysetClusterAnalysisConfig", None, Unset] = UNSET
46
-
47
- def to_dict(self) -> dict[str, Any]:
48
- from ..models.memoryset_cluster_analysis_config import MemorysetClusterAnalysisConfig
49
- from ..models.memoryset_duplicate_analysis_config import MemorysetDuplicateAnalysisConfig
50
- from ..models.memoryset_label_analysis_config import MemorysetLabelAnalysisConfig
51
- from ..models.memoryset_neighbor_analysis_config import MemorysetNeighborAnalysisConfig
52
- from ..models.memoryset_projection_analysis_config import MemorysetProjectionAnalysisConfig
53
-
54
- neighbor: Union[Dict[str, Any], None, Unset]
55
- if isinstance(self.neighbor, Unset):
56
- neighbor = UNSET
57
- elif isinstance(self.neighbor, MemorysetNeighborAnalysisConfig):
58
- neighbor = self.neighbor.to_dict()
59
- else:
60
- neighbor = self.neighbor
61
-
62
- label: Union[Dict[str, Any], None, Unset]
63
- if isinstance(self.label, Unset):
64
- label = UNSET
65
- elif isinstance(self.label, MemorysetLabelAnalysisConfig):
66
- label = self.label.to_dict()
67
- else:
68
- label = self.label
69
-
70
- duplicate: Union[Dict[str, Any], None, Unset]
71
- if isinstance(self.duplicate, Unset):
72
- duplicate = UNSET
73
- elif isinstance(self.duplicate, MemorysetDuplicateAnalysisConfig):
74
- duplicate = self.duplicate.to_dict()
75
- else:
76
- duplicate = self.duplicate
77
-
78
- projection: Union[Dict[str, Any], None, Unset]
79
- if isinstance(self.projection, Unset):
80
- projection = UNSET
81
- elif isinstance(self.projection, MemorysetProjectionAnalysisConfig):
82
- projection = self.projection.to_dict()
83
- else:
84
- projection = self.projection
85
-
86
- cluster: Union[Dict[str, Any], None, Unset]
87
- if isinstance(self.cluster, Unset):
88
- cluster = UNSET
89
- elif isinstance(self.cluster, MemorysetClusterAnalysisConfig):
90
- cluster = self.cluster.to_dict()
91
- else:
92
- cluster = self.cluster
93
-
94
- field_dict: dict[str, Any] = {}
95
- field_dict.update({})
96
- if neighbor is not UNSET:
97
- field_dict["neighbor"] = neighbor
98
- if label is not UNSET:
99
- field_dict["label"] = label
100
- if duplicate is not UNSET:
101
- field_dict["duplicate"] = duplicate
102
- if projection is not UNSET:
103
- field_dict["projection"] = projection
104
- if cluster is not UNSET:
105
- field_dict["cluster"] = cluster
106
-
107
- return field_dict
108
-
109
- @classmethod
110
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
111
- from ..models.memoryset_cluster_analysis_config import MemorysetClusterAnalysisConfig
112
- from ..models.memoryset_duplicate_analysis_config import MemorysetDuplicateAnalysisConfig
113
- from ..models.memoryset_label_analysis_config import MemorysetLabelAnalysisConfig
114
- from ..models.memoryset_neighbor_analysis_config import MemorysetNeighborAnalysisConfig
115
- from ..models.memoryset_projection_analysis_config import MemorysetProjectionAnalysisConfig
116
-
117
- d = src_dict.copy()
118
-
119
- def _parse_neighbor(data: object) -> Union["MemorysetNeighborAnalysisConfig", None, Unset]:
120
- if data is None:
121
- return data
122
- if isinstance(data, Unset):
123
- return data
124
- try:
125
- if not isinstance(data, dict):
126
- raise TypeError()
127
- neighbor_type_0 = MemorysetNeighborAnalysisConfig.from_dict(data)
128
-
129
- return neighbor_type_0
130
- except: # noqa: E722
131
- pass
132
- return cast(Union["MemorysetNeighborAnalysisConfig", None, Unset], data)
133
-
134
- neighbor = _parse_neighbor(d.pop("neighbor", UNSET))
135
-
136
- def _parse_label(data: object) -> Union["MemorysetLabelAnalysisConfig", None, Unset]:
137
- if data is None:
138
- return data
139
- if isinstance(data, Unset):
140
- return data
141
- try:
142
- if not isinstance(data, dict):
143
- raise TypeError()
144
- label_type_0 = MemorysetLabelAnalysisConfig.from_dict(data)
145
-
146
- return label_type_0
147
- except: # noqa: E722
148
- pass
149
- return cast(Union["MemorysetLabelAnalysisConfig", None, Unset], data)
150
-
151
- label = _parse_label(d.pop("label", UNSET))
152
-
153
- def _parse_duplicate(data: object) -> Union["MemorysetDuplicateAnalysisConfig", None, Unset]:
154
- if data is None:
155
- return data
156
- if isinstance(data, Unset):
157
- return data
158
- try:
159
- if not isinstance(data, dict):
160
- raise TypeError()
161
- duplicate_type_0 = MemorysetDuplicateAnalysisConfig.from_dict(data)
162
-
163
- return duplicate_type_0
164
- except: # noqa: E722
165
- pass
166
- return cast(Union["MemorysetDuplicateAnalysisConfig", None, Unset], data)
167
-
168
- duplicate = _parse_duplicate(d.pop("duplicate", UNSET))
169
-
170
- def _parse_projection(data: object) -> Union["MemorysetProjectionAnalysisConfig", None, Unset]:
171
- if data is None:
172
- return data
173
- if isinstance(data, Unset):
174
- return data
175
- try:
176
- if not isinstance(data, dict):
177
- raise TypeError()
178
- projection_type_0 = MemorysetProjectionAnalysisConfig.from_dict(data)
179
-
180
- return projection_type_0
181
- except: # noqa: E722
182
- pass
183
- return cast(Union["MemorysetProjectionAnalysisConfig", None, Unset], data)
184
-
185
- projection = _parse_projection(d.pop("projection", UNSET))
186
-
187
- def _parse_cluster(data: object) -> Union["MemorysetClusterAnalysisConfig", None, Unset]:
188
- if data is None:
189
- return data
190
- if isinstance(data, Unset):
191
- return data
192
- try:
193
- if not isinstance(data, dict):
194
- raise TypeError()
195
- cluster_type_0 = MemorysetClusterAnalysisConfig.from_dict(data)
196
-
197
- return cluster_type_0
198
- except: # noqa: E722
199
- pass
200
- return cast(Union["MemorysetClusterAnalysisConfig", None, Unset], data)
201
-
202
- cluster = _parse_cluster(d.pop("cluster", UNSET))
203
-
204
- memoryset_analysis_configs = cls(
205
- neighbor=neighbor,
206
- label=label,
207
- duplicate=duplicate,
208
- projection=projection,
209
- cluster=cluster,
210
- )
211
-
212
- 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
@@ -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,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