rapidata 1.4.6__py3-none-any.whl → 1.6.0__py3-none-any.whl

rapidata/api_client/models/validation_set_model_paged_result.py

@@ -0,0 +1,105 @@
+# coding: utf-8
+
+"""
+    Rapidata.Dataset
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: v1
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, Field, StrictInt
+from typing import Any, ClassVar, Dict, List, Optional
+from rapidata.api_client.models.validation_set_model import ValidationSetModel
+from typing import Optional, Set
+from typing_extensions import Self
+
+class ValidationSetModelPagedResult(BaseModel):
+    """
+    ValidationSetModelPagedResult
+    """ # noqa: E501
+    total: StrictInt
+    page: StrictInt
+    page_size: StrictInt = Field(alias="pageSize")
+    items: List[ValidationSetModel]
+    total_pages: Optional[StrictInt] = Field(default=None, alias="totalPages")
+    __properties: ClassVar[List[str]] = ["total", "page", "pageSize", "items", "totalPages"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of ValidationSetModelPagedResult from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        * OpenAPI `readOnly` fields are excluded.
+        """
+        excluded_fields: Set[str] = set([
+            "total_pages",
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of each item in items (list)
+        _items = []
+        if self.items:
+            for _item_items in self.items:
+                if _item_items:
+                    _items.append(_item_items.to_dict())
+            _dict['items'] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of ValidationSetModelPagedResult from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "total": obj.get("total"),
+            "page": obj.get("page"),
+            "pageSize": obj.get("pageSize"),
+            "items": [ValidationSetModel.from_dict(_item) for _item in obj["items"]] if obj.get("items") is not None else None,
+            "totalPages": obj.get("totalPages")
+        })
+        return _obj
+
+

rapidata/api_client_README.md

@@ -128,6 +128,7 @@ Class | Method | HTTP request | Description
 *ValidationApi* | [**validation_add_validation_text_rapid_post**](rapidata/api_client/docs/ValidationApi.md#validation_add_validation_text_rapid_post) | **POST** /Validation/AddValidationTextRapid | Adds a new validation rapid to the specified validation set.
 *ValidationApi* | [**validation_create_validation_set_post**](rapidata/api_client/docs/ValidationApi.md#validation_create_validation_set_post) | **POST** /Validation/CreateValidationSet | Creates a new empty validation set.
 *ValidationApi* | [**validation_get_available_validation_sets_get**](rapidata/api_client/docs/ValidationApi.md#validation_get_available_validation_sets_get) | **GET** /Validation/GetAvailableValidationSets | Gets the available validation sets for the current user.
+*ValidationApi* | [**validation_get_by_id_get**](rapidata/api_client/docs/ValidationApi.md#validation_get_by_id_get) | **GET** /Validation/GetById | Gets a validation set by the id.
 *ValidationApi* | [**validation_import_compare_post**](rapidata/api_client/docs/ValidationApi.md#validation_import_compare_post) | **POST** /Validation/ImportCompare | Imports a compare validation set from a zip file.
 *ValidationApi* | [**validation_import_post**](rapidata/api_client/docs/ValidationApi.md#validation_import_post) | **POST** /Validation/Import | Imports a validation set from a zip file.
 *ValidationApi* | [**validation_query_validation_sets_get**](rapidata/api_client/docs/ValidationApi.md#validation_query_validation_sets_get) | **GET** /Validation/QueryValidationSets | Queries validation sets based on the provided filter, paging and sorting criteria.
@@ -255,6 +256,7 @@ Class | Method | HTTP request | Description
  - [GetPipelineByIdResultArtifactsValue](rapidata/api_client/docs/GetPipelineByIdResultArtifactsValue.md)
  - [GetPublicOrdersResult](rapidata/api_client/docs/GetPublicOrdersResult.md)
  - [GetSimpleWorkflowResultOverviewResult](rapidata/api_client/docs/GetSimpleWorkflowResultOverviewResult.md)
+ - [GetValidationSetByIdResult](rapidata/api_client/docs/GetValidationSetByIdResult.md)
  - [GetWorkflowByIdResult](rapidata/api_client/docs/GetWorkflowByIdResult.md)
  - [GetWorkflowByIdResultWorkflow](rapidata/api_client/docs/GetWorkflowByIdResultWorkflow.md)
  - [GetWorkflowProgressResult](rapidata/api_client/docs/GetWorkflowProgressResult.md)
@@ -380,9 +382,9 @@ Class | Method | HTTP request | Description
  - [ValidationChance](rapidata/api_client/docs/ValidationChance.md)
  - [ValidationImportPostRequestBlueprint](rapidata/api_client/docs/ValidationImportPostRequestBlueprint.md)
  - [ValidationSelection](rapidata/api_client/docs/ValidationSelection.md)
- - [ValidationSet](rapidata/api_client/docs/ValidationSet.md)
+ - [ValidationSetModel](rapidata/api_client/docs/ValidationSetModel.md)
+ - [ValidationSetModelPagedResult](rapidata/api_client/docs/ValidationSetModelPagedResult.md)
  - [ValidationSetOverviewModel](rapidata/api_client/docs/ValidationSetOverviewModel.md)
- - [ValidationSetPagedResult](rapidata/api_client/docs/ValidationSetPagedResult.md)
  - [WorkflowAggregationStepModel](rapidata/api_client/docs/WorkflowAggregationStepModel.md)
  - [WorkflowArtifactModel](rapidata/api_client/docs/WorkflowArtifactModel.md)
  - [WorkflowConfigArtifactModel](rapidata/api_client/docs/WorkflowConfigArtifactModel.md)

rapidata/rapidata_client/__init__.py

@@ -19,7 +19,7 @@ from .metadata import (
     PromptMetadata,
     TranscriptionMetadata,
 )
-from .feature_flags import FeatureFlags
+from .settings import Settings, FeatureFlags # remove FeatureFlags next major version
 from .country_codes import CountryCodes
 from .assets import MediaAsset, TextAsset, MultiAsset
 from .filter import (

rapidata/rapidata_client/dataset/rapidata_validation_set.py

@@ -47,9 +47,10 @@ class RapidataValidationSet:
     Get a `ValidationSet` either by using `rapi.get_validation_set(id)` to get an existing validation set or by using `rapi.new_validation_set(name)` to create a new validation set.
     """
 
-    def __init__(self, validation_set_id, openapi_service: OpenAPIService):
+    def __init__(self, validation_set_id, openapi_service: OpenAPIService, name: str):
         self.id = validation_set_id
         self.openapi_service = openapi_service
+        self.name = name
 
     def add_general_validation_rapid(
         self,
@@ -153,7 +154,7 @@ class RapidataValidationSet:
         else:
             raise ValueError("Invalid asset type")
 
-    def add_classify_validation_rapid(
+    def add_classify_rapid(
         self,
         asset: MediaAsset | TextAsset,
         question: str,
@@ -188,7 +189,7 @@ class RapidataValidationSet:
             randomCorrectProbability=len(truths) / len(categories),
         )
 
-    def add_compare_validation_rapid(
+    def add_compare_rapid(
         self,
         asset: MultiAsset,
         question: str,
@@ -225,7 +226,7 @@ class RapidataValidationSet:
             randomCorrectProbability=1 / len(asset),
         )
 
-    def add_transcription_validation_rapid(
+    def add_transcription_rapid(
         self,
         asset: MediaAsset | TextAsset,
         question: str,

rapidata/rapidata_client/dataset/validation_set_builder.py

@@ -58,6 +58,7 @@ class ValidationSetBuilder:
         validation_set = RapidataValidationSet(
             validation_set_id=self.validation_set_id,
             openapi_service=self.openapi_service,
+            name=self.name,
         )
 
         for rapid_part in self._rapid_parts:

rapidata/rapidata_client/order/rapidata_order.py

@@ -3,7 +3,7 @@ from rapidata.rapidata_client.dataset.rapidata_dataset import RapidataDataset
 from rapidata.service.openapi_service import OpenAPIService
 import json
 from rapidata.api_client.exceptions import ApiException
-from typing import Optional, cast
+from typing import Optional, cast, Any
 from rapidata.api_client.models.workflow_artifact_model import WorkflowArtifactModel
 from tqdm import tqdm
 
@@ -11,12 +11,11 @@ class RapidataOrder:
     """
     Represents a Rapidata order.
 
-    :param name: The name of the order.
-    :type name: str
-    :param workflow: The workflow associated with the order.
-    :type workflow: Workflow
-    :param rapidata_service: The Rapidata service used to create and manage the order.
-    :type rapidata_service: RapidataService
+    Args:
+        The ID of the order.
+        The optional Dataset associated with the order.
+        The OpenAPIService instance used to interact with the Rapidata API.
+        The name of the order.
     """
 
     def __init__(
@@ -36,30 +35,23 @@ class RapidataOrder:
         """
         Submits the order for processing.
         """
-
         self.openapi_service.order_api.order_submit_post(self.order_id)
 
-    def approve(self):
-        """
-        Approves the order for execution.
-        """
-        self.openapi_service.order_api.order_approve_post(self.order_id)
-
-    def get_status(self):
+    def get_status(self) -> str:
         """
         Gets the status of the order.
-
-        :return: The status of the order.
-        :rtype: str
+        
+        Returns: 
+            The status of the order.
         """
-        return self.openapi_service.order_api.order_get_by_id_get(self.order_id)
+        return self.openapi_service.order_api.order_get_by_id_get(self.order_id).state
 
     def display_progress_bar(self, refresh_rate=5):
         """
         Displays a progress bar for the order processing using tqdm.
         
-        :param refresh_rate: How often to refresh the progress bar, in seconds.
-        :type refresh_rate: float
+        Prameter: 
+            How often to refresh the progress bar, in seconds.
         """
         total_rapids = self._get_workflow_progress().total
         with tqdm(total=total_rapids, desc="Processing order", unit="rapids") as pbar:
@@ -107,15 +99,15 @@ class RapidataOrder:
         return progress
 
         
-    def get_results(self):
+    def get_results(self) -> dict[str, Any]:
         """
         Gets the results of the order. 
         If the order is still processing, this method will block until the order is completed and then return the results.
 
-        :return: The results of the order.
-        :rtype: dict
+        Returns: 
+            The results of the order.
         """
-        while self.get_status().state == "Processing":
+        while self.get_status() == "Processing":
             sleep(5)
 
         try:
@@ -131,10 +123,10 @@ class RapidataOrder:
             raise Exception(f"Failed to parse order results: {str(e)}") from e
 
     @property
-    def dataset(self):
+    def dataset(self) -> RapidataDataset | None:
         """
         The dataset associated with the order.
-        :return: The RapidataDataset instance.
-        :rtype: RapidataDataset
+        Returns: 
+            The RapidataDataset instance.
         """
         return self._dataset

rapidata/rapidata_client/order/rapidata_order_builder.py

@@ -14,11 +14,7 @@ from rapidata.api_client.models.create_order_model_user_filters_inner import (
 from rapidata.api_client.models.create_order_model_workflow import (
     CreateOrderModelWorkflow,
 )
-from rapidata.api_client.models.country_user_filter_model import CountryUserFilterModel
-from rapidata.api_client.models.language_user_filter_model import (
-    LanguageUserFilterModel,
-)
-from rapidata.rapidata_client.feature_flags import FeatureFlags
+from rapidata.rapidata_client.settings import FeatureFlags, Settings
 from rapidata.rapidata_client.metadata.base_metadata import Metadata
 from rapidata.rapidata_client.dataset.rapidata_dataset import RapidataDataset
 from rapidata.rapidata_client.referee.naive_referee import NaiveReferee
@@ -35,6 +31,8 @@ from rapidata.rapidata_client.assets import MediaAsset, TextAsset, MultiAsset
 
 from typing import Optional, cast, Sequence
 
+from deprecated import deprecated
+
 
 class RapidataOrderBuilder:
     """Builder object for creating Rapidata orders.
@@ -66,7 +64,7 @@ class RapidataOrderBuilder:
         self._metadata: list[Metadata] | None = None
         self._aggregator: AggregatorType | None = None
         self._validation_set_id: str | None = None
-        self._feature_flags: FeatureFlags | None = None
+        self._settings: Settings | FeatureFlags | None = None # remove featureflag next big release
         self._user_filters: list[Filter] = []
         self._selections: list[Selection] = []
         self._rapids_per_bag: int = 2
@@ -101,8 +99,8 @@ class RapidataOrderBuilder:
             referee=CreateOrderModelReferee(self._referee.to_model()),
             validationSetId=self._validation_set_id,
             featureFlags=(
-                self._feature_flags.to_list()
-                if self._feature_flags is not None
+                self._settings.to_list()
+                if self._settings is not None
                 else None
             ),
             selections=[
@@ -269,6 +267,26 @@ class RapidataOrderBuilder:
         self._metadata = metadata  # type: ignore
         return self
 
+    def settings(self, settings: Settings) -> "RapidataOrderBuilder":
+        """
+        Set the settings for the order.
+
+        Args:
+            settings (Settings): The settings to be set.
+
+        Returns:
+            RapidataOrderBuilder: The updated RapidataOrderBuilder instance.
+        """
+        if not isinstance(settings, Settings):
+            raise TypeError("Settings must be of type Settings.")
+
+        self._settings = settings
+        return self
+
+    @deprecated(
+        version="1.6.0",
+        reason="The feature_flags method is deprecated, use settings instead.",
+    )
     def feature_flags(self, feature_flags: FeatureFlags) -> "RapidataOrderBuilder":
         """
         Set the feature flags for the order.
@@ -282,7 +300,7 @@ class RapidataOrderBuilder:
         if not isinstance(feature_flags, FeatureFlags):
             raise TypeError("Feature flags must be of type FeatureFlags.")
 
-        self._feature_flags = feature_flags
+        self._settings = feature_flags
         return self
 
     def filters(self, filters: Sequence[Filter]) -> "RapidataOrderBuilder":

rapidata/rapidata_client/rapidata_client.py

@@ -3,7 +3,6 @@ from rapidata.rapidata_client.dataset.rapidata_validation_set import (
 )
 from rapidata.rapidata_client.dataset.validation_set_builder import ValidationSetBuilder
 from rapidata.rapidata_client.order.rapidata_order_builder import RapidataOrderBuilder
-from rapidata.rapidata_client.utils.utils import Utils
 from rapidata.service.openapi_service import OpenAPIService
 from rapidata.rapidata_client.order.rapidata_order import RapidataOrder
 from rapidata.rapidata_client.dataset.rapidata_dataset import RapidataDataset
@@ -20,6 +19,8 @@ from rapidata.api_client.models.root_filter import RootFilter
 from rapidata.api_client.models.filter import Filter
 from rapidata.api_client.models.sort_criterion import SortCriterion
 
+from rapidata.api_client.models.query_validation_set_model import QueryValidationSetModel
+
 
 class RapidataClient:
     """The Rapidata client is the main entry point for interacting with the Rapidata API. It allows you to create orders and validation sets. For creating a new order, check out `new_order()`. For creating a new validation set, check out `new_validation_set()`."""
@@ -71,17 +72,6 @@ class RapidataClient:
         """
         return ValidationSetBuilder(name=name, openapi_service=self.openapi_service)
 
-    def get_validation_set(self, validation_set_id: str) -> RapidataValidationSet:
-        """Get a validation set by ID.
-
-        Args:
-            validation_set_id (str): The ID of the validation set.
-
-        Returns:
-            RapidataValidationSet: The ValidationSet instance.
-        """
-        return RapidataValidationSet(validation_set_id, self.openapi_service)
-
     def get_order(self, order_id: str) -> RapidataOrder:
         """Get an order by ID.
 
@@ -132,6 +122,39 @@ class RapidataClient:
 
         orders = [self.get_order(order.id) for order in order_page_result.items]
         return orders
+    
+    def get_validation_set(self, validation_set_id: str) -> RapidataValidationSet:
+        """Get a validation set by ID.
+
+        Args:
+            validation_set_id (str): The ID of the validation set.
+
+        Returns:
+            RapidataValidationSet: The ValidationSet instance.
+        """
+        try:
+            validation_set = self.openapi_service.validation_api.validation_get_by_id_get(id=validation_set_id)
+        except Exception:
+            raise ValueError(f"ValidationSet with ID {validation_set_id} not found.")
+        
+        return RapidataValidationSet(validation_set_id, self.openapi_service, validation_set.name)
+    
+    def find_validation_sets(self, name: str = "", amount: int = 1) -> list[RapidataValidationSet]:
+        try:
+            validation_page_result = self.openapi_service.validation_api.validation_query_validation_sets_get(QueryValidationSetModel(
+                pageInfo=PageInfo(index=1, size=amount),
+                filter=RootFilter(filters=[Filter(field="Name", operator="Contains", value=name)]),
+                sortCriteria=[SortCriterion(direction="Desc", propertyName="CreatedAt")]
+                ))
+
+        except BadRequestException as e:
+            raise ValueError(f"Error occured during request. \nError: {e.body} \nTraceid: {e.headers.get('X-Trace-Id') if isinstance(e.headers, HTTPHeaderDict) else 'Unknown'}")
+
+        except Exception as e:
+            raise ValueError(f"Unknown error occured: {e}")
+
+        orders = [self.get_validation_set(validation_set.id) for validation_set in validation_page_result.items] # type: ignore # will be fixed with the next backend deployment
+        return orders
 
     def create_classify_order(self, name: str) -> ClassificationQuestionBuilder:
         """Create a new classification order where people are asked to classify an image.
@@ -154,12 +177,3 @@ class RapidataClient:
             CompareQuestionBuilder: A CompareQuestionBuilder instance.
         """
         return CompareCriteriaBuilder(name=name, openapi_service=self.openapi_service)
-
-    @property
-    def utils(self) -> Utils:
-        """Get the Utils instance.
-
-        Returns:
-            Utils: The Utils instance associated with this client.
-        """
-        return Utils(openapi_service=self.openapi_service)

rapidata/rapidata_client/{feature_flags → settings}/__init__.py

@@ -1 +1,2 @@
 from .feature_flags import FeatureFlags
+from .settings import Settings

rapidata/rapidata_client/{feature_flags → settings}/feature_flags.py

@@ -1,12 +1,13 @@
 from rapidata.api_client.models.feature_flag_model import FeatureFlagModel
 from enum import Enum
+from deprecated import deprecated
 
 class TranslationBehaviour(Enum):
     BOTH = "both"
     ONLY_ORIGINAL = "only original"
     ONLY_TRANSLATED = "only translated"
 
-
+@deprecated(version="1.6.0", reason="Use the Settings class instead.")
 class FeatureFlags:
     """A class to manage feature flags.
 
@@ -17,6 +18,7 @@ class FeatureFlags:
         _flags (dict[str, str]): A dictionary to store feature flags.
     """
 
+    @deprecated(version="1.6.0", reason="Use the Settings class instead.")
     def __init__(self):
         """Initialize the FeatureFlags object with an empty flags dictionary."""
         self._flags: dict[str, str] = {}

rapidata/rapidata_client/settings/settings.py

@@ -0,0 +1,123 @@
+from rapidata.api_client.models.feature_flag_model import FeatureFlagModel
+from enum import Enum
+
+class TranslationBehaviour(Enum):
+    BOTH = "both"
+    ONLY_ORIGINAL = "only original"
+    ONLY_TRANSLATED = "only translated"
+
+
+class Settings:
+    """A class to manage settings.
+
+    This class provides methods to set and manage various settings
+    used in the application.
+
+    Attributes:
+        _settings (dict[str, str]): A dictionary to store settings.
+    """
+
+    def __init__(self):
+        """Initialize the Settings object with an empty flags dictionary."""
+        self._settings: dict[str, str] = {}
+
+    def to_list(self) -> list[FeatureFlagModel]:
+        """Convert the settings to a list of FeatureFlagModel objects.
+
+        Returns:
+            list[FeatureFlagModel]: A list of FeatureFlagModel objects
+                representing the current settings.
+        """
+        return [
+            FeatureFlagModel(key=name, value=value)
+            for name, value in self._settings.items()
+        ]
+
+    def alert_on_fast_response(self, milliseconds: int):
+        """Gives an alert as a pop up on the UI when the response time is less than the threshold.
+
+        Args:
+            milliseconds (int): if the user responds in less than this time, an alert will be shown.
+
+        Returns:
+            Settings: The current Settings instance for method chaining.
+        """
+        if milliseconds < 10:
+            print(
+                f"Warning: Are you sure you want to set the threshold so low ({milliseconds} milliseconds)?"
+            )
+
+        if milliseconds > 30000:
+            print(
+                f"Warning: Are you sure you want to set the threshold so high ({milliseconds/1000} seconds)?"
+            )
+
+        self._settings["alert_on_fast_response"] = str(milliseconds)
+        return self
+
+
+    def translation_behaviour(self, behaviour: TranslationBehaviour = TranslationBehaviour.BOTH):
+        """Defines what's the behaviour of the translation in the UI.
+
+        The behaviour can be set to:
+            - TranslationBehaviour.BOTH: Show both the original and the translated text.
+            - TranslationBehaviour.ONLY_ORIGINAL: Show only the original text.
+            - TranslationBehaviour.ONLY_TRANSLATED: Show only the translated text.
+
+        Args:
+            behaviour (TranslationBehaviour): The translation behaviour. Defaults to TranslationBehaviour.BOTH.
+
+        Returns:
+            Settings: The current Settings instance for method chaining.
+        """
+        self._settings["translation_behaviour"] = behaviour.value
+        return self
+
+    def free_text_minimum_characters(self, value: int):
+        """Set the minimum number of characters a user has to type before the task can be successfully submitted.
+
+        Args:
+            value (int): The minimum number of characters for free text.
+
+        Returns:
+            Settings: The current Settings instance for method chaining.
+        """
+        self._settings["free_text_minimum_characters"] = str(value)
+        return self
+
+    def no_shuffle(self, value: bool = True):
+        """Only for classify tasks. If true, the order of the categories will be the same as the one in the workflow.
+
+        Args:
+            value (bool, optional): Whether to disable shuffling. Defaults to True.
+
+        Returns:
+            Settings: The current Settings instance for method chaining.
+        """
+        self._settings["no_shuffle"] = str(value)
+        return self
+
+    def compare_with_prompt_design(self, value: bool = True):
+        """A special design to compare two texts/images based on a criteria and a given prompt.
+
+        Args:
+            value (bool, optional): Whether to enable compare with prompt design. Defaults to True.
+
+        Returns:
+            Settings: The current Settings instance for method chaining.
+        """
+        self._settings["claire"] = str(value)
+        return self
+
+    def key_value(self, key: str, value: str):
+        """Set a custom setting with the given key and value. Use this to enable features that do not have a dedicated method (yet).
+
+        Args:
+            key (str): The key for the custom setting.
+            value (str): The value for the custom setting.
+
+        Returns:
+            Settings: The current Settings instance for method chaining.
+        """
+        self._settings[key] = value
+        return self

rapidata/rapidata_client/workflow/base_workflow.py

@@ -4,7 +4,7 @@ from typing import Any
 from rapidata.api_client.models.simple_workflow_model import SimpleWorkflowModel
 from rapidata.api_client.models.evaluation_workflow_model import EvaluationWorkflowModel
 from rapidata.api_client.models.compare_workflow_model import CompareWorkflowModel
-from rapidata.rapidata_client.feature_flags import FeatureFlags
+from rapidata.rapidata_client.settings import Settings
 from rapidata.rapidata_client.referee.base_referee import Referee
 
 
@@ -13,7 +13,7 @@ class Workflow(ABC):
     def __init__(self, type: str):
         self._type = type
         self._target_country_codes: list[str] = []
-        self._feature_flags: FeatureFlags = FeatureFlags()
+        self._feature_flags: Settings = Settings()
 
     def to_dict(self) -> dict[str, Any]:
         return {
@@ -37,6 +37,6 @@ class Workflow(ABC):
         self._target_country_codes = target_country_codes
         return self
 
-    def feature_flags(self, feature_flags: FeatureFlags):
+    def feature_flags(self, feature_flags: Settings):
         self._feature_flags = feature_flags
         return self