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

rapidata/rapidata_client/workflow/_select_words_workflow.py

@@ -1,14 +1,26 @@
-from rapidata.api_client.models.simple_workflow_model import SimpleWorkflowModel
-from rapidata.api_client.models.simple_workflow_model_blueprint import SimpleWorkflowModelBlueprint
-from rapidata.api_client.models.transcription_rapid_blueprint import TranscriptionRapidBlueprint
+from rapidata.api_client.models.i_order_workflow_model import IOrderWorkflowModel
+from rapidata.api_client.models.i_order_workflow_model_simple_workflow_model import (
+    IOrderWorkflowModelSimpleWorkflowModel,
+)
+from rapidata.api_client.models.i_rapid_blueprint import IRapidBlueprint
+from rapidata.api_client.models.i_rapid_blueprint_transcription_rapid_blueprint import (
+    IRapidBlueprintTranscriptionRapidBlueprint,
+)
 from rapidata.rapidata_client.workflow._base_workflow import Workflow
+from rapidata.api_client.models.i_rapid_payload_transcription_payload import (
+    IRapidPayloadTranscriptionPayload,
+)
+from rapidata.api_client.models.transcription_word import TranscriptionWord
+from rapidata.rapidata_client.datapoints._datapoint import Datapoint
+from rapidata.api_client.models.rapid_modality import RapidModality
+from rapidata.api_client.models.i_rapid_payload import IRapidPayload
 
 
 class SelectWordsWorkflow(Workflow):
     """
     A workflow for select words tasks.
 
-    This class represents a select words workflow 
+    This class represents a select words workflow
     where datapoints have a sentence attached to them where words can be selected.
 
     Attributes:
@@ -18,17 +30,45 @@ class SelectWordsWorkflow(Workflow):
         instruction (str): The instruction to be provided for the select words task.
     """
 
+    modality = RapidModality.TRANSCRIPTION
+
     def __init__(self, instruction: str):
         super().__init__(type="SimpleWorkflowConfig")
         self._instruction = instruction
 
-    def _to_model(self) -> SimpleWorkflowModel:
-        blueprint = TranscriptionRapidBlueprint(
-            _t="TranscriptionBlueprint",
-            title=self._instruction
+    def _get_instruction(self) -> str:
+        return self._instruction
+
+    def _to_model(self) -> IOrderWorkflowModel:
+        blueprint = IRapidBlueprintTranscriptionRapidBlueprint(
+            _t="TranscriptionBlueprint", title=self._instruction
         )
 
-        return SimpleWorkflowModel(
-            _t="SimpleWorkflow",
-            blueprint=SimpleWorkflowModelBlueprint(blueprint)
+        return IOrderWorkflowModel(
+            actual_instance=IOrderWorkflowModelSimpleWorkflowModel(
+                _t="SimpleWorkflow",
+                blueprint=IRapidBlueprint(actual_instance=blueprint),
+            )
         )
+
+    def _to_payload(self, datapoint: Datapoint) -> IRapidPayload:
+        assert (
+            datapoint.sentence is not None
+        ), "SelectWordsWorkflow requires a sentence datapoint"
+
+        return IRapidPayload(
+            actual_instance=IRapidPayloadTranscriptionPayload(
+                _t="TranscriptionPayload",
+                title=self._instruction,
+                transcription=[
+                    TranscriptionWord(word=word, wordIndex=i)
+                    for i, word in enumerate(datapoint.sentence.split())
+                ],
+            )
+        )
+
+    def __str__(self) -> str:
+        return f"SelectWordsWorkflow(instruction='{self._instruction}')"
+
+    def __repr__(self) -> str:
+        return f"SelectWordsWorkflow(instruction={self._instruction!r})"

rapidata/rapidata_client/workflow/_timestamp_workflow.py

@@ -1,14 +1,25 @@
-from rapidata.api_client.models.simple_workflow_model import SimpleWorkflowModel
-from rapidata.api_client.models.simple_workflow_model_blueprint import SimpleWorkflowModelBlueprint
-from rapidata.api_client.models.scrub_rapid_blueprint import ScrubRapidBlueprint
+from rapidata.api_client.models.i_order_workflow_model import IOrderWorkflowModel
+from rapidata.api_client.models.i_order_workflow_model_simple_workflow_model import (
+    IOrderWorkflowModelSimpleWorkflowModel,
+)
+from rapidata.api_client.models.i_rapid_blueprint import IRapidBlueprint
+from rapidata.api_client.models.i_rapid_blueprint_scrub_rapid_blueprint import (
+    IRapidBlueprintScrubRapidBlueprint,
+)
 from rapidata.rapidata_client.workflow._base_workflow import Workflow
+from rapidata.api_client.models.i_rapid_payload_scrub_payload import (
+    IRapidPayloadScrubPayload,
+)
+from rapidata.rapidata_client.datapoints._datapoint import Datapoint
+from rapidata.api_client.models.rapid_modality import RapidModality
+from rapidata.api_client.models.i_rapid_payload import IRapidPayload
 
 
 class TimestampWorkflow(Workflow):
     """
     A workflow for timestamp tasks.
 
-    This class represents a timestamp workflow 
+    This class represents a timestamp workflow
     where audio or video content receives timestamps based on the instruction.
 
     Attributes:
@@ -18,17 +29,37 @@ class TimestampWorkflow(Workflow):
         instruction (str): The instruction to be provided for the timestamp task.
     """
 
+    modality = RapidModality.SCRUB
+
     def __init__(self, instruction: str):
         super().__init__(type="SimpleWorkflowConfig")
         self._instruction = instruction
 
-    def _to_model(self) -> SimpleWorkflowModel:
-        blueprint = ScrubRapidBlueprint(
-            _t="ScrubBlueprint",
-            target=self._instruction
+    def _get_instruction(self) -> str:
+        return self._instruction
+
+    def _to_model(self) -> IOrderWorkflowModel:
+        blueprint = IRapidBlueprintScrubRapidBlueprint(
+            _t="ScrubBlueprint", target=self._instruction
+        )
+
+        return IOrderWorkflowModel(
+            actual_instance=IOrderWorkflowModelSimpleWorkflowModel(
+                _t="SimpleWorkflow",
+                blueprint=IRapidBlueprint(actual_instance=blueprint),
+            )
         )
 
-        return SimpleWorkflowModel(
-            _t="SimpleWorkflow",
-            blueprint=SimpleWorkflowModelBlueprint(blueprint)
+    def _to_payload(self, datapoint: Datapoint) -> IRapidPayload:
+        return IRapidPayload(
+            actual_instance=IRapidPayloadScrubPayload(
+                _t="ScrubPayload",
+                target=self._instruction,
+            )
         )
+
+    def __str__(self) -> str:
+        return f"TimestampWorkflow(instruction='{self._instruction}')"
+
+    def __repr__(self) -> str:
+        return f"TimestampWorkflow(instruction={self._instruction!r})"

rapidata/service/__init__.py

@@ -1 +0,0 @@
-from .local_file_service import LocalFileService as LocalFileService

rapidata/service/credential_manager.py

@@ -11,7 +11,7 @@ from typing import Dict, List, Optional, Tuple
 import requests
 from colorama import Fore
 from pydantic import BaseModel
-from rapidata.rapidata_client.logging import logger, managed_print
+from rapidata.rapidata_client.config import logger, managed_print
 
 
 class ClientCredential(BaseModel):
@@ -52,7 +52,7 @@ class CredentialManager:
 
     def _read_credentials(self) -> Dict[str, List[ClientCredential]]:
         """Read all stored credentials from the config file."""
-        logger.debug(f"Reading credentials from {self.config_path}")
+        logger.debug("Reading credentials from %s", self.config_path)
         if not self.config_path.exists():
             return {}
 
@@ -77,11 +77,13 @@ class CredentialManager:
         with open(self.config_path, "w") as f:
             json.dump(data, f, indent=2)
 
-        logger.debug(f"Credentials written to {self.config_path} with data: {data}")
+        logger.debug("Credentials written to %s with data: %s", self.config_path, data)
 
         # Ensure file is only readable by the user
         os.chmod(self.config_path, 0o600)
-        logger.debug(f"Set permissions for {self.config_path} to read/write for user only.")
+        logger.debug(
+            "Set permissions for %s to read/write for user only.", self.config_path
+        )
 
     def _store_credential(self, credential: ClientCredential) -> None:
         credentials = self._read_credentials()
@@ -107,19 +109,19 @@ class CredentialManager:
     def get_client_credentials(self) -> Optional[ClientCredential]:
         """Gets stored client credentials or create new ones via browser auth."""
         credentials = self._read_credentials()
-        logger.debug(f"Stored credentials: {credentials}")
+        logger.debug("Stored credentials: %s", credentials)
         env_credentials = credentials.get(self.endpoint, [])
 
         if env_credentials:
-            logger.debug(f"Found credentials for {self.endpoint}: {env_credentials}")
+            logger.debug("Found credentials for %s: %s", self.endpoint, env_credentials)
             credential = self._select_credential(env_credentials)
-            logger.debug(f"Selected credential: {credential}")
+            logger.debug("Selected credential: %s", credential)
             if credential:
                 credential.last_used = datetime.now(timezone.utc)
                 self._write_credentials(credentials)
                 return credential
 
-        logger.debug(f"No credentials found for {self.endpoint}. Creating new ones.")
+        logger.debug("No credentials found for %s. Creating new ones.", self.endpoint)
         return self._create_new_credentials()
 
     def reset_credentials(self) -> None:
@@ -128,31 +130,29 @@ class CredentialManager:
         if self.endpoint in credentials:
             del credentials[self.endpoint]
             self._write_credentials(credentials)
-            logger.info(
-                f"Credentials for {self.endpoint} have been reset."
-            )
+            logger.info("Credentials for %s have been reset.", self.endpoint)
 
     def _get_bridge_tokens(self) -> Optional[BridgeToken]:
         """Get bridge tokens from the identity endpoint."""
         logger.debug("Getting bridge tokens")
         try:
             bridge_endpoint = (
-                f"{self.endpoint}/Identity/CreateBridgeToken?clientId=rapidata-cli"
+                f"{self.endpoint}/identity/bridge-token?clientId=rapidata-cli"
             )
             response = requests.post(bridge_endpoint, verify=self.cert_path)
             if not response.ok:
-                logger.error(f"Failed to get bridge tokens: {response.status_code}")
+                logger.error("Failed to get bridge tokens: %s", response.status_code)
                 return None
 
             data = response.json()
             return BridgeToken(read_key=data["readKey"], write_key=data["writeKey"])
         except requests.RequestException as e:
-            logger.error(f"Failed to get bridge tokens: {e}")
+            logger.error("Failed to get bridge tokens: %s", e)
             return None
 
     def _poll_read_key(self, read_key: str) -> Optional[str]:
         """Poll the read key endpoint until we get an access token."""
-        read_endpoint = f"{self.endpoint}/Identity/ReadBridgeToken"
+        read_endpoint = f"{self.endpoint}/identity/bridge-token"
         start_time = time.time()
 
         while time.time() - start_time < self.poll_timeout:
@@ -169,11 +169,11 @@ class CredentialManager:
                     continue
                 else:
                     # Error occurred
-                    logger.error(f"Error polling read key: {response.status_code}")
+                    logger.error("Error polling read key: %s", response.status_code)
                     return None
 
             except requests.RequestException as e:
-                logger.error(f"Error polling read key: {e}")
+                logger.error("Error polling read key: %s", e)
                 return None
 
         logger.error("Polling timed out")
@@ -198,7 +198,7 @@ class CredentialManager:
             data = response.json()
             return data.get("clientId"), data.get("clientSecret"), display_name
         except requests.RequestException as e:
-            logger.error(f"Failed to create client: {e}")
+            logger.error("Failed to create client: %s", e)
             return None
 
     def _create_new_credentials(self) -> Optional[ClientCredential]:

rapidata/service/openapi_service.py

@@ -1,29 +1,41 @@
+from __future__ import annotations
+
 import subprocess
 from importlib.metadata import version, PackageNotFoundError
+from typing import TYPE_CHECKING
 
-from rapidata.api_client.api.campaign_api import CampaignApi
-from rapidata.api_client.api.dataset_api import DatasetApi
-from rapidata.api_client.api.order_api import OrderApi
-from rapidata.api_client.api.pipeline_api import PipelineApi
-from rapidata.api_client.api.rapid_api import RapidApi
-from rapidata.api_client.api.validation_set_api import ValidationSetApi
-from rapidata.api_client.api.workflow_api import WorkflowApi
 from rapidata.api_client.configuration import Configuration
 from rapidata.service.credential_manager import CredentialManager
-from rapidata.rapidata_client.api.rapidata_exception import RapidataApiClient
-from rapidata.rapidata_client.logging import logger
+from rapidata.rapidata_client.api.rapidata_api_client import RapidataApiClient
+from rapidata.rapidata_client.config import logger, managed_print
+from authlib.integrations.httpx_client import OAuthError
+
+if TYPE_CHECKING:
+    from rapidata.api_client.api.job_api import JobApi
+    from rapidata.api_client import CustomerRapidApi
+    from rapidata.api_client.api.campaign_api import CampaignApi
+    from rapidata.api_client.api.asset_api import AssetApi
+    from rapidata.api_client.api.dataset_api import DatasetApi
+    from rapidata.api_client.api.benchmark_api import BenchmarkApi
+    from rapidata.api_client.api.order_api import OrderApi
+    from rapidata.api_client.api.pipeline_api import PipelineApi
+    from rapidata.api_client.api.leaderboard_api import LeaderboardApi
+    from rapidata.api_client.api.validation_set_api import ValidationSetApi
+    from rapidata.api_client.api.workflow_api import WorkflowApi
+    from rapidata.api_client.api.participant_api import ParticipantApi
+    from rapidata.api_client.api.audience_api import AudienceApi
 
 
 class OpenAPIService:
     def __init__(
-            self,
-            client_id: str | None,
-            client_secret: str | None,
-            environment: str,
-            oauth_scope: str,
-            cert_path: str | None = None,
-            token: dict | None = None,
-            leeway: int = 60,
+        self,
+        client_id: str | None,
+        client_secret: str | None,
+        environment: str,
+        oauth_scope: str,
+        cert_path: str | None = None,
+        token: dict | None = None,
+        leeway: int = 60,
     ):
         self.environment = environment
         endpoint = f"https://api.{environment}"
@@ -32,7 +44,12 @@ class OpenAPIService:
         if environment == "rapidata.dev" and not cert_path:
             cert_path = _get_local_certificate()
 
-        logger.debug(f"Using cert_path: {cert_path} environment: {environment} token: {token} client_id: {client_id} client_secret: {client_secret}")
+        logger.debug(
+            "Using cert_path: %s environment: %s client_id: %s",
+            cert_path,
+            environment,
+            client_id,
+        )
         logger.debug("Initializing OpenAPIService")
         self.credential_manager = CredentialManager(
             endpoint=auth_endpoint, cert_path=cert_path
@@ -41,7 +58,7 @@ class OpenAPIService:
 
         logger.debug("Initializing RapidataApiClient")
         client_configuration = Configuration(host=endpoint, ssl_ca_cert=cert_path)
-        logger.debug(f"Client configuration: {client_configuration}")
+        logger.debug("Client configuration: %s", client_configuration)
         self.api_client = RapidataApiClient(
             configuration=client_configuration,
             header_name="X-Client",
@@ -62,52 +79,129 @@ class OpenAPIService:
             return
 
         if not client_id or not client_secret:
-            logger.debug("Client ID and secret not provided, fetching from credential manager")
+            logger.debug(
+                "Client ID and secret not provided, fetching from credential manager"
+            )
             credentials = self.credential_manager.get_client_credentials()
             if not credentials:
                 raise ValueError("Failed to fetch client credentials")
             client_id = credentials.client_id
             client_secret = credentials.client_secret
+        try:
+            self.api_client.rest_client.setup_oauth_client_credentials(
+                client_id=client_id,
+                client_secret=client_secret,
+                token_endpoint=f"{auth_endpoint}/connect/token",
+                scope=oauth_scope,
+            )
+        except OAuthError as e:
+            if e.error != "invalid_client":
+                raise
+            logger.warning(
+                "Invalid client credentials detected, resetting and retrying: %s", e
+            )
+            self.reset_credentials()
+
+            # Retry with fresh credentials
+            credentials = self.credential_manager.get_client_credentials()
+            if not credentials:
+                raise ValueError(
+                    "Failed to fetch client credentials after reset"
+                ) from e
+
+            self.api_client.rest_client.setup_oauth_client_credentials(
+                client_id=credentials.client_id,
+                client_secret=credentials.client_secret,
+                token_endpoint=f"{auth_endpoint}/connect/token",
+                scope=oauth_scope,
+            )
+            managed_print("Credentials were reset and re-authenticated successfully")
 
-        self.api_client.rest_client.setup_oauth_client_credentials(
-            client_id=client_id,
-            client_secret=client_secret,
-            token_endpoint=f"{auth_endpoint}/connect/token",
-            scope=oauth_scope,
-        )
         logger.debug("Client credentials authentication setup complete")
 
     def reset_credentials(self):
+        logger.info("Resetting credentials in OpenAPIService")
         self.credential_manager.reset_credentials()
+        logger.info("Credentials reset in OpenAPIService")
 
     @property
     def order_api(self) -> OrderApi:
+        from rapidata.api_client.api.order_api import OrderApi
+
         return OrderApi(self.api_client)
 
+    @property
+    def job_api(self) -> JobApi:
+        from rapidata.api_client.api.job_api import JobApi
+
+        return JobApi(self.api_client)
+
+    @property
+    def asset_api(self) -> AssetApi:
+        from rapidata.api_client.api.asset_api import AssetApi
+
+        return AssetApi(self.api_client)
+
     @property
     def dataset_api(self) -> DatasetApi:
+        from rapidata.api_client.api.dataset_api import DatasetApi
+
         return DatasetApi(self.api_client)
 
     @property
     def validation_api(self) -> ValidationSetApi:
+        from rapidata.api_client.api.validation_set_api import ValidationSetApi
+
         return ValidationSetApi(self.api_client)
 
     @property
-    def rapid_api(self) -> RapidApi:
-        return RapidApi(self.api_client)
+    def rapid_api(self) -> CustomerRapidApi:
+        from rapidata.api_client.api.customer_rapid_api import CustomerRapidApi
+
+        return CustomerRapidApi(self.api_client)
 
     @property
     def campaign_api(self) -> CampaignApi:
+        from rapidata.api_client.api.campaign_api import CampaignApi
+
         return CampaignApi(self.api_client)
 
     @property
     def pipeline_api(self) -> PipelineApi:
+        from rapidata.api_client.api.pipeline_api import PipelineApi
+
         return PipelineApi(self.api_client)
 
     @property
     def workflow_api(self) -> WorkflowApi:
+        from rapidata.api_client.api.workflow_api import WorkflowApi
+
         return WorkflowApi(self.api_client)
 
+    @property
+    def leaderboard_api(self) -> LeaderboardApi:
+        from rapidata.api_client.api.leaderboard_api import LeaderboardApi
+
+        return LeaderboardApi(self.api_client)
+
+    @property
+    def benchmark_api(self) -> BenchmarkApi:
+        from rapidata.api_client.api.benchmark_api import BenchmarkApi
+
+        return BenchmarkApi(self.api_client)
+
+    @property
+    def participant_api(self) -> ParticipantApi:
+        from rapidata.api_client.api.participant_api import ParticipantApi
+
+        return ParticipantApi(self.api_client)
+
+    @property
+    def audience_api(self) -> AudienceApi:
+        from rapidata.api_client.api.audience_api import AudienceApi
+
+        return AudienceApi(self.api_client)
+
     def _get_rapidata_package_version(self):
         """
         Returns the version of the currently installed rapidata package.
@@ -121,6 +215,12 @@ class OpenAPIService:
         except PackageNotFoundError:
             return None
 
+    def __repr__(self) -> str:
+        return self.__str__()
+
+    def __str__(self) -> str:
+        return f"OpenAPIService(environment={self.environment})"
+
 
 def _get_local_certificate() -> str | None:
     result = subprocess.run(["mkcert", "-CAROOT"], capture_output=True)

rapidata/types/__init__.py

@@ -0,0 +1,149 @@
+# Main Client and Core Types
+from rapidata.rapidata_client.rapidata_client import RapidataClient
+
+# Audience Types
+from rapidata.rapidata_client.audience.rapidata_audience import RapidataAudience
+from rapidata.rapidata_client.audience.rapidata_audience_manager import (
+    RapidataAudienceManager,
+)
+
+# Benchmark Types
+from rapidata.rapidata_client.benchmark.rapidata_benchmark import RapidataBenchmark
+from rapidata.rapidata_client.benchmark.rapidata_benchmark_manager import (
+    RapidataBenchmarkManager,
+)
+from rapidata.rapidata_client.benchmark.leaderboard.rapidata_leaderboard import (
+    RapidataLeaderboard,
+)
+
+# Selection Types
+from rapidata.rapidata_client.selection.ab_test_selection import AbTestSelection
+from rapidata.rapidata_client.selection.capped_selection import CappedSelection
+from rapidata.rapidata_client.selection.conditional_validation_selection import (
+    ConditionalValidationSelection,
+)
+from rapidata.rapidata_client.selection.demographic_selection import (
+    DemographicSelection,
+)
+from rapidata.rapidata_client.selection.effort_selection import EffortSelection
+from rapidata.rapidata_client.selection.labeling_selection import LabelingSelection
+from rapidata.rapidata_client.selection.shuffling_selection import ShufflingSelection
+from rapidata.rapidata_client.selection.static_selection import StaticSelection
+from rapidata.rapidata_client.selection.validation_selection import ValidationSelection
+from rapidata.rapidata_client.selection.rapidata_retrieval_modes import (
+    RapidataRetrievalMode,
+)
+
+# Filter Types
+from rapidata.rapidata_client.filter.age_filter import AgeFilter
+from rapidata.rapidata_client.filter.and_filter import AndFilter
+from rapidata.rapidata_client.filter.campaign_filter import CampaignFilter
+from rapidata.rapidata_client.filter.country_filter import CountryFilter
+from rapidata.rapidata_client.filter.custom_filter import CustomFilter
+from rapidata.rapidata_client.filter.gender_filter import GenderFilter
+from rapidata.rapidata_client.filter.language_filter import LanguageFilter
+from rapidata.rapidata_client.filter.new_user_filter import NewUserFilter
+from rapidata.rapidata_client.filter.not_filter import NotFilter
+from rapidata.rapidata_client.filter.or_filter import OrFilter
+from rapidata.rapidata_client.filter.response_count_filter import ResponseCountFilter
+from rapidata.rapidata_client.filter.user_score_filter import UserScoreFilter
+
+# Filter Model Types
+from rapidata.rapidata_client.filter.models.age_group import AgeGroup
+from rapidata.rapidata_client.filter.models.gender import Gender
+
+# Settings Types
+from rapidata.rapidata_client.settings.alert_on_fast_response import AlertOnFastResponse
+from rapidata.rapidata_client.settings.allow_neither_both import AllowNeitherBoth
+from rapidata.rapidata_client.settings.custom_setting import CustomSetting
+from rapidata.rapidata_client.settings.free_text_minimum_characters import (
+    FreeTextMinimumCharacters,
+)
+from rapidata.rapidata_client.settings.no_shuffle import NoShuffle
+from rapidata.rapidata_client.settings.play_video_until_the_end import (
+    PlayVideoUntilTheEnd,
+)
+from rapidata.rapidata_client.settings.rapidata_settings import RapidataSettings
+from rapidata.rapidata_client.settings.translation_behaviour import TranslationBehaviour
+from rapidata.rapidata_client.settings.models.translation_behaviour_options import (
+    TranslationBehaviourOptions,
+)
+
+
+# Configuration Types
+from rapidata.rapidata_client.config.order_config import OrderConfig
+from rapidata.rapidata_client.config.upload_config import UploadConfig
+from rapidata.rapidata_client.config.rapidata_config import RapidataConfig
+from rapidata.rapidata_client.config.logging_config import LoggingConfig
+
+# Exception Types
+from rapidata.rapidata_client.exceptions.failed_upload_exception import (
+    FailedUploadException,
+)
+
+# Utility Types
+from rapidata.rapidata_client.demographic.demographic_manager import DemographicManager
+
+# API Client Types
+from rapidata.rapidata_client.api.rapidata_api_client import RapidataApiClient
+
+__all__ = [
+    # Main Client and Core Types
+    "RapidataClient",
+    # Audience Types
+    "RapidataAudience",
+    "RapidataAudienceManager",
+    # Benchmark Types
+    "RapidataBenchmark",
+    "RapidataBenchmarkManager",
+    "RapidataLeaderboard",
+    # Selection Types
+    "AbTestSelection",
+    "CappedSelection",
+    "ConditionalValidationSelection",
+    "DemographicSelection",
+    "EffortSelection",
+    "LabelingSelection",
+    "ShufflingSelection",
+    "StaticSelection",
+    "ValidationSelection",
+    "RapidataRetrievalMode",
+    # Filter Types
+    "AgeFilter",
+    "AndFilter",
+    "CampaignFilter",
+    "CountryFilter",
+    "CustomFilter",
+    "GenderFilter",
+    "LanguageFilter",
+    "NewUserFilter",
+    "NotFilter",
+    "OrFilter",
+    "ResponseCountFilter",
+    "UserScoreFilter",
+    # Filter Model Types
+    "AgeGroup",
+    "Gender",
+    # Settings Types
+    "AlertOnFastResponse",
+    "AllowNeitherBoth",
+    "CustomSetting",
+    "FreeTextMinimumCharacters",
+    "NoShuffle",
+    "PlayVideoUntilTheEnd",
+    "RapidataSettings",
+    "TranslationBehaviour",
+    "TranslationBehaviourOptions",
+    # Configuration Types
+    "OrderConfig",
+    "UploadConfig",
+    "RapidataConfig",
+    "LoggingConfig",
+    # Exception Types
+    "FailedUploadException",
+    # Utility Types
+    "CountryCodes",
+    "DemographicManager",
+    # API Client Types
+    "RapidataApiClient",
+]