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

rapidata/rapidata_client/workflow/__init__.py

@@ -1,10 +1,10 @@
 from ._locate_workflow import LocateWorkflow
 from ._draw_workflow import DrawWorkflow
-from ._base_workflow import Workflow 
+from ._base_workflow import Workflow
 from ._classify_workflow import ClassifyWorkflow
 from ._compare_workflow import CompareWorkflow
 from ._free_text_workflow import FreeTextWorkflow
 from ._select_words_workflow import SelectWordsWorkflow
 from ._evaluation_workflow import EvaluationWorkflow
 from ._timestamp_workflow import TimestampWorkflow
-from ._ranking_workflow import RankingWorkflow
+from ._multi_ranking_workflow import MultiRankingWorkflow

rapidata/rapidata_client/workflow/_base_workflow.py

@@ -1,19 +1,8 @@
 from abc import ABC, abstractmethod
 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.referee._base_referee import Referee
-from rapidata.api_client import (
-    ClassifyPayload,
-    ComparePayload,
-    LocatePayload,
-    ScrubPayload,
-    TranscriptionPayload,
-    LinePayload,
-    FreeTextPayload,
-)
+from rapidata.api_client.models.i_order_workflow_model import IOrderWorkflowModel
+from rapidata.api_client.models.i_rapid_payload import IRapidPayload
 from rapidata.rapidata_client.datapoints._datapoint import Datapoint
 from rapidata.api_client.models.rapid_modality import RapidModality
 
@@ -23,7 +12,6 @@ class Workflow(ABC):
 
     def __init__(self, type: str):
         self._type = type
-        self._target_country_codes: list[str] = []
 
     def _to_dict(self) -> dict[str, Any]:
         return {
@@ -34,19 +22,22 @@ class Workflow(ABC):
     def _to_payload(
         self,
         datapoint: Datapoint,
-    ) -> (
-        ClassifyPayload
-        | ComparePayload
-        | LocatePayload
-        | ScrubPayload
-        | LinePayload
-        | FreeTextPayload
-        | TranscriptionPayload
-    ):
+    ) -> IRapidPayload:
         pass
 
     @abstractmethod
-    def _to_model(
-        self,
-    ) -> SimpleWorkflowModel | CompareWorkflowModel | EvaluationWorkflowModel:
+    def _get_instruction(self) -> str:
+        pass
+
+    @abstractmethod
+    def _to_model(self) -> IOrderWorkflowModel:
         pass
+
+    def _format_datapoints(self, datapoints: list[Datapoint]) -> list[Datapoint]:
+        return datapoints
+
+    def __str__(self) -> str:
+        return self._type
+
+    def __repr__(self) -> str:
+        return self._type

rapidata/rapidata_client/workflow/_classify_workflow.py

@@ -1,15 +1,23 @@
 from typing import Any
-from rapidata.api_client.models.attach_category_rapid_blueprint import (
-    AttachCategoryRapidBlueprint,
+from rapidata.api_client.models.i_rapid_blueprint import IRapidBlueprint
+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.simple_workflow_model import SimpleWorkflowModel
-from rapidata.api_client.models.simple_workflow_model_blueprint import (
-    SimpleWorkflowModelBlueprint,
+from rapidata.api_client.models.attach_category_rapid_blueprint_category import (
+    AttachCategoryRapidBlueprintCategory,
 )
+from rapidata.api_client.models.i_rapid_payload import IRapidPayload
 from rapidata.rapidata_client.workflow import Workflow
-from rapidata.api_client import ClassifyPayload
+from rapidata.api_client.models.i_rapid_payload_classify_payload import (
+    IRapidPayloadClassifyPayload,
+)
 from rapidata.rapidata_client.datapoints._datapoint import Datapoint
 from rapidata.api_client.models.rapid_modality import RapidModality
+from rapidata.api_client.models.classify_payload_category import ClassifyPayloadCategory
+from rapidata.api_client.models.i_rapid_blueprint_attach_category_rapid_blueprint import (
+    IRapidBlueprintAttachCategoryRapidBlueprint,
+)
 
 
 class ClassifyWorkflow(Workflow):
@@ -35,6 +43,9 @@ class ClassifyWorkflow(Workflow):
         self._instruction = instruction
         self._answer_options = answer_options
 
+    def _get_instruction(self) -> str:
+        return self._instruction
+
     def _to_dict(self) -> dict[str, Any]:
         return {
             **super()._to_dict(),
@@ -45,21 +56,37 @@ class ClassifyWorkflow(Workflow):
             },
         }
 
-    def _to_model(self) -> SimpleWorkflowModel:
-        blueprint = AttachCategoryRapidBlueprint(
+    def _to_model(self) -> IOrderWorkflowModel:
+        blueprint = IRapidBlueprintAttachCategoryRapidBlueprint(
             _t="ClassifyBlueprint",
             title=self._instruction,
-            possibleCategories=self._answer_options,
+            categories=[
+                AttachCategoryRapidBlueprintCategory(label=option, value=option)
+                for option in self._answer_options
+            ],
         )
 
-        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) -> ClassifyPayload:
-        return ClassifyPayload(
-            _t="ClassifyPayload",
-            possibleCategories=self._answer_options,
-            title=self._instruction,
+    def _to_payload(self, datapoint: Datapoint) -> IRapidPayload:
+        return IRapidPayload(
+            actual_instance=IRapidPayloadClassifyPayload(
+                _t="ClassifyPayload",
+                categories=[
+                    ClassifyPayloadCategory(label=option, value=option)
+                    for option in self._answer_options
+                ],
+                title=self._instruction,
+            )
         )
+
+    def __str__(self) -> str:
+        return f"ClassifyWorkflow(instruction='{self._instruction}', options={self._answer_options})"
+
+    def __repr__(self) -> str:
+        return f"ClassifyWorkflow(instruction='{self._instruction}', answer_options={self._answer_options!r})"

rapidata/rapidata_client/workflow/_compare_workflow.py

@@ -1,12 +1,18 @@
 from typing import Any
-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_compare_rapid_blueprint import (
+    IRapidBlueprintCompareRapidBlueprint,
+)
+from rapidata.api_client.models.i_rapid_payload import IRapidPayload
 from rapidata.rapidata_client.workflow import Workflow
-from rapidata.api_client.models.compare_rapid_blueprint import CompareRapidBlueprint
-from rapidata.api_client.models.simple_workflow_model import SimpleWorkflowModel
-from rapidata.api_client import ComparePayload
+from rapidata.api_client.models.i_rapid_payload_compare_payload import (
+    IRapidPayloadComparePayload,
+)
 from rapidata.rapidata_client.datapoints._datapoint import Datapoint
+from rapidata.api_client.models.i_rapid_blueprint import IRapidBlueprint
 from rapidata.api_client.models.rapid_modality import RapidModality
 
 
@@ -31,6 +37,9 @@ class CompareWorkflow(Workflow):
         self._instruction = instruction
         self._a_b_names = a_b_names
 
+    def _get_instruction(self) -> str:
+        return self._instruction
+
     def _to_dict(self) -> dict[str, Any]:
         return {
             **super()._to_dict(),
@@ -38,20 +47,30 @@ class CompareWorkflow(Workflow):
             "indexIdentifiers": self._a_b_names,
         }
 
-    def _to_model(self) -> SimpleWorkflowModel:
-        blueprint = CompareRapidBlueprint(
+    def _to_model(self) -> IOrderWorkflowModel:
+        blueprint = IRapidBlueprintCompareRapidBlueprint(
             _t="CompareBlueprint",
             criteria=self._instruction,
             indexIdentifiers=self._a_b_names,
         )
 
-        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) -> ComparePayload:
-        return ComparePayload(
-            _t="ComparePayload",
-            criteria=self._instruction,
+    def _to_payload(self, datapoint: Datapoint) -> IRapidPayload:
+        return IRapidPayload(
+            actual_instance=IRapidPayloadComparePayload(
+                _t="ComparePayload",
+                criteria=self._instruction,
+            )
         )
+
+    def __str__(self) -> str:
+        return f"CompareWorkflow(instruction='{self._instruction}', a_b_names={self._a_b_names})"
+
+    def __repr__(self) -> str:
+        return f"CompareWorkflow(instruction={self._instruction!r}, a_b_names={self._a_b_names!r})"

rapidata/rapidata_client/workflow/_draw_workflow.py

@@ -1,10 +1,16 @@
-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.add_validation_rapid_model import IRapidPayload
+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_line_rapid_blueprint import (
+    IRapidBlueprintLineRapidBlueprint,
 )
-from rapidata.api_client.models.line_rapid_blueprint import LineRapidBlueprint
 from rapidata.rapidata_client.workflow._base_workflow import Workflow
-from rapidata.api_client import LinePayload
+from rapidata.api_client.models.i_rapid_payload_line_payload import (
+    IRapidPayloadLinePayload,
+)
 from rapidata.rapidata_client.datapoints._datapoint import Datapoint
 from rapidata.api_client.models.rapid_modality import RapidModality
 
@@ -16,15 +22,31 @@ class DrawWorkflow(Workflow):
         super().__init__(type="SimpleWorkflowConfig")
         self._target = target
 
-    def _to_model(self) -> SimpleWorkflowModel:
-        blueprint = LineRapidBlueprint(_t="LineBlueprint", target=self._target)
+    def _get_instruction(self) -> str:
+        return self._target
+
+    def _to_model(self) -> IOrderWorkflowModel:
+        blueprint = IRapidBlueprintLineRapidBlueprint(
+            _t="LineBlueprint", target=self._target
+        )
 
-        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) -> LinePayload:
-        return LinePayload(
-            _t="LinePayload",
-            target=self._target,
+    def _to_payload(self, datapoint: Datapoint) -> IRapidPayload:
+        return IRapidPayload(
+            actual_instance=IRapidPayloadLinePayload(
+                _t="LinePayload",
+                target=self._target,
+            )
         )
+
+    def __str__(self) -> str:
+        return f"DrawWorkflow(target='{self._target}')"
+
+    def __repr__(self) -> str:
+        return f"DrawWorkflow(target={self._target!r})"

rapidata/rapidata_client/workflow/_evaluation_workflow.py

@@ -1,7 +1,10 @@
-from rapidata.api_client.models.evaluation_workflow_model import EvaluationWorkflowModel
 from rapidata.rapidata_client.workflow._base_workflow import Workflow
 from rapidata.rapidata_client.datapoints._datapoint import Datapoint
 from rapidata.api_client.models.rapid_modality import RapidModality
+from rapidata.api_client.models.i_order_workflow_model import IOrderWorkflowModel
+from rapidata.api_client.models.i_order_workflow_model_evaluation_workflow_model import (
+    IOrderWorkflowModelEvaluationWorkflowModel,
+)
 
 
 class EvaluationWorkflow(Workflow):
@@ -22,12 +25,23 @@ class EvaluationWorkflow(Workflow):
         self.validation_set_id = validation_set_id
         self.should_accept_incorrect = should_accept_incorrect
 
-    def _to_model(self):
-        return EvaluationWorkflowModel(
-            _t="EvaluationWorkflow",
-            validationSetId=self.validation_set_id,
-            shouldAcceptIncorrect=self.should_accept_incorrect,
+    def _get_instruction(self) -> str:
+        return ""
+
+    def _to_model(self) -> IOrderWorkflowModel:
+        return IOrderWorkflowModel(
+            actual_instance=IOrderWorkflowModelEvaluationWorkflowModel(
+                _t="EvaluationWorkflow",
+                validationSetId=self.validation_set_id,
+                shouldAcceptIncorrect=self.should_accept_incorrect,
+            )
         )
 
     def _to_payload(self, datapoint: Datapoint):
         raise NotImplementedError("EvaluationWorkflow does not have a payload")
+
+    def __str__(self) -> str:
+        return f"EvaluationWorkflow(validation_set_id='{self.validation_set_id}', should_accept_incorrect={self.should_accept_incorrect})"
+
+    def __repr__(self) -> str:
+        return f"EvaluationWorkflow(validation_set_id={self.validation_set_id!r}, should_accept_incorrect={self.should_accept_incorrect!r})"

rapidata/rapidata_client/workflow/_free_text_workflow.py

@@ -1,13 +1,19 @@
 from typing import Any
-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_rapid_payload import IRapidPayload
 from rapidata.rapidata_client.workflow import Workflow
-from rapidata.api_client.models.free_text_rapid_blueprint import FreeTextRapidBlueprint
-from rapidata.api_client import FreeTextPayload
+from rapidata.api_client.models.i_rapid_payload_free_text_payload import (
+    IRapidPayloadFreeTextPayload,
+)
 from rapidata.rapidata_client.datapoints._datapoint import Datapoint
 from rapidata.api_client.models.rapid_modality import RapidModality
+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_free_text_rapid_blueprint import (
+    IRapidBlueprintFreeTextRapidBlueprint,
+)
 
 
 class FreeTextWorkflow(Workflow):
@@ -33,6 +39,9 @@ class FreeTextWorkflow(Workflow):
         self._instruction = instruction
         self._validation_system_prompt = validation_system_prompt
 
+    def _get_instruction(self) -> str:
+        return self._instruction
+
     def _to_dict(self) -> dict[str, Any]:
         return {
             **super()._to_dict(),
@@ -43,20 +52,30 @@ class FreeTextWorkflow(Workflow):
             },
         }
 
-    def _to_model(self) -> SimpleWorkflowModel:
-        blueprint = FreeTextRapidBlueprint(
+    def _to_model(self) -> IOrderWorkflowModel:
+        blueprint = IRapidBlueprintFreeTextRapidBlueprint(
             _t="FreeTextBlueprint",
             question=self._instruction,
             validationSystemPrompt=self._validation_system_prompt,
         )
 
-        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) -> FreeTextPayload:
-        return FreeTextPayload(
-            _t="FreeTextPayload",
-            question=self._instruction,
+    def _to_payload(self, datapoint: Datapoint) -> IRapidPayload:
+        return IRapidPayload(
+            actual_instance=IRapidPayloadFreeTextPayload(
+                _t="FreeTextPayload",
+                question=self._instruction,
+            )
         )
+
+    def __str__(self) -> str:
+        return f"FreeTextWorkflow(instruction='{self._instruction}', validation_system_prompt={self._validation_system_prompt})"
+
+    def __repr__(self) -> str:
+        return f"FreeTextWorkflow(instruction={self._instruction!r}, validation_system_prompt={self._validation_system_prompt!r})"

rapidata/rapidata_client/workflow/_locate_workflow.py

@@ -1,10 +1,16 @@
-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.locate_rapid_blueprint import LocateRapidBlueprint
+from rapidata.api_client.models.i_rapid_blueprint import IRapidBlueprint
+from rapidata.api_client.models.i_rapid_blueprint_locate_rapid_blueprint import (
+    IRapidBlueprintLocateRapidBlueprint,
+)
+from rapidata.api_client.models.i_rapid_payload import IRapidPayload
 from rapidata.rapidata_client.workflow._base_workflow import Workflow
-from rapidata.api_client import LocatePayload
+from rapidata.api_client.models.i_rapid_payload_locate_payload import (
+    IRapidPayloadLocatePayload,
+)
 from rapidata.rapidata_client.datapoints._datapoint import Datapoint
 from rapidata.api_client.models.rapid_modality import RapidModality
 
@@ -16,15 +22,31 @@ class LocateWorkflow(Workflow):
         super().__init__(type="SimpleWorkflowConfig")
         self._target = target
 
-    def _to_model(self) -> SimpleWorkflowModel:
-        blueprint = LocateRapidBlueprint(_t="LocateBlueprint", target=self._target)
+    def _get_instruction(self) -> str:
+        return self._target
+
+    def _to_model(self) -> IOrderWorkflowModel:
+        blueprint = IRapidBlueprintLocateRapidBlueprint(
+            _t="LocateBlueprint", target=self._target
+        )
 
-        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) -> LocatePayload:
-        return LocatePayload(
-            _t="LocatePayload",
-            target=self._target,
+    def _to_payload(self, datapoint: Datapoint) -> IRapidPayload:
+        return IRapidPayload(
+            actual_instance=IRapidPayloadLocatePayload(
+                _t="LocatePayload",
+                target=self._target,
+            )
         )
+
+    def __str__(self) -> str:
+        return f"LocateWorkflow(target='{self._target}')"
+
+    def __repr__(self) -> str:
+        return f"LocateWorkflow(target={self._target!r})"

rapidata/rapidata_client/workflow/_multi_ranking_workflow.py

@@ -0,0 +1,90 @@
+from rapidata.api_client.models.add_validation_rapid_model import IRapidPayload
+from rapidata.api_client.models.i_order_workflow_model import IOrderWorkflowModel
+from rapidata.api_client.models.i_order_workflow_model_grouped_ranking_workflow_model import (
+    IOrderWorkflowModelGroupedRankingWorkflowModel,
+)
+from rapidata.rapidata_client.workflow._base_workflow import Workflow
+from rapidata.api_client.models.i_rapid_payload_compare_payload import (
+    IRapidPayloadComparePayload,
+)
+from rapidata.rapidata_client.datapoints._datapoint import Datapoint
+from rapidata.api_client.models.rapid_modality import RapidModality
+from rapidata.api_client.models.i_asset_input import IAssetInput
+
+
+class MultiRankingWorkflow(Workflow):
+    modality = RapidModality.COMPARE
+
+    def __init__(
+        self,
+        instruction: str,
+        comparison_budget_per_ranking: int,
+        random_comparisons_ratio,
+        elo_start: int = 1200,
+        elo_k_factor: int = 40,
+        elo_scaling_factor: int = 400,
+        contexts: dict[str, str] | None = None,
+        media_contexts: dict[str, IAssetInput] | None = None,
+    ):
+        from rapidata.api_client.models.i_pair_maker_config_model import (
+            IPairMakerConfigModel,
+        )
+        from rapidata.api_client.models.i_pair_maker_config_model_online_pair_maker_config_model import (
+            IPairMakerConfigModelOnlinePairMakerConfigModel,
+        )
+        from rapidata.api_client.models.elo_config_model import EloConfigModel
+
+        super().__init__(type="CompareWorkflowConfig")
+
+        self.contexts = contexts
+        self.media_contexts = media_contexts
+
+        self.instruction = instruction
+        self.comparison_budget_per_ranking = comparison_budget_per_ranking
+        self.random_comparisons_ratio = random_comparisons_ratio
+        self.elo_start = elo_start
+        self.elo_k_factor = elo_k_factor
+        self.elo_scaling_factor = elo_scaling_factor
+
+        self.pair_maker_config = IPairMakerConfigModel(
+            actual_instance=IPairMakerConfigModelOnlinePairMakerConfigModel(
+                _t="OnlinePairMaker",
+                totalComparisonBudget=comparison_budget_per_ranking,
+                randomMatchesRatio=random_comparisons_ratio,
+            ),
+        )
+
+        self.elo_config = EloConfigModel(
+            startingElo=elo_start,
+            kFactor=elo_k_factor,
+            scalingFactor=elo_scaling_factor,
+        )
+
+    def _to_model(self) -> IOrderWorkflowModel:
+        return IOrderWorkflowModel(
+            actual_instance=IOrderWorkflowModelGroupedRankingWorkflowModel(
+                _t="GroupedRankingWorkflow",
+                criteria=self.instruction,
+                eloConfig=self.elo_config,
+                pairMakerConfig=self.pair_maker_config,
+                contexts=self.contexts,
+                contextAssets=self.media_contexts,
+            )
+        )
+
+    def _get_instruction(self) -> str:
+        return self.instruction
+
+    def _to_payload(self, datapoint: Datapoint) -> IRapidPayload:
+        return IRapidPayload(
+            actual_instance=IRapidPayloadComparePayload(
+                _t="ComparePayload",
+                criteria=self.instruction,
+            )
+        )
+
+    def __str__(self) -> str:
+        return f"MultiRankingWorkflow(instruction='{self.instruction}')"
+
+    def __repr__(self) -> str:
+        return f"MultiRankingWorkflow(instruction={self.instruction!r}, comparison_budget_per_ranking={self.comparison_budget_per_ranking!r}, random_comparisons_ratio={self.random_comparisons_ratio!r}, elo_start={self.elo_start!r}, elo_k_factor={self.elo_k_factor!r}, elo_scaling_factor={self.elo_scaling_factor!r})"

rapidata/rapidata_client/workflow/_select_words_workflow.py

@@ -1,17 +1,19 @@
-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.transcription_rapid_blueprint import (
-    TranscriptionRapidBlueprint,
+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 import TranscriptionPayload, TranscriptionWord
-from rapidata.rapidata_client.datapoints._datapoint import Datapoint
-from rapidata.rapidata_client.datapoints.metadata._select_words_metadata import (
-    SelectWordsMetadata,
+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):
@@ -34,35 +36,39 @@ class SelectWordsWorkflow(Workflow):
         super().__init__(type="SimpleWorkflowConfig")
         self._instruction = instruction
 
-    def _to_model(self) -> SimpleWorkflowModel:
-        blueprint = TranscriptionRapidBlueprint(
+    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) -> TranscriptionPayload:
+    def _to_payload(self, datapoint: Datapoint) -> IRapidPayload:
         assert (
-            datapoint.metadata is not None
-        ), "SelectWordsWorkflow requires a metadata datapoint"
+            datapoint.sentence is not None
+        ), "SelectWordsWorkflow requires a sentence datapoint"
 
-        assert any(
-            isinstance(metadata, SelectWordsMetadata) for metadata in datapoint.metadata
-        ), "SelectWordsWorkflow requires a SelectWordsMetadata datapoint"
-
-        select_words_metadata = next(
-            metadata
-            for metadata in datapoint.metadata
-            if isinstance(metadata, SelectWordsMetadata)
+        return IRapidPayload(
+            actual_instance=IRapidPayloadTranscriptionPayload(
+                _t="TranscriptionPayload",
+                title=self._instruction,
+                transcription=[
+                    TranscriptionWord(word=word, wordIndex=i)
+                    for i, word in enumerate(datapoint.sentence.split())
+                ],
+            )
         )
 
-        return TranscriptionPayload(
-            _t="TranscriptionPayload",
-            title=self._instruction,
-            transcription=[
-                TranscriptionWord(word=word, wordIndex=i)
-                for i, word in enumerate(select_words_metadata.select_words.split())
-            ],
-        )
+    def __str__(self) -> str:
+        return f"SelectWordsWorkflow(instruction='{self._instruction}')"
+
+    def __repr__(self) -> str:
+        return f"SelectWordsWorkflow(instruction={self._instruction!r})"