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,252 +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 typing import TYPE_CHECKING, Any, List, Type, TypeVar, Union, cast
15
-
16
- from attrs import define as _attrs_define
17
- from attrs import field as _attrs_field
18
- from dateutil.parser import isoparse
19
-
20
- if TYPE_CHECKING:
21
- from ..models.prediction_feedback import PredictionFeedback
22
- from ..models.score_prediction_memory_lookup import ScorePredictionMemoryLookup
23
-
24
-
25
- T = TypeVar("T", bound="ScorePredictionWithMemoriesAndFeedback")
26
-
27
-
28
- @_attrs_define
29
- class ScorePredictionWithMemoriesAndFeedback:
30
- """
31
- Attributes:
32
- prediction_id (str):
33
- confidence (float):
34
- anomaly_score (Union[None, float]):
35
- score (float):
36
- timestamp (datetime.datetime):
37
- input_value (str):
38
- input_embedding (List[float]):
39
- expected_score (Union[None, float]):
40
- memories (List['ScorePredictionMemoryLookup']):
41
- org_id (str):
42
- memoryset_id (str):
43
- model_id (str):
44
- updated_at (datetime.datetime):
45
- tags (List[str]):
46
- explanation (Union[None, str]):
47
- memory_id (Union[None, str]):
48
- feedbacks (List['PredictionFeedback']):
49
- """
50
-
51
- prediction_id: str
52
- confidence: float
53
- anomaly_score: Union[None, float]
54
- score: float
55
- timestamp: datetime.datetime
56
- input_value: str
57
- input_embedding: List[float]
58
- expected_score: Union[None, float]
59
- memories: List["ScorePredictionMemoryLookup"]
60
- org_id: str
61
- memoryset_id: str
62
- model_id: str
63
- updated_at: datetime.datetime
64
- tags: List[str]
65
- explanation: Union[None, str]
66
- memory_id: Union[None, str]
67
- feedbacks: List["PredictionFeedback"]
68
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
69
-
70
- def to_dict(self) -> dict[str, Any]:
71
- prediction_id = self.prediction_id
72
-
73
- confidence = self.confidence
74
-
75
- anomaly_score: Union[None, float]
76
- anomaly_score = self.anomaly_score
77
-
78
- score = self.score
79
-
80
- timestamp = self.timestamp.isoformat()
81
-
82
- input_value: str
83
- input_value = self.input_value
84
-
85
- input_embedding = self.input_embedding
86
-
87
- expected_score: Union[None, float]
88
- expected_score = self.expected_score
89
-
90
- memories = []
91
- for memories_item_data in self.memories:
92
- memories_item = memories_item_data.to_dict()
93
- memories.append(memories_item)
94
-
95
- org_id = self.org_id
96
-
97
- memoryset_id = self.memoryset_id
98
-
99
- model_id = self.model_id
100
-
101
- updated_at = self.updated_at.isoformat()
102
-
103
- tags = self.tags
104
-
105
- explanation: Union[None, str]
106
- explanation = self.explanation
107
-
108
- memory_id: Union[None, str]
109
- memory_id = self.memory_id
110
-
111
- feedbacks = []
112
- for feedbacks_item_data in self.feedbacks:
113
- feedbacks_item = feedbacks_item_data.to_dict()
114
- feedbacks.append(feedbacks_item)
115
-
116
- field_dict: dict[str, Any] = {}
117
- field_dict.update(self.additional_properties)
118
- field_dict.update(
119
- {
120
- "prediction_id": prediction_id,
121
- "confidence": confidence,
122
- "anomaly_score": anomaly_score,
123
- "score": score,
124
- "timestamp": timestamp,
125
- "input_value": input_value,
126
- "input_embedding": input_embedding,
127
- "expected_score": expected_score,
128
- "memories": memories,
129
- "org_id": org_id,
130
- "memoryset_id": memoryset_id,
131
- "model_id": model_id,
132
- "updated_at": updated_at,
133
- "tags": tags,
134
- "explanation": explanation,
135
- "memory_id": memory_id,
136
- "feedbacks": feedbacks,
137
- }
138
- )
139
-
140
- return field_dict
141
-
142
- @classmethod
143
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
144
- from ..models.prediction_feedback import PredictionFeedback
145
- from ..models.score_prediction_memory_lookup import ScorePredictionMemoryLookup
146
-
147
- d = src_dict.copy()
148
- prediction_id = d.pop("prediction_id")
149
-
150
- confidence = d.pop("confidence")
151
-
152
- def _parse_anomaly_score(data: object) -> Union[None, float]:
153
- if data is None:
154
- return data
155
- return cast(Union[None, float], data)
156
-
157
- anomaly_score = _parse_anomaly_score(d.pop("anomaly_score"))
158
-
159
- score = d.pop("score")
160
-
161
- timestamp = isoparse(d.pop("timestamp"))
162
-
163
- def _parse_input_value(data: object) -> str:
164
- return cast(str, data)
165
-
166
- input_value = _parse_input_value(d.pop("input_value"))
167
-
168
- input_embedding = cast(List[float], d.pop("input_embedding"))
169
-
170
- def _parse_expected_score(data: object) -> Union[None, float]:
171
- if data is None:
172
- return data
173
- return cast(Union[None, float], data)
174
-
175
- expected_score = _parse_expected_score(d.pop("expected_score"))
176
-
177
- memories = []
178
- _memories = d.pop("memories")
179
- for memories_item_data in _memories:
180
- memories_item = ScorePredictionMemoryLookup.from_dict(memories_item_data)
181
-
182
- memories.append(memories_item)
183
-
184
- org_id = d.pop("org_id")
185
-
186
- memoryset_id = d.pop("memoryset_id")
187
-
188
- model_id = d.pop("model_id")
189
-
190
- updated_at = isoparse(d.pop("updated_at"))
191
-
192
- tags = cast(List[str], d.pop("tags"))
193
-
194
- def _parse_explanation(data: object) -> Union[None, str]:
195
- if data is None:
196
- return data
197
- return cast(Union[None, str], data)
198
-
199
- explanation = _parse_explanation(d.pop("explanation"))
200
-
201
- def _parse_memory_id(data: object) -> Union[None, str]:
202
- if data is None:
203
- return data
204
- return cast(Union[None, str], data)
205
-
206
- memory_id = _parse_memory_id(d.pop("memory_id"))
207
-
208
- feedbacks = []
209
- _feedbacks = d.pop("feedbacks")
210
- for feedbacks_item_data in _feedbacks:
211
- feedbacks_item = PredictionFeedback.from_dict(feedbacks_item_data)
212
-
213
- feedbacks.append(feedbacks_item)
214
-
215
- score_prediction_with_memories_and_feedback = cls(
216
- prediction_id=prediction_id,
217
- confidence=confidence,
218
- anomaly_score=anomaly_score,
219
- score=score,
220
- timestamp=timestamp,
221
- input_value=input_value,
222
- input_embedding=input_embedding,
223
- expected_score=expected_score,
224
- memories=memories,
225
- org_id=org_id,
226
- memoryset_id=memoryset_id,
227
- model_id=model_id,
228
- updated_at=updated_at,
229
- tags=tags,
230
- explanation=explanation,
231
- memory_id=memory_id,
232
- feedbacks=feedbacks,
233
- )
234
-
235
- score_prediction_with_memories_and_feedback.additional_properties = d
236
- return score_prediction_with_memories_and_feedback
237
-
238
- @property
239
- def additional_keys(self) -> list[str]:
240
- return list(self.additional_properties.keys())
241
-
242
- def __getitem__(self, key: str) -> Any:
243
- return self.additional_properties[key]
244
-
245
- def __setitem__(self, key: str, value: Any) -> None:
246
- self.additional_properties[key] = value
247
-
248
- def __delitem__(self, key: str) -> None:
249
- del self.additional_properties[key]
250
-
251
- def __contains__(self, key: str) -> bool:
252
- return key in self.additional_properties
@@ -1,172 +0,0 @@
1
- """
2
- This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
-
4
- It is a customized template from the openapi-python-client tool's default template:
5
- https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
6
-
7
- The main change is:
8
- - Fix typing issues
9
- """
10
-
11
- # flake8: noqa: C901
12
-
13
- import datetime
14
- from typing import TYPE_CHECKING, Any, List, Type, TypeVar, Union, cast
15
-
16
- from attrs import define as _attrs_define
17
- from attrs import field as _attrs_field
18
- from dateutil.parser import isoparse
19
-
20
- if TYPE_CHECKING:
21
- from ..models.memory_metrics import MemoryMetrics
22
- from ..models.scored_memory_metadata import ScoredMemoryMetadata
23
-
24
-
25
- T = TypeVar("T", bound="ScoredMemory")
26
-
27
-
28
- @_attrs_define
29
- class ScoredMemory:
30
- """A scored memory is a single item stored in the database with a score.
31
-
32
- Attributes:
33
- value (str):
34
- embedding (List[float]):
35
- source_id (Union[None, str]):
36
- metadata (ScoredMemoryMetadata):
37
- memory_id (str):
38
- memory_version (int):
39
- created_at (datetime.datetime):
40
- updated_at (datetime.datetime):
41
- edited_at (datetime.datetime):
42
- metrics (MemoryMetrics):
43
- score (float):
44
- """
45
-
46
- value: str
47
- embedding: List[float]
48
- source_id: Union[None, str]
49
- metadata: "ScoredMemoryMetadata"
50
- memory_id: str
51
- memory_version: int
52
- created_at: datetime.datetime
53
- updated_at: datetime.datetime
54
- edited_at: datetime.datetime
55
- metrics: "MemoryMetrics"
56
- score: float
57
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
58
-
59
- def to_dict(self) -> dict[str, Any]:
60
- value: str
61
- value = self.value
62
-
63
- embedding = self.embedding
64
-
65
- source_id: Union[None, str]
66
- source_id = self.source_id
67
-
68
- metadata = self.metadata.to_dict()
69
-
70
- memory_id = self.memory_id
71
-
72
- memory_version = self.memory_version
73
-
74
- created_at = self.created_at.isoformat()
75
-
76
- updated_at = self.updated_at.isoformat()
77
-
78
- edited_at = self.edited_at.isoformat()
79
-
80
- metrics = self.metrics.to_dict()
81
-
82
- score = self.score
83
-
84
- field_dict: dict[str, Any] = {}
85
- field_dict.update(self.additional_properties)
86
- field_dict.update(
87
- {
88
- "value": value,
89
- "embedding": embedding,
90
- "source_id": source_id,
91
- "metadata": metadata,
92
- "memory_id": memory_id,
93
- "memory_version": memory_version,
94
- "created_at": created_at,
95
- "updated_at": updated_at,
96
- "edited_at": edited_at,
97
- "metrics": metrics,
98
- "score": score,
99
- }
100
- )
101
-
102
- return field_dict
103
-
104
- @classmethod
105
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
106
- from ..models.memory_metrics import MemoryMetrics
107
- from ..models.scored_memory_metadata import ScoredMemoryMetadata
108
-
109
- d = src_dict.copy()
110
-
111
- def _parse_value(data: object) -> str:
112
- return cast(str, data)
113
-
114
- value = _parse_value(d.pop("value"))
115
-
116
- embedding = cast(List[float], d.pop("embedding"))
117
-
118
- def _parse_source_id(data: object) -> Union[None, str]:
119
- if data is None:
120
- return data
121
- return cast(Union[None, str], data)
122
-
123
- source_id = _parse_source_id(d.pop("source_id"))
124
-
125
- metadata = ScoredMemoryMetadata.from_dict(d.pop("metadata"))
126
-
127
- memory_id = d.pop("memory_id")
128
-
129
- memory_version = d.pop("memory_version")
130
-
131
- created_at = isoparse(d.pop("created_at"))
132
-
133
- updated_at = isoparse(d.pop("updated_at"))
134
-
135
- edited_at = isoparse(d.pop("edited_at"))
136
-
137
- metrics = MemoryMetrics.from_dict(d.pop("metrics"))
138
-
139
- score = d.pop("score")
140
-
141
- scored_memory = cls(
142
- value=value,
143
- embedding=embedding,
144
- source_id=source_id,
145
- metadata=metadata,
146
- memory_id=memory_id,
147
- memory_version=memory_version,
148
- created_at=created_at,
149
- updated_at=updated_at,
150
- edited_at=edited_at,
151
- metrics=metrics,
152
- score=score,
153
- )
154
-
155
- scored_memory.additional_properties = d
156
- return scored_memory
157
-
158
- @property
159
- def additional_keys(self) -> list[str]:
160
- return list(self.additional_properties.keys())
161
-
162
- def __getitem__(self, key: str) -> Any:
163
- return self.additional_properties[key]
164
-
165
- def __setitem__(self, key: str, value: Any) -> None:
166
- self.additional_properties[key] = value
167
-
168
- def __delitem__(self, key: str) -> None:
169
- del self.additional_properties[key]
170
-
171
- def __contains__(self, key: str) -> bool:
172
- return key in self.additional_properties
@@ -1,128 +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.scored_memory_insert_metadata import ScoredMemoryInsertMetadata
21
-
22
-
23
- T = TypeVar("T", bound="ScoredMemoryInsert")
24
-
25
-
26
- @_attrs_define
27
- class ScoredMemoryInsert:
28
- """Model of fields required for inserting a scored memory.
29
-
30
- Attributes:
31
- value (str):
32
- score (float):
33
- memory_id (Union[None, Unset, str]):
34
- metadata (Union[Unset, ScoredMemoryInsertMetadata]):
35
- source_id (Union[None, Unset, str]):
36
- """
37
-
38
- value: str
39
- score: float
40
- memory_id: Union[None, Unset, str] = UNSET
41
- metadata: Union[Unset, "ScoredMemoryInsertMetadata"] = UNSET
42
- source_id: Union[None, Unset, str] = UNSET
43
-
44
- def to_dict(self) -> dict[str, Any]:
45
- value: str
46
- value = self.value
47
-
48
- score = self.score
49
-
50
- memory_id: Union[None, Unset, str]
51
- if isinstance(self.memory_id, Unset):
52
- memory_id = UNSET
53
- else:
54
- memory_id = self.memory_id
55
-
56
- metadata: Union[Unset, Dict[str, Any]] = UNSET
57
- if not isinstance(self.metadata, Unset):
58
- metadata = self.metadata.to_dict()
59
-
60
- source_id: Union[None, Unset, str]
61
- if isinstance(self.source_id, Unset):
62
- source_id = UNSET
63
- else:
64
- source_id = self.source_id
65
-
66
- field_dict: dict[str, Any] = {}
67
- field_dict.update(
68
- {
69
- "value": value,
70
- "score": score,
71
- }
72
- )
73
- if memory_id is not UNSET:
74
- field_dict["memory_id"] = memory_id
75
- if metadata is not UNSET:
76
- field_dict["metadata"] = metadata
77
- if source_id is not UNSET:
78
- field_dict["source_id"] = source_id
79
-
80
- return field_dict
81
-
82
- @classmethod
83
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
84
- from ..models.scored_memory_insert_metadata import ScoredMemoryInsertMetadata
85
-
86
- d = src_dict.copy()
87
-
88
- def _parse_value(data: object) -> str:
89
- return cast(str, data)
90
-
91
- value = _parse_value(d.pop("value"))
92
-
93
- score = d.pop("score")
94
-
95
- def _parse_memory_id(data: object) -> Union[None, Unset, str]:
96
- if data is None:
97
- return data
98
- if isinstance(data, Unset):
99
- return data
100
- return cast(Union[None, Unset, str], data)
101
-
102
- memory_id = _parse_memory_id(d.pop("memory_id", UNSET))
103
-
104
- _metadata = d.pop("metadata", UNSET)
105
- metadata: Union[Unset, ScoredMemoryInsertMetadata]
106
- if isinstance(_metadata, Unset):
107
- metadata = UNSET
108
- else:
109
- metadata = ScoredMemoryInsertMetadata.from_dict(_metadata)
110
-
111
- def _parse_source_id(data: object) -> Union[None, Unset, str]:
112
- if data is None:
113
- return data
114
- if isinstance(data, Unset):
115
- return data
116
- return cast(Union[None, Unset, str], data)
117
-
118
- source_id = _parse_source_id(d.pop("source_id", UNSET))
119
-
120
- scored_memory_insert = cls(
121
- value=value,
122
- score=score,
123
- memory_id=memory_id,
124
- metadata=metadata,
125
- source_id=source_id,
126
- )
127
-
128
- return scored_memory_insert
@@ -1,68 +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
- T = TypeVar("T", bound="ScoredMemoryInsertMetadata")
19
-
20
-
21
- @_attrs_define
22
- class ScoredMemoryInsertMetadata:
23
- """ """
24
-
25
- additional_properties: dict[str, Union[None, bool, float, int, str]] = _attrs_field(init=False, factory=dict)
26
-
27
- def to_dict(self) -> dict[str, Any]:
28
- field_dict: dict[str, Any] = {}
29
- for prop_name, prop in self.additional_properties.items():
30
- field_dict[prop_name] = prop
31
-
32
- return field_dict
33
-
34
- @classmethod
35
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
36
- d = src_dict.copy()
37
- scored_memory_insert_metadata = cls()
38
-
39
- additional_properties = {}
40
- for prop_name, prop_dict in d.items():
41
-
42
- def _parse_additional_property(data: object) -> Union[None, bool, float, int, str]:
43
- if data is None:
44
- return data
45
- return cast(Union[None, bool, float, int, str], data)
46
-
47
- additional_property = _parse_additional_property(prop_dict)
48
-
49
- additional_properties[prop_name] = additional_property
50
-
51
- scored_memory_insert_metadata.additional_properties = additional_properties
52
- return scored_memory_insert_metadata
53
-
54
- @property
55
- def additional_keys(self) -> list[str]:
56
- return list(self.additional_properties.keys())
57
-
58
- def __getitem__(self, key: str) -> Union[None, bool, float, int, str]:
59
- return self.additional_properties[key]
60
-
61
- def __setitem__(self, key: str, value: Union[None, bool, float, int, str]) -> None:
62
- self.additional_properties[key] = value
63
-
64
- def __delitem__(self, key: str) -> None:
65
- del self.additional_properties[key]
66
-
67
- def __contains__(self, key: str) -> bool:
68
- return key in self.additional_properties