orca-sdk 0.0.78__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 (188) hide show
  1. orca_sdk/__init__.py +24 -0
  2. orca_sdk/_generated_api_client/__init__.py +3 -0
  3. orca_sdk/_generated_api_client/api/__init__.py +205 -0
  4. orca_sdk/_generated_api_client/api/auth/__init__.py +0 -0
  5. orca_sdk/_generated_api_client/api/auth/check_authentication_auth_get.py +130 -0
  6. orca_sdk/_generated_api_client/api/auth/create_api_key_auth_api_key_post.py +172 -0
  7. orca_sdk/_generated_api_client/api/auth/delete_api_key_auth_api_key_name_or_id_delete.py +158 -0
  8. orca_sdk/_generated_api_client/api/auth/delete_org_auth_org_delete.py +132 -0
  9. orca_sdk/_generated_api_client/api/auth/list_api_keys_auth_api_key_get.py +129 -0
  10. orca_sdk/_generated_api_client/api/classification_model/__init__.py +0 -0
  11. orca_sdk/_generated_api_client/api/classification_model/create_evaluation_classification_model_model_name_or_id_evaluation_post.py +185 -0
  12. orca_sdk/_generated_api_client/api/classification_model/create_model_classification_model_post.py +172 -0
  13. orca_sdk/_generated_api_client/api/classification_model/delete_evaluation_classification_model_model_name_or_id_evaluation_task_id_delete.py +170 -0
  14. orca_sdk/_generated_api_client/api/classification_model/delete_model_classification_model_name_or_id_delete.py +156 -0
  15. orca_sdk/_generated_api_client/api/classification_model/get_evaluation_classification_model_model_name_or_id_evaluation_task_id_get.py +172 -0
  16. orca_sdk/_generated_api_client/api/classification_model/get_model_classification_model_name_or_id_get.py +158 -0
  17. orca_sdk/_generated_api_client/api/classification_model/list_evaluations_classification_model_model_name_or_id_evaluation_get.py +163 -0
  18. orca_sdk/_generated_api_client/api/classification_model/list_models_classification_model_get.py +129 -0
  19. orca_sdk/_generated_api_client/api/classification_model/predict_gpu_classification_model_name_or_id_prediction_post.py +192 -0
  20. orca_sdk/_generated_api_client/api/datasource/__init__.py +0 -0
  21. orca_sdk/_generated_api_client/api/datasource/create_datasource_datasource_post.py +169 -0
  22. orca_sdk/_generated_api_client/api/datasource/create_embedding_evaluation_datasource_name_or_id_embedding_evaluation_post.py +185 -0
  23. orca_sdk/_generated_api_client/api/datasource/delete_datasource_datasource_name_or_id_delete.py +158 -0
  24. orca_sdk/_generated_api_client/api/datasource/get_datasource_datasource_name_or_id_get.py +158 -0
  25. orca_sdk/_generated_api_client/api/datasource/get_embedding_evaluation_datasource_name_or_id_embedding_evaluation_task_id_get.py +171 -0
  26. orca_sdk/_generated_api_client/api/datasource/list_datasources_datasource_get.py +129 -0
  27. orca_sdk/_generated_api_client/api/datasource/list_embedding_evaluations_datasource_name_or_id_embedding_evaluation_get.py +237 -0
  28. orca_sdk/_generated_api_client/api/default/__init__.py +0 -0
  29. orca_sdk/_generated_api_client/api/default/healthcheck_get.py +120 -0
  30. orca_sdk/_generated_api_client/api/default/healthcheck_gpu_get.py +120 -0
  31. orca_sdk/_generated_api_client/api/finetuned_embedding_model/__init__.py +0 -0
  32. orca_sdk/_generated_api_client/api/finetuned_embedding_model/create_finetuned_embedding_model_finetuned_embedding_model_post.py +170 -0
  33. orca_sdk/_generated_api_client/api/finetuned_embedding_model/delete_finetuned_embedding_model_finetuned_embedding_model_name_or_id_delete.py +158 -0
  34. orca_sdk/_generated_api_client/api/finetuned_embedding_model/embed_with_finetuned_model_gpu_finetuned_embedding_model_name_or_id_embedding_post.py +191 -0
  35. orca_sdk/_generated_api_client/api/finetuned_embedding_model/get_finetuned_embedding_model_finetuned_embedding_model_name_or_id_get.py +158 -0
  36. orca_sdk/_generated_api_client/api/finetuned_embedding_model/list_finetuned_embedding_models_finetuned_embedding_model_get.py +129 -0
  37. orca_sdk/_generated_api_client/api/memoryset/__init__.py +0 -0
  38. orca_sdk/_generated_api_client/api/memoryset/clone_memoryset_memoryset_name_or_id_clone_post.py +183 -0
  39. orca_sdk/_generated_api_client/api/memoryset/create_analysis_memoryset_name_or_id_analysis_post.py +185 -0
  40. orca_sdk/_generated_api_client/api/memoryset/create_memoryset_memoryset_post.py +170 -0
  41. orca_sdk/_generated_api_client/api/memoryset/delete_memories_memoryset_name_or_id_memories_delete_post.py +183 -0
  42. orca_sdk/_generated_api_client/api/memoryset/delete_memory_memoryset_name_or_id_memory_memory_id_delete.py +169 -0
  43. orca_sdk/_generated_api_client/api/memoryset/delete_memoryset_memoryset_name_or_id_delete.py +158 -0
  44. orca_sdk/_generated_api_client/api/memoryset/get_analysis_memoryset_name_or_id_analysis_analysis_task_id_get.py +171 -0
  45. orca_sdk/_generated_api_client/api/memoryset/get_memories_memoryset_name_or_id_memories_get_post.py +190 -0
  46. orca_sdk/_generated_api_client/api/memoryset/get_memory_memoryset_name_or_id_memory_memory_id_get.py +171 -0
  47. orca_sdk/_generated_api_client/api/memoryset/get_memoryset_memoryset_name_or_id_get.py +158 -0
  48. orca_sdk/_generated_api_client/api/memoryset/insert_memories_gpu_memoryset_name_or_id_memory_post.py +186 -0
  49. orca_sdk/_generated_api_client/api/memoryset/list_analyses_memoryset_name_or_id_analysis_get.py +262 -0
  50. orca_sdk/_generated_api_client/api/memoryset/list_memorysets_memoryset_get.py +129 -0
  51. orca_sdk/_generated_api_client/api/memoryset/memoryset_lookup_gpu_memoryset_name_or_id_lookup_post.py +195 -0
  52. orca_sdk/_generated_api_client/api/memoryset/query_memoryset_memoryset_name_or_id_memories_post.py +190 -0
  53. orca_sdk/_generated_api_client/api/memoryset/update_memories_gpu_memoryset_name_or_id_memories_patch.py +193 -0
  54. orca_sdk/_generated_api_client/api/memoryset/update_memory_gpu_memoryset_name_or_id_memory_patch.py +189 -0
  55. orca_sdk/_generated_api_client/api/pretrained_embedding_model/__init__.py +0 -0
  56. orca_sdk/_generated_api_client/api/pretrained_embedding_model/embed_with_pretrained_model_gpu_pretrained_embedding_model_model_name_embedding_post.py +194 -0
  57. orca_sdk/_generated_api_client/api/pretrained_embedding_model/get_pretrained_embedding_model_pretrained_embedding_model_model_name_get.py +163 -0
  58. orca_sdk/_generated_api_client/api/pretrained_embedding_model/list_pretrained_embedding_models_pretrained_embedding_model_get.py +129 -0
  59. orca_sdk/_generated_api_client/api/task/__init__.py +0 -0
  60. orca_sdk/_generated_api_client/api/task/abort_task_task_task_id_abort_delete.py +156 -0
  61. orca_sdk/_generated_api_client/api/task/get_task_status_task_task_id_status_get.py +158 -0
  62. orca_sdk/_generated_api_client/api/task/list_tasks_task_get.py +245 -0
  63. orca_sdk/_generated_api_client/api/telemetry/__init__.py +0 -0
  64. orca_sdk/_generated_api_client/api/telemetry/drop_feedback_category_with_data_telemetry_feedback_category_name_or_id_delete.py +164 -0
  65. orca_sdk/_generated_api_client/api/telemetry/get_feedback_category_telemetry_feedback_category_name_or_id_get.py +158 -0
  66. orca_sdk/_generated_api_client/api/telemetry/get_prediction_telemetry_prediction_prediction_id_get.py +159 -0
  67. orca_sdk/_generated_api_client/api/telemetry/list_feedback_categories_telemetry_feedback_category_get.py +129 -0
  68. orca_sdk/_generated_api_client/api/telemetry/list_predictions_telemetry_prediction_post.py +177 -0
  69. orca_sdk/_generated_api_client/api/telemetry/record_prediction_feedback_telemetry_prediction_feedback_put.py +173 -0
  70. orca_sdk/_generated_api_client/api/telemetry/update_prediction_telemetry_prediction_prediction_id_patch.py +183 -0
  71. orca_sdk/_generated_api_client/client.py +216 -0
  72. orca_sdk/_generated_api_client/errors.py +38 -0
  73. orca_sdk/_generated_api_client/models/__init__.py +179 -0
  74. orca_sdk/_generated_api_client/models/analyze_neighbor_labels_result.py +116 -0
  75. orca_sdk/_generated_api_client/models/api_key_metadata.py +137 -0
  76. orca_sdk/_generated_api_client/models/api_key_metadata_scope_item.py +9 -0
  77. orca_sdk/_generated_api_client/models/base_model.py +55 -0
  78. orca_sdk/_generated_api_client/models/body_create_datasource_datasource_post.py +176 -0
  79. orca_sdk/_generated_api_client/models/classification_evaluation_result.py +147 -0
  80. orca_sdk/_generated_api_client/models/clone_labeled_memoryset_request.py +150 -0
  81. orca_sdk/_generated_api_client/models/column_info.py +114 -0
  82. orca_sdk/_generated_api_client/models/column_type.py +14 -0
  83. orca_sdk/_generated_api_client/models/conflict_error_response.py +80 -0
  84. orca_sdk/_generated_api_client/models/create_api_key_request.py +120 -0
  85. orca_sdk/_generated_api_client/models/create_api_key_request_scope_item.py +9 -0
  86. orca_sdk/_generated_api_client/models/create_api_key_response.py +145 -0
  87. orca_sdk/_generated_api_client/models/create_api_key_response_scope_item.py +9 -0
  88. orca_sdk/_generated_api_client/models/create_labeled_memoryset_request.py +279 -0
  89. orca_sdk/_generated_api_client/models/create_rac_model_request.py +209 -0
  90. orca_sdk/_generated_api_client/models/datasource_metadata.py +142 -0
  91. orca_sdk/_generated_api_client/models/delete_memories_request.py +70 -0
  92. orca_sdk/_generated_api_client/models/embed_request.py +127 -0
  93. orca_sdk/_generated_api_client/models/embedding_evaluation_request.py +179 -0
  94. orca_sdk/_generated_api_client/models/embedding_evaluation_response.py +148 -0
  95. orca_sdk/_generated_api_client/models/embedding_evaluation_result.py +86 -0
  96. orca_sdk/_generated_api_client/models/embedding_finetuning_method.py +9 -0
  97. orca_sdk/_generated_api_client/models/embedding_model_result.py +114 -0
  98. orca_sdk/_generated_api_client/models/evaluation_request.py +180 -0
  99. orca_sdk/_generated_api_client/models/evaluation_response.py +140 -0
  100. orca_sdk/_generated_api_client/models/feedback_type.py +9 -0
  101. orca_sdk/_generated_api_client/models/field_validation_error.py +103 -0
  102. orca_sdk/_generated_api_client/models/filter_item.py +231 -0
  103. orca_sdk/_generated_api_client/models/filter_item_field_type_0_item.py +15 -0
  104. orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_1.py +20 -0
  105. orca_sdk/_generated_api_client/models/filter_item_op.py +16 -0
  106. orca_sdk/_generated_api_client/models/find_duplicates_analysis_result.py +70 -0
  107. orca_sdk/_generated_api_client/models/finetune_embedding_model_request.py +259 -0
  108. orca_sdk/_generated_api_client/models/finetune_embedding_model_request_training_args.py +66 -0
  109. orca_sdk/_generated_api_client/models/finetuned_embedding_model_metadata.py +166 -0
  110. orca_sdk/_generated_api_client/models/get_memories_request.py +70 -0
  111. orca_sdk/_generated_api_client/models/internal_server_error_response.py +80 -0
  112. orca_sdk/_generated_api_client/models/label_class_metrics.py +108 -0
  113. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup.py +274 -0
  114. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup_metadata.py +68 -0
  115. orca_sdk/_generated_api_client/models/label_prediction_result.py +115 -0
  116. orca_sdk/_generated_api_client/models/label_prediction_with_memories_and_feedback.py +246 -0
  117. orca_sdk/_generated_api_client/models/labeled_memory.py +197 -0
  118. orca_sdk/_generated_api_client/models/labeled_memory_insert.py +128 -0
  119. orca_sdk/_generated_api_client/models/labeled_memory_insert_metadata.py +68 -0
  120. orca_sdk/_generated_api_client/models/labeled_memory_lookup.py +258 -0
  121. orca_sdk/_generated_api_client/models/labeled_memory_lookup_metadata.py +68 -0
  122. orca_sdk/_generated_api_client/models/labeled_memory_metadata.py +68 -0
  123. orca_sdk/_generated_api_client/models/labeled_memory_metrics.py +237 -0
  124. orca_sdk/_generated_api_client/models/labeled_memory_update.py +171 -0
  125. orca_sdk/_generated_api_client/models/labeled_memory_update_metadata_type_0.py +68 -0
  126. orca_sdk/_generated_api_client/models/labeled_memoryset_metadata.py +195 -0
  127. orca_sdk/_generated_api_client/models/list_analyses_memoryset_name_or_id_analysis_get_type_type_0.py +9 -0
  128. orca_sdk/_generated_api_client/models/list_memories_request.py +104 -0
  129. orca_sdk/_generated_api_client/models/list_predictions_request.py +257 -0
  130. orca_sdk/_generated_api_client/models/lookup_request.py +81 -0
  131. orca_sdk/_generated_api_client/models/memory_metrics.py +156 -0
  132. orca_sdk/_generated_api_client/models/memoryset_analysis_request.py +83 -0
  133. orca_sdk/_generated_api_client/models/memoryset_analysis_request_type.py +9 -0
  134. orca_sdk/_generated_api_client/models/memoryset_analysis_response.py +180 -0
  135. orca_sdk/_generated_api_client/models/memoryset_analysis_response_config.py +66 -0
  136. orca_sdk/_generated_api_client/models/memoryset_analysis_response_type.py +9 -0
  137. orca_sdk/_generated_api_client/models/not_found_error_response.py +100 -0
  138. orca_sdk/_generated_api_client/models/not_found_error_response_resource_type_0.py +21 -0
  139. orca_sdk/_generated_api_client/models/precision_recall_curve.py +94 -0
  140. orca_sdk/_generated_api_client/models/prediction_feedback.py +157 -0
  141. orca_sdk/_generated_api_client/models/prediction_feedback_category.py +115 -0
  142. orca_sdk/_generated_api_client/models/prediction_feedback_request.py +122 -0
  143. orca_sdk/_generated_api_client/models/prediction_feedback_result.py +102 -0
  144. orca_sdk/_generated_api_client/models/prediction_request.py +169 -0
  145. orca_sdk/_generated_api_client/models/prediction_sort_item_item_type_0.py +10 -0
  146. orca_sdk/_generated_api_client/models/prediction_sort_item_item_type_1.py +9 -0
  147. orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py +97 -0
  148. orca_sdk/_generated_api_client/models/pretrained_embedding_model_name.py +12 -0
  149. orca_sdk/_generated_api_client/models/rac_head_type.py +11 -0
  150. orca_sdk/_generated_api_client/models/rac_model_metadata.py +191 -0
  151. orca_sdk/_generated_api_client/models/roc_curve.py +94 -0
  152. orca_sdk/_generated_api_client/models/service_unavailable_error_response.py +80 -0
  153. orca_sdk/_generated_api_client/models/task.py +198 -0
  154. orca_sdk/_generated_api_client/models/task_status.py +14 -0
  155. orca_sdk/_generated_api_client/models/task_status_info.py +133 -0
  156. orca_sdk/_generated_api_client/models/unauthenticated_error_response.py +72 -0
  157. orca_sdk/_generated_api_client/models/unauthorized_error_response.py +80 -0
  158. orca_sdk/_generated_api_client/models/unprocessable_input_error_response.py +94 -0
  159. orca_sdk/_generated_api_client/models/update_prediction_request.py +93 -0
  160. orca_sdk/_generated_api_client/py.typed +1 -0
  161. orca_sdk/_generated_api_client/types.py +56 -0
  162. orca_sdk/_utils/__init__.py +0 -0
  163. orca_sdk/_utils/analysis_ui.py +192 -0
  164. orca_sdk/_utils/analysis_ui_style.css +54 -0
  165. orca_sdk/_utils/auth.py +68 -0
  166. orca_sdk/_utils/auth_test.py +31 -0
  167. orca_sdk/_utils/common.py +37 -0
  168. orca_sdk/_utils/data_parsing.py +99 -0
  169. orca_sdk/_utils/data_parsing_test.py +244 -0
  170. orca_sdk/_utils/prediction_result_ui.css +18 -0
  171. orca_sdk/_utils/prediction_result_ui.py +64 -0
  172. orca_sdk/_utils/task.py +73 -0
  173. orca_sdk/classification_model.py +508 -0
  174. orca_sdk/classification_model_test.py +272 -0
  175. orca_sdk/conftest.py +116 -0
  176. orca_sdk/credentials.py +126 -0
  177. orca_sdk/credentials_test.py +37 -0
  178. orca_sdk/datasource.py +333 -0
  179. orca_sdk/datasource_test.py +96 -0
  180. orca_sdk/embedding_model.py +347 -0
  181. orca_sdk/embedding_model_test.py +176 -0
  182. orca_sdk/memoryset.py +1209 -0
  183. orca_sdk/memoryset_test.py +287 -0
  184. orca_sdk/telemetry.py +398 -0
  185. orca_sdk/telemetry_test.py +109 -0
  186. orca_sdk-0.0.78.dist-info/METADATA +79 -0
  187. orca_sdk-0.0.78.dist-info/RECORD +188 -0
  188. orca_sdk-0.0.78.dist-info/WHEEL +4 -0
@@ -0,0 +1,179 @@
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 .api_key_metadata_scope_item import ApiKeyMetadataScopeItem
6
+ from .base_model import BaseModel
7
+ from .body_create_datasource_datasource_post import BodyCreateDatasourceDatasourcePost
8
+ from .classification_evaluation_result import ClassificationEvaluationResult
9
+ from .clone_labeled_memoryset_request import CloneLabeledMemorysetRequest
10
+ from .column_info import ColumnInfo
11
+ from .column_type import ColumnType
12
+ from .conflict_error_response import ConflictErrorResponse
13
+ from .create_api_key_request import CreateApiKeyRequest
14
+ from .create_api_key_request_scope_item import CreateApiKeyRequestScopeItem
15
+ from .create_api_key_response import CreateApiKeyResponse
16
+ from .create_api_key_response_scope_item import CreateApiKeyResponseScopeItem
17
+ from .create_labeled_memoryset_request import CreateLabeledMemorysetRequest
18
+ from .create_rac_model_request import CreateRACModelRequest
19
+ from .datasource_metadata import DatasourceMetadata
20
+ from .delete_memories_request import DeleteMemoriesRequest
21
+ from .embed_request import EmbedRequest
22
+ from .embedding_evaluation_request import EmbeddingEvaluationRequest
23
+ from .embedding_evaluation_response import EmbeddingEvaluationResponse
24
+ from .embedding_evaluation_result import EmbeddingEvaluationResult
25
+ from .embedding_finetuning_method import EmbeddingFinetuningMethod
26
+ from .embedding_model_result import EmbeddingModelResult
27
+ from .evaluation_request import EvaluationRequest
28
+ from .evaluation_response import EvaluationResponse
29
+ from .feedback_type import FeedbackType
30
+ from .field_validation_error import FieldValidationError
31
+ from .filter_item import FilterItem
32
+ from .filter_item_field_type_0_item import FilterItemFieldType0Item
33
+ from .filter_item_field_type_2_item_type_1 import FilterItemFieldType2ItemType1
34
+ from .filter_item_op import FilterItemOp
35
+ from .find_duplicates_analysis_result import FindDuplicatesAnalysisResult
36
+ from .finetune_embedding_model_request import FinetuneEmbeddingModelRequest
37
+ from .finetune_embedding_model_request_training_args import FinetuneEmbeddingModelRequestTrainingArgs
38
+ from .finetuned_embedding_model_metadata import FinetunedEmbeddingModelMetadata
39
+ from .get_memories_request import GetMemoriesRequest
40
+ from .internal_server_error_response import InternalServerErrorResponse
41
+ from .label_class_metrics import LabelClassMetrics
42
+ from .label_prediction_memory_lookup import LabelPredictionMemoryLookup
43
+ from .label_prediction_memory_lookup_metadata import LabelPredictionMemoryLookupMetadata
44
+ from .label_prediction_result import LabelPredictionResult
45
+ from .label_prediction_with_memories_and_feedback import LabelPredictionWithMemoriesAndFeedback
46
+ from .labeled_memory import LabeledMemory
47
+ from .labeled_memory_insert import LabeledMemoryInsert
48
+ from .labeled_memory_insert_metadata import LabeledMemoryInsertMetadata
49
+ from .labeled_memory_lookup import LabeledMemoryLookup
50
+ from .labeled_memory_lookup_metadata import LabeledMemoryLookupMetadata
51
+ from .labeled_memory_metadata import LabeledMemoryMetadata
52
+ from .labeled_memory_metrics import LabeledMemoryMetrics
53
+ from .labeled_memory_update import LabeledMemoryUpdate
54
+ from .labeled_memory_update_metadata_type_0 import LabeledMemoryUpdateMetadataType0
55
+ from .labeled_memoryset_metadata import LabeledMemorysetMetadata
56
+ from .list_analyses_memoryset_name_or_id_analysis_get_type_type_0 import (
57
+ ListAnalysesMemorysetNameOrIdAnalysisGetTypeType0,
58
+ )
59
+ from .list_memories_request import ListMemoriesRequest
60
+ from .list_predictions_request import ListPredictionsRequest
61
+ from .lookup_request import LookupRequest
62
+ from .memory_metrics import MemoryMetrics
63
+ from .memoryset_analysis_request import MemorysetAnalysisRequest
64
+ from .memoryset_analysis_request_type import MemorysetAnalysisRequestType
65
+ from .memoryset_analysis_response import MemorysetAnalysisResponse
66
+ from .memoryset_analysis_response_config import MemorysetAnalysisResponseConfig
67
+ from .memoryset_analysis_response_type import MemorysetAnalysisResponseType
68
+ from .not_found_error_response import NotFoundErrorResponse
69
+ from .not_found_error_response_resource_type_0 import NotFoundErrorResponseResourceType0
70
+ from .precision_recall_curve import PrecisionRecallCurve
71
+ from .prediction_feedback import PredictionFeedback
72
+ from .prediction_feedback_category import PredictionFeedbackCategory
73
+ from .prediction_feedback_request import PredictionFeedbackRequest
74
+ from .prediction_feedback_result import PredictionFeedbackResult
75
+ from .prediction_request import PredictionRequest
76
+ from .prediction_sort_item_item_type_0 import PredictionSortItemItemType0
77
+ from .prediction_sort_item_item_type_1 import PredictionSortItemItemType1
78
+ from .pretrained_embedding_model_metadata import PretrainedEmbeddingModelMetadata
79
+ from .pretrained_embedding_model_name import PretrainedEmbeddingModelName
80
+ from .rac_head_type import RACHeadType
81
+ from .rac_model_metadata import RACModelMetadata
82
+ from .roc_curve import ROCCurve
83
+ from .service_unavailable_error_response import ServiceUnavailableErrorResponse
84
+ from .task import Task
85
+ from .task_status import TaskStatus
86
+ from .task_status_info import TaskStatusInfo
87
+ from .unauthenticated_error_response import UnauthenticatedErrorResponse
88
+ from .unauthorized_error_response import UnauthorizedErrorResponse
89
+ from .unprocessable_input_error_response import UnprocessableInputErrorResponse
90
+ from .update_prediction_request import UpdatePredictionRequest
91
+
92
+ __all__ = (
93
+ "AnalyzeNeighborLabelsResult",
94
+ "ApiKeyMetadata",
95
+ "ApiKeyMetadataScopeItem",
96
+ "BaseModel",
97
+ "BodyCreateDatasourceDatasourcePost",
98
+ "ClassificationEvaluationResult",
99
+ "CloneLabeledMemorysetRequest",
100
+ "ColumnInfo",
101
+ "ColumnType",
102
+ "ConflictErrorResponse",
103
+ "CreateApiKeyRequest",
104
+ "CreateApiKeyRequestScopeItem",
105
+ "CreateApiKeyResponse",
106
+ "CreateApiKeyResponseScopeItem",
107
+ "CreateLabeledMemorysetRequest",
108
+ "CreateRACModelRequest",
109
+ "DatasourceMetadata",
110
+ "DeleteMemoriesRequest",
111
+ "EmbeddingEvaluationRequest",
112
+ "EmbeddingEvaluationResponse",
113
+ "EmbeddingEvaluationResult",
114
+ "EmbeddingFinetuningMethod",
115
+ "EmbeddingModelResult",
116
+ "EmbedRequest",
117
+ "EvaluationRequest",
118
+ "EvaluationResponse",
119
+ "FeedbackType",
120
+ "FieldValidationError",
121
+ "FilterItem",
122
+ "FilterItemFieldType0Item",
123
+ "FilterItemFieldType2ItemType1",
124
+ "FilterItemOp",
125
+ "FindDuplicatesAnalysisResult",
126
+ "FinetunedEmbeddingModelMetadata",
127
+ "FinetuneEmbeddingModelRequest",
128
+ "FinetuneEmbeddingModelRequestTrainingArgs",
129
+ "GetMemoriesRequest",
130
+ "InternalServerErrorResponse",
131
+ "LabelClassMetrics",
132
+ "LabeledMemory",
133
+ "LabeledMemoryInsert",
134
+ "LabeledMemoryInsertMetadata",
135
+ "LabeledMemoryLookup",
136
+ "LabeledMemoryLookupMetadata",
137
+ "LabeledMemoryMetadata",
138
+ "LabeledMemoryMetrics",
139
+ "LabeledMemorysetMetadata",
140
+ "LabeledMemoryUpdate",
141
+ "LabeledMemoryUpdateMetadataType0",
142
+ "LabelPredictionMemoryLookup",
143
+ "LabelPredictionMemoryLookupMetadata",
144
+ "LabelPredictionResult",
145
+ "LabelPredictionWithMemoriesAndFeedback",
146
+ "ListAnalysesMemorysetNameOrIdAnalysisGetTypeType0",
147
+ "ListMemoriesRequest",
148
+ "ListPredictionsRequest",
149
+ "LookupRequest",
150
+ "MemoryMetrics",
151
+ "MemorysetAnalysisRequest",
152
+ "MemorysetAnalysisRequestType",
153
+ "MemorysetAnalysisResponse",
154
+ "MemorysetAnalysisResponseConfig",
155
+ "MemorysetAnalysisResponseType",
156
+ "NotFoundErrorResponse",
157
+ "NotFoundErrorResponseResourceType0",
158
+ "PrecisionRecallCurve",
159
+ "PredictionFeedback",
160
+ "PredictionFeedbackCategory",
161
+ "PredictionFeedbackRequest",
162
+ "PredictionFeedbackResult",
163
+ "PredictionRequest",
164
+ "PredictionSortItemItemType0",
165
+ "PredictionSortItemItemType1",
166
+ "PretrainedEmbeddingModelMetadata",
167
+ "PretrainedEmbeddingModelName",
168
+ "RACHeadType",
169
+ "RACModelMetadata",
170
+ "ROCCurve",
171
+ "ServiceUnavailableErrorResponse",
172
+ "Task",
173
+ "TaskStatus",
174
+ "TaskStatusInfo",
175
+ "UnauthenticatedErrorResponse",
176
+ "UnauthorizedErrorResponse",
177
+ "UnprocessableInputErrorResponse",
178
+ "UpdatePredictionRequest",
179
+ )
@@ -0,0 +1,116 @@
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
+ neighbor_prediction_accuracy (float):
31
+ mean_neighbor_label_confidence (float):
32
+ mean_neighbor_label_entropy (float):
33
+ mean_neighbor_predicted_label_ambiguity (float):
34
+ """
35
+
36
+ label_metrics: List["LabelClassMetrics"]
37
+ neighbor_prediction_accuracy: float
38
+ mean_neighbor_label_confidence: float
39
+ mean_neighbor_label_entropy: float
40
+ mean_neighbor_predicted_label_ambiguity: float
41
+ additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
42
+
43
+ def to_dict(self) -> dict[str, Any]:
44
+ label_metrics = []
45
+ for label_metrics_item_data in self.label_metrics:
46
+ label_metrics_item = label_metrics_item_data.to_dict()
47
+ label_metrics.append(label_metrics_item)
48
+
49
+ neighbor_prediction_accuracy = self.neighbor_prediction_accuracy
50
+
51
+ mean_neighbor_label_confidence = self.mean_neighbor_label_confidence
52
+
53
+ mean_neighbor_label_entropy = self.mean_neighbor_label_entropy
54
+
55
+ mean_neighbor_predicted_label_ambiguity = self.mean_neighbor_predicted_label_ambiguity
56
+
57
+ field_dict: dict[str, Any] = {}
58
+ field_dict.update(self.additional_properties)
59
+ field_dict.update(
60
+ {
61
+ "label_metrics": label_metrics,
62
+ "neighbor_prediction_accuracy": neighbor_prediction_accuracy,
63
+ "mean_neighbor_label_confidence": mean_neighbor_label_confidence,
64
+ "mean_neighbor_label_entropy": mean_neighbor_label_entropy,
65
+ "mean_neighbor_predicted_label_ambiguity": mean_neighbor_predicted_label_ambiguity,
66
+ }
67
+ )
68
+
69
+ return field_dict
70
+
71
+ @classmethod
72
+ def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
73
+ from ..models.label_class_metrics import LabelClassMetrics
74
+
75
+ d = src_dict.copy()
76
+ label_metrics = []
77
+ _label_metrics = d.pop("label_metrics")
78
+ for label_metrics_item_data in _label_metrics:
79
+ label_metrics_item = LabelClassMetrics.from_dict(label_metrics_item_data)
80
+
81
+ label_metrics.append(label_metrics_item)
82
+
83
+ neighbor_prediction_accuracy = d.pop("neighbor_prediction_accuracy")
84
+
85
+ mean_neighbor_label_confidence = d.pop("mean_neighbor_label_confidence")
86
+
87
+ mean_neighbor_label_entropy = d.pop("mean_neighbor_label_entropy")
88
+
89
+ mean_neighbor_predicted_label_ambiguity = d.pop("mean_neighbor_predicted_label_ambiguity")
90
+
91
+ analyze_neighbor_labels_result = cls(
92
+ label_metrics=label_metrics,
93
+ neighbor_prediction_accuracy=neighbor_prediction_accuracy,
94
+ mean_neighbor_label_confidence=mean_neighbor_label_confidence,
95
+ mean_neighbor_label_entropy=mean_neighbor_label_entropy,
96
+ mean_neighbor_predicted_label_ambiguity=mean_neighbor_predicted_label_ambiguity,
97
+ )
98
+
99
+ analyze_neighbor_labels_result.additional_properties = d
100
+ return analyze_neighbor_labels_result
101
+
102
+ @property
103
+ def additional_keys(self) -> list[str]:
104
+ return list(self.additional_properties.keys())
105
+
106
+ def __getitem__(self, key: str) -> Any:
107
+ return self.additional_properties[key]
108
+
109
+ def __setitem__(self, key: str, value: Any) -> None:
110
+ self.additional_properties[key] = value
111
+
112
+ def __delitem__(self, key: str) -> None:
113
+ del self.additional_properties[key]
114
+
115
+ def __contains__(self, key: str) -> bool:
116
+ return key in self.additional_properties
@@ -0,0 +1,137 @@
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 enum import Enum
15
+ from typing import Any, List, Type, TypeVar, Union, cast
16
+
17
+ from attrs import define as _attrs_define
18
+ from attrs import field as _attrs_field
19
+ from dateutil.parser import isoparse
20
+
21
+ from ..models.api_key_metadata_scope_item import ApiKeyMetadataScopeItem
22
+
23
+ T = TypeVar("T", bound="ApiKeyMetadata")
24
+
25
+
26
+ @_attrs_define
27
+ class ApiKeyMetadata:
28
+ """
29
+ Attributes:
30
+ id (str):
31
+ org_id (str):
32
+ name (str):
33
+ created_by (Union[None, str]):
34
+ created_at (datetime.datetime):
35
+ updated_at (datetime.datetime):
36
+ scope (List[ApiKeyMetadataScopeItem]):
37
+ """
38
+
39
+ id: str
40
+ org_id: str
41
+ name: str
42
+ created_by: Union[None, str]
43
+ created_at: datetime.datetime
44
+ updated_at: datetime.datetime
45
+ scope: List[ApiKeyMetadataScopeItem]
46
+ additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
47
+
48
+ def to_dict(self) -> dict[str, Any]:
49
+ id = self.id
50
+
51
+ org_id = self.org_id
52
+
53
+ name = self.name
54
+
55
+ created_by: Union[None, str]
56
+ created_by = self.created_by
57
+
58
+ created_at = self.created_at.isoformat()
59
+
60
+ updated_at = self.updated_at.isoformat()
61
+
62
+ scope = []
63
+ for scope_item_data in self.scope:
64
+ scope_item = scope_item_data.value if isinstance(scope_item_data, Enum) else scope_item_data
65
+ scope.append(scope_item)
66
+
67
+ field_dict: dict[str, Any] = {}
68
+ field_dict.update(self.additional_properties)
69
+ field_dict.update(
70
+ {
71
+ "id": id,
72
+ "org_id": org_id,
73
+ "name": name,
74
+ "created_by": created_by,
75
+ "created_at": created_at,
76
+ "updated_at": updated_at,
77
+ "scope": scope,
78
+ }
79
+ )
80
+
81
+ return field_dict
82
+
83
+ @classmethod
84
+ def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
85
+ d = src_dict.copy()
86
+ id = d.pop("id")
87
+
88
+ org_id = d.pop("org_id")
89
+
90
+ name = d.pop("name")
91
+
92
+ def _parse_created_by(data: object) -> Union[None, str]:
93
+ if data is None:
94
+ return data
95
+ return cast(Union[None, str], data)
96
+
97
+ created_by = _parse_created_by(d.pop("created_by"))
98
+
99
+ created_at = isoparse(d.pop("created_at"))
100
+
101
+ updated_at = isoparse(d.pop("updated_at"))
102
+
103
+ scope = []
104
+ _scope = d.pop("scope")
105
+ for scope_item_data in _scope:
106
+ scope_item = ApiKeyMetadataScopeItem(scope_item_data)
107
+
108
+ scope.append(scope_item)
109
+
110
+ api_key_metadata = cls(
111
+ id=id,
112
+ org_id=org_id,
113
+ name=name,
114
+ created_by=created_by,
115
+ created_at=created_at,
116
+ updated_at=updated_at,
117
+ scope=scope,
118
+ )
119
+
120
+ api_key_metadata.additional_properties = d
121
+ return api_key_metadata
122
+
123
+ @property
124
+ def additional_keys(self) -> list[str]:
125
+ return list(self.additional_properties.keys())
126
+
127
+ def __getitem__(self, key: str) -> Any:
128
+ return self.additional_properties[key]
129
+
130
+ def __setitem__(self, key: str, value: Any) -> None:
131
+ self.additional_properties[key] = value
132
+
133
+ def __delitem__(self, key: str) -> None:
134
+ del self.additional_properties[key]
135
+
136
+ def __contains__(self, key: str) -> bool:
137
+ return key in self.additional_properties
@@ -0,0 +1,9 @@
1
+ from enum import Enum
2
+
3
+
4
+ class ApiKeyMetadataScopeItem(str, Enum):
5
+ ADMINISTER = "ADMINISTER"
6
+ PREDICT = "PREDICT"
7
+
8
+ def __str__(self) -> str:
9
+ return str(self.value)
@@ -0,0 +1,55 @@
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
@@ -0,0 +1,176 @@
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