rapidata 2.37.0__py3-none-any.whl → 3.0.0a8__py3-none-any.whl

rapidata/rapidata_client/workflow/_timestamp_workflow.py

@@ -1,12 +1,18 @@
-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.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.api_client.models.scrub_rapid_blueprint import ScrubRapidBlueprint
 from rapidata.rapidata_client.workflow._base_workflow import Workflow
-from rapidata.api_client import ScrubPayload
+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):
@@ -29,15 +35,31 @@ class TimestampWorkflow(Workflow):
         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 SimpleWorkflowModel(
-            _t="SimpleWorkflow", blueprint=SimpleWorkflowModelBlueprint(blueprint)
+        return IOrderWorkflowModel(
+            actual_instance=IOrderWorkflowModelSimpleWorkflowModel(
+                _t="SimpleWorkflow",
+                blueprint=IRapidBlueprint(actual_instance=blueprint),
+            )
         )
 
-    def _to_payload(self, datapoint: Datapoint) -> ScrubPayload:
-        return ScrubPayload(
-            _t="ScrubPayload",
-            target=self._instruction,
+    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):

rapidata/service/openapi_service.py

@@ -1,20 +1,29 @@
+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.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.rapid_api import RapidApi
-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.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:
@@ -36,12 +45,10 @@ class OpenAPIService:
             cert_path = _get_local_certificate()
 
         logger.debug(
-            "Using cert_path: %s environment: %s token: %s client_id: %s client_secret: %s",
+            "Using cert_path: %s environment: %s client_id: %s",
             cert_path,
             environment,
-            token,
             client_id,
-            client_secret,
         )
         logger.debug("Initializing OpenAPIService")
         self.credential_manager = CredentialManager(
@@ -80,58 +87,121 @@ class OpenAPIService:
                 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.
@@ -145,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",
+]

rapidata-3.0.0a8.dist-info/METADATA

@@ -0,0 +1,32 @@
+Metadata-Version: 2.4
+Name: rapidata
+Version: 3.0.0a8
+Summary: Rapidata package containing the Rapidata Python Client to interact with the Rapidata Web API in an easy way.
+Author-email: Rapidata AG <info@rapidata.ai>
+License-Expression: Apache-2.0
+License-File: LICENSE
+Requires-Python: <4,>=3.10
+Requires-Dist: authlib==1.6.5
+Requires-Dist: colorama==0.4.6
+Requires-Dist: deprecated<2,>=1.2.14
+Requires-Dist: diskcache<6,>=5.6.3
+Requires-Dist: hatchling>=1.28.0
+Requires-Dist: httpx<0.29,>=0.28.1
+Requires-Dist: opentelemetry-api>=1.27.0
+Requires-Dist: opentelemetry-exporter-otlp>=1.27.0
+Requires-Dist: opentelemetry-sdk>=1.27.0
+Requires-Dist: packaging>=16.0.0
+Requires-Dist: pandas<3,>=2.2.3
+Requires-Dist: pydantic<3,>=2.8.2
+Requires-Dist: pyjwt<3,>=2.9.0
+Requires-Dist: python-dateutil<3,>=2.9.0.post0
+Requires-Dist: requests<3,>=2.32.3
+Requires-Dist: tinytag<3,>=2.0.0
+Requires-Dist: tqdm<5,>=4.66.5
+Description-Content-Type: text/markdown
+
+# Rapidata Python Client
+
+Python client to interface with the Rapidata API.
+
+Docs: https://docs.rapidata.ai/