orca-sdk 0.0.97__py3-none-any.whl → 0.0.98__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (298) 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 +3515 -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 +42 -76
  14. orca_sdk/embedding_model.py +225 -71
  15. orca_sdk/embedding_model_test.py +27 -36
  16. orca_sdk/job.py +49 -45
  17. orca_sdk/job_test.py +16 -0
  18. orca_sdk/memoryset.py +340 -353
  19. orca_sdk/memoryset_test.py +7 -11
  20. orca_sdk/regression_model.py +120 -111
  21. orca_sdk/regression_model_test.py +15 -0
  22. orca_sdk/telemetry.py +162 -139
  23. {orca_sdk-0.0.97.dist-info → orca_sdk-0.0.98.dist-info}/METADATA +2 -5
  24. orca_sdk-0.0.98.dist-info/RECORD +40 -0
  25. orca_sdk/_generated_api_client/__init__.py +0 -3
  26. orca_sdk/_generated_api_client/api/__init__.py +0 -307
  27. orca_sdk/_generated_api_client/api/auth/__init__.py +0 -0
  28. orca_sdk/_generated_api_client/api/auth/check_authentication_auth_get.py +0 -128
  29. orca_sdk/_generated_api_client/api/auth/create_api_key_auth_api_key_post.py +0 -170
  30. orca_sdk/_generated_api_client/api/auth/create_org_plan_auth_org_plan_post.py +0 -168
  31. orca_sdk/_generated_api_client/api/auth/delete_api_key_auth_api_key_name_or_id_delete.py +0 -156
  32. orca_sdk/_generated_api_client/api/auth/delete_org_auth_org_delete.py +0 -130
  33. orca_sdk/_generated_api_client/api/auth/get_org_plan_auth_org_plan_get.py +0 -122
  34. orca_sdk/_generated_api_client/api/auth/list_api_keys_auth_api_key_get.py +0 -127
  35. orca_sdk/_generated_api_client/api/auth/update_org_plan_auth_org_plan_put.py +0 -168
  36. orca_sdk/_generated_api_client/api/classification_model/__init__.py +0 -0
  37. orca_sdk/_generated_api_client/api/classification_model/create_classification_model_classification_model_post.py +0 -170
  38. orca_sdk/_generated_api_client/api/classification_model/delete_classification_model_classification_model_name_or_id_delete.py +0 -156
  39. 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
  40. orca_sdk/_generated_api_client/api/classification_model/evaluate_classification_model_classification_model_model_name_or_id_evaluation_post.py +0 -183
  41. orca_sdk/_generated_api_client/api/classification_model/get_classification_model_classification_model_name_or_id_get.py +0 -156
  42. 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
  43. orca_sdk/_generated_api_client/api/classification_model/list_classification_model_evaluations_classification_model_model_name_or_id_evaluation_get.py +0 -161
  44. orca_sdk/_generated_api_client/api/classification_model/list_classification_models_classification_model_get.py +0 -127
  45. orca_sdk/_generated_api_client/api/classification_model/predict_label_gpu_classification_model_name_or_id_prediction_post.py +0 -190
  46. orca_sdk/_generated_api_client/api/classification_model/update_classification_model_classification_model_name_or_id_patch.py +0 -183
  47. orca_sdk/_generated_api_client/api/datasource/__init__.py +0 -0
  48. orca_sdk/_generated_api_client/api/datasource/create_datasource_from_content_datasource_post.py +0 -224
  49. orca_sdk/_generated_api_client/api/datasource/create_datasource_from_files_datasource_upload_post.py +0 -229
  50. orca_sdk/_generated_api_client/api/datasource/create_embedding_evaluation_datasource_name_or_id_embedding_evaluation_post.py +0 -183
  51. orca_sdk/_generated_api_client/api/datasource/delete_datasource_datasource_name_or_id_delete.py +0 -156
  52. orca_sdk/_generated_api_client/api/datasource/download_datasource_datasource_name_or_id_download_get.py +0 -172
  53. orca_sdk/_generated_api_client/api/datasource/get_datasource_datasource_name_or_id_get.py +0 -156
  54. orca_sdk/_generated_api_client/api/datasource/get_embedding_evaluation_datasource_name_or_id_embedding_evaluation_task_id_get.py +0 -169
  55. orca_sdk/_generated_api_client/api/datasource/list_datasources_datasource_get.py +0 -127
  56. orca_sdk/_generated_api_client/api/datasource/list_embedding_evaluations_datasource_name_or_id_embedding_evaluation_get.py +0 -235
  57. orca_sdk/_generated_api_client/api/default/__init__.py +0 -0
  58. orca_sdk/_generated_api_client/api/default/healthcheck_get.py +0 -118
  59. orca_sdk/_generated_api_client/api/default/healthcheck_gpu_get.py +0 -118
  60. orca_sdk/_generated_api_client/api/finetuned_embedding_model/__init__.py +0 -0
  61. orca_sdk/_generated_api_client/api/finetuned_embedding_model/create_finetuned_embedding_model_finetuned_embedding_model_post.py +0 -168
  62. orca_sdk/_generated_api_client/api/finetuned_embedding_model/delete_finetuned_embedding_model_finetuned_embedding_model_name_or_id_delete.py +0 -156
  63. 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
  64. orca_sdk/_generated_api_client/api/finetuned_embedding_model/get_finetuned_embedding_model_finetuned_embedding_model_name_or_id_get.py +0 -156
  65. orca_sdk/_generated_api_client/api/finetuned_embedding_model/list_finetuned_embedding_models_finetuned_embedding_model_get.py +0 -127
  66. orca_sdk/_generated_api_client/api/memoryset/__init__.py +0 -0
  67. orca_sdk/_generated_api_client/api/memoryset/analyze_memoryset_memoryset_name_or_id_analysis_post.py +0 -183
  68. orca_sdk/_generated_api_client/api/memoryset/batch_delete_memoryset_batch_delete_memoryset_post.py +0 -168
  69. orca_sdk/_generated_api_client/api/memoryset/clone_memoryset_memoryset_name_or_id_clone_post.py +0 -181
  70. orca_sdk/_generated_api_client/api/memoryset/create_memoryset_memoryset_post.py +0 -168
  71. orca_sdk/_generated_api_client/api/memoryset/delete_memories_memoryset_name_or_id_memories_delete_post.py +0 -181
  72. orca_sdk/_generated_api_client/api/memoryset/delete_memory_memoryset_name_or_id_memory_memory_id_delete.py +0 -167
  73. orca_sdk/_generated_api_client/api/memoryset/delete_memoryset_memoryset_name_or_id_delete.py +0 -156
  74. orca_sdk/_generated_api_client/api/memoryset/get_analysis_memoryset_name_or_id_analysis_analysis_task_id_get.py +0 -169
  75. orca_sdk/_generated_api_client/api/memoryset/get_memories_memoryset_name_or_id_memories_get_post.py +0 -210
  76. orca_sdk/_generated_api_client/api/memoryset/get_memory_memoryset_name_or_id_memory_memory_id_get.py +0 -186
  77. orca_sdk/_generated_api_client/api/memoryset/get_memoryset_memoryset_name_or_id_get.py +0 -156
  78. orca_sdk/_generated_api_client/api/memoryset/insert_memories_gpu_memoryset_name_or_id_memory_post.py +0 -188
  79. orca_sdk/_generated_api_client/api/memoryset/list_analyses_memoryset_name_or_id_analysis_get.py +0 -235
  80. orca_sdk/_generated_api_client/api/memoryset/list_memorysets_memoryset_get.py +0 -180
  81. orca_sdk/_generated_api_client/api/memoryset/memoryset_lookup_gpu_memoryset_name_or_id_lookup_post.py +0 -212
  82. orca_sdk/_generated_api_client/api/memoryset/potential_duplicate_groups_memoryset_name_or_id_potential_duplicate_groups_get.py +0 -195
  83. orca_sdk/_generated_api_client/api/memoryset/query_memoryset_memoryset_name_or_id_memories_post.py +0 -210
  84. orca_sdk/_generated_api_client/api/memoryset/suggest_cascading_edits_memoryset_name_or_id_memory_memory_id_cascading_edits_post.py +0 -233
  85. orca_sdk/_generated_api_client/api/memoryset/update_memories_gpu_memoryset_name_or_id_memories_patch.py +0 -216
  86. orca_sdk/_generated_api_client/api/memoryset/update_memory_gpu_memoryset_name_or_id_memory_patch.py +0 -205
  87. orca_sdk/_generated_api_client/api/memoryset/update_memoryset_memoryset_name_or_id_patch.py +0 -183
  88. orca_sdk/_generated_api_client/api/predictive_model/__init__.py +0 -0
  89. orca_sdk/_generated_api_client/api/predictive_model/list_predictive_models_predictive_model_get.py +0 -150
  90. orca_sdk/_generated_api_client/api/pretrained_embedding_model/__init__.py +0 -0
  91. orca_sdk/_generated_api_client/api/pretrained_embedding_model/embed_with_pretrained_model_gpu_pretrained_embedding_model_model_name_embedding_post.py +0 -192
  92. orca_sdk/_generated_api_client/api/pretrained_embedding_model/get_pretrained_embedding_model_pretrained_embedding_model_model_name_get.py +0 -161
  93. orca_sdk/_generated_api_client/api/pretrained_embedding_model/list_pretrained_embedding_models_pretrained_embedding_model_get.py +0 -127
  94. orca_sdk/_generated_api_client/api/regression_model/__init__.py +0 -0
  95. orca_sdk/_generated_api_client/api/regression_model/create_regression_model_regression_model_post.py +0 -170
  96. 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
  97. orca_sdk/_generated_api_client/api/regression_model/delete_regression_model_regression_model_name_or_id_delete.py +0 -154
  98. orca_sdk/_generated_api_client/api/regression_model/evaluate_regression_model_regression_model_model_name_or_id_evaluation_post.py +0 -183
  99. 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
  100. orca_sdk/_generated_api_client/api/regression_model/get_regression_model_regression_model_name_or_id_get.py +0 -156
  101. orca_sdk/_generated_api_client/api/regression_model/list_regression_model_evaluations_regression_model_model_name_or_id_evaluation_get.py +0 -161
  102. orca_sdk/_generated_api_client/api/regression_model/list_regression_models_regression_model_get.py +0 -127
  103. orca_sdk/_generated_api_client/api/regression_model/predict_score_gpu_regression_model_name_or_id_prediction_post.py +0 -190
  104. orca_sdk/_generated_api_client/api/regression_model/update_regression_model_regression_model_name_or_id_patch.py +0 -183
  105. orca_sdk/_generated_api_client/api/task/__init__.py +0 -0
  106. orca_sdk/_generated_api_client/api/task/abort_task_task_task_id_abort_delete.py +0 -154
  107. orca_sdk/_generated_api_client/api/task/get_task_status_task_task_id_status_get.py +0 -156
  108. orca_sdk/_generated_api_client/api/task/get_task_task_task_id_get.py +0 -156
  109. orca_sdk/_generated_api_client/api/task/list_tasks_task_get.py +0 -288
  110. orca_sdk/_generated_api_client/api/telemetry/__init__.py +0 -0
  111. orca_sdk/_generated_api_client/api/telemetry/count_predictions_telemetry_prediction_count_post.py +0 -168
  112. orca_sdk/_generated_api_client/api/telemetry/drop_feedback_category_with_data_telemetry_feedback_category_name_or_id_delete.py +0 -162
  113. orca_sdk/_generated_api_client/api/telemetry/explain_prediction_telemetry_prediction_prediction_id_explanation_get.py +0 -182
  114. orca_sdk/_generated_api_client/api/telemetry/generate_memory_suggestions_telemetry_prediction_prediction_id_memory_suggestions_post.py +0 -239
  115. orca_sdk/_generated_api_client/api/telemetry/get_action_recommendation_telemetry_prediction_prediction_id_action_get.py +0 -192
  116. orca_sdk/_generated_api_client/api/telemetry/get_feedback_category_telemetry_feedback_category_name_or_id_get.py +0 -156
  117. orca_sdk/_generated_api_client/api/telemetry/get_prediction_telemetry_prediction_prediction_id_get.py +0 -180
  118. orca_sdk/_generated_api_client/api/telemetry/list_feedback_categories_telemetry_feedback_category_get.py +0 -127
  119. orca_sdk/_generated_api_client/api/telemetry/list_memories_with_feedback_telemetry_memories_post.py +0 -198
  120. orca_sdk/_generated_api_client/api/telemetry/list_predictions_telemetry_prediction_post.py +0 -198
  121. orca_sdk/_generated_api_client/api/telemetry/record_prediction_feedback_telemetry_prediction_feedback_put.py +0 -171
  122. orca_sdk/_generated_api_client/api/telemetry/update_prediction_telemetry_prediction_prediction_id_patch.py +0 -181
  123. orca_sdk/_generated_api_client/client.py +0 -216
  124. orca_sdk/_generated_api_client/errors.py +0 -38
  125. orca_sdk/_generated_api_client/models/__init__.py +0 -345
  126. orca_sdk/_generated_api_client/models/action_recommendation.py +0 -82
  127. orca_sdk/_generated_api_client/models/action_recommendation_action.py +0 -11
  128. orca_sdk/_generated_api_client/models/add_memory_recommendations.py +0 -85
  129. orca_sdk/_generated_api_client/models/add_memory_suggestion.py +0 -79
  130. orca_sdk/_generated_api_client/models/analyze_neighbor_labels_result.py +0 -116
  131. orca_sdk/_generated_api_client/models/api_key_metadata.py +0 -137
  132. orca_sdk/_generated_api_client/models/api_key_metadata_scope_item.py +0 -9
  133. orca_sdk/_generated_api_client/models/base_label_prediction_result.py +0 -130
  134. orca_sdk/_generated_api_client/models/base_model.py +0 -55
  135. orca_sdk/_generated_api_client/models/base_score_prediction_result.py +0 -108
  136. orca_sdk/_generated_api_client/models/body_create_datasource_from_files_datasource_upload_post.py +0 -145
  137. orca_sdk/_generated_api_client/models/cascade_edit_suggestions_request.py +0 -154
  138. orca_sdk/_generated_api_client/models/cascading_edit_suggestion.py +0 -92
  139. orca_sdk/_generated_api_client/models/class_representatives.py +0 -92
  140. orca_sdk/_generated_api_client/models/classification_evaluation_request.py +0 -148
  141. orca_sdk/_generated_api_client/models/classification_metrics.py +0 -259
  142. orca_sdk/_generated_api_client/models/classification_model_metadata.py +0 -227
  143. orca_sdk/_generated_api_client/models/classification_prediction_request.py +0 -220
  144. orca_sdk/_generated_api_client/models/clone_memoryset_request.py +0 -210
  145. orca_sdk/_generated_api_client/models/cluster_metrics.py +0 -78
  146. orca_sdk/_generated_api_client/models/column_info.py +0 -145
  147. orca_sdk/_generated_api_client/models/column_type.py +0 -14
  148. orca_sdk/_generated_api_client/models/constraint_violation_error_response.py +0 -81
  149. orca_sdk/_generated_api_client/models/constraint_violation_error_response_status_code.py +0 -8
  150. orca_sdk/_generated_api_client/models/count_predictions_request.py +0 -195
  151. orca_sdk/_generated_api_client/models/create_api_key_request.py +0 -120
  152. orca_sdk/_generated_api_client/models/create_api_key_request_scope_item.py +0 -9
  153. orca_sdk/_generated_api_client/models/create_api_key_response.py +0 -145
  154. orca_sdk/_generated_api_client/models/create_api_key_response_scope_item.py +0 -9
  155. orca_sdk/_generated_api_client/models/create_classification_model_request.py +0 -237
  156. orca_sdk/_generated_api_client/models/create_datasource_from_content_request.py +0 -101
  157. orca_sdk/_generated_api_client/models/create_memoryset_request.py +0 -365
  158. orca_sdk/_generated_api_client/models/create_memoryset_request_index_params.py +0 -66
  159. orca_sdk/_generated_api_client/models/create_memoryset_request_index_type.py +0 -13
  160. orca_sdk/_generated_api_client/models/create_org_plan_request.py +0 -73
  161. orca_sdk/_generated_api_client/models/create_org_plan_request_tier.py +0 -11
  162. orca_sdk/_generated_api_client/models/create_regression_model_request.py +0 -157
  163. orca_sdk/_generated_api_client/models/datasource_metadata.py +0 -156
  164. orca_sdk/_generated_api_client/models/delete_memories_request.py +0 -70
  165. orca_sdk/_generated_api_client/models/delete_memorysets_request.py +0 -70
  166. orca_sdk/_generated_api_client/models/embed_request.py +0 -155
  167. orca_sdk/_generated_api_client/models/embedding_evaluation_payload.py +0 -205
  168. orca_sdk/_generated_api_client/models/embedding_evaluation_request.py +0 -197
  169. orca_sdk/_generated_api_client/models/embedding_evaluation_response.py +0 -158
  170. orca_sdk/_generated_api_client/models/embedding_evaluation_result.py +0 -86
  171. orca_sdk/_generated_api_client/models/embedding_finetuning_method.py +0 -9
  172. orca_sdk/_generated_api_client/models/embedding_model_result.py +0 -123
  173. orca_sdk/_generated_api_client/models/evaluation_response.py +0 -153
  174. orca_sdk/_generated_api_client/models/evaluation_response_classification_metrics.py +0 -140
  175. orca_sdk/_generated_api_client/models/evaluation_response_regression_metrics.py +0 -140
  176. orca_sdk/_generated_api_client/models/feedback_metrics.py +0 -85
  177. orca_sdk/_generated_api_client/models/feedback_type.py +0 -9
  178. orca_sdk/_generated_api_client/models/filter_item.py +0 -239
  179. orca_sdk/_generated_api_client/models/filter_item_field_type_0_item.py +0 -17
  180. orca_sdk/_generated_api_client/models/filter_item_field_type_1_item_type_0.py +0 -8
  181. orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_0.py +0 -8
  182. orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_1.py +0 -22
  183. orca_sdk/_generated_api_client/models/filter_item_op.py +0 -16
  184. orca_sdk/_generated_api_client/models/finetune_embedding_model_request.py +0 -259
  185. orca_sdk/_generated_api_client/models/finetune_embedding_model_request_training_args.py +0 -66
  186. orca_sdk/_generated_api_client/models/finetuned_embedding_model_metadata.py +0 -166
  187. orca_sdk/_generated_api_client/models/get_memories_request.py +0 -70
  188. orca_sdk/_generated_api_client/models/http_validation_error.py +0 -86
  189. orca_sdk/_generated_api_client/models/internal_server_error_response.py +0 -81
  190. orca_sdk/_generated_api_client/models/internal_server_error_response_status_code.py +0 -8
  191. orca_sdk/_generated_api_client/models/label_class_metrics.py +0 -108
  192. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup.py +0 -210
  193. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup_metadata.py +0 -68
  194. orca_sdk/_generated_api_client/models/label_prediction_with_memories_and_feedback.py +0 -288
  195. orca_sdk/_generated_api_client/models/labeled_memory.py +0 -186
  196. orca_sdk/_generated_api_client/models/labeled_memory_insert.py +0 -128
  197. orca_sdk/_generated_api_client/models/labeled_memory_insert_metadata.py +0 -68
  198. orca_sdk/_generated_api_client/models/labeled_memory_lookup.py +0 -194
  199. orca_sdk/_generated_api_client/models/labeled_memory_lookup_metadata.py +0 -68
  200. orca_sdk/_generated_api_client/models/labeled_memory_metadata.py +0 -68
  201. orca_sdk/_generated_api_client/models/labeled_memory_update.py +0 -171
  202. orca_sdk/_generated_api_client/models/labeled_memory_update_metadata_type_0.py +0 -68
  203. orca_sdk/_generated_api_client/models/labeled_memory_with_feedback_metrics.py +0 -207
  204. orca_sdk/_generated_api_client/models/labeled_memory_with_feedback_metrics_feedback_metrics.py +0 -68
  205. orca_sdk/_generated_api_client/models/labeled_memory_with_feedback_metrics_metadata.py +0 -68
  206. orca_sdk/_generated_api_client/models/list_memories_request.py +0 -104
  207. orca_sdk/_generated_api_client/models/list_predictions_request.py +0 -319
  208. orca_sdk/_generated_api_client/models/lookup_request.py +0 -101
  209. orca_sdk/_generated_api_client/models/lookup_score_metrics.py +0 -94
  210. orca_sdk/_generated_api_client/models/memory_metrics.py +0 -263
  211. orca_sdk/_generated_api_client/models/memory_type.py +0 -9
  212. orca_sdk/_generated_api_client/models/memoryset_analysis_configs.py +0 -245
  213. orca_sdk/_generated_api_client/models/memoryset_analysis_request.py +0 -105
  214. orca_sdk/_generated_api_client/models/memoryset_analysis_response.py +0 -182
  215. orca_sdk/_generated_api_client/models/memoryset_class_patterns_analysis_config.py +0 -79
  216. orca_sdk/_generated_api_client/models/memoryset_class_patterns_metrics.py +0 -138
  217. orca_sdk/_generated_api_client/models/memoryset_cluster_analysis_config.py +0 -202
  218. orca_sdk/_generated_api_client/models/memoryset_cluster_analysis_config_clustering_method.py +0 -9
  219. orca_sdk/_generated_api_client/models/memoryset_cluster_analysis_config_partitioning_method.py +0 -10
  220. orca_sdk/_generated_api_client/models/memoryset_cluster_metrics.py +0 -100
  221. orca_sdk/_generated_api_client/models/memoryset_duplicate_analysis_config.py +0 -70
  222. orca_sdk/_generated_api_client/models/memoryset_duplicate_metrics.py +0 -70
  223. orca_sdk/_generated_api_client/models/memoryset_label_analysis_config.py +0 -70
  224. orca_sdk/_generated_api_client/models/memoryset_label_metrics.py +0 -116
  225. orca_sdk/_generated_api_client/models/memoryset_metadata.py +0 -333
  226. orca_sdk/_generated_api_client/models/memoryset_metadata_index_params.py +0 -55
  227. orca_sdk/_generated_api_client/models/memoryset_metadata_index_type.py +0 -13
  228. orca_sdk/_generated_api_client/models/memoryset_metrics.py +0 -265
  229. orca_sdk/_generated_api_client/models/memoryset_neighbor_analysis_config.py +0 -83
  230. orca_sdk/_generated_api_client/models/memoryset_neighbor_metrics.py +0 -76
  231. orca_sdk/_generated_api_client/models/memoryset_neighbor_metrics_lookup_score_metrics.py +0 -68
  232. orca_sdk/_generated_api_client/models/memoryset_projection_analysis_config.py +0 -79
  233. orca_sdk/_generated_api_client/models/memoryset_projection_metrics.py +0 -55
  234. orca_sdk/_generated_api_client/models/memoryset_update.py +0 -121
  235. orca_sdk/_generated_api_client/models/not_found_error_response.py +0 -99
  236. orca_sdk/_generated_api_client/models/not_found_error_response_resource_type_0.py +0 -23
  237. orca_sdk/_generated_api_client/models/not_found_error_response_status_code.py +0 -8
  238. orca_sdk/_generated_api_client/models/org_plan.py +0 -99
  239. orca_sdk/_generated_api_client/models/org_plan_tier.py +0 -11
  240. orca_sdk/_generated_api_client/models/paginated_task.py +0 -108
  241. orca_sdk/_generated_api_client/models/paginated_union_labeled_memory_with_feedback_metrics_scored_memory_with_feedback_metrics.py +0 -135
  242. orca_sdk/_generated_api_client/models/pr_curve.py +0 -86
  243. orca_sdk/_generated_api_client/models/prediction_feedback.py +0 -157
  244. orca_sdk/_generated_api_client/models/prediction_feedback_category.py +0 -115
  245. orca_sdk/_generated_api_client/models/prediction_feedback_request.py +0 -122
  246. orca_sdk/_generated_api_client/models/prediction_feedback_result.py +0 -102
  247. orca_sdk/_generated_api_client/models/prediction_sort_item_item_type_0.py +0 -10
  248. orca_sdk/_generated_api_client/models/prediction_sort_item_item_type_1.py +0 -9
  249. orca_sdk/_generated_api_client/models/predictive_model_update.py +0 -111
  250. orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py +0 -115
  251. orca_sdk/_generated_api_client/models/pretrained_embedding_model_name.py +0 -17
  252. orca_sdk/_generated_api_client/models/rac_head_type.py +0 -11
  253. orca_sdk/_generated_api_client/models/rar_head_type.py +0 -8
  254. orca_sdk/_generated_api_client/models/regression_evaluation_request.py +0 -148
  255. orca_sdk/_generated_api_client/models/regression_metrics.py +0 -172
  256. orca_sdk/_generated_api_client/models/regression_model_metadata.py +0 -191
  257. orca_sdk/_generated_api_client/models/regression_prediction_request.py +0 -195
  258. orca_sdk/_generated_api_client/models/roc_curve.py +0 -86
  259. orca_sdk/_generated_api_client/models/score_prediction_memory_lookup.py +0 -196
  260. orca_sdk/_generated_api_client/models/score_prediction_memory_lookup_metadata.py +0 -68
  261. orca_sdk/_generated_api_client/models/score_prediction_with_memories_and_feedback.py +0 -252
  262. orca_sdk/_generated_api_client/models/scored_memory.py +0 -172
  263. orca_sdk/_generated_api_client/models/scored_memory_insert.py +0 -128
  264. orca_sdk/_generated_api_client/models/scored_memory_insert_metadata.py +0 -68
  265. orca_sdk/_generated_api_client/models/scored_memory_lookup.py +0 -180
  266. orca_sdk/_generated_api_client/models/scored_memory_lookup_metadata.py +0 -68
  267. orca_sdk/_generated_api_client/models/scored_memory_metadata.py +0 -68
  268. orca_sdk/_generated_api_client/models/scored_memory_update.py +0 -171
  269. orca_sdk/_generated_api_client/models/scored_memory_update_metadata_type_0.py +0 -68
  270. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics.py +0 -193
  271. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics_feedback_metrics.py +0 -68
  272. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics_metadata.py +0 -68
  273. orca_sdk/_generated_api_client/models/service_unavailable_error_response.py +0 -81
  274. orca_sdk/_generated_api_client/models/service_unavailable_error_response_status_code.py +0 -8
  275. orca_sdk/_generated_api_client/models/task.py +0 -198
  276. orca_sdk/_generated_api_client/models/task_status.py +0 -14
  277. orca_sdk/_generated_api_client/models/task_status_info.py +0 -133
  278. orca_sdk/_generated_api_client/models/telemetry_field_type_0_item_type_0.py +0 -8
  279. orca_sdk/_generated_api_client/models/telemetry_field_type_0_item_type_2.py +0 -9
  280. orca_sdk/_generated_api_client/models/telemetry_field_type_1_item_type_0.py +0 -8
  281. orca_sdk/_generated_api_client/models/telemetry_field_type_1_item_type_1.py +0 -8
  282. orca_sdk/_generated_api_client/models/telemetry_filter_item.py +0 -217
  283. orca_sdk/_generated_api_client/models/telemetry_filter_item_op.py +0 -15
  284. orca_sdk/_generated_api_client/models/telemetry_memories_request.py +0 -181
  285. orca_sdk/_generated_api_client/models/telemetry_sort_options.py +0 -185
  286. orca_sdk/_generated_api_client/models/telemetry_sort_options_direction.py +0 -9
  287. orca_sdk/_generated_api_client/models/unauthenticated_error_response.py +0 -73
  288. orca_sdk/_generated_api_client/models/unauthenticated_error_response_status_code.py +0 -8
  289. orca_sdk/_generated_api_client/models/unauthorized_error_response.py +0 -81
  290. orca_sdk/_generated_api_client/models/unauthorized_error_response_status_code.py +0 -8
  291. orca_sdk/_generated_api_client/models/update_org_plan_request.py +0 -73
  292. orca_sdk/_generated_api_client/models/update_org_plan_request_tier.py +0 -11
  293. orca_sdk/_generated_api_client/models/update_prediction_request.py +0 -133
  294. orca_sdk/_generated_api_client/models/validation_error.py +0 -99
  295. orca_sdk/_generated_api_client/py.typed +0 -1
  296. orca_sdk/_generated_api_client/types.py +0 -56
  297. orca_sdk-0.0.97.dist-info/RECORD +0 -309
  298. {orca_sdk-0.0.97.dist-info → orca_sdk-0.0.98.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