orca-sdk 0.0.97__py3-none-any.whl → 0.0.100__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 (299) hide show
  1. orca_sdk/__init__.py +1 -0
  2. orca_sdk/_shared/__init__.py +1 -0
  3. orca_sdk/_utils/analysis_ui.py +5 -5
  4. orca_sdk/_utils/auth.py +23 -33
  5. orca_sdk/_utils/pagination.py +126 -0
  6. orca_sdk/_utils/pagination_test.py +132 -0
  7. orca_sdk/classification_model.py +188 -126
  8. orca_sdk/classification_model_test.py +57 -8
  9. orca_sdk/client.py +3563 -0
  10. orca_sdk/conftest.py +10 -0
  11. orca_sdk/credentials.py +59 -21
  12. orca_sdk/credentials_test.py +20 -0
  13. orca_sdk/datasource.py +80 -93
  14. orca_sdk/datasource_test.py +41 -7
  15. orca_sdk/embedding_model.py +225 -71
  16. orca_sdk/embedding_model_test.py +27 -36
  17. orca_sdk/job.py +49 -45
  18. orca_sdk/job_test.py +16 -0
  19. orca_sdk/memoryset.py +340 -353
  20. orca_sdk/memoryset_test.py +7 -11
  21. orca_sdk/regression_model.py +120 -111
  22. orca_sdk/regression_model_test.py +15 -0
  23. orca_sdk/telemetry.py +162 -139
  24. {orca_sdk-0.0.97.dist-info → orca_sdk-0.0.100.dist-info}/METADATA +2 -5
  25. orca_sdk-0.0.100.dist-info/RECORD +40 -0
  26. orca_sdk/_generated_api_client/__init__.py +0 -3
  27. orca_sdk/_generated_api_client/api/__init__.py +0 -307
  28. orca_sdk/_generated_api_client/api/auth/__init__.py +0 -0
  29. orca_sdk/_generated_api_client/api/auth/check_authentication_auth_get.py +0 -128
  30. orca_sdk/_generated_api_client/api/auth/create_api_key_auth_api_key_post.py +0 -170
  31. orca_sdk/_generated_api_client/api/auth/create_org_plan_auth_org_plan_post.py +0 -168
  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/get_org_plan_auth_org_plan_get.py +0 -122
  35. orca_sdk/_generated_api_client/api/auth/list_api_keys_auth_api_key_get.py +0 -127
  36. orca_sdk/_generated_api_client/api/auth/update_org_plan_auth_org_plan_put.py +0 -168
  37. orca_sdk/_generated_api_client/api/classification_model/__init__.py +0 -0
  38. orca_sdk/_generated_api_client/api/classification_model/create_classification_model_classification_model_post.py +0 -170
  39. orca_sdk/_generated_api_client/api/classification_model/delete_classification_model_classification_model_name_or_id_delete.py +0 -156
  40. 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
  41. orca_sdk/_generated_api_client/api/classification_model/evaluate_classification_model_classification_model_model_name_or_id_evaluation_post.py +0 -183
  42. orca_sdk/_generated_api_client/api/classification_model/get_classification_model_classification_model_name_or_id_get.py +0 -156
  43. 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
  44. orca_sdk/_generated_api_client/api/classification_model/list_classification_model_evaluations_classification_model_model_name_or_id_evaluation_get.py +0 -161
  45. orca_sdk/_generated_api_client/api/classification_model/list_classification_models_classification_model_get.py +0 -127
  46. orca_sdk/_generated_api_client/api/classification_model/predict_label_gpu_classification_model_name_or_id_prediction_post.py +0 -190
  47. orca_sdk/_generated_api_client/api/classification_model/update_classification_model_classification_model_name_or_id_patch.py +0 -183
  48. orca_sdk/_generated_api_client/api/datasource/__init__.py +0 -0
  49. orca_sdk/_generated_api_client/api/datasource/create_datasource_from_content_datasource_post.py +0 -224
  50. orca_sdk/_generated_api_client/api/datasource/create_datasource_from_files_datasource_upload_post.py +0 -229
  51. orca_sdk/_generated_api_client/api/datasource/create_embedding_evaluation_datasource_name_or_id_embedding_evaluation_post.py +0 -183
  52. orca_sdk/_generated_api_client/api/datasource/delete_datasource_datasource_name_or_id_delete.py +0 -156
  53. orca_sdk/_generated_api_client/api/datasource/download_datasource_datasource_name_or_id_download_get.py +0 -172
  54. orca_sdk/_generated_api_client/api/datasource/get_datasource_datasource_name_or_id_get.py +0 -156
  55. orca_sdk/_generated_api_client/api/datasource/get_embedding_evaluation_datasource_name_or_id_embedding_evaluation_task_id_get.py +0 -169
  56. orca_sdk/_generated_api_client/api/datasource/list_datasources_datasource_get.py +0 -127
  57. orca_sdk/_generated_api_client/api/datasource/list_embedding_evaluations_datasource_name_or_id_embedding_evaluation_get.py +0 -235
  58. orca_sdk/_generated_api_client/api/default/__init__.py +0 -0
  59. orca_sdk/_generated_api_client/api/default/healthcheck_get.py +0 -118
  60. orca_sdk/_generated_api_client/api/default/healthcheck_gpu_get.py +0 -118
  61. orca_sdk/_generated_api_client/api/finetuned_embedding_model/__init__.py +0 -0
  62. orca_sdk/_generated_api_client/api/finetuned_embedding_model/create_finetuned_embedding_model_finetuned_embedding_model_post.py +0 -168
  63. orca_sdk/_generated_api_client/api/finetuned_embedding_model/delete_finetuned_embedding_model_finetuned_embedding_model_name_or_id_delete.py +0 -156
  64. 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
  65. orca_sdk/_generated_api_client/api/finetuned_embedding_model/get_finetuned_embedding_model_finetuned_embedding_model_name_or_id_get.py +0 -156
  66. orca_sdk/_generated_api_client/api/finetuned_embedding_model/list_finetuned_embedding_models_finetuned_embedding_model_get.py +0 -127
  67. orca_sdk/_generated_api_client/api/memoryset/__init__.py +0 -0
  68. orca_sdk/_generated_api_client/api/memoryset/analyze_memoryset_memoryset_name_or_id_analysis_post.py +0 -183
  69. orca_sdk/_generated_api_client/api/memoryset/batch_delete_memoryset_batch_delete_memoryset_post.py +0 -168
  70. orca_sdk/_generated_api_client/api/memoryset/clone_memoryset_memoryset_name_or_id_clone_post.py +0 -181
  71. orca_sdk/_generated_api_client/api/memoryset/create_memoryset_memoryset_post.py +0 -168
  72. orca_sdk/_generated_api_client/api/memoryset/delete_memories_memoryset_name_or_id_memories_delete_post.py +0 -181
  73. orca_sdk/_generated_api_client/api/memoryset/delete_memory_memoryset_name_or_id_memory_memory_id_delete.py +0 -167
  74. orca_sdk/_generated_api_client/api/memoryset/delete_memoryset_memoryset_name_or_id_delete.py +0 -156
  75. orca_sdk/_generated_api_client/api/memoryset/get_analysis_memoryset_name_or_id_analysis_analysis_task_id_get.py +0 -169
  76. orca_sdk/_generated_api_client/api/memoryset/get_memories_memoryset_name_or_id_memories_get_post.py +0 -210
  77. orca_sdk/_generated_api_client/api/memoryset/get_memory_memoryset_name_or_id_memory_memory_id_get.py +0 -186
  78. orca_sdk/_generated_api_client/api/memoryset/get_memoryset_memoryset_name_or_id_get.py +0 -156
  79. orca_sdk/_generated_api_client/api/memoryset/insert_memories_gpu_memoryset_name_or_id_memory_post.py +0 -188
  80. orca_sdk/_generated_api_client/api/memoryset/list_analyses_memoryset_name_or_id_analysis_get.py +0 -235
  81. orca_sdk/_generated_api_client/api/memoryset/list_memorysets_memoryset_get.py +0 -180
  82. orca_sdk/_generated_api_client/api/memoryset/memoryset_lookup_gpu_memoryset_name_or_id_lookup_post.py +0 -212
  83. orca_sdk/_generated_api_client/api/memoryset/potential_duplicate_groups_memoryset_name_or_id_potential_duplicate_groups_get.py +0 -195
  84. orca_sdk/_generated_api_client/api/memoryset/query_memoryset_memoryset_name_or_id_memories_post.py +0 -210
  85. orca_sdk/_generated_api_client/api/memoryset/suggest_cascading_edits_memoryset_name_or_id_memory_memory_id_cascading_edits_post.py +0 -233
  86. orca_sdk/_generated_api_client/api/memoryset/update_memories_gpu_memoryset_name_or_id_memories_patch.py +0 -216
  87. orca_sdk/_generated_api_client/api/memoryset/update_memory_gpu_memoryset_name_or_id_memory_patch.py +0 -205
  88. orca_sdk/_generated_api_client/api/memoryset/update_memoryset_memoryset_name_or_id_patch.py +0 -183
  89. orca_sdk/_generated_api_client/api/predictive_model/__init__.py +0 -0
  90. orca_sdk/_generated_api_client/api/predictive_model/list_predictive_models_predictive_model_get.py +0 -150
  91. orca_sdk/_generated_api_client/api/pretrained_embedding_model/__init__.py +0 -0
  92. orca_sdk/_generated_api_client/api/pretrained_embedding_model/embed_with_pretrained_model_gpu_pretrained_embedding_model_model_name_embedding_post.py +0 -192
  93. orca_sdk/_generated_api_client/api/pretrained_embedding_model/get_pretrained_embedding_model_pretrained_embedding_model_model_name_get.py +0 -161
  94. orca_sdk/_generated_api_client/api/pretrained_embedding_model/list_pretrained_embedding_models_pretrained_embedding_model_get.py +0 -127
  95. orca_sdk/_generated_api_client/api/regression_model/__init__.py +0 -0
  96. orca_sdk/_generated_api_client/api/regression_model/create_regression_model_regression_model_post.py +0 -170
  97. 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
  98. orca_sdk/_generated_api_client/api/regression_model/delete_regression_model_regression_model_name_or_id_delete.py +0 -154
  99. orca_sdk/_generated_api_client/api/regression_model/evaluate_regression_model_regression_model_model_name_or_id_evaluation_post.py +0 -183
  100. 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
  101. orca_sdk/_generated_api_client/api/regression_model/get_regression_model_regression_model_name_or_id_get.py +0 -156
  102. orca_sdk/_generated_api_client/api/regression_model/list_regression_model_evaluations_regression_model_model_name_or_id_evaluation_get.py +0 -161
  103. orca_sdk/_generated_api_client/api/regression_model/list_regression_models_regression_model_get.py +0 -127
  104. orca_sdk/_generated_api_client/api/regression_model/predict_score_gpu_regression_model_name_or_id_prediction_post.py +0 -190
  105. orca_sdk/_generated_api_client/api/regression_model/update_regression_model_regression_model_name_or_id_patch.py +0 -183
  106. orca_sdk/_generated_api_client/api/task/__init__.py +0 -0
  107. orca_sdk/_generated_api_client/api/task/abort_task_task_task_id_abort_delete.py +0 -154
  108. orca_sdk/_generated_api_client/api/task/get_task_status_task_task_id_status_get.py +0 -156
  109. orca_sdk/_generated_api_client/api/task/get_task_task_task_id_get.py +0 -156
  110. orca_sdk/_generated_api_client/api/task/list_tasks_task_get.py +0 -288
  111. orca_sdk/_generated_api_client/api/telemetry/__init__.py +0 -0
  112. orca_sdk/_generated_api_client/api/telemetry/count_predictions_telemetry_prediction_count_post.py +0 -168
  113. orca_sdk/_generated_api_client/api/telemetry/drop_feedback_category_with_data_telemetry_feedback_category_name_or_id_delete.py +0 -162
  114. orca_sdk/_generated_api_client/api/telemetry/explain_prediction_telemetry_prediction_prediction_id_explanation_get.py +0 -182
  115. orca_sdk/_generated_api_client/api/telemetry/generate_memory_suggestions_telemetry_prediction_prediction_id_memory_suggestions_post.py +0 -239
  116. orca_sdk/_generated_api_client/api/telemetry/get_action_recommendation_telemetry_prediction_prediction_id_action_get.py +0 -192
  117. orca_sdk/_generated_api_client/api/telemetry/get_feedback_category_telemetry_feedback_category_name_or_id_get.py +0 -156
  118. orca_sdk/_generated_api_client/api/telemetry/get_prediction_telemetry_prediction_prediction_id_get.py +0 -180
  119. orca_sdk/_generated_api_client/api/telemetry/list_feedback_categories_telemetry_feedback_category_get.py +0 -127
  120. orca_sdk/_generated_api_client/api/telemetry/list_memories_with_feedback_telemetry_memories_post.py +0 -198
  121. orca_sdk/_generated_api_client/api/telemetry/list_predictions_telemetry_prediction_post.py +0 -198
  122. orca_sdk/_generated_api_client/api/telemetry/record_prediction_feedback_telemetry_prediction_feedback_put.py +0 -171
  123. orca_sdk/_generated_api_client/api/telemetry/update_prediction_telemetry_prediction_prediction_id_patch.py +0 -181
  124. orca_sdk/_generated_api_client/client.py +0 -216
  125. orca_sdk/_generated_api_client/errors.py +0 -38
  126. orca_sdk/_generated_api_client/models/__init__.py +0 -345
  127. orca_sdk/_generated_api_client/models/action_recommendation.py +0 -82
  128. orca_sdk/_generated_api_client/models/action_recommendation_action.py +0 -11
  129. orca_sdk/_generated_api_client/models/add_memory_recommendations.py +0 -85
  130. orca_sdk/_generated_api_client/models/add_memory_suggestion.py +0 -79
  131. orca_sdk/_generated_api_client/models/analyze_neighbor_labels_result.py +0 -116
  132. orca_sdk/_generated_api_client/models/api_key_metadata.py +0 -137
  133. orca_sdk/_generated_api_client/models/api_key_metadata_scope_item.py +0 -9
  134. orca_sdk/_generated_api_client/models/base_label_prediction_result.py +0 -130
  135. orca_sdk/_generated_api_client/models/base_model.py +0 -55
  136. orca_sdk/_generated_api_client/models/base_score_prediction_result.py +0 -108
  137. orca_sdk/_generated_api_client/models/body_create_datasource_from_files_datasource_upload_post.py +0 -145
  138. orca_sdk/_generated_api_client/models/cascade_edit_suggestions_request.py +0 -154
  139. orca_sdk/_generated_api_client/models/cascading_edit_suggestion.py +0 -92
  140. orca_sdk/_generated_api_client/models/class_representatives.py +0 -92
  141. orca_sdk/_generated_api_client/models/classification_evaluation_request.py +0 -148
  142. orca_sdk/_generated_api_client/models/classification_metrics.py +0 -259
  143. orca_sdk/_generated_api_client/models/classification_model_metadata.py +0 -227
  144. orca_sdk/_generated_api_client/models/classification_prediction_request.py +0 -220
  145. orca_sdk/_generated_api_client/models/clone_memoryset_request.py +0 -210
  146. orca_sdk/_generated_api_client/models/cluster_metrics.py +0 -78
  147. orca_sdk/_generated_api_client/models/column_info.py +0 -145
  148. orca_sdk/_generated_api_client/models/column_type.py +0 -14
  149. orca_sdk/_generated_api_client/models/constraint_violation_error_response.py +0 -81
  150. orca_sdk/_generated_api_client/models/constraint_violation_error_response_status_code.py +0 -8
  151. orca_sdk/_generated_api_client/models/count_predictions_request.py +0 -195
  152. orca_sdk/_generated_api_client/models/create_api_key_request.py +0 -120
  153. orca_sdk/_generated_api_client/models/create_api_key_request_scope_item.py +0 -9
  154. orca_sdk/_generated_api_client/models/create_api_key_response.py +0 -145
  155. orca_sdk/_generated_api_client/models/create_api_key_response_scope_item.py +0 -9
  156. orca_sdk/_generated_api_client/models/create_classification_model_request.py +0 -237
  157. orca_sdk/_generated_api_client/models/create_datasource_from_content_request.py +0 -101
  158. orca_sdk/_generated_api_client/models/create_memoryset_request.py +0 -365
  159. orca_sdk/_generated_api_client/models/create_memoryset_request_index_params.py +0 -66
  160. orca_sdk/_generated_api_client/models/create_memoryset_request_index_type.py +0 -13
  161. orca_sdk/_generated_api_client/models/create_org_plan_request.py +0 -73
  162. orca_sdk/_generated_api_client/models/create_org_plan_request_tier.py +0 -11
  163. orca_sdk/_generated_api_client/models/create_regression_model_request.py +0 -157
  164. orca_sdk/_generated_api_client/models/datasource_metadata.py +0 -156
  165. orca_sdk/_generated_api_client/models/delete_memories_request.py +0 -70
  166. orca_sdk/_generated_api_client/models/delete_memorysets_request.py +0 -70
  167. orca_sdk/_generated_api_client/models/embed_request.py +0 -155
  168. orca_sdk/_generated_api_client/models/embedding_evaluation_payload.py +0 -205
  169. orca_sdk/_generated_api_client/models/embedding_evaluation_request.py +0 -197
  170. orca_sdk/_generated_api_client/models/embedding_evaluation_response.py +0 -158
  171. orca_sdk/_generated_api_client/models/embedding_evaluation_result.py +0 -86
  172. orca_sdk/_generated_api_client/models/embedding_finetuning_method.py +0 -9
  173. orca_sdk/_generated_api_client/models/embedding_model_result.py +0 -123
  174. orca_sdk/_generated_api_client/models/evaluation_response.py +0 -153
  175. orca_sdk/_generated_api_client/models/evaluation_response_classification_metrics.py +0 -140
  176. orca_sdk/_generated_api_client/models/evaluation_response_regression_metrics.py +0 -140
  177. orca_sdk/_generated_api_client/models/feedback_metrics.py +0 -85
  178. orca_sdk/_generated_api_client/models/feedback_type.py +0 -9
  179. orca_sdk/_generated_api_client/models/filter_item.py +0 -239
  180. orca_sdk/_generated_api_client/models/filter_item_field_type_0_item.py +0 -17
  181. orca_sdk/_generated_api_client/models/filter_item_field_type_1_item_type_0.py +0 -8
  182. orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_0.py +0 -8
  183. orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_1.py +0 -22
  184. orca_sdk/_generated_api_client/models/filter_item_op.py +0 -16
  185. orca_sdk/_generated_api_client/models/finetune_embedding_model_request.py +0 -259
  186. orca_sdk/_generated_api_client/models/finetune_embedding_model_request_training_args.py +0 -66
  187. orca_sdk/_generated_api_client/models/finetuned_embedding_model_metadata.py +0 -166
  188. orca_sdk/_generated_api_client/models/get_memories_request.py +0 -70
  189. orca_sdk/_generated_api_client/models/http_validation_error.py +0 -86
  190. orca_sdk/_generated_api_client/models/internal_server_error_response.py +0 -81
  191. orca_sdk/_generated_api_client/models/internal_server_error_response_status_code.py +0 -8
  192. orca_sdk/_generated_api_client/models/label_class_metrics.py +0 -108
  193. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup.py +0 -210
  194. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup_metadata.py +0 -68
  195. orca_sdk/_generated_api_client/models/label_prediction_with_memories_and_feedback.py +0 -288
  196. orca_sdk/_generated_api_client/models/labeled_memory.py +0 -186
  197. orca_sdk/_generated_api_client/models/labeled_memory_insert.py +0 -128
  198. orca_sdk/_generated_api_client/models/labeled_memory_insert_metadata.py +0 -68
  199. orca_sdk/_generated_api_client/models/labeled_memory_lookup.py +0 -194
  200. orca_sdk/_generated_api_client/models/labeled_memory_lookup_metadata.py +0 -68
  201. orca_sdk/_generated_api_client/models/labeled_memory_metadata.py +0 -68
  202. orca_sdk/_generated_api_client/models/labeled_memory_update.py +0 -171
  203. orca_sdk/_generated_api_client/models/labeled_memory_update_metadata_type_0.py +0 -68
  204. orca_sdk/_generated_api_client/models/labeled_memory_with_feedback_metrics.py +0 -207
  205. orca_sdk/_generated_api_client/models/labeled_memory_with_feedback_metrics_feedback_metrics.py +0 -68
  206. orca_sdk/_generated_api_client/models/labeled_memory_with_feedback_metrics_metadata.py +0 -68
  207. orca_sdk/_generated_api_client/models/list_memories_request.py +0 -104
  208. orca_sdk/_generated_api_client/models/list_predictions_request.py +0 -319
  209. orca_sdk/_generated_api_client/models/lookup_request.py +0 -101
  210. orca_sdk/_generated_api_client/models/lookup_score_metrics.py +0 -94
  211. orca_sdk/_generated_api_client/models/memory_metrics.py +0 -263
  212. orca_sdk/_generated_api_client/models/memory_type.py +0 -9
  213. orca_sdk/_generated_api_client/models/memoryset_analysis_configs.py +0 -245
  214. orca_sdk/_generated_api_client/models/memoryset_analysis_request.py +0 -105
  215. orca_sdk/_generated_api_client/models/memoryset_analysis_response.py +0 -182
  216. orca_sdk/_generated_api_client/models/memoryset_class_patterns_analysis_config.py +0 -79
  217. orca_sdk/_generated_api_client/models/memoryset_class_patterns_metrics.py +0 -138
  218. orca_sdk/_generated_api_client/models/memoryset_cluster_analysis_config.py +0 -202
  219. orca_sdk/_generated_api_client/models/memoryset_cluster_analysis_config_clustering_method.py +0 -9
  220. orca_sdk/_generated_api_client/models/memoryset_cluster_analysis_config_partitioning_method.py +0 -10
  221. orca_sdk/_generated_api_client/models/memoryset_cluster_metrics.py +0 -100
  222. orca_sdk/_generated_api_client/models/memoryset_duplicate_analysis_config.py +0 -70
  223. orca_sdk/_generated_api_client/models/memoryset_duplicate_metrics.py +0 -70
  224. orca_sdk/_generated_api_client/models/memoryset_label_analysis_config.py +0 -70
  225. orca_sdk/_generated_api_client/models/memoryset_label_metrics.py +0 -116
  226. orca_sdk/_generated_api_client/models/memoryset_metadata.py +0 -333
  227. orca_sdk/_generated_api_client/models/memoryset_metadata_index_params.py +0 -55
  228. orca_sdk/_generated_api_client/models/memoryset_metadata_index_type.py +0 -13
  229. orca_sdk/_generated_api_client/models/memoryset_metrics.py +0 -265
  230. orca_sdk/_generated_api_client/models/memoryset_neighbor_analysis_config.py +0 -83
  231. orca_sdk/_generated_api_client/models/memoryset_neighbor_metrics.py +0 -76
  232. orca_sdk/_generated_api_client/models/memoryset_neighbor_metrics_lookup_score_metrics.py +0 -68
  233. orca_sdk/_generated_api_client/models/memoryset_projection_analysis_config.py +0 -79
  234. orca_sdk/_generated_api_client/models/memoryset_projection_metrics.py +0 -55
  235. orca_sdk/_generated_api_client/models/memoryset_update.py +0 -121
  236. orca_sdk/_generated_api_client/models/not_found_error_response.py +0 -99
  237. orca_sdk/_generated_api_client/models/not_found_error_response_resource_type_0.py +0 -23
  238. orca_sdk/_generated_api_client/models/not_found_error_response_status_code.py +0 -8
  239. orca_sdk/_generated_api_client/models/org_plan.py +0 -99
  240. orca_sdk/_generated_api_client/models/org_plan_tier.py +0 -11
  241. orca_sdk/_generated_api_client/models/paginated_task.py +0 -108
  242. orca_sdk/_generated_api_client/models/paginated_union_labeled_memory_with_feedback_metrics_scored_memory_with_feedback_metrics.py +0 -135
  243. orca_sdk/_generated_api_client/models/pr_curve.py +0 -86
  244. orca_sdk/_generated_api_client/models/prediction_feedback.py +0 -157
  245. orca_sdk/_generated_api_client/models/prediction_feedback_category.py +0 -115
  246. orca_sdk/_generated_api_client/models/prediction_feedback_request.py +0 -122
  247. orca_sdk/_generated_api_client/models/prediction_feedback_result.py +0 -102
  248. orca_sdk/_generated_api_client/models/prediction_sort_item_item_type_0.py +0 -10
  249. orca_sdk/_generated_api_client/models/prediction_sort_item_item_type_1.py +0 -9
  250. orca_sdk/_generated_api_client/models/predictive_model_update.py +0 -111
  251. orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py +0 -115
  252. orca_sdk/_generated_api_client/models/pretrained_embedding_model_name.py +0 -17
  253. orca_sdk/_generated_api_client/models/rac_head_type.py +0 -11
  254. orca_sdk/_generated_api_client/models/rar_head_type.py +0 -8
  255. orca_sdk/_generated_api_client/models/regression_evaluation_request.py +0 -148
  256. orca_sdk/_generated_api_client/models/regression_metrics.py +0 -172
  257. orca_sdk/_generated_api_client/models/regression_model_metadata.py +0 -191
  258. orca_sdk/_generated_api_client/models/regression_prediction_request.py +0 -195
  259. orca_sdk/_generated_api_client/models/roc_curve.py +0 -86
  260. orca_sdk/_generated_api_client/models/score_prediction_memory_lookup.py +0 -196
  261. orca_sdk/_generated_api_client/models/score_prediction_memory_lookup_metadata.py +0 -68
  262. orca_sdk/_generated_api_client/models/score_prediction_with_memories_and_feedback.py +0 -252
  263. orca_sdk/_generated_api_client/models/scored_memory.py +0 -172
  264. orca_sdk/_generated_api_client/models/scored_memory_insert.py +0 -128
  265. orca_sdk/_generated_api_client/models/scored_memory_insert_metadata.py +0 -68
  266. orca_sdk/_generated_api_client/models/scored_memory_lookup.py +0 -180
  267. orca_sdk/_generated_api_client/models/scored_memory_lookup_metadata.py +0 -68
  268. orca_sdk/_generated_api_client/models/scored_memory_metadata.py +0 -68
  269. orca_sdk/_generated_api_client/models/scored_memory_update.py +0 -171
  270. orca_sdk/_generated_api_client/models/scored_memory_update_metadata_type_0.py +0 -68
  271. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics.py +0 -193
  272. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics_feedback_metrics.py +0 -68
  273. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics_metadata.py +0 -68
  274. orca_sdk/_generated_api_client/models/service_unavailable_error_response.py +0 -81
  275. orca_sdk/_generated_api_client/models/service_unavailable_error_response_status_code.py +0 -8
  276. orca_sdk/_generated_api_client/models/task.py +0 -198
  277. orca_sdk/_generated_api_client/models/task_status.py +0 -14
  278. orca_sdk/_generated_api_client/models/task_status_info.py +0 -133
  279. orca_sdk/_generated_api_client/models/telemetry_field_type_0_item_type_0.py +0 -8
  280. orca_sdk/_generated_api_client/models/telemetry_field_type_0_item_type_2.py +0 -9
  281. orca_sdk/_generated_api_client/models/telemetry_field_type_1_item_type_0.py +0 -8
  282. orca_sdk/_generated_api_client/models/telemetry_field_type_1_item_type_1.py +0 -8
  283. orca_sdk/_generated_api_client/models/telemetry_filter_item.py +0 -217
  284. orca_sdk/_generated_api_client/models/telemetry_filter_item_op.py +0 -15
  285. orca_sdk/_generated_api_client/models/telemetry_memories_request.py +0 -181
  286. orca_sdk/_generated_api_client/models/telemetry_sort_options.py +0 -185
  287. orca_sdk/_generated_api_client/models/telemetry_sort_options_direction.py +0 -9
  288. orca_sdk/_generated_api_client/models/unauthenticated_error_response.py +0 -73
  289. orca_sdk/_generated_api_client/models/unauthenticated_error_response_status_code.py +0 -8
  290. orca_sdk/_generated_api_client/models/unauthorized_error_response.py +0 -81
  291. orca_sdk/_generated_api_client/models/unauthorized_error_response_status_code.py +0 -8
  292. orca_sdk/_generated_api_client/models/update_org_plan_request.py +0 -73
  293. orca_sdk/_generated_api_client/models/update_org_plan_request_tier.py +0 -11
  294. orca_sdk/_generated_api_client/models/update_prediction_request.py +0 -133
  295. orca_sdk/_generated_api_client/models/validation_error.py +0 -99
  296. orca_sdk/_generated_api_client/py.typed +0 -1
  297. orca_sdk/_generated_api_client/types.py +0 -56
  298. orca_sdk-0.0.97.dist-info/RECORD +0 -309
  299. {orca_sdk-0.0.97.dist-info → orca_sdk-0.0.100.dist-info}/WHEEL +0 -0
@@ -1,216 +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/5cfe4e1d594951725844ea470fc9d61f40c08093/openapi_python_client/templates/client.py.jinja
6
-
7
- The main changes are:
8
- - add a lazy initialized global client singleton that is injected into every api method via a context manager
9
- - add methods for setting headers and base URL on the global client singleton
10
- """
11
-
12
- import os
13
- import ssl
14
- from contextlib import contextmanager
15
- from typing import Any, Optional, Union
16
-
17
- import httpx
18
- from attrs import define, evolve, field
19
-
20
-
21
- @define
22
- class Client:
23
- """A class for keeping track of data related to the API
24
-
25
- The following are accepted as keyword arguments and will be used to construct httpx Clients internally:
26
-
27
- ``base_url``: The base URL for the API, all requests are made to a relative path to this URL
28
-
29
- ``cookies``: A dictionary of cookies to be sent with every request
30
-
31
- ``headers``: A dictionary of headers to be sent with every request
32
-
33
- ``timeout``: The maximum amount of a time a request can take. API functions will raise
34
- httpx.TimeoutException if this is exceeded.
35
-
36
- ``verify_ssl``: Whether or not to verify the SSL certificate of the API server. This should be True in production,
37
- but can be set to False for testing purposes.
38
-
39
- ``follow_redirects``: Whether or not to follow redirects. Default value is False.
40
-
41
- ``httpx_args``: A dictionary of additional arguments to be passed to the ``httpx.Client`` and ``httpx.AsyncClient`` constructor.
42
-
43
-
44
- Attributes:
45
- raise_on_unexpected_status: Whether or not to raise an errors.UnexpectedStatus if the API returns a
46
- status code that was not documented in the source OpenAPI document. Can also be provided as a keyword
47
- argument to the constructor.
48
- """
49
-
50
- raise_on_unexpected_status: bool = field(default=False, kw_only=True)
51
- _base_url: str = field(alias="base_url")
52
- _cookies: dict[str, str] = field(factory=dict, kw_only=True, alias="cookies")
53
- _headers: dict[str, str] = field(factory=dict, kw_only=True, alias="headers")
54
- _timeout: Optional[httpx.Timeout] = field(default=None, kw_only=True, alias="timeout")
55
- _verify_ssl: Union[str, bool, ssl.SSLContext] = field(default=True, kw_only=True, alias="verify_ssl")
56
- _follow_redirects: bool = field(default=False, kw_only=True, alias="follow_redirects")
57
- _httpx_args: dict[str, Any] = field(factory=dict, kw_only=True, alias="httpx_args")
58
- _client: Optional[httpx.Client] = field(default=None, init=False)
59
- _async_client: Optional[httpx.AsyncClient] = field(default=None, init=False)
60
-
61
- def with_headers(self, headers: dict[str, str]) -> "Client":
62
- """Get a new client matching this one with additional headers"""
63
- if self._client is not None:
64
- self._client.headers.update(headers)
65
- if self._async_client is not None:
66
- self._async_client.headers.update(headers)
67
- return evolve(self, headers={**self._headers, **headers})
68
-
69
- def with_cookies(self, cookies: dict[str, str]) -> "Client":
70
- """Get a new client matching this one with additional cookies"""
71
- if self._client is not None:
72
- self._client.cookies.update(cookies)
73
- if self._async_client is not None:
74
- self._async_client.cookies.update(cookies)
75
- return evolve(self, cookies={**self._cookies, **cookies})
76
-
77
- def with_timeout(self, timeout: httpx.Timeout) -> "Client":
78
- """Get a new client matching this one with a new timeout (in seconds)"""
79
- if self._client is not None:
80
- self._client.timeout = timeout
81
- if self._async_client is not None:
82
- self._async_client.timeout = timeout
83
- return evolve(self, timeout=timeout)
84
-
85
- def set_httpx_client(self, client: httpx.Client) -> "Client":
86
- """Manually set the underlying httpx.Client
87
-
88
- **NOTE**: This will override any other settings on the client, including cookies, headers, and timeout.
89
- """
90
- self._client = client
91
- return self
92
-
93
- def get_httpx_client(self) -> httpx.Client:
94
- """Get the underlying httpx.Client, constructing a new one if not previously set"""
95
- if self._client is None:
96
- self._client = httpx.Client(
97
- base_url=self._base_url,
98
- cookies=self._cookies,
99
- headers=self._headers,
100
- timeout=self._timeout,
101
- verify=self._verify_ssl,
102
- follow_redirects=self._follow_redirects,
103
- **self._httpx_args,
104
- )
105
- return self._client
106
-
107
- def __enter__(self) -> "Client":
108
- """Enter a context manager for self.client—you cannot enter twice (see httpx docs)"""
109
- self.get_httpx_client().__enter__()
110
- return self
111
-
112
- def __exit__(self, *args: Any, **kwargs: Any) -> None:
113
- """Exit a context manager for internal httpx.Client (see httpx docs)"""
114
- self.get_httpx_client().__exit__(*args, **kwargs)
115
-
116
- def set_async_httpx_client(self, async_client: httpx.AsyncClient) -> "Client":
117
- """Manually the underlying httpx.AsyncClient
118
-
119
- **NOTE**: This will override any other settings on the client, including cookies, headers, and timeout.
120
- """
121
- self._async_client = async_client
122
- return self
123
-
124
- def get_async_httpx_client(self) -> httpx.AsyncClient:
125
- """Get the underlying httpx.AsyncClient, constructing a new one if not previously set"""
126
- if self._async_client is None:
127
- self._async_client = httpx.AsyncClient(
128
- base_url=self._base_url,
129
- cookies=self._cookies,
130
- headers=self._headers,
131
- timeout=self._timeout,
132
- verify=self._verify_ssl,
133
- follow_redirects=self._follow_redirects,
134
- **self._httpx_args,
135
- )
136
- return self._async_client
137
-
138
- async def __aenter__(self) -> "Client":
139
- """Enter a context manager for underlying httpx.AsyncClient—you cannot enter twice (see httpx docs)"""
140
- await self.get_async_httpx_client().__aenter__()
141
- return self
142
-
143
- async def __aexit__(self, *args: Any, **kwargs: Any) -> None:
144
- """Exit a context manager for underlying httpx.AsyncClient (see httpx docs)"""
145
- await self.get_async_httpx_client().__aexit__(*args, **kwargs)
146
-
147
-
148
- # this must be initialized lazily to ensure that the environment variables are set
149
- _client: Optional["Client"] = None
150
-
151
-
152
- def get_base_url():
153
- """Get the base URL for the global client singleton"""
154
- global _client
155
- if _client is not None:
156
- return _client._base_url
157
- return os.environ.get("ORCA_API_URL", "https://api.orcadb.ai/")
158
-
159
-
160
- def get_headers():
161
- """Get the headers for the global client singleton"""
162
- global _client
163
- if _client is not None:
164
- return _client._headers
165
- headers = {}
166
- api_key = os.environ.get("ORCA_API_KEY")
167
- if api_key is not None:
168
- headers["Api-Key"] = api_key
169
- return headers
170
-
171
-
172
- def get_client():
173
- """Get the global client singleton"""
174
- global _client
175
- if _client is None:
176
- _client = Client(base_url=get_base_url(), headers=get_headers())
177
- return _client
178
-
179
-
180
- def set_headers(headers: dict[str, str]):
181
- """Set the headers for the global client singleton"""
182
- global _client
183
- _client = Client(base_url=get_base_url(), headers=headers)
184
- return _client
185
-
186
-
187
- def set_base_url(base_url: str):
188
- """Set the base URL for the global client singleton"""
189
- global _client
190
- _client = Client(base_url=base_url, headers=_client._headers if _client is not None else {})
191
- return _client
192
-
193
-
194
- @contextmanager
195
- def headers_context(headers: dict[str, str]):
196
- """Set the headers for the global client singleton within a context manager"""
197
- global _client
198
- old_client = get_client()
199
- set_headers(headers=headers)
200
- yield
201
- _client = old_client
202
-
203
-
204
- @contextmanager
205
- def base_url_context(base_url: str):
206
- """Set the base URL for the global client singleton within a context manager"""
207
- global _client
208
- old_client = get_client()
209
- set_base_url(base_url=base_url)
210
- yield
211
- _client = old_client
212
-
213
-
214
- @contextmanager
215
- def _client_context():
216
- yield get_client()
@@ -1,38 +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/errors.py.jinja
6
-
7
- The main change is:
8
- - Returning the appropriate python exception for Orca API's error responses
9
- """
10
-
11
- import httpx
12
-
13
-
14
- def get_error_for_response(response: httpx.Response):
15
- """Raise an appropriate exception based on the response status code and content"""
16
- if response.status_code == 401:
17
- return ValueError("Invalid API key")
18
- elif response.status_code == 403:
19
- return PermissionError(response.json()["reason"])
20
- elif response.status_code == 404:
21
- resource = response.json()["resource"]
22
- if resource is not None:
23
- return LookupError(f"The {resource} you are looking for does not exist")
24
- else:
25
- return RuntimeError(f"Unknown API route: {response.url}")
26
- elif response.status_code == 405:
27
- return RuntimeError(f"Unknown method {response.request.method} for API route: {response.url}")
28
- elif response.status_code == 409:
29
- return RuntimeError(response.json()["constraint"])
30
- elif response.status_code == 422:
31
- issues = [f"{issue['loc'][-1]}: {issue['msg']}" for issue in response.json()["validation_issues"]]
32
- return ValueError("Invalid input:\n\t" + "\n\t".join(issues))
33
- elif response.status_code == 500:
34
- return RuntimeError(f"Unexpected server error: {response.json()['message']}")
35
- elif response.status_code == 503:
36
- return RuntimeError("Orca API is currently unavailable, please try again later")
37
- else:
38
- return RuntimeError(f"Unexpected status code: {response.status_code}")
@@ -1,345 +0,0 @@
1
- """Contains all the data models used in inputs/outputs"""
2
-
3
- from .action_recommendation import ActionRecommendation
4
- from .action_recommendation_action import ActionRecommendationAction
5
- from .add_memory_recommendations import AddMemoryRecommendations
6
- from .add_memory_suggestion import AddMemorySuggestion
7
- from .analyze_neighbor_labels_result import AnalyzeNeighborLabelsResult
8
- from .api_key_metadata import ApiKeyMetadata
9
- from .api_key_metadata_scope_item import ApiKeyMetadataScopeItem
10
- from .base_label_prediction_result import BaseLabelPredictionResult
11
- from .base_model import BaseModel
12
- from .base_score_prediction_result import BaseScorePredictionResult
13
- from .body_create_datasource_from_files_datasource_upload_post import BodyCreateDatasourceFromFilesDatasourceUploadPost
14
- from .cascade_edit_suggestions_request import CascadeEditSuggestionsRequest
15
- from .cascading_edit_suggestion import CascadingEditSuggestion
16
- from .class_representatives import ClassRepresentatives
17
- from .classification_evaluation_request import ClassificationEvaluationRequest
18
- from .classification_metrics import ClassificationMetrics
19
- from .classification_model_metadata import ClassificationModelMetadata
20
- from .classification_prediction_request import ClassificationPredictionRequest
21
- from .clone_memoryset_request import CloneMemorysetRequest
22
- from .cluster_metrics import ClusterMetrics
23
- from .column_info import ColumnInfo
24
- from .column_type import ColumnType
25
- from .constraint_violation_error_response import ConstraintViolationErrorResponse
26
- from .constraint_violation_error_response_status_code import ConstraintViolationErrorResponseStatusCode
27
- from .count_predictions_request import CountPredictionsRequest
28
- from .create_api_key_request import CreateApiKeyRequest
29
- from .create_api_key_request_scope_item import CreateApiKeyRequestScopeItem
30
- from .create_api_key_response import CreateApiKeyResponse
31
- from .create_api_key_response_scope_item import CreateApiKeyResponseScopeItem
32
- from .create_classification_model_request import CreateClassificationModelRequest
33
- from .create_datasource_from_content_request import CreateDatasourceFromContentRequest
34
- from .create_memoryset_request import CreateMemorysetRequest
35
- from .create_memoryset_request_index_params import CreateMemorysetRequestIndexParams
36
- from .create_memoryset_request_index_type import CreateMemorysetRequestIndexType
37
- from .create_org_plan_request import CreateOrgPlanRequest
38
- from .create_org_plan_request_tier import CreateOrgPlanRequestTier
39
- from .create_regression_model_request import CreateRegressionModelRequest
40
- from .datasource_metadata import DatasourceMetadata
41
- from .delete_memories_request import DeleteMemoriesRequest
42
- from .delete_memorysets_request import DeleteMemorysetsRequest
43
- from .embed_request import EmbedRequest
44
- from .embedding_evaluation_payload import EmbeddingEvaluationPayload
45
- from .embedding_evaluation_request import EmbeddingEvaluationRequest
46
- from .embedding_evaluation_response import EmbeddingEvaluationResponse
47
- from .embedding_evaluation_result import EmbeddingEvaluationResult
48
- from .embedding_finetuning_method import EmbeddingFinetuningMethod
49
- from .embedding_model_result import EmbeddingModelResult
50
- from .evaluation_response import EvaluationResponse
51
- from .evaluation_response_classification_metrics import EvaluationResponseClassificationMetrics
52
- from .evaluation_response_regression_metrics import EvaluationResponseRegressionMetrics
53
- from .feedback_metrics import FeedbackMetrics
54
- from .feedback_type import FeedbackType
55
- from .filter_item import FilterItem
56
- from .filter_item_field_type_0_item import FilterItemFieldType0Item
57
- from .filter_item_field_type_1_item_type_0 import FilterItemFieldType1ItemType0
58
- from .filter_item_field_type_2_item_type_0 import FilterItemFieldType2ItemType0
59
- from .filter_item_field_type_2_item_type_1 import FilterItemFieldType2ItemType1
60
- from .filter_item_op import FilterItemOp
61
- from .finetune_embedding_model_request import FinetuneEmbeddingModelRequest
62
- from .finetune_embedding_model_request_training_args import FinetuneEmbeddingModelRequestTrainingArgs
63
- from .finetuned_embedding_model_metadata import FinetunedEmbeddingModelMetadata
64
- from .get_memories_request import GetMemoriesRequest
65
- from .http_validation_error import HTTPValidationError
66
- from .internal_server_error_response import InternalServerErrorResponse
67
- from .internal_server_error_response_status_code import InternalServerErrorResponseStatusCode
68
- from .label_class_metrics import LabelClassMetrics
69
- from .label_prediction_memory_lookup import LabelPredictionMemoryLookup
70
- from .label_prediction_memory_lookup_metadata import LabelPredictionMemoryLookupMetadata
71
- from .label_prediction_with_memories_and_feedback import LabelPredictionWithMemoriesAndFeedback
72
- from .labeled_memory import LabeledMemory
73
- from .labeled_memory_insert import LabeledMemoryInsert
74
- from .labeled_memory_insert_metadata import LabeledMemoryInsertMetadata
75
- from .labeled_memory_lookup import LabeledMemoryLookup
76
- from .labeled_memory_lookup_metadata import LabeledMemoryLookupMetadata
77
- from .labeled_memory_metadata import LabeledMemoryMetadata
78
- from .labeled_memory_update import LabeledMemoryUpdate
79
- from .labeled_memory_update_metadata_type_0 import LabeledMemoryUpdateMetadataType0
80
- from .labeled_memory_with_feedback_metrics import LabeledMemoryWithFeedbackMetrics
81
- from .labeled_memory_with_feedback_metrics_feedback_metrics import LabeledMemoryWithFeedbackMetricsFeedbackMetrics
82
- from .labeled_memory_with_feedback_metrics_metadata import LabeledMemoryWithFeedbackMetricsMetadata
83
- from .list_memories_request import ListMemoriesRequest
84
- from .list_predictions_request import ListPredictionsRequest
85
- from .lookup_request import LookupRequest
86
- from .lookup_score_metrics import LookupScoreMetrics
87
- from .memory_metrics import MemoryMetrics
88
- from .memory_type import MemoryType
89
- from .memoryset_analysis_configs import MemorysetAnalysisConfigs
90
- from .memoryset_analysis_request import MemorysetAnalysisRequest
91
- from .memoryset_analysis_response import MemorysetAnalysisResponse
92
- from .memoryset_class_patterns_analysis_config import MemorysetClassPatternsAnalysisConfig
93
- from .memoryset_class_patterns_metrics import MemorysetClassPatternsMetrics
94
- from .memoryset_cluster_analysis_config import MemorysetClusterAnalysisConfig
95
- from .memoryset_cluster_analysis_config_clustering_method import MemorysetClusterAnalysisConfigClusteringMethod
96
- from .memoryset_cluster_analysis_config_partitioning_method import MemorysetClusterAnalysisConfigPartitioningMethod
97
- from .memoryset_cluster_metrics import MemorysetClusterMetrics
98
- from .memoryset_duplicate_analysis_config import MemorysetDuplicateAnalysisConfig
99
- from .memoryset_duplicate_metrics import MemorysetDuplicateMetrics
100
- from .memoryset_label_analysis_config import MemorysetLabelAnalysisConfig
101
- from .memoryset_label_metrics import MemorysetLabelMetrics
102
- from .memoryset_metadata import MemorysetMetadata
103
- from .memoryset_metadata_index_params import MemorysetMetadataIndexParams
104
- from .memoryset_metadata_index_type import MemorysetMetadataIndexType
105
- from .memoryset_metrics import MemorysetMetrics
106
- from .memoryset_neighbor_analysis_config import MemorysetNeighborAnalysisConfig
107
- from .memoryset_neighbor_metrics import MemorysetNeighborMetrics
108
- from .memoryset_neighbor_metrics_lookup_score_metrics import MemorysetNeighborMetricsLookupScoreMetrics
109
- from .memoryset_projection_analysis_config import MemorysetProjectionAnalysisConfig
110
- from .memoryset_projection_metrics import MemorysetProjectionMetrics
111
- from .memoryset_update import MemorysetUpdate
112
- from .not_found_error_response import NotFoundErrorResponse
113
- from .not_found_error_response_resource_type_0 import NotFoundErrorResponseResourceType0
114
- from .not_found_error_response_status_code import NotFoundErrorResponseStatusCode
115
- from .org_plan import OrgPlan
116
- from .org_plan_tier import OrgPlanTier
117
- from .paginated_task import PaginatedTask
118
- from .paginated_union_labeled_memory_with_feedback_metrics_scored_memory_with_feedback_metrics import (
119
- PaginatedUnionLabeledMemoryWithFeedbackMetricsScoredMemoryWithFeedbackMetrics,
120
- )
121
- from .pr_curve import PRCurve
122
- from .prediction_feedback import PredictionFeedback
123
- from .prediction_feedback_category import PredictionFeedbackCategory
124
- from .prediction_feedback_request import PredictionFeedbackRequest
125
- from .prediction_feedback_result import PredictionFeedbackResult
126
- from .prediction_sort_item_item_type_0 import PredictionSortItemItemType0
127
- from .prediction_sort_item_item_type_1 import PredictionSortItemItemType1
128
- from .predictive_model_update import PredictiveModelUpdate
129
- from .pretrained_embedding_model_metadata import PretrainedEmbeddingModelMetadata
130
- from .pretrained_embedding_model_name import PretrainedEmbeddingModelName
131
- from .rac_head_type import RACHeadType
132
- from .rar_head_type import RARHeadType
133
- from .regression_evaluation_request import RegressionEvaluationRequest
134
- from .regression_metrics import RegressionMetrics
135
- from .regression_model_metadata import RegressionModelMetadata
136
- from .regression_prediction_request import RegressionPredictionRequest
137
- from .roc_curve import ROCCurve
138
- from .score_prediction_memory_lookup import ScorePredictionMemoryLookup
139
- from .score_prediction_memory_lookup_metadata import ScorePredictionMemoryLookupMetadata
140
- from .score_prediction_with_memories_and_feedback import ScorePredictionWithMemoriesAndFeedback
141
- from .scored_memory import ScoredMemory
142
- from .scored_memory_insert import ScoredMemoryInsert
143
- from .scored_memory_insert_metadata import ScoredMemoryInsertMetadata
144
- from .scored_memory_lookup import ScoredMemoryLookup
145
- from .scored_memory_lookup_metadata import ScoredMemoryLookupMetadata
146
- from .scored_memory_metadata import ScoredMemoryMetadata
147
- from .scored_memory_update import ScoredMemoryUpdate
148
- from .scored_memory_update_metadata_type_0 import ScoredMemoryUpdateMetadataType0
149
- from .scored_memory_with_feedback_metrics import ScoredMemoryWithFeedbackMetrics
150
- from .scored_memory_with_feedback_metrics_feedback_metrics import ScoredMemoryWithFeedbackMetricsFeedbackMetrics
151
- from .scored_memory_with_feedback_metrics_metadata import ScoredMemoryWithFeedbackMetricsMetadata
152
- from .service_unavailable_error_response import ServiceUnavailableErrorResponse
153
- from .service_unavailable_error_response_status_code import ServiceUnavailableErrorResponseStatusCode
154
- from .task import Task
155
- from .task_status import TaskStatus
156
- from .task_status_info import TaskStatusInfo
157
- from .telemetry_field_type_0_item_type_0 import TelemetryFieldType0ItemType0
158
- from .telemetry_field_type_0_item_type_2 import TelemetryFieldType0ItemType2
159
- from .telemetry_field_type_1_item_type_0 import TelemetryFieldType1ItemType0
160
- from .telemetry_field_type_1_item_type_1 import TelemetryFieldType1ItemType1
161
- from .telemetry_filter_item import TelemetryFilterItem
162
- from .telemetry_filter_item_op import TelemetryFilterItemOp
163
- from .telemetry_memories_request import TelemetryMemoriesRequest
164
- from .telemetry_sort_options import TelemetrySortOptions
165
- from .telemetry_sort_options_direction import TelemetrySortOptionsDirection
166
- from .unauthenticated_error_response import UnauthenticatedErrorResponse
167
- from .unauthenticated_error_response_status_code import UnauthenticatedErrorResponseStatusCode
168
- from .unauthorized_error_response import UnauthorizedErrorResponse
169
- from .unauthorized_error_response_status_code import UnauthorizedErrorResponseStatusCode
170
- from .update_org_plan_request import UpdateOrgPlanRequest
171
- from .update_org_plan_request_tier import UpdateOrgPlanRequestTier
172
- from .update_prediction_request import UpdatePredictionRequest
173
- from .validation_error import ValidationError
174
-
175
- __all__ = (
176
- "ActionRecommendation",
177
- "ActionRecommendationAction",
178
- "AddMemoryRecommendations",
179
- "AddMemorySuggestion",
180
- "AnalyzeNeighborLabelsResult",
181
- "ApiKeyMetadata",
182
- "ApiKeyMetadataScopeItem",
183
- "BaseLabelPredictionResult",
184
- "BaseModel",
185
- "BaseScorePredictionResult",
186
- "BodyCreateDatasourceFromFilesDatasourceUploadPost",
187
- "CascadeEditSuggestionsRequest",
188
- "CascadingEditSuggestion",
189
- "ClassificationEvaluationRequest",
190
- "ClassificationMetrics",
191
- "ClassificationModelMetadata",
192
- "ClassificationPredictionRequest",
193
- "ClassRepresentatives",
194
- "CloneMemorysetRequest",
195
- "ClusterMetrics",
196
- "ColumnInfo",
197
- "ColumnType",
198
- "ConstraintViolationErrorResponse",
199
- "ConstraintViolationErrorResponseStatusCode",
200
- "CountPredictionsRequest",
201
- "CreateApiKeyRequest",
202
- "CreateApiKeyRequestScopeItem",
203
- "CreateApiKeyResponse",
204
- "CreateApiKeyResponseScopeItem",
205
- "CreateClassificationModelRequest",
206
- "CreateDatasourceFromContentRequest",
207
- "CreateMemorysetRequest",
208
- "CreateMemorysetRequestIndexParams",
209
- "CreateMemorysetRequestIndexType",
210
- "CreateOrgPlanRequest",
211
- "CreateOrgPlanRequestTier",
212
- "CreateRegressionModelRequest",
213
- "DatasourceMetadata",
214
- "DeleteMemoriesRequest",
215
- "DeleteMemorysetsRequest",
216
- "EmbeddingEvaluationPayload",
217
- "EmbeddingEvaluationRequest",
218
- "EmbeddingEvaluationResponse",
219
- "EmbeddingEvaluationResult",
220
- "EmbeddingFinetuningMethod",
221
- "EmbeddingModelResult",
222
- "EmbedRequest",
223
- "EvaluationResponse",
224
- "EvaluationResponseClassificationMetrics",
225
- "EvaluationResponseRegressionMetrics",
226
- "FeedbackMetrics",
227
- "FeedbackType",
228
- "FilterItem",
229
- "FilterItemFieldType0Item",
230
- "FilterItemFieldType1ItemType0",
231
- "FilterItemFieldType2ItemType0",
232
- "FilterItemFieldType2ItemType1",
233
- "FilterItemOp",
234
- "FinetunedEmbeddingModelMetadata",
235
- "FinetuneEmbeddingModelRequest",
236
- "FinetuneEmbeddingModelRequestTrainingArgs",
237
- "GetMemoriesRequest",
238
- "HTTPValidationError",
239
- "InternalServerErrorResponse",
240
- "InternalServerErrorResponseStatusCode",
241
- "LabelClassMetrics",
242
- "LabeledMemory",
243
- "LabeledMemoryInsert",
244
- "LabeledMemoryInsertMetadata",
245
- "LabeledMemoryLookup",
246
- "LabeledMemoryLookupMetadata",
247
- "LabeledMemoryMetadata",
248
- "LabeledMemoryUpdate",
249
- "LabeledMemoryUpdateMetadataType0",
250
- "LabeledMemoryWithFeedbackMetrics",
251
- "LabeledMemoryWithFeedbackMetricsFeedbackMetrics",
252
- "LabeledMemoryWithFeedbackMetricsMetadata",
253
- "LabelPredictionMemoryLookup",
254
- "LabelPredictionMemoryLookupMetadata",
255
- "LabelPredictionWithMemoriesAndFeedback",
256
- "ListMemoriesRequest",
257
- "ListPredictionsRequest",
258
- "LookupRequest",
259
- "LookupScoreMetrics",
260
- "MemoryMetrics",
261
- "MemorysetAnalysisConfigs",
262
- "MemorysetAnalysisRequest",
263
- "MemorysetAnalysisResponse",
264
- "MemorysetClassPatternsAnalysisConfig",
265
- "MemorysetClassPatternsMetrics",
266
- "MemorysetClusterAnalysisConfig",
267
- "MemorysetClusterAnalysisConfigClusteringMethod",
268
- "MemorysetClusterAnalysisConfigPartitioningMethod",
269
- "MemorysetClusterMetrics",
270
- "MemorysetDuplicateAnalysisConfig",
271
- "MemorysetDuplicateMetrics",
272
- "MemorysetLabelAnalysisConfig",
273
- "MemorysetLabelMetrics",
274
- "MemorysetMetadata",
275
- "MemorysetMetadataIndexParams",
276
- "MemorysetMetadataIndexType",
277
- "MemorysetMetrics",
278
- "MemorysetNeighborAnalysisConfig",
279
- "MemorysetNeighborMetrics",
280
- "MemorysetNeighborMetricsLookupScoreMetrics",
281
- "MemorysetProjectionAnalysisConfig",
282
- "MemorysetProjectionMetrics",
283
- "MemorysetUpdate",
284
- "MemoryType",
285
- "NotFoundErrorResponse",
286
- "NotFoundErrorResponseResourceType0",
287
- "NotFoundErrorResponseStatusCode",
288
- "OrgPlan",
289
- "OrgPlanTier",
290
- "PaginatedTask",
291
- "PaginatedUnionLabeledMemoryWithFeedbackMetricsScoredMemoryWithFeedbackMetrics",
292
- "PRCurve",
293
- "PredictionFeedback",
294
- "PredictionFeedbackCategory",
295
- "PredictionFeedbackRequest",
296
- "PredictionFeedbackResult",
297
- "PredictionSortItemItemType0",
298
- "PredictionSortItemItemType1",
299
- "PredictiveModelUpdate",
300
- "PretrainedEmbeddingModelMetadata",
301
- "PretrainedEmbeddingModelName",
302
- "RACHeadType",
303
- "RARHeadType",
304
- "RegressionEvaluationRequest",
305
- "RegressionMetrics",
306
- "RegressionModelMetadata",
307
- "RegressionPredictionRequest",
308
- "ROCCurve",
309
- "ScoredMemory",
310
- "ScoredMemoryInsert",
311
- "ScoredMemoryInsertMetadata",
312
- "ScoredMemoryLookup",
313
- "ScoredMemoryLookupMetadata",
314
- "ScoredMemoryMetadata",
315
- "ScoredMemoryUpdate",
316
- "ScoredMemoryUpdateMetadataType0",
317
- "ScoredMemoryWithFeedbackMetrics",
318
- "ScoredMemoryWithFeedbackMetricsFeedbackMetrics",
319
- "ScoredMemoryWithFeedbackMetricsMetadata",
320
- "ScorePredictionMemoryLookup",
321
- "ScorePredictionMemoryLookupMetadata",
322
- "ScorePredictionWithMemoriesAndFeedback",
323
- "ServiceUnavailableErrorResponse",
324
- "ServiceUnavailableErrorResponseStatusCode",
325
- "Task",
326
- "TaskStatus",
327
- "TaskStatusInfo",
328
- "TelemetryFieldType0ItemType0",
329
- "TelemetryFieldType0ItemType2",
330
- "TelemetryFieldType1ItemType0",
331
- "TelemetryFieldType1ItemType1",
332
- "TelemetryFilterItem",
333
- "TelemetryFilterItemOp",
334
- "TelemetryMemoriesRequest",
335
- "TelemetrySortOptions",
336
- "TelemetrySortOptionsDirection",
337
- "UnauthenticatedErrorResponse",
338
- "UnauthenticatedErrorResponseStatusCode",
339
- "UnauthorizedErrorResponse",
340
- "UnauthorizedErrorResponseStatusCode",
341
- "UpdateOrgPlanRequest",
342
- "UpdateOrgPlanRequestTier",
343
- "UpdatePredictionRequest",
344
- "ValidationError",
345
- )
@@ -1,82 +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 enum import Enum
14
- from typing import Any, Type, TypeVar
15
-
16
- from attrs import define as _attrs_define
17
- from attrs import field as _attrs_field
18
-
19
- from ..models.action_recommendation_action import ActionRecommendationAction
20
-
21
- T = TypeVar("T", bound="ActionRecommendation")
22
-
23
-
24
- @_attrs_define
25
- class ActionRecommendation:
26
- """Recommended actions to take based on the prediction and explanation
27
-
28
- Attributes:
29
- action (ActionRecommendationAction): The recommended action to take
30
- rationale (str): Explanation for why this action was recommended
31
- """
32
-
33
- action: ActionRecommendationAction
34
- rationale: str
35
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
36
-
37
- def to_dict(self) -> dict[str, Any]:
38
- action = self.action.value if isinstance(self.action, Enum) else self.action
39
-
40
- rationale = self.rationale
41
-
42
- field_dict: dict[str, Any] = {}
43
- field_dict.update(self.additional_properties)
44
- field_dict.update(
45
- {
46
- "action": action,
47
- "rationale": rationale,
48
- }
49
- )
50
-
51
- return field_dict
52
-
53
- @classmethod
54
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
55
- d = src_dict.copy()
56
- action = ActionRecommendationAction(d.pop("action"))
57
-
58
- rationale = d.pop("rationale")
59
-
60
- action_recommendation = cls(
61
- action=action,
62
- rationale=rationale,
63
- )
64
-
65
- action_recommendation.additional_properties = d
66
- return action_recommendation
67
-
68
- @property
69
- def additional_keys(self) -> list[str]:
70
- return list(self.additional_properties.keys())
71
-
72
- def __getitem__(self, key: str) -> Any:
73
- return self.additional_properties[key]
74
-
75
- def __setitem__(self, key: str, value: Any) -> None:
76
- self.additional_properties[key] = value
77
-
78
- def __delitem__(self, key: str) -> None:
79
- del self.additional_properties[key]
80
-
81
- def __contains__(self, key: str) -> bool:
82
- return key in self.additional_properties