rapidata 0.2.1__tar.gz → 0.4.0__tar.gz

{rapidata-0.2.1 → rapidata-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rapidata
-Version: 0.2.1
+Version: 0.4.0
 Summary: Rapidata package containing the Rapidata Python Client to interact with the Rapidata Web API in an easy way.
 License: Apache-2.0
 Author: Rapidata AG

{rapidata-0.2.1 → rapidata-0.4.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "rapidata"
-version = "0.2.1"
+version = "0.4.0"
 description = "Rapidata package containing the Rapidata Python Client to interact with the Rapidata Web API in an easy way."
 authors = ["Rapidata AG <info@rapidata.ai>"]
 license = "Apache-2.0"

rapidata-0.4.0/src/rapidata/__init__.py

@@ -0,0 +1,18 @@
+from .rapidata_client import (
+    RapidataClient, 
+    ClassifyWorkflow,
+    TranscriptionWorkflow,
+    CompareWorkflow,
+    FreeTextWorkflow,
+    DemographicSelection,
+    LabelingSelection,
+    ValidationSelection,
+    NaiveReferee,
+    ClassifyEarlyStoppingReferee,
+    PrivateTextMetadata,
+    PublicTextMetadata,
+    PromptMetadata,
+    TranscriptionMetadata,
+    FeatureFlags,
+    CountryCodes
+    )

rapidata-0.4.0/src/rapidata/rapidata_client/__init__.py

@@ -0,0 +1,7 @@
+from .rapidata_client import RapidataClient
+from .workflow import ClassifyWorkflow, TranscriptionWorkflow, CompareWorkflow, FreeTextWorkflow
+from .selection import DemographicSelection, LabelingSelection, ValidationSelection
+from .referee import NaiveReferee, ClassifyEarlyStoppingReferee
+from .metadata import PrivateTextMetadata, PublicTextMetadata, PromptMetadata, TranscriptionMetadata
+from .feature_flags import FeatureFlags
+from .country_codes import CountryCodes

rapidata-0.4.0/src/rapidata/rapidata_client/country_codes/__init__.py

@@ -0,0 +1 @@
+from .country_codes import CountryCodes

rapidata-0.4.0/src/rapidata/rapidata_client/feature_flags/__init__.py

@@ -0,0 +1 @@
+from .feature_flags import FeatureFlags

rapidata-0.4.0/src/rapidata/rapidata_client/metadata/__init__.py

@@ -0,0 +1,5 @@
+from .base_metadata import Metadata
+from .private_text_metadata import PrivateTextMetadata
+from .public_text_metadata import PublicTextMetadata
+from .prompt_metadata import PromptMetadata
+from .transcription_metadata import TranscriptionMetadata

rapidata-0.4.0/src/rapidata/rapidata_client/metadata/public_text_metadata.py

@@ -0,0 +1,16 @@
+from rapidata.api_client.models.public_text_metadata_input import (
+    PublicTextMetadataInput,
+)
+from rapidata.rapidata_client.metadata.base_metadata import Metadata
+
+
+class PublicTextMetadata(Metadata):
+
+    def __init__(self, text: str, identifier: str = "public_text"):
+        super().__init__(identifier=identifier)
+        self._text = text
+
+    def to_model(self):
+        return PublicTextMetadataInput(
+            _t="PublicTextMetadataInput", identifier=self._identifier, text=self._text
+        )

{rapidata-0.2.1 → rapidata-0.4.0}/src/rapidata/rapidata_client/order/rapidata_order_builder.py

@@ -23,6 +23,8 @@ from rapidata.rapidata_client.order.rapidata_order import RapidataOrder
 from rapidata.rapidata_client.referee import Referee
 from rapidata.service.openapi_service import OpenAPIService
 
+from rapidata.rapidata_client.workflow.compare_workflow import CompareWorkflow
+
 
 class RapidataOrderBuilder:
     """Builder object for creating Rapidata orders.
@@ -51,10 +53,11 @@ class RapidataOrderBuilder:
         self._country_codes: list[str] | None = None
         self._selections: list[Selection] = []
         self._rapids_per_bag: int = 2
+        self._priority: int = 50
 
     def _to_model(self) -> CreateOrderModel:
         if self._workflow is None:
-            raise ValueError("You must provide a blueprint to create an order.")
+            raise ValueError("You must provide a workflow to create an order.")
 
         if self._referee is None:
             print("No referee provided, using default NaiveReferee.")
@@ -85,6 +88,7 @@ class RapidataOrderBuilder:
                 CreateOrderModelSelectionsInner(selection.to_model())
                 for selection in self._selections
             ],
+            priority=self._priority,
         )
 
     def create(self, submit=True, max_workers=10) -> RapidataOrder:
@@ -100,6 +104,8 @@ class RapidataOrderBuilder:
             ValueError: If no workflow is provided.
         """
         order_model = self._to_model()
+        if isinstance(self._workflow, CompareWorkflow): # temp fix, will be handeled by backend in the future
+            assert all([len(path) == 2 for path in self._media_paths]), "The media paths must come in pairs for comparison tasks."
 
         result = self._openapi_service.order_api.order_create_post(
             create_order_model=order_model
@@ -114,7 +120,6 @@ class RapidataOrderBuilder:
         )
 
         order.dataset.add_media_from_paths(self._media_paths, self._metadata, max_workers)
-
         if submit:
             order.submit()
 
@@ -235,3 +240,15 @@ class RapidataOrderBuilder:
         """
         self._selections = selections
         return self
+    
+    def priority(self, priority: int):
+        """Set the priority for the order.
+
+        Args:
+            priority (int): The priority to be set.
+
+        Returns:
+            RapidataOrderBuilder: The updated RapidataOrderBuilder instance.
+        """
+        self._priority = priority
+        return self

{rapidata-0.2.1 → rapidata-0.4.0}/src/rapidata/rapidata_client/rapidata_client.py

@@ -5,6 +5,9 @@ from rapidata.rapidata_client.dataset.validation_set_builder import ValidationSe
 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
+
 
 
 class RapidataClient:
@@ -59,6 +62,25 @@ class RapidataClient:
             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.
+
+        Args:
+            order_id (str): The ID of the order.
+
+        Returns:
+            RapidataOrder: The Order instance.
+        """
+
+        # TODO: check the pipeline for the dataset id - not really necessary atm
+        # order = self.openapi_service.order_api.order_get_by_id_get(order_id)
+        # pipeline = self.openapi_service..pipeline_get_by_id_get(order.pipeline_id)
+        temp_dataset = RapidataDataset("temp", self.openapi_service)
+        return RapidataOrder(
+            dataset=temp_dataset, 
+            order_id=order_id, 
+            openapi_service=self.openapi_service)
 
     @property
     def utils(self) -> Utils:

rapidata-0.4.0/src/rapidata/rapidata_client/referee/__init__.py

@@ -0,0 +1,3 @@
+from .base_referee import Referee
+from .naive_referee import NaiveReferee #as MaxVoteReferee
+from .classify_early_stopping_referee import ClassifyEarlyStoppingReferee

rapidata-0.4.0/src/rapidata/rapidata_client/selection/__init__.py

@@ -0,0 +1,4 @@
+from .base_selection import Selection
+from .demographic_selection import DemographicSelection
+from .labeling_selection import LabelingSelection
+from .validation_selection import ValidationSelection

rapidata-0.4.0/src/rapidata/rapidata_client/workflow/__init__.py

@@ -0,0 +1,5 @@
+from .base_workflow import Workflow 
+from .classify_workflow import ClassifyWorkflow
+from .compare_workflow import CompareWorkflow
+from .free_text_workflow import FreeTextWorkflow
+from .transcription_workflow import TranscriptionWorkflow

rapidata-0.2.1/src/rapidata/__init__.py

@@ -1 +0,0 @@
-from rapidata.rapidata_client import RapidataClient

rapidata-0.2.1/src/rapidata/rapidata_client/__init__.py

@@ -1 +0,0 @@
-from .rapidata_client import RapidataClient

rapidata-0.2.1/src/rapidata/rapidata_client/country_codes/__init__.py

@@ -1 +0,0 @@
-from .country_codes import CountryCodes as CountryCodes

rapidata-0.2.1/src/rapidata/rapidata_client/feature_flags/__init__.py

@@ -1 +0,0 @@
-from .feature_flags import FeatureFlags as FeatureFlags

rapidata-0.2.1/src/rapidata/rapidata_client/referee/__init__.py

@@ -1,3 +0,0 @@
-from .base_referee import Referee as Referee
-from .naive_referee import NaiveReferee as NaiveReferee
-from .classify_early_stopping_referee import ClassifyEarlyStoppingReferee as ClassifyEarlyStoppingReferee

rapidata-0.2.1/src/rapidata/rapidata_client/workflow/__init__.py

@@ -1,4 +0,0 @@
-from .base_workflow import Workflow as Workflow
-from .classify_workflow import ClassifyWorkflow as ClassifyWorkflow
-from .compare_workflow import CompareWorkflow as CompareWorkflow
-from .free_text_workflow import FreeTextWorkflow as FreeTextWorkflow