rapidata 2.27.0__py3-none-any.whl → 2.44.4__py3-none-any.whl

rapidata/__init__.py

@@ -1,10 +1,11 @@
-__version__ = "2.27.0"
+__version__ = "2.44.4"
 
 from .rapidata_client import (
     RapidataClient,
     DemographicSelection,
     LabelingSelection,
-    RetrievalMode,
+    EffortSelection,
+    RapidataRetrievalMode,
     ValidationSelection,
     ConditionalValidationSelection,
     CappedSelection,
@@ -16,7 +17,10 @@ from .rapidata_client import (
     FreeTextMinimumCharacters,
     NoShuffle,
     PlayVideoUntilTheEnd,
+    MuteVideo,
     CustomSetting,
+    AllowNeitherBoth,
+    SwapContextInstruction,
     CountryFilter,
     LanguageFilter,
     UserScoreFilter,
@@ -28,14 +32,13 @@ from .rapidata_client import (
     AgeGroup,
     GenderFilter,
     Gender,
-    CountryCodes,
-    MediaAsset,
-    TextAsset,
-    MultiAsset,
-    RapidataDataTypes,
     Box,
+    Datapoint,
     PromptMetadata,
+    FailedUploadException,
+    rapidata_config,
     logger,
-    configure_logger,
-    RapidataOutputManager,
+    managed_print,
 )
+
+from . import types

rapidata/api_client/__init__.py

@@ -3,9 +3,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)
@@ -16,28 +16,7 @@
 
 __version__ = "1.0.0"
 
-# import apis into sdk package
-from rapidata.api_client.api.campaign_api import CampaignApi
-from rapidata.api_client.api.client_api import ClientApi
-from rapidata.api_client.api.coco_api import CocoApi
-from rapidata.api_client.api.compare_workflow_api import CompareWorkflowApi
-from rapidata.api_client.api.customer_rapid_api import CustomerRapidApi
-from rapidata.api_client.api.datapoint_api import DatapointApi
-from rapidata.api_client.api.dataset_api import DatasetApi
-from rapidata.api_client.api.evaluation_workflow_api import EvaluationWorkflowApi
-from rapidata.api_client.api.feedback_api import FeedbackApi
-from rapidata.api_client.api.identity_api import IdentityApi
-from rapidata.api_client.api.newsletter_api import NewsletterApi
-from rapidata.api_client.api.order_api import OrderApi
-from rapidata.api_client.api.pipeline_api import PipelineApi
-from rapidata.api_client.api.rapidata_identity_api_api import RapidataIdentityAPIApi
-from rapidata.api_client.api.simple_workflow_api import SimpleWorkflowApi
-from rapidata.api_client.api.user_info_api import UserInfoApi
-from rapidata.api_client.api.user_rapid_api import UserRapidApi
-from rapidata.api_client.api.validation_set_api import ValidationSetApi
-from rapidata.api_client.api.workflow_api import WorkflowApi
-
-# import ApiClient
+# These are imported eagerly as they're commonly needed
 from rapidata.api_client.api_response import ApiResponse
 from rapidata.api_client.api_client import ApiClient
 from rapidata.api_client.configuration import Configuration
@@ -48,308 +27,83 @@ from rapidata.api_client.exceptions import ApiKeyError
 from rapidata.api_client.exceptions import ApiAttributeError
 from rapidata.api_client.exceptions import ApiException
 
-# import models into sdk 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.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_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_asset import DatapointAsset
-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_dataset_id_datapoints_post_request_metadata_inner import DatasetDatasetIdDatapointsPostRequestMetadataInner
-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.multi_compare_truth import MultiCompareTruth
-from rapidata.api_client.models.naive_referee_config import NaiveRefereeConfig
-from rapidata.api_client.models.naive_referee_info import NaiveRefereeInfo
-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.never_ending_referee_info import NeverEndingRefereeInfo
-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.probabilistic_attach_category_referee_info import ProbabilisticAttachCategoryRefereeInfo
-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_model import RapidModel
-from rapidata.api_client.models.rapid_model_paged_result import RapidModelPagedResult
-from rapidata.api_client.models.rapid_model_referee import RapidModelReferee
-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
+# Lazy loading for APIs
+_API_IMPORTS = {
+    "AssetApi": "rapidata.api_client.api.asset_api",
+    "AudienceApi": "rapidata.api_client.api.audience_api",
+    "BenchmarkApi": "rapidata.api_client.api.benchmark_api",
+    "CampaignApi": "rapidata.api_client.api.campaign_api",
+    "ClientApi": "rapidata.api_client.api.client_api",
+    "CompareWorkflowApi": "rapidata.api_client.api.compare_workflow_api",
+    "CustomerRapidApi": "rapidata.api_client.api.customer_rapid_api",
+    "DatapointApi": "rapidata.api_client.api.datapoint_api",
+    "DatasetApi": "rapidata.api_client.api.dataset_api",
+    "EvaluationWorkflowApi": "rapidata.api_client.api.evaluation_workflow_api",
+    "FeedbackApi": "rapidata.api_client.api.feedback_api",
+    "GroupedRankingWorkflowApi": "rapidata.api_client.api.grouped_ranking_workflow_api",
+    "IdentityApi": "rapidata.api_client.api.identity_api",
+    "LeaderboardApi": "rapidata.api_client.api.leaderboard_api",
+    "NewsletterApi": "rapidata.api_client.api.newsletter_api",
+    "OrderApi": "rapidata.api_client.api.order_api",
+    "ParticipantApi": "rapidata.api_client.api.participant_api",
+    "PipelineApi": "rapidata.api_client.api.pipeline_api",
+    "PromptApi": "rapidata.api_client.api.prompt_api",
+    "RapidataIdentityAPIApi": "rapidata.api_client.api.rapidata_identity_api_api",
+    "SampleApi": "rapidata.api_client.api.sample_api",
+    "SimpleWorkflowApi": "rapidata.api_client.api.simple_workflow_api",
+    "SurveyApi": "rapidata.api_client.api.survey_api",
+    "UserRapidApi": "rapidata.api_client.api.user_rapid_api",
+    "ValidationSetApi": "rapidata.api_client.api.validation_set_api",
+    "WorkflowApi": "rapidata.api_client.api.workflow_api",
+}
+
+# Import model mappings from the models subpackage (dynamically discovered)
+from rapidata.api_client.models import _MODEL_IMPORTS
+
+# Combine all lazy imports
+_LAZY_IMPORTS = {**_API_IMPORTS, **_MODEL_IMPORTS}
+
+# Cache for imported modules
+_IMPORTED: dict[str, type] = {}
+
+
+def __getattr__(name: str):
+    """Lazy import handler for APIs and models."""
+    if name in _IMPORTED:
+        return _IMPORTED[name]
+
+    if name in _LAZY_IMPORTS:
+        module_path = _LAZY_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(globals().keys()) + list(_LAZY_IMPORTS.keys())
+
+
+# For type checkers and IDE support
+__all__ = (
+    [
+        # Core
+        "__version__",
+        "ApiResponse",
+        "ApiClient",
+        "Configuration",
+        "OpenApiException",
+        "ApiTypeError",
+        "ApiValueError",
+        "ApiKeyError",
+        "ApiAttributeError",
+        "ApiException",
+    ]
+    + list(_API_IMPORTS.keys())
+    + list(_MODEL_IMPORTS.keys())
+)

rapidata/api_client/api/__init__.py

@@ -1,22 +1,29 @@
 # flake8: noqa
 
 # import apis into api package
+from rapidata.api_client.api.asset_api import AssetApi
+from rapidata.api_client.api.audience_api import AudienceApi
+from rapidata.api_client.api.benchmark_api import BenchmarkApi
 from rapidata.api_client.api.campaign_api import CampaignApi
 from rapidata.api_client.api.client_api import ClientApi
-from rapidata.api_client.api.coco_api import CocoApi
 from rapidata.api_client.api.compare_workflow_api import CompareWorkflowApi
 from rapidata.api_client.api.customer_rapid_api import CustomerRapidApi
 from rapidata.api_client.api.datapoint_api import DatapointApi
 from rapidata.api_client.api.dataset_api import DatasetApi
 from rapidata.api_client.api.evaluation_workflow_api import EvaluationWorkflowApi
 from rapidata.api_client.api.feedback_api import FeedbackApi
+from rapidata.api_client.api.grouped_ranking_workflow_api import GroupedRankingWorkflowApi
 from rapidata.api_client.api.identity_api import IdentityApi
+from rapidata.api_client.api.leaderboard_api import LeaderboardApi
 from rapidata.api_client.api.newsletter_api import NewsletterApi
 from rapidata.api_client.api.order_api import OrderApi
+from rapidata.api_client.api.participant_api import ParticipantApi
 from rapidata.api_client.api.pipeline_api import PipelineApi
+from rapidata.api_client.api.prompt_api import PromptApi
 from rapidata.api_client.api.rapidata_identity_api_api import RapidataIdentityAPIApi
+from rapidata.api_client.api.sample_api import SampleApi
 from rapidata.api_client.api.simple_workflow_api import SimpleWorkflowApi
-from rapidata.api_client.api.user_info_api import UserInfoApi
+from rapidata.api_client.api.survey_api import SurveyApi
 from rapidata.api_client.api.user_rapid_api import UserRapidApi
 from rapidata.api_client.api.validation_set_api import ValidationSetApi
 from rapidata.api_client.api.workflow_api import WorkflowApi