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,307 +0,0 @@
1
- """Contains methods for accessing the API"""
2
-
3
- from .auth.check_authentication_auth_get import sync as check_authentication
4
- from .auth.create_api_key_auth_api_key_post import sync as create_api_key
5
- from .auth.create_org_plan_auth_org_plan_post import sync as create_org_plan
6
- from .auth.delete_api_key_auth_api_key_name_or_id_delete import sync as delete_api_key
7
- from .auth.delete_org_auth_org_delete import sync as delete_org
8
- from .auth.get_org_plan_auth_org_plan_get import sync as get_org_plan
9
- from .auth.list_api_keys_auth_api_key_get import sync as list_api_keys
10
- from .auth.update_org_plan_auth_org_plan_put import sync as update_org_plan
11
- from .classification_model.create_classification_model_classification_model_post import (
12
- sync as create_classification_model,
13
- )
14
- from .classification_model.delete_classification_model_classification_model_name_or_id_delete import (
15
- sync as delete_classification_model,
16
- )
17
- from .classification_model.delete_classification_model_evaluation_classification_model_model_name_or_id_evaluation_task_id_delete import (
18
- sync as delete_classification_model_evaluation,
19
- )
20
- from .classification_model.evaluate_classification_model_classification_model_model_name_or_id_evaluation_post import (
21
- sync as evaluate_classification_model,
22
- )
23
- from .classification_model.get_classification_model_classification_model_name_or_id_get import (
24
- sync as get_classification_model,
25
- )
26
- from .classification_model.get_classification_model_evaluation_classification_model_model_name_or_id_evaluation_task_id_get import (
27
- sync as get_classification_model_evaluation,
28
- )
29
- from .classification_model.list_classification_model_evaluations_classification_model_model_name_or_id_evaluation_get import (
30
- sync as list_classification_model_evaluations,
31
- )
32
- from .classification_model.list_classification_models_classification_model_get import (
33
- sync as list_classification_models,
34
- )
35
- from .classification_model.predict_label_gpu_classification_model_name_or_id_prediction_post import (
36
- sync as predict_label_gpu,
37
- )
38
- from .classification_model.update_classification_model_classification_model_name_or_id_patch import (
39
- sync as update_classification_model,
40
- )
41
- from .datasource.create_datasource_from_content_datasource_post import (
42
- sync as create_datasource_from_content,
43
- )
44
- from .datasource.create_datasource_from_files_datasource_upload_post import (
45
- sync as create_datasource_from_files,
46
- )
47
- from .datasource.create_embedding_evaluation_datasource_name_or_id_embedding_evaluation_post import (
48
- sync as create_embedding_evaluation,
49
- )
50
- from .datasource.delete_datasource_datasource_name_or_id_delete import (
51
- sync as delete_datasource,
52
- )
53
- from .datasource.download_datasource_datasource_name_or_id_download_get import (
54
- sync as download_datasource,
55
- )
56
- from .datasource.get_datasource_datasource_name_or_id_get import sync as get_datasource
57
- from .datasource.get_embedding_evaluation_datasource_name_or_id_embedding_evaluation_task_id_get import (
58
- sync as get_embedding_evaluation,
59
- )
60
- from .datasource.list_datasources_datasource_get import sync as list_datasources
61
- from .datasource.list_embedding_evaluations_datasource_name_or_id_embedding_evaluation_get import (
62
- sync as list_embedding_evaluations,
63
- )
64
- from .default.healthcheck_get import sync as healthcheck
65
- from .default.healthcheck_gpu_get import sync as healthcheck_gpu
66
- from .finetuned_embedding_model.create_finetuned_embedding_model_finetuned_embedding_model_post import (
67
- sync as create_finetuned_embedding_model,
68
- )
69
- from .finetuned_embedding_model.delete_finetuned_embedding_model_finetuned_embedding_model_name_or_id_delete import (
70
- sync as delete_finetuned_embedding_model,
71
- )
72
- from .finetuned_embedding_model.embed_with_finetuned_model_gpu_finetuned_embedding_model_name_or_id_embedding_post import (
73
- sync as embed_with_finetuned_model_gpu,
74
- )
75
- from .finetuned_embedding_model.get_finetuned_embedding_model_finetuned_embedding_model_name_or_id_get import (
76
- sync as get_finetuned_embedding_model,
77
- )
78
- from .finetuned_embedding_model.list_finetuned_embedding_models_finetuned_embedding_model_get import (
79
- sync as list_finetuned_embedding_models,
80
- )
81
- from .memoryset.analyze_memoryset_memoryset_name_or_id_analysis_post import (
82
- sync as analyze_memoryset,
83
- )
84
- from .memoryset.batch_delete_memoryset_batch_delete_memoryset_post import (
85
- sync as batch_delete_memoryset_batch_delete,
86
- )
87
- from .memoryset.clone_memoryset_memoryset_name_or_id_clone_post import (
88
- sync as clone_memoryset,
89
- )
90
- from .memoryset.create_memoryset_memoryset_post import sync as create_memoryset
91
- from .memoryset.delete_memories_memoryset_name_or_id_memories_delete_post import (
92
- sync as delete_memories,
93
- )
94
- from .memoryset.delete_memory_memoryset_name_or_id_memory_memory_id_delete import (
95
- sync as delete_memory,
96
- )
97
- from .memoryset.delete_memoryset_memoryset_name_or_id_delete import (
98
- sync as delete_memoryset,
99
- )
100
- from .memoryset.get_analysis_memoryset_name_or_id_analysis_analysis_task_id_get import (
101
- sync as get_analysis,
102
- )
103
- from .memoryset.get_memories_memoryset_name_or_id_memories_get_post import (
104
- sync as get_memories,
105
- )
106
- from .memoryset.get_memory_memoryset_name_or_id_memory_memory_id_get import (
107
- sync as get_memory,
108
- )
109
- from .memoryset.get_memoryset_memoryset_name_or_id_get import sync as get_memoryset
110
- from .memoryset.insert_memories_gpu_memoryset_name_or_id_memory_post import (
111
- sync as insert_memories_gpu,
112
- )
113
- from .memoryset.list_analyses_memoryset_name_or_id_analysis_get import (
114
- sync as list_analyses,
115
- )
116
- from .memoryset.list_memorysets_memoryset_get import sync as list_memorysets
117
- from .memoryset.memoryset_lookup_gpu_memoryset_name_or_id_lookup_post import (
118
- sync as memoryset_lookup_gpu,
119
- )
120
- from .memoryset.potential_duplicate_groups_memoryset_name_or_id_potential_duplicate_groups_get import (
121
- sync as potential_duplicate_groups,
122
- )
123
- from .memoryset.query_memoryset_memoryset_name_or_id_memories_post import (
124
- sync as query_memoryset,
125
- )
126
- from .memoryset.suggest_cascading_edits_memoryset_name_or_id_memory_memory_id_cascading_edits_post import (
127
- sync as suggest_cascading_edits,
128
- )
129
- from .memoryset.update_memories_gpu_memoryset_name_or_id_memories_patch import (
130
- sync as update_memories_gpu,
131
- )
132
- from .memoryset.update_memory_gpu_memoryset_name_or_id_memory_patch import (
133
- sync as update_memory_gpu,
134
- )
135
- from .memoryset.update_memoryset_memoryset_name_or_id_patch import (
136
- sync as update_memoryset,
137
- )
138
- from .predictive_model.list_predictive_models_predictive_model_get import (
139
- sync as list_predictive_models,
140
- )
141
- from .pretrained_embedding_model.embed_with_pretrained_model_gpu_pretrained_embedding_model_model_name_embedding_post import (
142
- sync as embed_with_pretrained_model_gpu,
143
- )
144
- from .pretrained_embedding_model.get_pretrained_embedding_model_pretrained_embedding_model_model_name_get import (
145
- sync as get_pretrained_embedding_model,
146
- )
147
- from .pretrained_embedding_model.list_pretrained_embedding_models_pretrained_embedding_model_get import (
148
- sync as list_pretrained_embedding_models,
149
- )
150
- from .regression_model.create_regression_model_regression_model_post import (
151
- sync as create_regression_model,
152
- )
153
- from .regression_model.delete_regression_model_evaluation_regression_model_model_name_or_id_evaluation_task_id_delete import (
154
- sync as delete_regression_model_evaluation,
155
- )
156
- from .regression_model.delete_regression_model_regression_model_name_or_id_delete import (
157
- sync as delete_regression_model,
158
- )
159
- from .regression_model.evaluate_regression_model_regression_model_model_name_or_id_evaluation_post import (
160
- sync as evaluate_regression_model,
161
- )
162
- from .regression_model.get_regression_model_evaluation_regression_model_model_name_or_id_evaluation_task_id_get import (
163
- sync as get_regression_model_evaluation,
164
- )
165
- from .regression_model.get_regression_model_regression_model_name_or_id_get import (
166
- sync as get_regression_model,
167
- )
168
- from .regression_model.list_regression_model_evaluations_regression_model_model_name_or_id_evaluation_get import (
169
- sync as list_regression_model_evaluations,
170
- )
171
- from .regression_model.list_regression_models_regression_model_get import (
172
- sync as list_regression_models,
173
- )
174
- from .regression_model.predict_score_gpu_regression_model_name_or_id_prediction_post import (
175
- sync as predict_score_gpu,
176
- )
177
- from .regression_model.update_regression_model_regression_model_name_or_id_patch import (
178
- sync as update_regression_model,
179
- )
180
- from .task.abort_task_task_task_id_abort_delete import sync as abort_task
181
- from .task.get_task_status_task_task_id_status_get import sync as get_task_status
182
- from .task.get_task_task_task_id_get import sync as get_task
183
- from .task.list_tasks_task_get import sync as list_tasks
184
- from .telemetry.count_predictions_telemetry_prediction_count_post import (
185
- sync as count_predictions,
186
- )
187
- from .telemetry.drop_feedback_category_with_data_telemetry_feedback_category_name_or_id_delete import (
188
- sync as drop_feedback_category_with_data,
189
- )
190
- from .telemetry.explain_prediction_telemetry_prediction_prediction_id_explanation_get import (
191
- sync as explain_prediction,
192
- )
193
- from .telemetry.generate_memory_suggestions_telemetry_prediction_prediction_id_memory_suggestions_post import (
194
- sync as generate_memory_suggestions,
195
- )
196
- from .telemetry.get_action_recommendation_telemetry_prediction_prediction_id_action_get import (
197
- sync as get_action_recommendation,
198
- )
199
- from .telemetry.get_feedback_category_telemetry_feedback_category_name_or_id_get import (
200
- sync as get_feedback_category,
201
- )
202
- from .telemetry.get_prediction_telemetry_prediction_prediction_id_get import (
203
- sync as get_prediction,
204
- )
205
- from .telemetry.list_feedback_categories_telemetry_feedback_category_get import (
206
- sync as list_feedback_categories,
207
- )
208
- from .telemetry.list_memories_with_feedback_telemetry_memories_post import (
209
- sync as list_memories_with_feedback,
210
- )
211
- from .telemetry.list_predictions_telemetry_prediction_post import (
212
- sync as list_predictions,
213
- )
214
- from .telemetry.record_prediction_feedback_telemetry_prediction_feedback_put import (
215
- sync as record_prediction_feedback,
216
- )
217
- from .telemetry.update_prediction_telemetry_prediction_prediction_id_patch import (
218
- sync as update_prediction,
219
- )
220
-
221
- __all__ = [
222
- "abort_task",
223
- "analyze_memoryset",
224
- "batch_delete_memoryset_batch_delete",
225
- "check_authentication",
226
- "clone_memoryset",
227
- "count_predictions",
228
- "create_api_key",
229
- "create_classification_model",
230
- "create_datasource_from_content",
231
- "create_datasource_from_files",
232
- "create_embedding_evaluation",
233
- "create_finetuned_embedding_model",
234
- "create_memoryset",
235
- "create_org_plan",
236
- "create_regression_model",
237
- "delete_api_key",
238
- "delete_classification_model",
239
- "delete_classification_model_evaluation",
240
- "delete_datasource",
241
- "delete_finetuned_embedding_model",
242
- "delete_memories",
243
- "delete_memory",
244
- "delete_memoryset",
245
- "delete_org",
246
- "delete_regression_model",
247
- "delete_regression_model_evaluation",
248
- "download_datasource",
249
- "drop_feedback_category_with_data",
250
- "embed_with_finetuned_model_gpu",
251
- "embed_with_pretrained_model_gpu",
252
- "evaluate_classification_model",
253
- "evaluate_regression_model",
254
- "explain_prediction",
255
- "generate_memory_suggestions",
256
- "get_action_recommendation",
257
- "get_analysis",
258
- "get_classification_model",
259
- "get_classification_model_evaluation",
260
- "get_datasource",
261
- "get_embedding_evaluation",
262
- "get_feedback_category",
263
- "get_finetuned_embedding_model",
264
- "get_memories",
265
- "get_memory",
266
- "get_memoryset",
267
- "get_org_plan",
268
- "get_prediction",
269
- "get_pretrained_embedding_model",
270
- "get_regression_model",
271
- "get_regression_model_evaluation",
272
- "get_task",
273
- "get_task_status",
274
- "healthcheck",
275
- "healthcheck_gpu",
276
- "insert_memories_gpu",
277
- "list_analyses",
278
- "list_api_keys",
279
- "list_classification_model_evaluations",
280
- "list_classification_models",
281
- "list_datasources",
282
- "list_embedding_evaluations",
283
- "list_feedback_categories",
284
- "list_finetuned_embedding_models",
285
- "list_memories_with_feedback",
286
- "list_memorysets",
287
- "list_predictions",
288
- "list_predictive_models",
289
- "list_pretrained_embedding_models",
290
- "list_regression_model_evaluations",
291
- "list_regression_models",
292
- "list_tasks",
293
- "memoryset_lookup_gpu",
294
- "potential_duplicate_groups",
295
- "predict_label_gpu",
296
- "predict_score_gpu",
297
- "query_memoryset",
298
- "record_prediction_feedback",
299
- "suggest_cascading_edits",
300
- "update_classification_model",
301
- "update_memories_gpu",
302
- "update_memory_gpu",
303
- "update_memoryset",
304
- "update_org_plan",
305
- "update_prediction",
306
- "update_regression_model",
307
- ]
File without changes
@@ -1,128 +0,0 @@
1
- """
2
- This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
-
4
- It is a customized template from the openapi-python-client tool's default template:
5
- https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/endpoint_module.py.jinja
6
-
7
- The main changes are:
8
- - Update the API call responses to either return the successful response type or raise an error by:
9
- - Updating the _parse_response function to raise an error if the response status code is not in the 2xx range
10
- - Inject a client into every method via a context manager
11
- """
12
-
13
- from http import HTTPStatus
14
- from typing import Any, cast
15
-
16
- import httpx
17
-
18
- from ...client import _client_context
19
- from ...errors import get_error_for_response
20
- from ...types import Response
21
-
22
-
23
- def _get_kwargs() -> dict[str, Any]:
24
- _kwargs: dict[str, Any] = {
25
- "method": "get",
26
- "url": "/auth/",
27
- }
28
-
29
- return _kwargs
30
-
31
-
32
- def _parse_response(*, response: httpx.Response) -> bool:
33
- if response.status_code == 200:
34
- response_200 = cast(bool, response.json())
35
- return response_200
36
- if response.status_code == 401:
37
- raise get_error_for_response(response)
38
- if response.status_code == 500:
39
- raise get_error_for_response(response)
40
- if response.status_code == 503:
41
- raise get_error_for_response(response)
42
- else:
43
- raise get_error_for_response(response)
44
-
45
-
46
- def _build_response(*, response: httpx.Response) -> Response[bool]:
47
- return Response(
48
- status_code=HTTPStatus(response.status_code),
49
- content=response.content,
50
- headers=response.headers,
51
- parsed=_parse_response(response=response),
52
- )
53
-
54
-
55
- def sync_detailed() -> Response[bool]:
56
- """Check Authentication
57
-
58
- Returns true if the api key header is valid for the org (will be false for admin api key)
59
-
60
- Raises:
61
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
62
- httpx.TimeoutException: If the request takes longer than Client.timeout.
63
-
64
- Returns:
65
- Response[bool]
66
- """
67
-
68
- kwargs = _get_kwargs()
69
-
70
- with _client_context() as client:
71
- response = client.get_httpx_client().request(
72
- **kwargs,
73
- )
74
-
75
- return _build_response(response=response)
76
-
77
-
78
- def sync() -> bool:
79
- """Check Authentication
80
-
81
- Returns true if the api key header is valid for the org (will be false for admin api key)
82
-
83
- Raises:
84
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
85
- httpx.TimeoutException: If the request takes longer than Client.timeout.
86
-
87
- Returns:
88
- bool
89
- """
90
-
91
- return sync_detailed().parsed
92
-
93
-
94
- async def asyncio_detailed() -> Response[bool]:
95
- """Check Authentication
96
-
97
- Returns true if the api key header is valid for the org (will be false for admin api key)
98
-
99
- Raises:
100
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
101
- httpx.TimeoutException: If the request takes longer than Client.timeout.
102
-
103
- Returns:
104
- Response[bool]
105
- """
106
-
107
- kwargs = _get_kwargs()
108
-
109
- with _client_context() as client:
110
- response = await client.get_async_httpx_client().request(**kwargs)
111
-
112
- return _build_response(response=response)
113
-
114
-
115
- async def asyncio() -> bool:
116
- """Check Authentication
117
-
118
- Returns true if the api key header is valid for the org (will be false for admin api key)
119
-
120
- Raises:
121
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
122
- httpx.TimeoutException: If the request takes longer than Client.timeout.
123
-
124
- Returns:
125
- bool
126
- """
127
-
128
- return (await asyncio_detailed()).parsed
@@ -1,170 +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/endpoint_module.py.jinja
6
-
7
- The main changes are:
8
- - Update the API call responses to either return the successful response type or raise an error by:
9
- - Updating the _parse_response function to raise an error if the response status code is not in the 2xx range
10
- - Inject a client into every method via a context manager
11
- """
12
-
13
- from http import HTTPStatus
14
- from typing import Any, Dict
15
-
16
- import httpx
17
-
18
- from ...client import _client_context
19
- from ...errors import get_error_for_response
20
- from ...models.create_api_key_request import CreateApiKeyRequest
21
- from ...models.create_api_key_response import CreateApiKeyResponse
22
- from ...types import Response
23
-
24
-
25
- def _get_kwargs(
26
- *,
27
- body: CreateApiKeyRequest,
28
- ) -> dict[str, Any]:
29
- headers: Dict[str, Any] = {}
30
-
31
- _kwargs: dict[str, Any] = {
32
- "method": "post",
33
- "url": "/auth/api_key",
34
- }
35
-
36
- _body = body.to_dict()
37
-
38
- _kwargs["json"] = _body
39
- headers["Content-Type"] = "application/json"
40
-
41
- _kwargs["headers"] = headers
42
- return _kwargs
43
-
44
-
45
- def _parse_response(*, response: httpx.Response) -> CreateApiKeyResponse:
46
- if response.status_code == 201:
47
- response_201 = CreateApiKeyResponse.from_dict(response.json())
48
-
49
- return response_201
50
- if response.status_code == 401:
51
- raise get_error_for_response(response)
52
- if response.status_code == 500:
53
- raise get_error_for_response(response)
54
- if response.status_code == 503:
55
- raise get_error_for_response(response)
56
- if response.status_code == 403:
57
- raise get_error_for_response(response)
58
- else:
59
- raise get_error_for_response(response)
60
-
61
-
62
- def _build_response(*, response: httpx.Response) -> Response[CreateApiKeyResponse]:
63
- return Response(
64
- status_code=HTTPStatus(response.status_code),
65
- content=response.content,
66
- headers=response.headers,
67
- parsed=_parse_response(response=response),
68
- )
69
-
70
-
71
- def sync_detailed(
72
- *,
73
- body: CreateApiKeyRequest,
74
- ) -> Response[CreateApiKeyResponse]:
75
- """Create Api Key
76
-
77
- Args:
78
- body (CreateApiKeyRequest):
79
-
80
- Raises:
81
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
82
- httpx.TimeoutException: If the request takes longer than Client.timeout.
83
-
84
- Returns:
85
- Response[CreateApiKeyResponse]
86
- """
87
-
88
- kwargs = _get_kwargs(
89
- body=body,
90
- )
91
-
92
- with _client_context() as client:
93
- response = client.get_httpx_client().request(
94
- **kwargs,
95
- )
96
-
97
- return _build_response(response=response)
98
-
99
-
100
- def sync(
101
- *,
102
- body: CreateApiKeyRequest,
103
- ) -> CreateApiKeyResponse:
104
- """Create Api Key
105
-
106
- Args:
107
- body (CreateApiKeyRequest):
108
-
109
- Raises:
110
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
111
- httpx.TimeoutException: If the request takes longer than Client.timeout.
112
-
113
- Returns:
114
- CreateApiKeyResponse
115
- """
116
-
117
- return sync_detailed(
118
- body=body,
119
- ).parsed
120
-
121
-
122
- async def asyncio_detailed(
123
- *,
124
- body: CreateApiKeyRequest,
125
- ) -> Response[CreateApiKeyResponse]:
126
- """Create Api Key
127
-
128
- Args:
129
- body (CreateApiKeyRequest):
130
-
131
- Raises:
132
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
133
- httpx.TimeoutException: If the request takes longer than Client.timeout.
134
-
135
- Returns:
136
- Response[CreateApiKeyResponse]
137
- """
138
-
139
- kwargs = _get_kwargs(
140
- body=body,
141
- )
142
-
143
- with _client_context() as client:
144
- response = await client.get_async_httpx_client().request(**kwargs)
145
-
146
- return _build_response(response=response)
147
-
148
-
149
- async def asyncio(
150
- *,
151
- body: CreateApiKeyRequest,
152
- ) -> CreateApiKeyResponse:
153
- """Create Api Key
154
-
155
- Args:
156
- body (CreateApiKeyRequest):
157
-
158
- Raises:
159
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
160
- httpx.TimeoutException: If the request takes longer than Client.timeout.
161
-
162
- Returns:
163
- CreateApiKeyResponse
164
- """
165
-
166
- return (
167
- await asyncio_detailed(
168
- body=body,
169
- )
170
- ).parsed