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,193 +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_with_feedback_metrics_feedback_metrics import (
23
- ScoredMemoryWithFeedbackMetricsFeedbackMetrics,
24
- )
25
- from ..models.scored_memory_with_feedback_metrics_metadata import ScoredMemoryWithFeedbackMetricsMetadata
26
-
27
-
28
- T = TypeVar("T", bound="ScoredMemoryWithFeedbackMetrics")
29
-
30
-
31
- @_attrs_define
32
- class ScoredMemoryWithFeedbackMetrics:
33
- """
34
- Attributes:
35
- value (str):
36
- embedding (List[float]):
37
- source_id (Union[None, str]):
38
- metadata (ScoredMemoryWithFeedbackMetricsMetadata):
39
- memory_id (str):
40
- memory_version (int):
41
- created_at (datetime.datetime):
42
- updated_at (datetime.datetime):
43
- edited_at (datetime.datetime):
44
- metrics (MemoryMetrics):
45
- score (float):
46
- feedback_metrics (ScoredMemoryWithFeedbackMetricsFeedbackMetrics):
47
- lookup_count (int):
48
- """
49
-
50
- value: str
51
- embedding: List[float]
52
- source_id: Union[None, str]
53
- metadata: "ScoredMemoryWithFeedbackMetricsMetadata"
54
- memory_id: str
55
- memory_version: int
56
- created_at: datetime.datetime
57
- updated_at: datetime.datetime
58
- edited_at: datetime.datetime
59
- metrics: "MemoryMetrics"
60
- score: float
61
- feedback_metrics: "ScoredMemoryWithFeedbackMetricsFeedbackMetrics"
62
- lookup_count: int
63
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
64
-
65
- def to_dict(self) -> dict[str, Any]:
66
- value: str
67
- value = self.value
68
-
69
- embedding = self.embedding
70
-
71
- source_id: Union[None, str]
72
- source_id = self.source_id
73
-
74
- metadata = self.metadata.to_dict()
75
-
76
- memory_id = self.memory_id
77
-
78
- memory_version = self.memory_version
79
-
80
- created_at = self.created_at.isoformat()
81
-
82
- updated_at = self.updated_at.isoformat()
83
-
84
- edited_at = self.edited_at.isoformat()
85
-
86
- metrics = self.metrics.to_dict()
87
-
88
- score = self.score
89
-
90
- feedback_metrics = self.feedback_metrics.to_dict()
91
-
92
- lookup_count = self.lookup_count
93
-
94
- field_dict: dict[str, Any] = {}
95
- field_dict.update(self.additional_properties)
96
- field_dict.update(
97
- {
98
- "value": value,
99
- "embedding": embedding,
100
- "source_id": source_id,
101
- "metadata": metadata,
102
- "memory_id": memory_id,
103
- "memory_version": memory_version,
104
- "created_at": created_at,
105
- "updated_at": updated_at,
106
- "edited_at": edited_at,
107
- "metrics": metrics,
108
- "score": score,
109
- "feedback_metrics": feedback_metrics,
110
- "lookup_count": lookup_count,
111
- }
112
- )
113
-
114
- return field_dict
115
-
116
- @classmethod
117
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
118
- from ..models.memory_metrics import MemoryMetrics
119
- from ..models.scored_memory_with_feedback_metrics_feedback_metrics import (
120
- ScoredMemoryWithFeedbackMetricsFeedbackMetrics,
121
- )
122
- from ..models.scored_memory_with_feedback_metrics_metadata import ScoredMemoryWithFeedbackMetricsMetadata
123
-
124
- d = src_dict.copy()
125
-
126
- def _parse_value(data: object) -> str:
127
- return cast(str, data)
128
-
129
- value = _parse_value(d.pop("value"))
130
-
131
- embedding = cast(List[float], d.pop("embedding"))
132
-
133
- def _parse_source_id(data: object) -> Union[None, str]:
134
- if data is None:
135
- return data
136
- return cast(Union[None, str], data)
137
-
138
- source_id = _parse_source_id(d.pop("source_id"))
139
-
140
- metadata = ScoredMemoryWithFeedbackMetricsMetadata.from_dict(d.pop("metadata"))
141
-
142
- memory_id = d.pop("memory_id")
143
-
144
- memory_version = d.pop("memory_version")
145
-
146
- created_at = isoparse(d.pop("created_at"))
147
-
148
- updated_at = isoparse(d.pop("updated_at"))
149
-
150
- edited_at = isoparse(d.pop("edited_at"))
151
-
152
- metrics = MemoryMetrics.from_dict(d.pop("metrics"))
153
-
154
- score = d.pop("score")
155
-
156
- feedback_metrics = ScoredMemoryWithFeedbackMetricsFeedbackMetrics.from_dict(d.pop("feedback_metrics"))
157
-
158
- lookup_count = d.pop("lookup_count")
159
-
160
- scored_memory_with_feedback_metrics = cls(
161
- value=value,
162
- embedding=embedding,
163
- source_id=source_id,
164
- metadata=metadata,
165
- memory_id=memory_id,
166
- memory_version=memory_version,
167
- created_at=created_at,
168
- updated_at=updated_at,
169
- edited_at=edited_at,
170
- metrics=metrics,
171
- score=score,
172
- feedback_metrics=feedback_metrics,
173
- lookup_count=lookup_count,
174
- )
175
-
176
- scored_memory_with_feedback_metrics.additional_properties = d
177
- return scored_memory_with_feedback_metrics
178
-
179
- @property
180
- def additional_keys(self) -> list[str]:
181
- return list(self.additional_properties.keys())
182
-
183
- def __getitem__(self, key: str) -> Any:
184
- return self.additional_properties[key]
185
-
186
- def __setitem__(self, key: str, value: Any) -> None:
187
- self.additional_properties[key] = value
188
-
189
- def __delitem__(self, key: str) -> None:
190
- del self.additional_properties[key]
191
-
192
- def __contains__(self, key: str) -> bool:
193
- return key in self.additional_properties
@@ -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 TYPE_CHECKING, Any, Type, TypeVar
14
-
15
- from attrs import define as _attrs_define
16
- from attrs import field as _attrs_field
17
-
18
- if TYPE_CHECKING:
19
- from ..models.feedback_metrics import FeedbackMetrics
20
-
21
-
22
- T = TypeVar("T", bound="ScoredMemoryWithFeedbackMetricsFeedbackMetrics")
23
-
24
-
25
- @_attrs_define
26
- class ScoredMemoryWithFeedbackMetricsFeedbackMetrics:
27
- """ """
28
-
29
- additional_properties: dict[str, "FeedbackMetrics"] = _attrs_field(init=False, factory=dict)
30
-
31
- def to_dict(self) -> dict[str, Any]:
32
- field_dict: dict[str, Any] = {}
33
- for prop_name, prop in self.additional_properties.items():
34
- field_dict[prop_name] = prop.to_dict()
35
-
36
- return field_dict
37
-
38
- @classmethod
39
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
40
- from ..models.feedback_metrics import FeedbackMetrics
41
-
42
- d = src_dict.copy()
43
- scored_memory_with_feedback_metrics_feedback_metrics = cls()
44
-
45
- additional_properties = {}
46
- for prop_name, prop_dict in d.items():
47
- additional_property = FeedbackMetrics.from_dict(prop_dict)
48
-
49
- additional_properties[prop_name] = additional_property
50
-
51
- scored_memory_with_feedback_metrics_feedback_metrics.additional_properties = additional_properties
52
- return scored_memory_with_feedback_metrics_feedback_metrics
53
-
54
- @property
55
- def additional_keys(self) -> list[str]:
56
- return list(self.additional_properties.keys())
57
-
58
- def __getitem__(self, key: str) -> "FeedbackMetrics":
59
- return self.additional_properties[key]
60
-
61
- def __setitem__(self, key: str, value: "FeedbackMetrics") -> 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
@@ -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="ScoredMemoryWithFeedbackMetricsMetadata")
19
-
20
-
21
- @_attrs_define
22
- class ScoredMemoryWithFeedbackMetricsMetadata:
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_with_feedback_metrics_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_with_feedback_metrics_metadata.additional_properties = additional_properties
52
- return scored_memory_with_feedback_metrics_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
@@ -1,80 +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, Literal, Type, TypeVar, cast
14
-
15
- from attrs import define as _attrs_define
16
- from attrs import field as _attrs_field
17
-
18
- T = TypeVar("T", bound="ServiceUnavailableErrorResponse")
19
-
20
-
21
- @_attrs_define
22
- class ServiceUnavailableErrorResponse:
23
- """
24
- Attributes:
25
- status_code (Literal[503]):
26
- service (str):
27
- """
28
-
29
- status_code: Literal[503]
30
- service: str
31
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
32
-
33
- def to_dict(self) -> dict[str, Any]:
34
- status_code = self.status_code
35
-
36
- service = self.service
37
-
38
- field_dict: dict[str, Any] = {}
39
- field_dict.update(self.additional_properties)
40
- field_dict.update(
41
- {
42
- "status_code": status_code,
43
- "service": service,
44
- }
45
- )
46
-
47
- return field_dict
48
-
49
- @classmethod
50
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
51
- d = src_dict.copy()
52
- status_code = cast(Literal[503], d.pop("status_code"))
53
- if status_code != 503:
54
- raise ValueError(f"status_code must match const 503, got '{status_code}'")
55
-
56
- service = d.pop("service")
57
-
58
- service_unavailable_error_response = cls(
59
- status_code=status_code,
60
- service=service,
61
- )
62
-
63
- service_unavailable_error_response.additional_properties = d
64
- return service_unavailable_error_response
65
-
66
- @property
67
- def additional_keys(self) -> list[str]:
68
- return list(self.additional_properties.keys())
69
-
70
- def __getitem__(self, key: str) -> Any:
71
- return self.additional_properties[key]
72
-
73
- def __setitem__(self, key: str, value: Any) -> None:
74
- self.additional_properties[key] = value
75
-
76
- def __delitem__(self, key: str) -> None:
77
- del self.additional_properties[key]
78
-
79
- def __contains__(self, key: str) -> bool:
80
- return key in self.additional_properties
@@ -1,198 +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.base_model import BaseModel
25
-
26
-
27
- T = TypeVar("T", bound="Task")
28
-
29
-
30
- @_attrs_define
31
- class Task:
32
- """
33
- Attributes:
34
- status (TaskStatus): Status of task in the task queue
35
- steps_total (Union[None, int]):
36
- steps_completed (Union[None, int]):
37
- exception (Union[None, str]):
38
- updated_at (datetime.datetime):
39
- created_at (datetime.datetime):
40
- id (str):
41
- org_id (str):
42
- type (str):
43
- payload (BaseModel):
44
- result (Union['BaseModel', None]):
45
- """
46
-
47
- status: TaskStatus
48
- steps_total: Union[None, int]
49
- steps_completed: Union[None, int]
50
- exception: Union[None, str]
51
- updated_at: datetime.datetime
52
- created_at: datetime.datetime
53
- id: str
54
- org_id: str
55
- type: str
56
- payload: "BaseModel"
57
- result: Union["BaseModel", None]
58
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
59
-
60
- def to_dict(self) -> dict[str, Any]:
61
- from ..models.base_model import BaseModel
62
-
63
- status = self.status.value if isinstance(self.status, Enum) else self.status
64
-
65
- steps_total: Union[None, int]
66
- steps_total = self.steps_total
67
-
68
- steps_completed: Union[None, int]
69
- steps_completed = self.steps_completed
70
-
71
- exception: Union[None, str]
72
- exception = self.exception
73
-
74
- updated_at = self.updated_at.isoformat()
75
-
76
- created_at = self.created_at.isoformat()
77
-
78
- id = self.id
79
-
80
- org_id = self.org_id
81
-
82
- type = self.type
83
-
84
- payload = self.payload.to_dict()
85
-
86
- result: Union[Dict[str, Any], None]
87
- if isinstance(self.result, BaseModel):
88
- result = self.result.to_dict()
89
- else:
90
- result = self.result
91
-
92
- field_dict: dict[str, Any] = {}
93
- field_dict.update(self.additional_properties)
94
- field_dict.update(
95
- {
96
- "status": status,
97
- "steps_total": steps_total,
98
- "steps_completed": steps_completed,
99
- "exception": exception,
100
- "updated_at": updated_at,
101
- "created_at": created_at,
102
- "id": id,
103
- "org_id": org_id,
104
- "type": type,
105
- "payload": payload,
106
- "result": result,
107
- }
108
- )
109
-
110
- return field_dict
111
-
112
- @classmethod
113
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
114
- from ..models.base_model import BaseModel
115
-
116
- d = src_dict.copy()
117
- status = TaskStatus(d.pop("status"))
118
-
119
- def _parse_steps_total(data: object) -> Union[None, int]:
120
- if data is None:
121
- return data
122
- return cast(Union[None, int], data)
123
-
124
- steps_total = _parse_steps_total(d.pop("steps_total"))
125
-
126
- def _parse_steps_completed(data: object) -> Union[None, int]:
127
- if data is None:
128
- return data
129
- return cast(Union[None, int], data)
130
-
131
- steps_completed = _parse_steps_completed(d.pop("steps_completed"))
132
-
133
- def _parse_exception(data: object) -> Union[None, str]:
134
- if data is None:
135
- return data
136
- return cast(Union[None, str], data)
137
-
138
- exception = _parse_exception(d.pop("exception"))
139
-
140
- updated_at = isoparse(d.pop("updated_at"))
141
-
142
- created_at = isoparse(d.pop("created_at"))
143
-
144
- id = d.pop("id")
145
-
146
- org_id = d.pop("org_id")
147
-
148
- type = d.pop("type")
149
-
150
- payload = BaseModel.from_dict(d.pop("payload"))
151
-
152
- def _parse_result(data: object) -> Union["BaseModel", None]:
153
- if data is None:
154
- return data
155
- try:
156
- if not isinstance(data, dict):
157
- raise TypeError()
158
- result_type_0 = BaseModel.from_dict(data)
159
-
160
- return result_type_0
161
- except: # noqa: E722
162
- pass
163
- return cast(Union["BaseModel", None], data)
164
-
165
- result = _parse_result(d.pop("result"))
166
-
167
- task = cls(
168
- status=status,
169
- steps_total=steps_total,
170
- steps_completed=steps_completed,
171
- exception=exception,
172
- updated_at=updated_at,
173
- created_at=created_at,
174
- id=id,
175
- org_id=org_id,
176
- type=type,
177
- payload=payload,
178
- result=result,
179
- )
180
-
181
- task.additional_properties = d
182
- return task
183
-
184
- @property
185
- def additional_keys(self) -> list[str]:
186
- return list(self.additional_properties.keys())
187
-
188
- def __getitem__(self, key: str) -> Any:
189
- return self.additional_properties[key]
190
-
191
- def __setitem__(self, key: str, value: Any) -> None:
192
- self.additional_properties[key] = value
193
-
194
- def __delitem__(self, key: str) -> None:
195
- del self.additional_properties[key]
196
-
197
- def __contains__(self, key: str) -> bool:
198
- return key in self.additional_properties
@@ -1,14 +0,0 @@
1
- from enum import Enum
2
-
3
-
4
- class TaskStatus(str, Enum):
5
- ABORTED = "ABORTED"
6
- ABORTING = "ABORTING"
7
- COMPLETED = "COMPLETED"
8
- DISPATCHED = "DISPATCHED"
9
- FAILED = "FAILED"
10
- INITIALIZED = "INITIALIZED"
11
- PROCESSING = "PROCESSING"
12
-
13
- def __str__(self) -> str:
14
- return str(self.value)