orca-sdk 0.1.1__py3-none-any.whl → 0.1.3__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 (186) 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 +31 -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/async_client.py +3795 -0
  17. orca_sdk/classification_model.py +601 -129
  18. orca_sdk/classification_model_test.py +415 -117
  19. orca_sdk/client.py +3787 -0
  20. orca_sdk/conftest.py +184 -38
  21. orca_sdk/credentials.py +162 -20
  22. orca_sdk/credentials_test.py +100 -16
  23. orca_sdk/datasource.py +268 -68
  24. orca_sdk/datasource_test.py +266 -18
  25. orca_sdk/embedding_model.py +434 -82
  26. orca_sdk/embedding_model_test.py +66 -33
  27. orca_sdk/job.py +343 -0
  28. orca_sdk/job_test.py +108 -0
  29. orca_sdk/memoryset.py +1690 -324
  30. orca_sdk/memoryset_test.py +456 -119
  31. orca_sdk/regression_model.py +694 -0
  32. orca_sdk/regression_model_test.py +378 -0
  33. orca_sdk/telemetry.py +460 -143
  34. orca_sdk/telemetry_test.py +43 -24
  35. {orca_sdk-0.1.1.dist-info → orca_sdk-0.1.3.dist-info}/METADATA +34 -16
  36. orca_sdk-0.1.3.dist-info/RECORD +41 -0
  37. {orca_sdk-0.1.1.dist-info → orca_sdk-0.1.3.dist-info}/WHEEL +1 -1
  38. orca_sdk/_generated_api_client/__init__.py +0 -3
  39. orca_sdk/_generated_api_client/api/__init__.py +0 -193
  40. orca_sdk/_generated_api_client/api/auth/__init__.py +0 -0
  41. orca_sdk/_generated_api_client/api/auth/check_authentication_auth_get.py +0 -128
  42. orca_sdk/_generated_api_client/api/auth/create_api_key_auth_api_key_post.py +0 -170
  43. orca_sdk/_generated_api_client/api/auth/delete_api_key_auth_api_key_name_or_id_delete.py +0 -156
  44. orca_sdk/_generated_api_client/api/auth/delete_org_auth_org_delete.py +0 -130
  45. orca_sdk/_generated_api_client/api/auth/list_api_keys_auth_api_key_get.py +0 -127
  46. orca_sdk/_generated_api_client/api/classification_model/__init__.py +0 -0
  47. orca_sdk/_generated_api_client/api/classification_model/create_evaluation_classification_model_model_name_or_id_evaluation_post.py +0 -183
  48. orca_sdk/_generated_api_client/api/classification_model/create_model_classification_model_post.py +0 -170
  49. orca_sdk/_generated_api_client/api/classification_model/delete_evaluation_classification_model_model_name_or_id_evaluation_task_id_delete.py +0 -168
  50. orca_sdk/_generated_api_client/api/classification_model/delete_model_classification_model_name_or_id_delete.py +0 -154
  51. orca_sdk/_generated_api_client/api/classification_model/get_evaluation_classification_model_model_name_or_id_evaluation_task_id_get.py +0 -170
  52. orca_sdk/_generated_api_client/api/classification_model/get_model_classification_model_name_or_id_get.py +0 -156
  53. orca_sdk/_generated_api_client/api/classification_model/list_evaluations_classification_model_model_name_or_id_evaluation_get.py +0 -161
  54. orca_sdk/_generated_api_client/api/classification_model/list_models_classification_model_get.py +0 -127
  55. orca_sdk/_generated_api_client/api/classification_model/predict_gpu_classification_model_name_or_id_prediction_post.py +0 -190
  56. orca_sdk/_generated_api_client/api/datasource/__init__.py +0 -0
  57. orca_sdk/_generated_api_client/api/datasource/create_datasource_datasource_post.py +0 -167
  58. orca_sdk/_generated_api_client/api/datasource/delete_datasource_datasource_name_or_id_delete.py +0 -156
  59. orca_sdk/_generated_api_client/api/datasource/get_datasource_datasource_name_or_id_get.py +0 -156
  60. orca_sdk/_generated_api_client/api/datasource/list_datasources_datasource_get.py +0 -127
  61. orca_sdk/_generated_api_client/api/default/__init__.py +0 -0
  62. orca_sdk/_generated_api_client/api/default/healthcheck_get.py +0 -118
  63. orca_sdk/_generated_api_client/api/default/healthcheck_gpu_get.py +0 -118
  64. orca_sdk/_generated_api_client/api/finetuned_embedding_model/__init__.py +0 -0
  65. orca_sdk/_generated_api_client/api/finetuned_embedding_model/create_finetuned_embedding_model_finetuned_embedding_model_post.py +0 -168
  66. orca_sdk/_generated_api_client/api/finetuned_embedding_model/delete_finetuned_embedding_model_finetuned_embedding_model_name_or_id_delete.py +0 -156
  67. 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
  68. orca_sdk/_generated_api_client/api/finetuned_embedding_model/get_finetuned_embedding_model_finetuned_embedding_model_name_or_id_get.py +0 -156
  69. orca_sdk/_generated_api_client/api/finetuned_embedding_model/list_finetuned_embedding_models_finetuned_embedding_model_get.py +0 -127
  70. orca_sdk/_generated_api_client/api/memoryset/__init__.py +0 -0
  71. orca_sdk/_generated_api_client/api/memoryset/clone_memoryset_memoryset_name_or_id_clone_post.py +0 -181
  72. orca_sdk/_generated_api_client/api/memoryset/create_analysis_memoryset_name_or_id_analysis_post.py +0 -183
  73. orca_sdk/_generated_api_client/api/memoryset/create_memoryset_memoryset_post.py +0 -168
  74. orca_sdk/_generated_api_client/api/memoryset/delete_memories_memoryset_name_or_id_memories_delete_post.py +0 -181
  75. orca_sdk/_generated_api_client/api/memoryset/delete_memory_memoryset_name_or_id_memory_memory_id_delete.py +0 -167
  76. orca_sdk/_generated_api_client/api/memoryset/delete_memoryset_memoryset_name_or_id_delete.py +0 -156
  77. orca_sdk/_generated_api_client/api/memoryset/get_analysis_memoryset_name_or_id_analysis_analysis_task_id_get.py +0 -169
  78. orca_sdk/_generated_api_client/api/memoryset/get_memories_memoryset_name_or_id_memories_get_post.py +0 -188
  79. orca_sdk/_generated_api_client/api/memoryset/get_memory_memoryset_name_or_id_memory_memory_id_get.py +0 -169
  80. orca_sdk/_generated_api_client/api/memoryset/get_memoryset_memoryset_name_or_id_get.py +0 -156
  81. orca_sdk/_generated_api_client/api/memoryset/insert_memories_gpu_memoryset_name_or_id_memory_post.py +0 -184
  82. orca_sdk/_generated_api_client/api/memoryset/list_analyses_memoryset_name_or_id_analysis_get.py +0 -260
  83. orca_sdk/_generated_api_client/api/memoryset/list_memorysets_memoryset_get.py +0 -127
  84. orca_sdk/_generated_api_client/api/memoryset/memoryset_lookup_gpu_memoryset_name_or_id_lookup_post.py +0 -193
  85. orca_sdk/_generated_api_client/api/memoryset/query_memoryset_memoryset_name_or_id_memories_post.py +0 -188
  86. orca_sdk/_generated_api_client/api/memoryset/update_memories_gpu_memoryset_name_or_id_memories_patch.py +0 -191
  87. orca_sdk/_generated_api_client/api/memoryset/update_memory_gpu_memoryset_name_or_id_memory_patch.py +0 -187
  88. orca_sdk/_generated_api_client/api/pretrained_embedding_model/__init__.py +0 -0
  89. orca_sdk/_generated_api_client/api/pretrained_embedding_model/embed_with_pretrained_model_gpu_pretrained_embedding_model_model_name_embedding_post.py +0 -188
  90. orca_sdk/_generated_api_client/api/pretrained_embedding_model/get_pretrained_embedding_model_pretrained_embedding_model_model_name_get.py +0 -157
  91. orca_sdk/_generated_api_client/api/pretrained_embedding_model/list_pretrained_embedding_models_pretrained_embedding_model_get.py +0 -127
  92. orca_sdk/_generated_api_client/api/task/__init__.py +0 -0
  93. orca_sdk/_generated_api_client/api/task/abort_task_task_task_id_abort_delete.py +0 -154
  94. orca_sdk/_generated_api_client/api/task/get_task_status_task_task_id_status_get.py +0 -156
  95. orca_sdk/_generated_api_client/api/task/list_tasks_task_get.py +0 -243
  96. orca_sdk/_generated_api_client/api/telemetry/__init__.py +0 -0
  97. orca_sdk/_generated_api_client/api/telemetry/drop_feedback_category_with_data_telemetry_feedback_category_name_or_id_delete.py +0 -162
  98. orca_sdk/_generated_api_client/api/telemetry/get_feedback_category_telemetry_feedback_category_name_or_id_get.py +0 -156
  99. orca_sdk/_generated_api_client/api/telemetry/get_prediction_telemetry_prediction_prediction_id_get.py +0 -157
  100. orca_sdk/_generated_api_client/api/telemetry/list_feedback_categories_telemetry_feedback_category_get.py +0 -127
  101. orca_sdk/_generated_api_client/api/telemetry/list_predictions_telemetry_prediction_post.py +0 -175
  102. orca_sdk/_generated_api_client/api/telemetry/record_prediction_feedback_telemetry_prediction_feedback_put.py +0 -171
  103. orca_sdk/_generated_api_client/api/telemetry/update_prediction_telemetry_prediction_prediction_id_patch.py +0 -181
  104. orca_sdk/_generated_api_client/client.py +0 -216
  105. orca_sdk/_generated_api_client/errors.py +0 -38
  106. orca_sdk/_generated_api_client/models/__init__.py +0 -159
  107. orca_sdk/_generated_api_client/models/analyze_neighbor_labels_result.py +0 -84
  108. orca_sdk/_generated_api_client/models/api_key_metadata.py +0 -118
  109. orca_sdk/_generated_api_client/models/base_model.py +0 -55
  110. orca_sdk/_generated_api_client/models/body_create_datasource_datasource_post.py +0 -176
  111. orca_sdk/_generated_api_client/models/classification_evaluation_result.py +0 -114
  112. orca_sdk/_generated_api_client/models/clone_labeled_memoryset_request.py +0 -150
  113. orca_sdk/_generated_api_client/models/column_info.py +0 -114
  114. orca_sdk/_generated_api_client/models/column_type.py +0 -14
  115. orca_sdk/_generated_api_client/models/conflict_error_response.py +0 -80
  116. orca_sdk/_generated_api_client/models/create_api_key_request.py +0 -99
  117. orca_sdk/_generated_api_client/models/create_api_key_response.py +0 -126
  118. orca_sdk/_generated_api_client/models/create_labeled_memoryset_request.py +0 -259
  119. orca_sdk/_generated_api_client/models/create_rac_model_request.py +0 -209
  120. orca_sdk/_generated_api_client/models/datasource_metadata.py +0 -142
  121. orca_sdk/_generated_api_client/models/delete_memories_request.py +0 -70
  122. orca_sdk/_generated_api_client/models/embed_request.py +0 -127
  123. orca_sdk/_generated_api_client/models/embedding_finetuning_method.py +0 -9
  124. orca_sdk/_generated_api_client/models/evaluation_request.py +0 -180
  125. orca_sdk/_generated_api_client/models/evaluation_response.py +0 -140
  126. orca_sdk/_generated_api_client/models/feedback_type.py +0 -9
  127. orca_sdk/_generated_api_client/models/field_validation_error.py +0 -103
  128. orca_sdk/_generated_api_client/models/filter_item.py +0 -231
  129. orca_sdk/_generated_api_client/models/filter_item_field_type_0_item.py +0 -15
  130. orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_1.py +0 -16
  131. orca_sdk/_generated_api_client/models/filter_item_op.py +0 -16
  132. orca_sdk/_generated_api_client/models/find_duplicates_analysis_result.py +0 -70
  133. orca_sdk/_generated_api_client/models/finetune_embedding_model_request.py +0 -259
  134. orca_sdk/_generated_api_client/models/finetune_embedding_model_request_training_args.py +0 -66
  135. orca_sdk/_generated_api_client/models/finetuned_embedding_model_metadata.py +0 -166
  136. orca_sdk/_generated_api_client/models/get_memories_request.py +0 -70
  137. orca_sdk/_generated_api_client/models/internal_server_error_response.py +0 -80
  138. orca_sdk/_generated_api_client/models/label_class_metrics.py +0 -108
  139. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup.py +0 -274
  140. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup_metadata.py +0 -68
  141. orca_sdk/_generated_api_client/models/label_prediction_result.py +0 -101
  142. orca_sdk/_generated_api_client/models/label_prediction_with_memories_and_feedback.py +0 -232
  143. orca_sdk/_generated_api_client/models/labeled_memory.py +0 -197
  144. orca_sdk/_generated_api_client/models/labeled_memory_insert.py +0 -108
  145. orca_sdk/_generated_api_client/models/labeled_memory_insert_metadata.py +0 -68
  146. orca_sdk/_generated_api_client/models/labeled_memory_lookup.py +0 -258
  147. orca_sdk/_generated_api_client/models/labeled_memory_lookup_metadata.py +0 -68
  148. orca_sdk/_generated_api_client/models/labeled_memory_metadata.py +0 -68
  149. orca_sdk/_generated_api_client/models/labeled_memory_metrics.py +0 -277
  150. orca_sdk/_generated_api_client/models/labeled_memory_update.py +0 -171
  151. orca_sdk/_generated_api_client/models/labeled_memory_update_metadata_type_0.py +0 -68
  152. orca_sdk/_generated_api_client/models/labeled_memoryset_metadata.py +0 -195
  153. orca_sdk/_generated_api_client/models/list_analyses_memoryset_name_or_id_analysis_get_type_type_0.py +0 -9
  154. orca_sdk/_generated_api_client/models/list_memories_request.py +0 -104
  155. orca_sdk/_generated_api_client/models/list_predictions_request.py +0 -234
  156. orca_sdk/_generated_api_client/models/list_predictions_request_sort_item_item_type_0.py +0 -9
  157. orca_sdk/_generated_api_client/models/list_predictions_request_sort_item_item_type_1.py +0 -9
  158. orca_sdk/_generated_api_client/models/lookup_request.py +0 -81
  159. orca_sdk/_generated_api_client/models/memoryset_analysis_request.py +0 -83
  160. orca_sdk/_generated_api_client/models/memoryset_analysis_request_type.py +0 -9
  161. orca_sdk/_generated_api_client/models/memoryset_analysis_response.py +0 -180
  162. orca_sdk/_generated_api_client/models/memoryset_analysis_response_config.py +0 -66
  163. orca_sdk/_generated_api_client/models/memoryset_analysis_response_type.py +0 -9
  164. orca_sdk/_generated_api_client/models/not_found_error_response.py +0 -100
  165. orca_sdk/_generated_api_client/models/not_found_error_response_resource_type_0.py +0 -20
  166. orca_sdk/_generated_api_client/models/prediction_feedback.py +0 -157
  167. orca_sdk/_generated_api_client/models/prediction_feedback_category.py +0 -115
  168. orca_sdk/_generated_api_client/models/prediction_feedback_request.py +0 -122
  169. orca_sdk/_generated_api_client/models/prediction_feedback_result.py +0 -102
  170. orca_sdk/_generated_api_client/models/prediction_request.py +0 -169
  171. orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py +0 -97
  172. orca_sdk/_generated_api_client/models/pretrained_embedding_model_name.py +0 -11
  173. orca_sdk/_generated_api_client/models/rac_head_type.py +0 -11
  174. orca_sdk/_generated_api_client/models/rac_model_metadata.py +0 -191
  175. orca_sdk/_generated_api_client/models/service_unavailable_error_response.py +0 -80
  176. orca_sdk/_generated_api_client/models/task.py +0 -198
  177. orca_sdk/_generated_api_client/models/task_status.py +0 -14
  178. orca_sdk/_generated_api_client/models/task_status_info.py +0 -133
  179. orca_sdk/_generated_api_client/models/unauthenticated_error_response.py +0 -72
  180. orca_sdk/_generated_api_client/models/unauthorized_error_response.py +0 -80
  181. orca_sdk/_generated_api_client/models/unprocessable_input_error_response.py +0 -94
  182. orca_sdk/_generated_api_client/models/update_prediction_request.py +0 -93
  183. orca_sdk/_generated_api_client/py.typed +0 -1
  184. orca_sdk/_generated_api_client/types.py +0 -56
  185. orca_sdk/_utils/task.py +0 -73
  186. 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