orca-sdk 0.1.1__py3-none-any.whl → 0.1.2__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 (185) hide show
  1. orca_sdk/__init__.py +10 -4
  2. orca_sdk/_shared/__init__.py +10 -0
  3. orca_sdk/_shared/metrics.py +393 -0
  4. orca_sdk/_shared/metrics_test.py +273 -0
  5. orca_sdk/_utils/analysis_ui.py +12 -10
  6. orca_sdk/_utils/analysis_ui_style.css +0 -3
  7. orca_sdk/_utils/auth.py +27 -29
  8. orca_sdk/_utils/data_parsing.py +28 -2
  9. orca_sdk/_utils/data_parsing_test.py +15 -15
  10. orca_sdk/_utils/pagination.py +126 -0
  11. orca_sdk/_utils/pagination_test.py +132 -0
  12. orca_sdk/_utils/prediction_result_ui.py +67 -21
  13. orca_sdk/_utils/tqdm_file_reader.py +12 -0
  14. orca_sdk/_utils/value_parser.py +45 -0
  15. orca_sdk/_utils/value_parser_test.py +39 -0
  16. orca_sdk/classification_model.py +439 -129
  17. orca_sdk/classification_model_test.py +334 -104
  18. orca_sdk/client.py +3747 -0
  19. orca_sdk/conftest.py +164 -19
  20. orca_sdk/credentials.py +120 -18
  21. orca_sdk/credentials_test.py +20 -0
  22. orca_sdk/datasource.py +259 -68
  23. orca_sdk/datasource_test.py +242 -0
  24. orca_sdk/embedding_model.py +425 -82
  25. orca_sdk/embedding_model_test.py +39 -13
  26. orca_sdk/job.py +337 -0
  27. orca_sdk/job_test.py +108 -0
  28. orca_sdk/memoryset.py +1341 -305
  29. orca_sdk/memoryset_test.py +350 -111
  30. orca_sdk/regression_model.py +684 -0
  31. orca_sdk/regression_model_test.py +369 -0
  32. orca_sdk/telemetry.py +449 -143
  33. orca_sdk/telemetry_test.py +43 -24
  34. {orca_sdk-0.1.1.dist-info → orca_sdk-0.1.2.dist-info}/METADATA +34 -16
  35. orca_sdk-0.1.2.dist-info/RECORD +40 -0
  36. {orca_sdk-0.1.1.dist-info → orca_sdk-0.1.2.dist-info}/WHEEL +1 -1
  37. orca_sdk/_generated_api_client/__init__.py +0 -3
  38. orca_sdk/_generated_api_client/api/__init__.py +0 -193
  39. orca_sdk/_generated_api_client/api/auth/__init__.py +0 -0
  40. orca_sdk/_generated_api_client/api/auth/check_authentication_auth_get.py +0 -128
  41. orca_sdk/_generated_api_client/api/auth/create_api_key_auth_api_key_post.py +0 -170
  42. orca_sdk/_generated_api_client/api/auth/delete_api_key_auth_api_key_name_or_id_delete.py +0 -156
  43. orca_sdk/_generated_api_client/api/auth/delete_org_auth_org_delete.py +0 -130
  44. orca_sdk/_generated_api_client/api/auth/list_api_keys_auth_api_key_get.py +0 -127
  45. orca_sdk/_generated_api_client/api/classification_model/__init__.py +0 -0
  46. orca_sdk/_generated_api_client/api/classification_model/create_evaluation_classification_model_model_name_or_id_evaluation_post.py +0 -183
  47. orca_sdk/_generated_api_client/api/classification_model/create_model_classification_model_post.py +0 -170
  48. orca_sdk/_generated_api_client/api/classification_model/delete_evaluation_classification_model_model_name_or_id_evaluation_task_id_delete.py +0 -168
  49. orca_sdk/_generated_api_client/api/classification_model/delete_model_classification_model_name_or_id_delete.py +0 -154
  50. orca_sdk/_generated_api_client/api/classification_model/get_evaluation_classification_model_model_name_or_id_evaluation_task_id_get.py +0 -170
  51. orca_sdk/_generated_api_client/api/classification_model/get_model_classification_model_name_or_id_get.py +0 -156
  52. orca_sdk/_generated_api_client/api/classification_model/list_evaluations_classification_model_model_name_or_id_evaluation_get.py +0 -161
  53. orca_sdk/_generated_api_client/api/classification_model/list_models_classification_model_get.py +0 -127
  54. orca_sdk/_generated_api_client/api/classification_model/predict_gpu_classification_model_name_or_id_prediction_post.py +0 -190
  55. orca_sdk/_generated_api_client/api/datasource/__init__.py +0 -0
  56. orca_sdk/_generated_api_client/api/datasource/create_datasource_datasource_post.py +0 -167
  57. orca_sdk/_generated_api_client/api/datasource/delete_datasource_datasource_name_or_id_delete.py +0 -156
  58. orca_sdk/_generated_api_client/api/datasource/get_datasource_datasource_name_or_id_get.py +0 -156
  59. orca_sdk/_generated_api_client/api/datasource/list_datasources_datasource_get.py +0 -127
  60. orca_sdk/_generated_api_client/api/default/__init__.py +0 -0
  61. orca_sdk/_generated_api_client/api/default/healthcheck_get.py +0 -118
  62. orca_sdk/_generated_api_client/api/default/healthcheck_gpu_get.py +0 -118
  63. orca_sdk/_generated_api_client/api/finetuned_embedding_model/__init__.py +0 -0
  64. orca_sdk/_generated_api_client/api/finetuned_embedding_model/create_finetuned_embedding_model_finetuned_embedding_model_post.py +0 -168
  65. orca_sdk/_generated_api_client/api/finetuned_embedding_model/delete_finetuned_embedding_model_finetuned_embedding_model_name_or_id_delete.py +0 -156
  66. 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
  67. orca_sdk/_generated_api_client/api/finetuned_embedding_model/get_finetuned_embedding_model_finetuned_embedding_model_name_or_id_get.py +0 -156
  68. orca_sdk/_generated_api_client/api/finetuned_embedding_model/list_finetuned_embedding_models_finetuned_embedding_model_get.py +0 -127
  69. orca_sdk/_generated_api_client/api/memoryset/__init__.py +0 -0
  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_analysis_memoryset_name_or_id_analysis_post.py +0 -183
  72. orca_sdk/_generated_api_client/api/memoryset/create_memoryset_memoryset_post.py +0 -168
  73. orca_sdk/_generated_api_client/api/memoryset/delete_memories_memoryset_name_or_id_memories_delete_post.py +0 -181
  74. orca_sdk/_generated_api_client/api/memoryset/delete_memory_memoryset_name_or_id_memory_memory_id_delete.py +0 -167
  75. orca_sdk/_generated_api_client/api/memoryset/delete_memoryset_memoryset_name_or_id_delete.py +0 -156
  76. orca_sdk/_generated_api_client/api/memoryset/get_analysis_memoryset_name_or_id_analysis_analysis_task_id_get.py +0 -169
  77. orca_sdk/_generated_api_client/api/memoryset/get_memories_memoryset_name_or_id_memories_get_post.py +0 -188
  78. orca_sdk/_generated_api_client/api/memoryset/get_memory_memoryset_name_or_id_memory_memory_id_get.py +0 -169
  79. orca_sdk/_generated_api_client/api/memoryset/get_memoryset_memoryset_name_or_id_get.py +0 -156
  80. orca_sdk/_generated_api_client/api/memoryset/insert_memories_gpu_memoryset_name_or_id_memory_post.py +0 -184
  81. orca_sdk/_generated_api_client/api/memoryset/list_analyses_memoryset_name_or_id_analysis_get.py +0 -260
  82. orca_sdk/_generated_api_client/api/memoryset/list_memorysets_memoryset_get.py +0 -127
  83. orca_sdk/_generated_api_client/api/memoryset/memoryset_lookup_gpu_memoryset_name_or_id_lookup_post.py +0 -193
  84. orca_sdk/_generated_api_client/api/memoryset/query_memoryset_memoryset_name_or_id_memories_post.py +0 -188
  85. orca_sdk/_generated_api_client/api/memoryset/update_memories_gpu_memoryset_name_or_id_memories_patch.py +0 -191
  86. orca_sdk/_generated_api_client/api/memoryset/update_memory_gpu_memoryset_name_or_id_memory_patch.py +0 -187
  87. orca_sdk/_generated_api_client/api/pretrained_embedding_model/__init__.py +0 -0
  88. orca_sdk/_generated_api_client/api/pretrained_embedding_model/embed_with_pretrained_model_gpu_pretrained_embedding_model_model_name_embedding_post.py +0 -188
  89. orca_sdk/_generated_api_client/api/pretrained_embedding_model/get_pretrained_embedding_model_pretrained_embedding_model_model_name_get.py +0 -157
  90. orca_sdk/_generated_api_client/api/pretrained_embedding_model/list_pretrained_embedding_models_pretrained_embedding_model_get.py +0 -127
  91. orca_sdk/_generated_api_client/api/task/__init__.py +0 -0
  92. orca_sdk/_generated_api_client/api/task/abort_task_task_task_id_abort_delete.py +0 -154
  93. orca_sdk/_generated_api_client/api/task/get_task_status_task_task_id_status_get.py +0 -156
  94. orca_sdk/_generated_api_client/api/task/list_tasks_task_get.py +0 -243
  95. orca_sdk/_generated_api_client/api/telemetry/__init__.py +0 -0
  96. orca_sdk/_generated_api_client/api/telemetry/drop_feedback_category_with_data_telemetry_feedback_category_name_or_id_delete.py +0 -162
  97. orca_sdk/_generated_api_client/api/telemetry/get_feedback_category_telemetry_feedback_category_name_or_id_get.py +0 -156
  98. orca_sdk/_generated_api_client/api/telemetry/get_prediction_telemetry_prediction_prediction_id_get.py +0 -157
  99. orca_sdk/_generated_api_client/api/telemetry/list_feedback_categories_telemetry_feedback_category_get.py +0 -127
  100. orca_sdk/_generated_api_client/api/telemetry/list_predictions_telemetry_prediction_post.py +0 -175
  101. orca_sdk/_generated_api_client/api/telemetry/record_prediction_feedback_telemetry_prediction_feedback_put.py +0 -171
  102. orca_sdk/_generated_api_client/api/telemetry/update_prediction_telemetry_prediction_prediction_id_patch.py +0 -181
  103. orca_sdk/_generated_api_client/client.py +0 -216
  104. orca_sdk/_generated_api_client/errors.py +0 -38
  105. orca_sdk/_generated_api_client/models/__init__.py +0 -159
  106. orca_sdk/_generated_api_client/models/analyze_neighbor_labels_result.py +0 -84
  107. orca_sdk/_generated_api_client/models/api_key_metadata.py +0 -118
  108. orca_sdk/_generated_api_client/models/base_model.py +0 -55
  109. orca_sdk/_generated_api_client/models/body_create_datasource_datasource_post.py +0 -176
  110. orca_sdk/_generated_api_client/models/classification_evaluation_result.py +0 -114
  111. orca_sdk/_generated_api_client/models/clone_labeled_memoryset_request.py +0 -150
  112. orca_sdk/_generated_api_client/models/column_info.py +0 -114
  113. orca_sdk/_generated_api_client/models/column_type.py +0 -14
  114. orca_sdk/_generated_api_client/models/conflict_error_response.py +0 -80
  115. orca_sdk/_generated_api_client/models/create_api_key_request.py +0 -99
  116. orca_sdk/_generated_api_client/models/create_api_key_response.py +0 -126
  117. orca_sdk/_generated_api_client/models/create_labeled_memoryset_request.py +0 -259
  118. orca_sdk/_generated_api_client/models/create_rac_model_request.py +0 -209
  119. orca_sdk/_generated_api_client/models/datasource_metadata.py +0 -142
  120. orca_sdk/_generated_api_client/models/delete_memories_request.py +0 -70
  121. orca_sdk/_generated_api_client/models/embed_request.py +0 -127
  122. orca_sdk/_generated_api_client/models/embedding_finetuning_method.py +0 -9
  123. orca_sdk/_generated_api_client/models/evaluation_request.py +0 -180
  124. orca_sdk/_generated_api_client/models/evaluation_response.py +0 -140
  125. orca_sdk/_generated_api_client/models/feedback_type.py +0 -9
  126. orca_sdk/_generated_api_client/models/field_validation_error.py +0 -103
  127. orca_sdk/_generated_api_client/models/filter_item.py +0 -231
  128. orca_sdk/_generated_api_client/models/filter_item_field_type_0_item.py +0 -15
  129. orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_1.py +0 -16
  130. orca_sdk/_generated_api_client/models/filter_item_op.py +0 -16
  131. orca_sdk/_generated_api_client/models/find_duplicates_analysis_result.py +0 -70
  132. orca_sdk/_generated_api_client/models/finetune_embedding_model_request.py +0 -259
  133. orca_sdk/_generated_api_client/models/finetune_embedding_model_request_training_args.py +0 -66
  134. orca_sdk/_generated_api_client/models/finetuned_embedding_model_metadata.py +0 -166
  135. orca_sdk/_generated_api_client/models/get_memories_request.py +0 -70
  136. orca_sdk/_generated_api_client/models/internal_server_error_response.py +0 -80
  137. orca_sdk/_generated_api_client/models/label_class_metrics.py +0 -108
  138. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup.py +0 -274
  139. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup_metadata.py +0 -68
  140. orca_sdk/_generated_api_client/models/label_prediction_result.py +0 -101
  141. orca_sdk/_generated_api_client/models/label_prediction_with_memories_and_feedback.py +0 -232
  142. orca_sdk/_generated_api_client/models/labeled_memory.py +0 -197
  143. orca_sdk/_generated_api_client/models/labeled_memory_insert.py +0 -108
  144. orca_sdk/_generated_api_client/models/labeled_memory_insert_metadata.py +0 -68
  145. orca_sdk/_generated_api_client/models/labeled_memory_lookup.py +0 -258
  146. orca_sdk/_generated_api_client/models/labeled_memory_lookup_metadata.py +0 -68
  147. orca_sdk/_generated_api_client/models/labeled_memory_metadata.py +0 -68
  148. orca_sdk/_generated_api_client/models/labeled_memory_metrics.py +0 -277
  149. orca_sdk/_generated_api_client/models/labeled_memory_update.py +0 -171
  150. orca_sdk/_generated_api_client/models/labeled_memory_update_metadata_type_0.py +0 -68
  151. orca_sdk/_generated_api_client/models/labeled_memoryset_metadata.py +0 -195
  152. orca_sdk/_generated_api_client/models/list_analyses_memoryset_name_or_id_analysis_get_type_type_0.py +0 -9
  153. orca_sdk/_generated_api_client/models/list_memories_request.py +0 -104
  154. orca_sdk/_generated_api_client/models/list_predictions_request.py +0 -234
  155. orca_sdk/_generated_api_client/models/list_predictions_request_sort_item_item_type_0.py +0 -9
  156. orca_sdk/_generated_api_client/models/list_predictions_request_sort_item_item_type_1.py +0 -9
  157. orca_sdk/_generated_api_client/models/lookup_request.py +0 -81
  158. orca_sdk/_generated_api_client/models/memoryset_analysis_request.py +0 -83
  159. orca_sdk/_generated_api_client/models/memoryset_analysis_request_type.py +0 -9
  160. orca_sdk/_generated_api_client/models/memoryset_analysis_response.py +0 -180
  161. orca_sdk/_generated_api_client/models/memoryset_analysis_response_config.py +0 -66
  162. orca_sdk/_generated_api_client/models/memoryset_analysis_response_type.py +0 -9
  163. orca_sdk/_generated_api_client/models/not_found_error_response.py +0 -100
  164. orca_sdk/_generated_api_client/models/not_found_error_response_resource_type_0.py +0 -20
  165. orca_sdk/_generated_api_client/models/prediction_feedback.py +0 -157
  166. orca_sdk/_generated_api_client/models/prediction_feedback_category.py +0 -115
  167. orca_sdk/_generated_api_client/models/prediction_feedback_request.py +0 -122
  168. orca_sdk/_generated_api_client/models/prediction_feedback_result.py +0 -102
  169. orca_sdk/_generated_api_client/models/prediction_request.py +0 -169
  170. orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py +0 -97
  171. orca_sdk/_generated_api_client/models/pretrained_embedding_model_name.py +0 -11
  172. orca_sdk/_generated_api_client/models/rac_head_type.py +0 -11
  173. orca_sdk/_generated_api_client/models/rac_model_metadata.py +0 -191
  174. orca_sdk/_generated_api_client/models/service_unavailable_error_response.py +0 -80
  175. orca_sdk/_generated_api_client/models/task.py +0 -198
  176. orca_sdk/_generated_api_client/models/task_status.py +0 -14
  177. orca_sdk/_generated_api_client/models/task_status_info.py +0 -133
  178. orca_sdk/_generated_api_client/models/unauthenticated_error_response.py +0 -72
  179. orca_sdk/_generated_api_client/models/unauthorized_error_response.py +0 -80
  180. orca_sdk/_generated_api_client/models/unprocessable_input_error_response.py +0 -94
  181. orca_sdk/_generated_api_client/models/update_prediction_request.py +0 -93
  182. orca_sdk/_generated_api_client/py.typed +0 -1
  183. orca_sdk/_generated_api_client/types.py +0 -56
  184. orca_sdk/_utils/task.py +0 -73
  185. orca_sdk-0.1.1.dist-info/RECORD +0 -175
@@ -1,159 +0,0 @@
1
- """Contains all the data models used in inputs/outputs"""
2
-
3
- from .analyze_neighbor_labels_result import AnalyzeNeighborLabelsResult
4
- from .api_key_metadata import ApiKeyMetadata
5
- from .base_model import BaseModel
6
- from .body_create_datasource_datasource_post import BodyCreateDatasourceDatasourcePost
7
- from .classification_evaluation_result import ClassificationEvaluationResult
8
- from .clone_labeled_memoryset_request import CloneLabeledMemorysetRequest
9
- from .column_info import ColumnInfo
10
- from .column_type import ColumnType
11
- from .conflict_error_response import ConflictErrorResponse
12
- from .create_api_key_request import CreateApiKeyRequest
13
- from .create_api_key_response import CreateApiKeyResponse
14
- from .create_labeled_memoryset_request import CreateLabeledMemorysetRequest
15
- from .create_rac_model_request import CreateRACModelRequest
16
- from .datasource_metadata import DatasourceMetadata
17
- from .delete_memories_request import DeleteMemoriesRequest
18
- from .embed_request import EmbedRequest
19
- from .embedding_finetuning_method import EmbeddingFinetuningMethod
20
- from .evaluation_request import EvaluationRequest
21
- from .evaluation_response import EvaluationResponse
22
- from .feedback_type import FeedbackType
23
- from .field_validation_error import FieldValidationError
24
- from .filter_item import FilterItem
25
- from .filter_item_field_type_0_item import FilterItemFieldType0Item
26
- from .filter_item_field_type_2_item_type_1 import FilterItemFieldType2ItemType1
27
- from .filter_item_op import FilterItemOp
28
- from .find_duplicates_analysis_result import FindDuplicatesAnalysisResult
29
- from .finetune_embedding_model_request import FinetuneEmbeddingModelRequest
30
- from .finetune_embedding_model_request_training_args import FinetuneEmbeddingModelRequestTrainingArgs
31
- from .finetuned_embedding_model_metadata import FinetunedEmbeddingModelMetadata
32
- from .get_memories_request import GetMemoriesRequest
33
- from .internal_server_error_response import InternalServerErrorResponse
34
- from .label_class_metrics import LabelClassMetrics
35
- from .label_prediction_memory_lookup import LabelPredictionMemoryLookup
36
- from .label_prediction_memory_lookup_metadata import LabelPredictionMemoryLookupMetadata
37
- from .label_prediction_result import LabelPredictionResult
38
- from .label_prediction_with_memories_and_feedback import LabelPredictionWithMemoriesAndFeedback
39
- from .labeled_memory import LabeledMemory
40
- from .labeled_memory_insert import LabeledMemoryInsert
41
- from .labeled_memory_insert_metadata import LabeledMemoryInsertMetadata
42
- from .labeled_memory_lookup import LabeledMemoryLookup
43
- from .labeled_memory_lookup_metadata import LabeledMemoryLookupMetadata
44
- from .labeled_memory_metadata import LabeledMemoryMetadata
45
- from .labeled_memory_metrics import LabeledMemoryMetrics
46
- from .labeled_memory_update import LabeledMemoryUpdate
47
- from .labeled_memory_update_metadata_type_0 import LabeledMemoryUpdateMetadataType0
48
- from .labeled_memoryset_metadata import LabeledMemorysetMetadata
49
- from .list_analyses_memoryset_name_or_id_analysis_get_type_type_0 import (
50
- ListAnalysesMemorysetNameOrIdAnalysisGetTypeType0,
51
- )
52
- from .list_memories_request import ListMemoriesRequest
53
- from .list_predictions_request import ListPredictionsRequest
54
- from .list_predictions_request_sort_item_item_type_0 import ListPredictionsRequestSortItemItemType0
55
- from .list_predictions_request_sort_item_item_type_1 import ListPredictionsRequestSortItemItemType1
56
- from .lookup_request import LookupRequest
57
- from .memoryset_analysis_request import MemorysetAnalysisRequest
58
- from .memoryset_analysis_request_type import MemorysetAnalysisRequestType
59
- from .memoryset_analysis_response import MemorysetAnalysisResponse
60
- from .memoryset_analysis_response_config import MemorysetAnalysisResponseConfig
61
- from .memoryset_analysis_response_type import MemorysetAnalysisResponseType
62
- from .not_found_error_response import NotFoundErrorResponse
63
- from .not_found_error_response_resource_type_0 import NotFoundErrorResponseResourceType0
64
- from .prediction_feedback import PredictionFeedback
65
- from .prediction_feedback_category import PredictionFeedbackCategory
66
- from .prediction_feedback_request import PredictionFeedbackRequest
67
- from .prediction_feedback_result import PredictionFeedbackResult
68
- from .prediction_request import PredictionRequest
69
- from .pretrained_embedding_model_metadata import PretrainedEmbeddingModelMetadata
70
- from .pretrained_embedding_model_name import PretrainedEmbeddingModelName
71
- from .rac_head_type import RACHeadType
72
- from .rac_model_metadata import RACModelMetadata
73
- from .service_unavailable_error_response import ServiceUnavailableErrorResponse
74
- from .task import Task
75
- from .task_status import TaskStatus
76
- from .task_status_info import TaskStatusInfo
77
- from .unauthenticated_error_response import UnauthenticatedErrorResponse
78
- from .unauthorized_error_response import UnauthorizedErrorResponse
79
- from .unprocessable_input_error_response import UnprocessableInputErrorResponse
80
- from .update_prediction_request import UpdatePredictionRequest
81
-
82
- __all__ = (
83
- "AnalyzeNeighborLabelsResult",
84
- "ApiKeyMetadata",
85
- "BaseModel",
86
- "BodyCreateDatasourceDatasourcePost",
87
- "ClassificationEvaluationResult",
88
- "CloneLabeledMemorysetRequest",
89
- "ColumnInfo",
90
- "ColumnType",
91
- "ConflictErrorResponse",
92
- "CreateApiKeyRequest",
93
- "CreateApiKeyResponse",
94
- "CreateLabeledMemorysetRequest",
95
- "CreateRACModelRequest",
96
- "DatasourceMetadata",
97
- "DeleteMemoriesRequest",
98
- "EmbeddingFinetuningMethod",
99
- "EmbedRequest",
100
- "EvaluationRequest",
101
- "EvaluationResponse",
102
- "FeedbackType",
103
- "FieldValidationError",
104
- "FilterItem",
105
- "FilterItemFieldType0Item",
106
- "FilterItemFieldType2ItemType1",
107
- "FilterItemOp",
108
- "FindDuplicatesAnalysisResult",
109
- "FinetunedEmbeddingModelMetadata",
110
- "FinetuneEmbeddingModelRequest",
111
- "FinetuneEmbeddingModelRequestTrainingArgs",
112
- "GetMemoriesRequest",
113
- "InternalServerErrorResponse",
114
- "LabelClassMetrics",
115
- "LabeledMemory",
116
- "LabeledMemoryInsert",
117
- "LabeledMemoryInsertMetadata",
118
- "LabeledMemoryLookup",
119
- "LabeledMemoryLookupMetadata",
120
- "LabeledMemoryMetadata",
121
- "LabeledMemoryMetrics",
122
- "LabeledMemorysetMetadata",
123
- "LabeledMemoryUpdate",
124
- "LabeledMemoryUpdateMetadataType0",
125
- "LabelPredictionMemoryLookup",
126
- "LabelPredictionMemoryLookupMetadata",
127
- "LabelPredictionResult",
128
- "LabelPredictionWithMemoriesAndFeedback",
129
- "ListAnalysesMemorysetNameOrIdAnalysisGetTypeType0",
130
- "ListMemoriesRequest",
131
- "ListPredictionsRequest",
132
- "ListPredictionsRequestSortItemItemType0",
133
- "ListPredictionsRequestSortItemItemType1",
134
- "LookupRequest",
135
- "MemorysetAnalysisRequest",
136
- "MemorysetAnalysisRequestType",
137
- "MemorysetAnalysisResponse",
138
- "MemorysetAnalysisResponseConfig",
139
- "MemorysetAnalysisResponseType",
140
- "NotFoundErrorResponse",
141
- "NotFoundErrorResponseResourceType0",
142
- "PredictionFeedback",
143
- "PredictionFeedbackCategory",
144
- "PredictionFeedbackRequest",
145
- "PredictionFeedbackResult",
146
- "PredictionRequest",
147
- "PretrainedEmbeddingModelMetadata",
148
- "PretrainedEmbeddingModelName",
149
- "RACHeadType",
150
- "RACModelMetadata",
151
- "ServiceUnavailableErrorResponse",
152
- "Task",
153
- "TaskStatus",
154
- "TaskStatusInfo",
155
- "UnauthenticatedErrorResponse",
156
- "UnauthorizedErrorResponse",
157
- "UnprocessableInputErrorResponse",
158
- "UpdatePredictionRequest",
159
- )
@@ -1,84 +0,0 @@
1
- """
2
- This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
-
4
- It is a customized template from the openapi-python-client tool's default template:
5
- https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
6
-
7
- The main change is:
8
- - Fix typing issues
9
- """
10
-
11
- # flake8: noqa: C901
12
-
13
- from typing import TYPE_CHECKING, Any, List, Type, TypeVar
14
-
15
- from attrs import define as _attrs_define
16
- from attrs import field as _attrs_field
17
-
18
- if TYPE_CHECKING:
19
- from ..models.label_class_metrics import LabelClassMetrics
20
-
21
-
22
- T = TypeVar("T", bound="AnalyzeNeighborLabelsResult")
23
-
24
-
25
- @_attrs_define
26
- class AnalyzeNeighborLabelsResult:
27
- """
28
- Attributes:
29
- label_metrics (List['LabelClassMetrics']):
30
- """
31
-
32
- label_metrics: List["LabelClassMetrics"]
33
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
34
-
35
- def to_dict(self) -> dict[str, Any]:
36
- label_metrics = []
37
- for label_metrics_item_data in self.label_metrics:
38
- label_metrics_item = label_metrics_item_data.to_dict()
39
- label_metrics.append(label_metrics_item)
40
-
41
- field_dict: dict[str, Any] = {}
42
- field_dict.update(self.additional_properties)
43
- field_dict.update(
44
- {
45
- "label_metrics": label_metrics,
46
- }
47
- )
48
-
49
- return field_dict
50
-
51
- @classmethod
52
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
53
- from ..models.label_class_metrics import LabelClassMetrics
54
-
55
- d = src_dict.copy()
56
- label_metrics = []
57
- _label_metrics = d.pop("label_metrics")
58
- for label_metrics_item_data in _label_metrics:
59
- label_metrics_item = LabelClassMetrics.from_dict(label_metrics_item_data)
60
-
61
- label_metrics.append(label_metrics_item)
62
-
63
- analyze_neighbor_labels_result = cls(
64
- label_metrics=label_metrics,
65
- )
66
-
67
- analyze_neighbor_labels_result.additional_properties = d
68
- return analyze_neighbor_labels_result
69
-
70
- @property
71
- def additional_keys(self) -> list[str]:
72
- return list(self.additional_properties.keys())
73
-
74
- def __getitem__(self, key: str) -> Any:
75
- return self.additional_properties[key]
76
-
77
- def __setitem__(self, key: str, value: Any) -> None:
78
- self.additional_properties[key] = value
79
-
80
- def __delitem__(self, key: str) -> None:
81
- del self.additional_properties[key]
82
-
83
- def __contains__(self, key: str) -> bool:
84
- return key in self.additional_properties
@@ -1,118 +0,0 @@
1
- """
2
- This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
-
4
- It is a customized template from the openapi-python-client tool's default template:
5
- https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
6
-
7
- The main change is:
8
- - Fix typing issues
9
- """
10
-
11
- # flake8: noqa: C901
12
-
13
- import datetime
14
- from typing import Any, Type, TypeVar, Union, cast
15
-
16
- from attrs import define as _attrs_define
17
- from attrs import field as _attrs_field
18
- from dateutil.parser import isoparse
19
-
20
- T = TypeVar("T", bound="ApiKeyMetadata")
21
-
22
-
23
- @_attrs_define
24
- class ApiKeyMetadata:
25
- """
26
- Attributes:
27
- id (str):
28
- org_id (str):
29
- name (str):
30
- created_by (Union[None, str]):
31
- created_at (datetime.datetime):
32
- updated_at (datetime.datetime):
33
- """
34
-
35
- id: str
36
- org_id: str
37
- name: str
38
- created_by: Union[None, str]
39
- created_at: datetime.datetime
40
- updated_at: datetime.datetime
41
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
42
-
43
- def to_dict(self) -> dict[str, Any]:
44
- id = self.id
45
-
46
- org_id = self.org_id
47
-
48
- name = self.name
49
-
50
- created_by: Union[None, str]
51
- created_by = self.created_by
52
-
53
- created_at = self.created_at.isoformat()
54
-
55
- updated_at = self.updated_at.isoformat()
56
-
57
- field_dict: dict[str, Any] = {}
58
- field_dict.update(self.additional_properties)
59
- field_dict.update(
60
- {
61
- "id": id,
62
- "org_id": org_id,
63
- "name": name,
64
- "created_by": created_by,
65
- "created_at": created_at,
66
- "updated_at": updated_at,
67
- }
68
- )
69
-
70
- return field_dict
71
-
72
- @classmethod
73
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
74
- d = src_dict.copy()
75
- id = d.pop("id")
76
-
77
- org_id = d.pop("org_id")
78
-
79
- name = d.pop("name")
80
-
81
- def _parse_created_by(data: object) -> Union[None, str]:
82
- if data is None:
83
- return data
84
- return cast(Union[None, str], data)
85
-
86
- created_by = _parse_created_by(d.pop("created_by"))
87
-
88
- created_at = isoparse(d.pop("created_at"))
89
-
90
- updated_at = isoparse(d.pop("updated_at"))
91
-
92
- api_key_metadata = cls(
93
- id=id,
94
- org_id=org_id,
95
- name=name,
96
- created_by=created_by,
97
- created_at=created_at,
98
- updated_at=updated_at,
99
- )
100
-
101
- api_key_metadata.additional_properties = d
102
- return api_key_metadata
103
-
104
- @property
105
- def additional_keys(self) -> list[str]:
106
- return list(self.additional_properties.keys())
107
-
108
- def __getitem__(self, key: str) -> Any:
109
- return self.additional_properties[key]
110
-
111
- def __setitem__(self, key: str, value: Any) -> None:
112
- self.additional_properties[key] = value
113
-
114
- def __delitem__(self, key: str) -> None:
115
- del self.additional_properties[key]
116
-
117
- def __contains__(self, key: str) -> bool:
118
- return key in self.additional_properties
@@ -1,55 +0,0 @@
1
- """
2
- This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
-
4
- It is a customized template from the openapi-python-client tool's default template:
5
- https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
6
-
7
- The main change is:
8
- - Fix typing issues
9
- """
10
-
11
- # flake8: noqa: C901
12
-
13
- from typing import Any, Type, TypeVar
14
-
15
- from attrs import define as _attrs_define
16
- from attrs import field as _attrs_field
17
-
18
- T = TypeVar("T", bound="BaseModel")
19
-
20
-
21
- @_attrs_define
22
- class BaseModel:
23
- """ """
24
-
25
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
26
-
27
- def to_dict(self) -> dict[str, Any]:
28
- field_dict: dict[str, Any] = {}
29
- field_dict.update(self.additional_properties)
30
-
31
- return field_dict
32
-
33
- @classmethod
34
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
35
- d = src_dict.copy()
36
- base_model = cls()
37
-
38
- base_model.additional_properties = d
39
- return base_model
40
-
41
- @property
42
- def additional_keys(self) -> list[str]:
43
- return list(self.additional_properties.keys())
44
-
45
- def __getitem__(self, key: str) -> Any:
46
- return self.additional_properties[key]
47
-
48
- def __setitem__(self, key: str, value: Any) -> None:
49
- self.additional_properties[key] = value
50
-
51
- def __delitem__(self, key: str) -> None:
52
- del self.additional_properties[key]
53
-
54
- def __contains__(self, key: str) -> bool:
55
- return key in self.additional_properties
@@ -1,176 +0,0 @@
1
- """
2
- This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
-
4
- It is a customized template from the openapi-python-client tool's default template:
5
- https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
6
-
7
- The main change is:
8
- - Fix typing issues
9
- """
10
-
11
- # flake8: noqa: C901
12
-
13
- import json
14
- from io import BytesIO
15
- from typing import Any, List, Tuple, Type, TypeVar, Union, cast
16
-
17
- from attrs import define as _attrs_define
18
- from attrs import field as _attrs_field
19
-
20
- from ..types import UNSET, File, FileJsonType, Unset
21
-
22
- T = TypeVar("T", bound="BodyCreateDatasourceDatasourcePost")
23
-
24
-
25
- @_attrs_define
26
- class BodyCreateDatasourceDatasourcePost:
27
- """
28
- Attributes:
29
- name (str):
30
- files (Union[List[File], None, Unset]):
31
- content (Union[None, Unset, str]):
32
- """
33
-
34
- name: str
35
- files: Union[List[File], None, Unset] = UNSET
36
- content: Union[None, Unset, str] = UNSET
37
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
38
-
39
- def to_dict(self) -> dict[str, Any]:
40
- name = self.name
41
-
42
- files: Union[List[FileJsonType], None, Unset]
43
- if isinstance(self.files, Unset):
44
- files = UNSET
45
- elif isinstance(self.files, list):
46
- files = []
47
- for files_type_0_item_data in self.files:
48
- files_type_0_item = files_type_0_item_data.to_tuple()
49
-
50
- files.append(files_type_0_item)
51
-
52
- else:
53
- files = self.files
54
-
55
- content: Union[None, Unset, str]
56
- if isinstance(self.content, Unset):
57
- content = UNSET
58
- else:
59
- content = self.content
60
-
61
- field_dict: dict[str, Any] = {}
62
- field_dict.update(self.additional_properties)
63
- field_dict.update(
64
- {
65
- "name": name,
66
- }
67
- )
68
- if files is not UNSET:
69
- field_dict["files"] = files
70
- if content is not UNSET:
71
- field_dict["content"] = content
72
-
73
- return field_dict
74
-
75
- def to_multipart(self) -> dict[str, Any]:
76
- name = (None, str(self.name).encode(), "text/plain")
77
-
78
- files: Union[Tuple[None, bytes, str], Unset]
79
-
80
- if isinstance(self.files, Unset):
81
- files = UNSET
82
- elif isinstance(self.files, list):
83
- _temp_files = []
84
- for files_type_0_item_data in self.files:
85
- files_type_0_item = files_type_0_item_data.to_tuple()
86
-
87
- _temp_files.append(files_type_0_item)
88
- files = (None, json.dumps(_temp_files).encode(), "application/json")
89
- else:
90
- files = (None, str(self.files).encode(), "text/plain")
91
-
92
- content: Union[Tuple[None, bytes, str], Unset]
93
-
94
- if isinstance(self.content, Unset):
95
- content = UNSET
96
- elif isinstance(self.content, str):
97
- content = (None, str(self.content).encode(), "text/plain")
98
- else:
99
- content = (None, str(self.content).encode(), "text/plain")
100
-
101
- field_dict: dict[str, Any] = {}
102
- for prop_name, prop in self.additional_properties.items():
103
- field_dict[prop_name] = (None, str(prop).encode(), "text/plain")
104
-
105
- field_dict.update(
106
- {
107
- "name": name,
108
- }
109
- )
110
- if files is not UNSET:
111
- field_dict["files"] = files
112
- if content is not UNSET:
113
- field_dict["content"] = content
114
-
115
- return field_dict
116
-
117
- @classmethod
118
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
119
- d = src_dict.copy()
120
- name = d.pop("name")
121
-
122
- def _parse_files(data: object) -> Union[List[File], None, Unset]:
123
- if data is None:
124
- return data
125
- if isinstance(data, Unset):
126
- return data
127
- try:
128
- if not isinstance(data, list):
129
- raise TypeError()
130
- files_type_0 = []
131
- _files_type_0 = data
132
- for files_type_0_item_data in _files_type_0:
133
- files_type_0_item = File(payload=BytesIO(files_type_0_item_data))
134
-
135
- files_type_0.append(files_type_0_item)
136
-
137
- return files_type_0
138
- except: # noqa: E722
139
- pass
140
- return cast(Union[List[File], None, Unset], data)
141
-
142
- files = _parse_files(d.pop("files", UNSET))
143
-
144
- def _parse_content(data: object) -> Union[None, Unset, str]:
145
- if data is None:
146
- return data
147
- if isinstance(data, Unset):
148
- return data
149
- return cast(Union[None, Unset, str], data)
150
-
151
- content = _parse_content(d.pop("content", UNSET))
152
-
153
- body_create_datasource_datasource_post = cls(
154
- name=name,
155
- files=files,
156
- content=content,
157
- )
158
-
159
- body_create_datasource_datasource_post.additional_properties = d
160
- return body_create_datasource_datasource_post
161
-
162
- @property
163
- def additional_keys(self) -> list[str]:
164
- return list(self.additional_properties.keys())
165
-
166
- def __getitem__(self, key: str) -> Any:
167
- return self.additional_properties[key]
168
-
169
- def __setitem__(self, key: str, value: Any) -> None:
170
- self.additional_properties[key] = value
171
-
172
- def __delitem__(self, key: str) -> None:
173
- del self.additional_properties[key]
174
-
175
- def __contains__(self, key: str) -> bool:
176
- return key in self.additional_properties