rapidata 2.24.0__py3-none-any.whl → 3.0.1__py3-none-any.whl

rapidata/api_client/api_client.py

@@ -1,9 +1,9 @@
 # coding: utf-8
 
 """
-    Rapidata.Dataset
+    Rapidata Asset API
 
-    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+    The API for the Rapidata Asset service
 
     The version of the OpenAPI document: v1
     Generated by OpenAPI Generator (https://openapi-generator.tech)

rapidata/api_client/configuration.py

@@ -1,9 +1,9 @@
 # coding: utf-8
 
 """
-    Rapidata.Dataset
+    Rapidata Asset API
 
-    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+    The API for the Rapidata Asset service
 
     The version of the OpenAPI document: v1
     Generated by OpenAPI Generator (https://openapi-generator.tech)
@@ -60,25 +60,6 @@ class Configuration:
     :param retries: Number of retries for API requests.
 
     :Example:
-
-    API Key Authentication Example.
-    Given the following security scheme in the OpenAPI specification:
-      components:
-        securitySchemes:
-          cookieAuth:         # name for the security scheme
-            type: apiKey
-            in: cookie
-            name: JSESSIONID  # cookie name
-
-    You can programmatically set the cookie:
-
-conf = rapidata.api_client.Configuration(
-    api_key={'cookieAuth': 'abc123'}
-    api_key_prefix={'cookieAuth': 'JSESSIONID'}
-)
-
-    The following cookie will be added to the HTTP request:
-       Cookie: JSESSIONID abc123
     """
 
     _default = None
@@ -97,7 +78,7 @@ conf = rapidata.api_client.Configuration(
                  ) -> None:
         """Constructor
         """
-        self._base_path = "http://localhost" if host is None else host
+        self._base_path = "https://api.rabbitdata.ch" if host is None else host
         """Default Base url
         """
         self.server_index = 0 if server_index is None and host is None else server_index
@@ -393,14 +374,20 @@ conf = rapidata.api_client.Configuration(
         :return: The Auth Settings information dict.
         """
         auth = {}
-        if 'bearer' in self.api_key:
-            auth['bearer'] = {
-                'type': 'api_key',
+        if self.access_token is not None:
+            auth['Bearer'] = {
+                'type': 'bearer',
+                'in': 'header',
+                'format': 'Json Web Token',
+                'key': 'Authorization',
+                'value': 'Bearer ' + self.access_token
+            }
+        if self.access_token is not None:
+            auth['OAuth2'] = {
+                'type': 'oauth2',
                 'in': 'header',
                 'key': 'Authorization',
-                'value': self.get_api_key_with_prefix(
-                    'bearer',
-                ),
+                'value': 'Bearer ' + self.access_token
             }
         return auth
 
@@ -423,7 +410,7 @@ conf = rapidata.api_client.Configuration(
         """
         return [
             {
-                'url': "",
+                'url': "https://api.rabbitdata.ch",
                 'description': "No description provided",
             }
         ]

rapidata/api_client/exceptions.py

@@ -1,9 +1,9 @@
 # coding: utf-8
 
 """
-    Rapidata.Dataset
+    Rapidata Asset API
 
-    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+    The API for the Rapidata Asset service
 
     The version of the OpenAPI document: v1
     Generated by OpenAPI Generator (https://openapi-generator.tech)

rapidata/api_client/models/__init__.py

@@ -2,9 +2,9 @@
 
 # flake8: noqa
 """
-    Rapidata.Dataset
+    Rapidata Asset API
 
-    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+    The API for the Rapidata Asset service
 
     The version of the OpenAPI document: v1
     Generated by OpenAPI Generator (https://openapi-generator.tech)
@@ -13,301 +13,66 @@
 """  # noqa: E501
 
 
-# import models into model package
-from rapidata.api_client.models.ab_test_selection import AbTestSelection
-from rapidata.api_client.models.ab_test_selection_a_inner import AbTestSelectionAInner
-from rapidata.api_client.models.add_campaign_artifact_result import AddCampaignArtifactResult
-from rapidata.api_client.models.add_campaign_model import AddCampaignModel
-from rapidata.api_client.models.add_user_response_result import AddUserResponseResult
-from rapidata.api_client.models.add_validation_rapid_model import AddValidationRapidModel
-from rapidata.api_client.models.add_validation_rapid_model_payload import AddValidationRapidModelPayload
-from rapidata.api_client.models.add_validation_rapid_model_truth import AddValidationRapidModelTruth
-from rapidata.api_client.models.add_validation_rapid_result import AddValidationRapidResult
-from rapidata.api_client.models.add_validation_text_rapid_model import AddValidationTextRapidModel
-from rapidata.api_client.models.age_group import AgeGroup
-from rapidata.api_client.models.age_user_filter_model import AgeUserFilterModel
-from rapidata.api_client.models.aggregator_type import AggregatorType
-from rapidata.api_client.models.are_rapids_active_result import AreRapidsActiveResult
-from rapidata.api_client.models.asset_metadata import AssetMetadata
-from rapidata.api_client.models.asset_metadata_asset import AssetMetadataAsset
-from rapidata.api_client.models.asset_metadata_model import AssetMetadataModel
-from rapidata.api_client.models.asset_metadata_model_asset import AssetMetadataModelAsset
-from rapidata.api_client.models.attach_category_rapid_blueprint import AttachCategoryRapidBlueprint
-from rapidata.api_client.models.attach_category_result import AttachCategoryResult
-from rapidata.api_client.models.attach_category_truth import AttachCategoryTruth
-from rapidata.api_client.models.base_error import BaseError
-from rapidata.api_client.models.bounding_box_payload import BoundingBoxPayload
-from rapidata.api_client.models.bounding_box_rapid_blueprint import BoundingBoxRapidBlueprint
-from rapidata.api_client.models.bounding_box_result import BoundingBoxResult
-from rapidata.api_client.models.bounding_box_truth import BoundingBoxTruth
-from rapidata.api_client.models.box_shape import BoxShape
-from rapidata.api_client.models.campaign_artifact_model import CampaignArtifactModel
-from rapidata.api_client.models.campaign_query_result import CampaignQueryResult
-from rapidata.api_client.models.campaign_query_result_paged_result import CampaignQueryResultPagedResult
-from rapidata.api_client.models.campaign_status import CampaignStatus
-from rapidata.api_client.models.campaign_user_filter_model import CampaignUserFilterModel
-from rapidata.api_client.models.capped_selection import CappedSelection
-from rapidata.api_client.models.classification_metadata import ClassificationMetadata
-from rapidata.api_client.models.classification_metadata_filter_config import ClassificationMetadataFilterConfig
-from rapidata.api_client.models.classification_metadata_model import ClassificationMetadataModel
-from rapidata.api_client.models.classify_payload import ClassifyPayload
-from rapidata.api_client.models.clients_query_result import ClientsQueryResult
-from rapidata.api_client.models.clients_query_result_paged_result import ClientsQueryResultPagedResult
-from rapidata.api_client.models.clone_dataset_model import CloneDatasetModel
-from rapidata.api_client.models.clone_order_model import CloneOrderModel
-from rapidata.api_client.models.clone_order_result import CloneOrderResult
-from rapidata.api_client.models.compare_payload import ComparePayload
-from rapidata.api_client.models.compare_rapid_blueprint import CompareRapidBlueprint
-from rapidata.api_client.models.compare_result import CompareResult
-from rapidata.api_client.models.compare_truth import CompareTruth
-from rapidata.api_client.models.compare_workflow_config import CompareWorkflowConfig
-from rapidata.api_client.models.compare_workflow_config_metadata_value import CompareWorkflowConfigMetadataValue
-from rapidata.api_client.models.compare_workflow_config_model import CompareWorkflowConfigModel
-from rapidata.api_client.models.compare_workflow_config_model_pair_maker_config import CompareWorkflowConfigModelPairMakerConfig
-from rapidata.api_client.models.compare_workflow_config_pair_maker_config import CompareWorkflowConfigPairMakerConfig
-from rapidata.api_client.models.compare_workflow_model import CompareWorkflowModel
-from rapidata.api_client.models.compare_workflow_model1 import CompareWorkflowModel1
-from rapidata.api_client.models.compare_workflow_model1_pair_maker_information import CompareWorkflowModel1PairMakerInformation
-from rapidata.api_client.models.compare_workflow_model1_referee import CompareWorkflowModel1Referee
-from rapidata.api_client.models.compare_workflow_model_pair_maker_config import CompareWorkflowModelPairMakerConfig
-from rapidata.api_client.models.comparison_operator import ComparisonOperator
-from rapidata.api_client.models.conditional_validation_selection import ConditionalValidationSelection
-from rapidata.api_client.models.coordinate import Coordinate
-from rapidata.api_client.models.count_classification_metadata_filter_config import CountClassificationMetadataFilterConfig
-from rapidata.api_client.models.count_metadata import CountMetadata
-from rapidata.api_client.models.count_metadata_model import CountMetadataModel
-from rapidata.api_client.models.country_user_filter_model import CountryUserFilterModel
-from rapidata.api_client.models.create_bridge_token_result import CreateBridgeTokenResult
-from rapidata.api_client.models.create_client_model import CreateClientModel
-from rapidata.api_client.models.create_client_result import CreateClientResult
-from rapidata.api_client.models.create_complex_order_model import CreateComplexOrderModel
-from rapidata.api_client.models.create_complex_order_model_pipeline import CreateComplexOrderModelPipeline
-from rapidata.api_client.models.create_complex_order_result import CreateComplexOrderResult
-from rapidata.api_client.models.create_datapoint_from_files_model import CreateDatapointFromFilesModel
-from rapidata.api_client.models.create_datapoint_from_files_model_metadata_inner import CreateDatapointFromFilesModelMetadataInner
-from rapidata.api_client.models.create_datapoint_from_text_sources_model import CreateDatapointFromTextSourcesModel
-from rapidata.api_client.models.create_datapoint_from_urls_model import CreateDatapointFromUrlsModel
-from rapidata.api_client.models.create_datapoint_result import CreateDatapointResult
-from rapidata.api_client.models.create_datapoints_from_s3_bucket_model import CreateDatapointsFromS3BucketModel
-from rapidata.api_client.models.create_dataset_artifact_model import CreateDatasetArtifactModel
-from rapidata.api_client.models.create_dataset_artifact_model_dataset import CreateDatasetArtifactModelDataset
-from rapidata.api_client.models.create_demographic_rapid_model import CreateDemographicRapidModel
-from rapidata.api_client.models.create_empty_validation_set_result import CreateEmptyValidationSetResult
-from rapidata.api_client.models.create_order_model import CreateOrderModel
-from rapidata.api_client.models.create_order_model_referee import CreateOrderModelReferee
-from rapidata.api_client.models.create_order_model_user_filters_inner import CreateOrderModelUserFiltersInner
-from rapidata.api_client.models.create_order_model_workflow import CreateOrderModelWorkflow
-from rapidata.api_client.models.create_order_result import CreateOrderResult
-from rapidata.api_client.models.create_rapid_result import CreateRapidResult
-from rapidata.api_client.models.create_simple_pipeline_model import CreateSimplePipelineModel
-from rapidata.api_client.models.create_simple_pipeline_model_artifacts_inner import CreateSimplePipelineModelArtifactsInner
-from rapidata.api_client.models.create_simple_pipeline_model_pipeline_steps_inner import CreateSimplePipelineModelPipelineStepsInner
-from rapidata.api_client.models.create_unsupported_order_model import CreateUnsupportedOrderModel
-from rapidata.api_client.models.create_validation_set_model import CreateValidationSetModel
-from rapidata.api_client.models.custom_user_filter_model import CustomUserFilterModel
-from rapidata.api_client.models.datapoint import Datapoint
-from rapidata.api_client.models.datapoint_metadata_model import DatapointMetadataModel
-from rapidata.api_client.models.datapoint_model import DatapointModel
-from rapidata.api_client.models.datapoint_state import DatapointState
-from rapidata.api_client.models.dataset_artifact_model import DatasetArtifactModel
-from rapidata.api_client.models.dataset_evaluation_step_model import DatasetEvaluationStepModel
-from rapidata.api_client.models.demographic import Demographic
-from rapidata.api_client.models.demographic_metadata_model import DemographicMetadataModel
-from rapidata.api_client.models.demographic_selection import DemographicSelection
-from rapidata.api_client.models.early_stopping_referee_model import EarlyStoppingRefereeModel
-from rapidata.api_client.models.elo_config import EloConfig
-from rapidata.api_client.models.elo_config_model import EloConfigModel
-from rapidata.api_client.models.empty_validation_truth import EmptyValidationTruth
-from rapidata.api_client.models.evaluation_workflow_config import EvaluationWorkflowConfig
-from rapidata.api_client.models.evaluation_workflow_model import EvaluationWorkflowModel
-from rapidata.api_client.models.evaluation_workflow_model1 import EvaluationWorkflowModel1
-from rapidata.api_client.models.feature_flag import FeatureFlag
-from rapidata.api_client.models.feature_flag_model import FeatureFlagModel
-from rapidata.api_client.models.feedback_model import FeedbackModel
-from rapidata.api_client.models.file_artifact_model import FileArtifactModel
-from rapidata.api_client.models.file_asset import FileAsset
-from rapidata.api_client.models.file_asset_model import FileAssetModel
-from rapidata.api_client.models.file_asset_model_metadata_value import FileAssetModelMetadataValue
-from rapidata.api_client.models.filter import Filter
-from rapidata.api_client.models.filter_operator import FilterOperator
-from rapidata.api_client.models.free_text_payload import FreeTextPayload
-from rapidata.api_client.models.free_text_rapid_blueprint import FreeTextRapidBlueprint
-from rapidata.api_client.models.free_text_result import FreeTextResult
-from rapidata.api_client.models.gender import Gender
-from rapidata.api_client.models.gender_user_filter_model import GenderUserFilterModel
-from rapidata.api_client.models.get_available_validation_sets_result import GetAvailableValidationSetsResult
-from rapidata.api_client.models.get_compare_ab_summary_result import GetCompareAbSummaryResult
-from rapidata.api_client.models.get_compare_workflow_results_model import GetCompareWorkflowResultsModel
-from rapidata.api_client.models.get_compare_workflow_results_result import GetCompareWorkflowResultsResult
-from rapidata.api_client.models.get_compare_workflow_results_result_paged_result import GetCompareWorkflowResultsResultPagedResult
-from rapidata.api_client.models.get_datapoint_by_id_result import GetDatapointByIdResult
-from rapidata.api_client.models.get_datapoints_by_dataset_id_result import GetDatapointsByDatasetIdResult
-from rapidata.api_client.models.get_dataset_by_id_result import GetDatasetByIdResult
-from rapidata.api_client.models.get_dataset_progress_result import GetDatasetProgressResult
-from rapidata.api_client.models.get_failed_datapoints_result import GetFailedDatapointsResult
-from rapidata.api_client.models.get_order_by_id_result import GetOrderByIdResult
-from rapidata.api_client.models.get_pipeline_by_id_result import GetPipelineByIdResult
-from rapidata.api_client.models.get_pipeline_by_id_result_artifacts_value import GetPipelineByIdResultArtifactsValue
-from rapidata.api_client.models.get_public_orders_result import GetPublicOrdersResult
-from rapidata.api_client.models.get_rapid_responses_result import GetRapidResponsesResult
-from rapidata.api_client.models.get_responses_result import GetResponsesResult
-from rapidata.api_client.models.get_simple_workflow_results_model import GetSimpleWorkflowResultsModel
-from rapidata.api_client.models.get_validation_rapids_result import GetValidationRapidsResult
-from rapidata.api_client.models.get_validation_rapids_result_asset import GetValidationRapidsResultAsset
-from rapidata.api_client.models.get_validation_rapids_result_paged_result import GetValidationRapidsResultPagedResult
-from rapidata.api_client.models.get_validation_rapids_result_payload import GetValidationRapidsResultPayload
-from rapidata.api_client.models.get_validation_rapids_result_truth import GetValidationRapidsResultTruth
-from rapidata.api_client.models.get_validation_set_by_id_result import GetValidationSetByIdResult
-from rapidata.api_client.models.get_workflow_by_id_result import GetWorkflowByIdResult
-from rapidata.api_client.models.get_workflow_by_id_result_workflow import GetWorkflowByIdResultWorkflow
-from rapidata.api_client.models.get_workflow_progress_result import GetWorkflowProgressResult
-from rapidata.api_client.models.get_workflow_results_result import GetWorkflowResultsResult
-from rapidata.api_client.models.get_workflow_results_result_paged_result import GetWorkflowResultsResultPagedResult
-from rapidata.api_client.models.google_one_tap_login_model import GoogleOneTapLoginModel
-from rapidata.api_client.models.i_workflow_model_paged_result import IWorkflowModelPagedResult
-from rapidata.api_client.models.image_dimension_metadata import ImageDimensionMetadata
-from rapidata.api_client.models.image_dimension_metadata_model import ImageDimensionMetadataModel
-from rapidata.api_client.models.import_from_file_result import ImportFromFileResult
-from rapidata.api_client.models.import_validation_set_from_file_result import ImportValidationSetFromFileResult
-from rapidata.api_client.models.inspect_report_result import InspectReportResult
-from rapidata.api_client.models.labeling_selection import LabelingSelection
-from rapidata.api_client.models.language_user_filter_model import LanguageUserFilterModel
-from rapidata.api_client.models.line import Line
-from rapidata.api_client.models.line_payload import LinePayload
-from rapidata.api_client.models.line_point import LinePoint
-from rapidata.api_client.models.line_rapid_blueprint import LineRapidBlueprint
-from rapidata.api_client.models.line_result import LineResult
-from rapidata.api_client.models.line_truth import LineTruth
-from rapidata.api_client.models.locate_box_truth import LocateBoxTruth
-from rapidata.api_client.models.locate_coordinate import LocateCoordinate
-from rapidata.api_client.models.locate_payload import LocatePayload
-from rapidata.api_client.models.locate_rapid_blueprint import LocateRapidBlueprint
-from rapidata.api_client.models.locate_result import LocateResult
-from rapidata.api_client.models.location_metadata import LocationMetadata
-from rapidata.api_client.models.location_metadata_exists_filter_config import LocationMetadataExistsFilterConfig
-from rapidata.api_client.models.location_metadata_model import LocationMetadataModel
-from rapidata.api_client.models.logic_operator import LogicOperator
-from rapidata.api_client.models.metadata_visibilities import MetadataVisibilities
-from rapidata.api_client.models.multi_asset import MultiAsset
-from rapidata.api_client.models.multi_asset_model import MultiAssetModel
-from rapidata.api_client.models.naive_referee_config import NaiveRefereeConfig
-from rapidata.api_client.models.naive_referee_model import NaiveRefereeModel
-from rapidata.api_client.models.named_classification import NamedClassification
-from rapidata.api_client.models.named_entity_payload import NamedEntityPayload
-from rapidata.api_client.models.named_entity_rapid_blueprint import NamedEntityRapidBlueprint
-from rapidata.api_client.models.named_entity_result import NamedEntityResult
-from rapidata.api_client.models.named_entity_truth import NamedEntityTruth
-from rapidata.api_client.models.never_ending_referee_config import NeverEndingRefereeConfig
-from rapidata.api_client.models.new_user_filter_model import NewUserFilterModel
-from rapidata.api_client.models.newsletter_model import NewsletterModel
-from rapidata.api_client.models.not_available_yet_result import NotAvailableYetResult
-from rapidata.api_client.models.not_user_filter_model import NotUserFilterModel
-from rapidata.api_client.models.null_asset import NullAsset
-from rapidata.api_client.models.null_asset_model import NullAssetModel
-from rapidata.api_client.models.online_pair_maker_config import OnlinePairMakerConfig
-from rapidata.api_client.models.online_pair_maker_config_model import OnlinePairMakerConfigModel
-from rapidata.api_client.models.online_pair_maker_information import OnlinePairMakerInformation
-from rapidata.api_client.models.or_user_filter_model import OrUserFilterModel
-from rapidata.api_client.models.order_model import OrderModel
-from rapidata.api_client.models.order_model_paged_result import OrderModelPagedResult
-from rapidata.api_client.models.order_state import OrderState
-from rapidata.api_client.models.original_filename_metadata import OriginalFilenameMetadata
-from rapidata.api_client.models.original_filename_metadata_model import OriginalFilenameMetadataModel
-from rapidata.api_client.models.page_info import PageInfo
-from rapidata.api_client.models.pipeline_id_workflow_artifact_id_put_request import PipelineIdWorkflowArtifactIdPutRequest
-from rapidata.api_client.models.polygon_payload import PolygonPayload
-from rapidata.api_client.models.polygon_rapid_blueprint import PolygonRapidBlueprint
-from rapidata.api_client.models.polygon_result import PolygonResult
-from rapidata.api_client.models.polygon_truth import PolygonTruth
-from rapidata.api_client.models.preliminary_download_model import PreliminaryDownloadModel
-from rapidata.api_client.models.preliminary_download_result import PreliminaryDownloadResult
-from rapidata.api_client.models.private_text_metadata_input import PrivateTextMetadataInput
-from rapidata.api_client.models.probabilistic_attach_category_referee_config import ProbabilisticAttachCategoryRefereeConfig
-from rapidata.api_client.models.prompt_asset_metadata_input import PromptAssetMetadataInput
-from rapidata.api_client.models.prompt_metadata import PromptMetadata
-from rapidata.api_client.models.prompt_metadata_input import PromptMetadataInput
-from rapidata.api_client.models.prompt_metadata_model import PromptMetadataModel
-from rapidata.api_client.models.public_order_model import PublicOrderModel
-from rapidata.api_client.models.public_text_metadata_input import PublicTextMetadataInput
-from rapidata.api_client.models.query_model import QueryModel
-from rapidata.api_client.models.query_workflows_model import QueryWorkflowsModel
-from rapidata.api_client.models.rapid_issue import RapidIssue
-from rapidata.api_client.models.rapid_response import RapidResponse
-from rapidata.api_client.models.rapid_response_result import RapidResponseResult
-from rapidata.api_client.models.rapid_result_model import RapidResultModel
-from rapidata.api_client.models.rapid_result_model_result import RapidResultModelResult
-from rapidata.api_client.models.rapid_skipped_model import RapidSkippedModel
-from rapidata.api_client.models.rapid_state import RapidState
-from rapidata.api_client.models.read_bridge_token_keys_result import ReadBridgeTokenKeysResult
-from rapidata.api_client.models.register_temporary_customer_model import RegisterTemporaryCustomerModel
-from rapidata.api_client.models.register_temporary_customer_result import RegisterTemporaryCustomerResult
-from rapidata.api_client.models.report_model import ReportModel
-from rapidata.api_client.models.response_count_user_filter_model import ResponseCountUserFilterModel
-from rapidata.api_client.models.retrieval_mode import RetrievalMode
-from rapidata.api_client.models.root_filter import RootFilter
-from rapidata.api_client.models.scrub_payload import ScrubPayload
-from rapidata.api_client.models.scrub_range import ScrubRange
-from rapidata.api_client.models.scrub_rapid_blueprint import ScrubRapidBlueprint
-from rapidata.api_client.models.scrub_result import ScrubResult
-from rapidata.api_client.models.scrub_truth import ScrubTruth
-from rapidata.api_client.models.send_completion_mail_step_model import SendCompletionMailStepModel
-from rapidata.api_client.models.shape import Shape
-from rapidata.api_client.models.shuffling_selection import ShufflingSelection
-from rapidata.api_client.models.simple_workflow_config import SimpleWorkflowConfig
-from rapidata.api_client.models.simple_workflow_config_model import SimpleWorkflowConfigModel
-from rapidata.api_client.models.simple_workflow_config_model_blueprint import SimpleWorkflowConfigModelBlueprint
-from rapidata.api_client.models.simple_workflow_model import SimpleWorkflowModel
-from rapidata.api_client.models.simple_workflow_model1 import SimpleWorkflowModel1
-from rapidata.api_client.models.simple_workflow_model_blueprint import SimpleWorkflowModelBlueprint
-from rapidata.api_client.models.skip_result import SkipResult
-from rapidata.api_client.models.skip_truth import SkipTruth
-from rapidata.api_client.models.sort_criterion import SortCriterion
-from rapidata.api_client.models.sort_direction import SortDirection
-from rapidata.api_client.models.source_url_metadata import SourceUrlMetadata
-from rapidata.api_client.models.source_url_metadata_model import SourceUrlMetadataModel
-from rapidata.api_client.models.static_selection import StaticSelection
-from rapidata.api_client.models.sticky_state import StickyState
-from rapidata.api_client.models.submit_coco_model import SubmitCocoModel
-from rapidata.api_client.models.submit_coco_result import SubmitCocoResult
-from rapidata.api_client.models.text_asset import TextAsset
-from rapidata.api_client.models.text_asset_model import TextAssetModel
-from rapidata.api_client.models.text_metadata import TextMetadata
-from rapidata.api_client.models.text_metadata_model import TextMetadataModel
-from rapidata.api_client.models.transcription_metadata import TranscriptionMetadata
-from rapidata.api_client.models.transcription_metadata_input import TranscriptionMetadataInput
-from rapidata.api_client.models.transcription_metadata_model import TranscriptionMetadataModel
-from rapidata.api_client.models.transcription_payload import TranscriptionPayload
-from rapidata.api_client.models.transcription_rapid_blueprint import TranscriptionRapidBlueprint
-from rapidata.api_client.models.transcription_result import TranscriptionResult
-from rapidata.api_client.models.transcription_truth import TranscriptionTruth
-from rapidata.api_client.models.transcription_word import TranscriptionWord
-from rapidata.api_client.models.translated_prompt_metadata_model import TranslatedPromptMetadataModel
-from rapidata.api_client.models.translated_string import TranslatedString
-from rapidata.api_client.models.unlock_order_result import UnlockOrderResult
-from rapidata.api_client.models.update_access_model import UpdateAccessModel
-from rapidata.api_client.models.update_campaign_model import UpdateCampaignModel
-from rapidata.api_client.models.update_dataset_name_model import UpdateDatasetNameModel
-from rapidata.api_client.models.update_dimensions_model import UpdateDimensionsModel
-from rapidata.api_client.models.update_order_name_model import UpdateOrderNameModel
-from rapidata.api_client.models.update_validation_rapid_model import UpdateValidationRapidModel
-from rapidata.api_client.models.update_validation_rapid_model_truth import UpdateValidationRapidModelTruth
-from rapidata.api_client.models.upload_coco_result import UploadCocoResult
-from rapidata.api_client.models.upload_files_from_s3_bucket_model import UploadFilesFromS3BucketModel
-from rapidata.api_client.models.upload_from_s3_result import UploadFromS3Result
-from rapidata.api_client.models.upload_text_sources_to_dataset_model import UploadTextSourcesToDatasetModel
-from rapidata.api_client.models.url_asset_input import UrlAssetInput
-from rapidata.api_client.models.user_score_user_filter_model import UserScoreUserFilterModel
-from rapidata.api_client.models.validation_chance import ValidationChance
-from rapidata.api_client.models.validation_import_post_request_blueprint import ValidationImportPostRequestBlueprint
-from rapidata.api_client.models.validation_selection import ValidationSelection
-from rapidata.api_client.models.validation_set_model import ValidationSetModel
-from rapidata.api_client.models.validation_set_model_paged_result import ValidationSetModelPagedResult
-from rapidata.api_client.models.validation_set_overview_model import ValidationSetOverviewModel
-from rapidata.api_client.models.workflow_aggregation_step_model import WorkflowAggregationStepModel
-from rapidata.api_client.models.workflow_artifact_model import WorkflowArtifactModel
-from rapidata.api_client.models.workflow_config_artifact_model import WorkflowConfigArtifactModel
-from rapidata.api_client.models.workflow_config_artifact_model_workflow_config import WorkflowConfigArtifactModelWorkflowConfig
-from rapidata.api_client.models.workflow_labeling_step_model import WorkflowLabelingStepModel
-from rapidata.api_client.models.workflow_split_model import WorkflowSplitModel
-from rapidata.api_client.models.workflow_split_model_filter_configs_inner import WorkflowSplitModelFilterConfigsInner
-from rapidata.api_client.models.workflow_state import WorkflowState
+import os as _os
+import re as _re
+
+# Dynamically discover all model modules in this package
+def _discover_models() -> dict[str, str]:
+    """Discover all model classes from .py files in this package directory."""
+    models = {}
+    package_dir = _os.path.dirname(_os.path.abspath(__file__))
+    package_name = __name__
+    
+    # Pattern to match class names (PascalCase)
+    class_pattern = _re.compile(r'^class\s+([A-Z][A-Za-z0-9]*)\s*[\(:]', _re.MULTILINE)
+    
+    for filename in _os.listdir(package_dir):
+        if filename.endswith('.py') and not filename.startswith('_'):
+            module_name = filename[:-3]  # Remove .py extension
+            filepath = _os.path.join(package_dir, filename)
+            
+            try:
+                with open(filepath, 'r', encoding='utf-8') as f:
+                    content = f.read()
+                    # Find all class definitions
+                    for match in class_pattern.finditer(content):
+                        class_name = match.group(1)
+                        # Skip private classes and base classes
+                        if not class_name.startswith('_'):
+                            models[class_name] = f"{package_name}.{module_name}"
+            except Exception:
+                pass  # Skip files that can't be read
+    
+    return models
+
+
+# Lazy loading for models - models are imported only when accessed
+_MODEL_IMPORTS = _discover_models()
+
+# Cache for imported models
+_IMPORTED: dict[str, type] = {}
+
+
+def __getattr__(name: str):
+    """Lazy import handler for models."""
+    if name in _IMPORTED:
+        return _IMPORTED[name]
+
+    if name in _MODEL_IMPORTS:
+        module_path = _MODEL_IMPORTS[name]
+        import importlib
+        module = importlib.import_module(module_path)
+        attr = getattr(module, name)
+        _IMPORTED[name] = attr
+        return attr
+
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+
+
+def __dir__():
+    """Return list of available attributes for autocompletion."""
+    return list(_MODEL_IMPORTS.keys())
+
+
+# For type checkers and IDE support
+__all__ = list(_MODEL_IMPORTS.keys())

rapidata/api_client/models/ab_test_selection_a_inner.py

@@ -17,8 +17,8 @@ import json
 import pprint
 from pydantic import BaseModel, ConfigDict, Field, StrictStr, ValidationError, field_validator
 from typing import Any, List, Optional
-from rapidata.api_client.models.conditional_validation_selection import ConditionalValidationSelection
 from rapidata.api_client.models.demographic_selection import DemographicSelection
+from rapidata.api_client.models.effort_capped_selection import EffortCappedSelection
 from rapidata.api_client.models.labeling_selection import LabelingSelection
 from rapidata.api_client.models.static_selection import StaticSelection
 from rapidata.api_client.models.validation_selection import ValidationSelection
@@ -26,7 +26,7 @@ from pydantic import StrictStr, Field
 from typing import Union, List, Set, Optional, Dict
 from typing_extensions import Literal, Self
 
-ABTESTSELECTIONAINNER_ONE_OF_SCHEMAS = ["AbTestSelection", "CappedSelection", "ConditionalValidationSelection", "DemographicSelection", "LabelingSelection", "ShufflingSelection", "StaticSelection", "ValidationSelection"]
+ABTESTSELECTIONAINNER_ONE_OF_SCHEMAS = ["AbTestSelection", "CappedSelection", "ConditionalValidationSelection", "DemographicSelection", "EffortCappedSelection", "LabelingSelection", "ShufflingSelection", "StaticSelection", "ValidationSelection"]
 
 class AbTestSelectionAInner(BaseModel):
     """
@@ -40,16 +40,18 @@ class AbTestSelectionAInner(BaseModel):
     oneof_schema_3_validator: Optional[ConditionalValidationSelection] = None
     # data type: DemographicSelection
     oneof_schema_4_validator: Optional[DemographicSelection] = None
+    # data type: EffortCappedSelection
+    oneof_schema_5_validator: Optional[EffortCappedSelection] = None
     # data type: LabelingSelection
-    oneof_schema_5_validator: Optional[LabelingSelection] = None
+    oneof_schema_6_validator: Optional[LabelingSelection] = None
     # data type: ShufflingSelection
-    oneof_schema_6_validator: Optional[ShufflingSelection] = None
+    oneof_schema_7_validator: Optional[ShufflingSelection] = None
     # data type: StaticSelection
-    oneof_schema_7_validator: Optional[StaticSelection] = None
+    oneof_schema_8_validator: Optional[StaticSelection] = None
     # data type: ValidationSelection
-    oneof_schema_8_validator: Optional[ValidationSelection] = None
-    actual_instance: Optional[Union[AbTestSelection, CappedSelection, ConditionalValidationSelection, DemographicSelection, LabelingSelection, ShufflingSelection, StaticSelection, ValidationSelection]] = None
-    one_of_schemas: Set[str] = { "AbTestSelection", "CappedSelection", "ConditionalValidationSelection", "DemographicSelection", "LabelingSelection", "ShufflingSelection", "StaticSelection", "ValidationSelection" }
+    oneof_schema_9_validator: Optional[ValidationSelection] = None
+    actual_instance: Optional[Union[AbTestSelection, CappedSelection, ConditionalValidationSelection, DemographicSelection, EffortCappedSelection, LabelingSelection, ShufflingSelection, StaticSelection, ValidationSelection]] = None
+    one_of_schemas: Set[str] = { "AbTestSelection", "CappedSelection", "ConditionalValidationSelection", "DemographicSelection", "EffortCappedSelection", "LabelingSelection", "ShufflingSelection", "StaticSelection", "ValidationSelection" }
 
     model_config = ConfigDict(
         validate_assignment=True,
@@ -95,6 +97,11 @@ class AbTestSelectionAInner(BaseModel):
             error_messages.append(f"Error! Input type `{type(v)}` is not `DemographicSelection`")
         else:
             match += 1
+        # validate data type: EffortCappedSelection
+        if not isinstance(v, EffortCappedSelection):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `EffortCappedSelection`")
+        else:
+            match += 1
         # validate data type: LabelingSelection
         if not isinstance(v, LabelingSelection):
             error_messages.append(f"Error! Input type `{type(v)}` is not `LabelingSelection`")
@@ -117,10 +124,10 @@ class AbTestSelectionAInner(BaseModel):
             match += 1
         if match > 1:
             # more than 1 match
-            raise ValueError("Multiple matches found when setting `actual_instance` in AbTestSelectionAInner with oneOf schemas: AbTestSelection, CappedSelection, ConditionalValidationSelection, DemographicSelection, LabelingSelection, ShufflingSelection, StaticSelection, ValidationSelection. Details: " + ", ".join(error_messages))
+            raise ValueError("Multiple matches found when setting `actual_instance` in AbTestSelectionAInner with oneOf schemas: AbTestSelection, CappedSelection, ConditionalValidationSelection, DemographicSelection, EffortCappedSelection, LabelingSelection, ShufflingSelection, StaticSelection, ValidationSelection. Details: " + ", ".join(error_messages))
         elif match == 0:
             # no match
-            raise ValueError("No match found when setting `actual_instance` in AbTestSelectionAInner with oneOf schemas: AbTestSelection, CappedSelection, ConditionalValidationSelection, DemographicSelection, LabelingSelection, ShufflingSelection, StaticSelection, ValidationSelection. Details: " + ", ".join(error_messages))
+            raise ValueError("No match found when setting `actual_instance` in AbTestSelectionAInner with oneOf schemas: AbTestSelection, CappedSelection, ConditionalValidationSelection, DemographicSelection, EffortCappedSelection, LabelingSelection, ShufflingSelection, StaticSelection, ValidationSelection. Details: " + ", ".join(error_messages))
         else:
             return v
 
@@ -159,6 +166,12 @@ class AbTestSelectionAInner(BaseModel):
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
+        # deserialize data into EffortCappedSelection
+        try:
+            instance.actual_instance = EffortCappedSelection.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
         # deserialize data into LabelingSelection
         try:
             instance.actual_instance = LabelingSelection.from_json(json_str)
@@ -186,10 +199,10 @@ class AbTestSelectionAInner(BaseModel):
 
         if match > 1:
             # more than 1 match
-            raise ValueError("Multiple matches found when deserializing the JSON string into AbTestSelectionAInner with oneOf schemas: AbTestSelection, CappedSelection, ConditionalValidationSelection, DemographicSelection, LabelingSelection, ShufflingSelection, StaticSelection, ValidationSelection. Details: " + ", ".join(error_messages))
+            raise ValueError("Multiple matches found when deserializing the JSON string into AbTestSelectionAInner with oneOf schemas: AbTestSelection, CappedSelection, ConditionalValidationSelection, DemographicSelection, EffortCappedSelection, LabelingSelection, ShufflingSelection, StaticSelection, ValidationSelection. Details: " + ", ".join(error_messages))
         elif match == 0:
             # no match
-            raise ValueError("No match found when deserializing the JSON string into AbTestSelectionAInner with oneOf schemas: AbTestSelection, CappedSelection, ConditionalValidationSelection, DemographicSelection, LabelingSelection, ShufflingSelection, StaticSelection, ValidationSelection. Details: " + ", ".join(error_messages))
+            raise ValueError("No match found when deserializing the JSON string into AbTestSelectionAInner with oneOf schemas: AbTestSelection, CappedSelection, ConditionalValidationSelection, DemographicSelection, EffortCappedSelection, LabelingSelection, ShufflingSelection, StaticSelection, ValidationSelection. Details: " + ", ".join(error_messages))
         else:
             return instance
 
@@ -203,7 +216,7 @@ class AbTestSelectionAInner(BaseModel):
         else:
             return json.dumps(self.actual_instance)
 
-    def to_dict(self) -> Optional[Union[Dict[str, Any], AbTestSelection, CappedSelection, ConditionalValidationSelection, DemographicSelection, LabelingSelection, ShufflingSelection, StaticSelection, ValidationSelection]]:
+    def to_dict(self) -> Optional[Union[Dict[str, Any], AbTestSelection, CappedSelection, ConditionalValidationSelection, DemographicSelection, EffortCappedSelection, LabelingSelection, ShufflingSelection, StaticSelection, ValidationSelection]]:
         """Returns the dict representation of the actual instance"""
         if self.actual_instance is None:
             return None
@@ -220,6 +233,7 @@ class AbTestSelectionAInner(BaseModel):
 
 from rapidata.api_client.models.ab_test_selection import AbTestSelection
 from rapidata.api_client.models.capped_selection import CappedSelection
+from rapidata.api_client.models.conditional_validation_selection import ConditionalValidationSelection
 from rapidata.api_client.models.shuffling_selection import ShufflingSelection
 # TODO: Rewrite to not use raise_errors
 AbTestSelectionAInner.model_rebuild(raise_errors=False)