rapidata 1.3.0__tar.gz → 1.4.0__tar.gz

{rapidata-1.3.0 → rapidata-1.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rapidata
-Version: 1.3.0
+Version: 1.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-1.3.0 → rapidata-1.4.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "rapidata"
-version = "1.3.0"
+version = "1.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-1.3.0 → rapidata-1.4.0}/src/rapidata/api_client/__init__.py

@@ -52,8 +52,6 @@ from rapidata.api_client.models.add_validation_rapid_model_payload import AddVal
 from rapidata.api_client.models.add_validation_rapid_model_truth import AddValidationRapidModelTruth
 from rapidata.api_client.models.add_validation_rapid_result import AddValidationRapidResult
 from rapidata.api_client.models.add_validation_text_rapid_model import AddValidationTextRapidModel
-from rapidata.api_client.models.admin_order_model import AdminOrderModel
-from rapidata.api_client.models.admin_order_model_paged_result import AdminOrderModelPagedResult
 from rapidata.api_client.models.age_group import AgeGroup
 from rapidata.api_client.models.age_user_filter_model import AgeUserFilterModel
 from rapidata.api_client.models.aggregator_type import AggregatorType
@@ -117,8 +115,6 @@ from rapidata.api_client.models.create_simple_pipeline_model import CreateSimple
 from rapidata.api_client.models.create_simple_pipeline_model_artifacts_inner import CreateSimplePipelineModelArtifactsInner
 from rapidata.api_client.models.create_simple_pipeline_model_pipeline_steps_inner import CreateSimplePipelineModelPipelineStepsInner
 from rapidata.api_client.models.create_unsupported_order_model import CreateUnsupportedOrderModel
-from rapidata.api_client.models.customer_order_model import CustomerOrderModel
-from rapidata.api_client.models.customer_order_model_paged_result import CustomerOrderModelPagedResult
 from rapidata.api_client.models.datapoint import Datapoint
 from rapidata.api_client.models.datapoint_asset import DatapointAsset
 from rapidata.api_client.models.datapoint_get_by_id_get200_response import DatapointGetByIdGet200Response
@@ -218,7 +214,7 @@ from rapidata.api_client.models.null_asset_model import NullAssetModel
 from rapidata.api_client.models.null_asset_model1 import NullAssetModel1
 from rapidata.api_client.models.null_asset_model2 import NullAssetModel2
 from rapidata.api_client.models.order_model import OrderModel
-from rapidata.api_client.models.order_query_get200_response import OrderQueryGet200Response
+from rapidata.api_client.models.order_model_paged_result import OrderModelPagedResult
 from rapidata.api_client.models.order_state import OrderState
 from rapidata.api_client.models.original_filename_metadata import OriginalFilenameMetadata
 from rapidata.api_client.models.original_filename_metadata_model import OriginalFilenameMetadataModel
@@ -232,6 +228,7 @@ from rapidata.api_client.models.probabilistic_attach_category_referee_config imp
 from rapidata.api_client.models.prompt_metadata import PromptMetadata
 from rapidata.api_client.models.prompt_metadata_input import PromptMetadataInput
 from rapidata.api_client.models.prompt_metadata_model import PromptMetadataModel
+from rapidata.api_client.models.public_order_model import PublicOrderModel
 from rapidata.api_client.models.public_text_metadata_input import PublicTextMetadataInput
 from rapidata.api_client.models.query_campaigns_model import QueryCampaignsModel
 from rapidata.api_client.models.query_model import QueryModel

{rapidata-1.3.0 → rapidata-1.4.0}/src/rapidata/api_client/api/order_api.py

@@ -30,7 +30,7 @@ from rapidata.api_client.models.feedback_model import FeedbackModel
 from rapidata.api_client.models.get_order_by_id_result import GetOrderByIdResult
 from rapidata.api_client.models.get_order_results_result import GetOrderResultsResult
 from rapidata.api_client.models.get_public_orders_result import GetPublicOrdersResult
-from rapidata.api_client.models.order_query_get200_response import OrderQueryGet200Response
+from rapidata.api_client.models.order_model_paged_result import OrderModelPagedResult
 from rapidata.api_client.models.query_orders_model import QueryOrdersModel
 from rapidata.api_client.models.unlock_order_result import UnlockOrderResult
 from rapidata.api_client.models.update_access_model import UpdateAccessModel
@@ -2972,7 +2972,7 @@ class OrderApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> OrderQueryGet200Response:
+    ) -> OrderModelPagedResult:
         """Queries orders based on a filter, page, and sort criteria.
 
 
@@ -3009,7 +3009,7 @@ class OrderApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "OrderQueryGet200Response",
+            '200': "OrderModelPagedResult",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -3038,7 +3038,7 @@ class OrderApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> ApiResponse[OrderQueryGet200Response]:
+    ) -> ApiResponse[OrderModelPagedResult]:
         """Queries orders based on a filter, page, and sort criteria.
 
 
@@ -3075,7 +3075,7 @@ class OrderApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "OrderQueryGet200Response",
+            '200': "OrderModelPagedResult",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -3141,7 +3141,7 @@ class OrderApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "OrderQueryGet200Response",
+            '200': "OrderModelPagedResult",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -3188,7 +3188,9 @@ class OrderApi:
         if 'Accept' not in _header_params:
             _header_params['Accept'] = self.api_client.select_header_accept(
                 [
-                    'application/json'
+                    'text/plain', 
+                    'application/json', 
+                    'text/json'
                 ]
             )
 

{rapidata-1.3.0 → rapidata-1.4.0}/src/rapidata/api_client/models/__init__.py

@@ -21,8 +21,6 @@ from rapidata.api_client.models.add_validation_rapid_model_payload import AddVal
 from rapidata.api_client.models.add_validation_rapid_model_truth import AddValidationRapidModelTruth
 from rapidata.api_client.models.add_validation_rapid_result import AddValidationRapidResult
 from rapidata.api_client.models.add_validation_text_rapid_model import AddValidationTextRapidModel
-from rapidata.api_client.models.admin_order_model import AdminOrderModel
-from rapidata.api_client.models.admin_order_model_paged_result import AdminOrderModelPagedResult
 from rapidata.api_client.models.age_group import AgeGroup
 from rapidata.api_client.models.age_user_filter_model import AgeUserFilterModel
 from rapidata.api_client.models.aggregator_type import AggregatorType
@@ -86,8 +84,6 @@ from rapidata.api_client.models.create_simple_pipeline_model import CreateSimple
 from rapidata.api_client.models.create_simple_pipeline_model_artifacts_inner import CreateSimplePipelineModelArtifactsInner
 from rapidata.api_client.models.create_simple_pipeline_model_pipeline_steps_inner import CreateSimplePipelineModelPipelineStepsInner
 from rapidata.api_client.models.create_unsupported_order_model import CreateUnsupportedOrderModel
-from rapidata.api_client.models.customer_order_model import CustomerOrderModel
-from rapidata.api_client.models.customer_order_model_paged_result import CustomerOrderModelPagedResult
 from rapidata.api_client.models.datapoint import Datapoint
 from rapidata.api_client.models.datapoint_asset import DatapointAsset
 from rapidata.api_client.models.datapoint_get_by_id_get200_response import DatapointGetByIdGet200Response
@@ -187,7 +183,7 @@ from rapidata.api_client.models.null_asset_model import NullAssetModel
 from rapidata.api_client.models.null_asset_model1 import NullAssetModel1
 from rapidata.api_client.models.null_asset_model2 import NullAssetModel2
 from rapidata.api_client.models.order_model import OrderModel
-from rapidata.api_client.models.order_query_get200_response import OrderQueryGet200Response
+from rapidata.api_client.models.order_model_paged_result import OrderModelPagedResult
 from rapidata.api_client.models.order_state import OrderState
 from rapidata.api_client.models.original_filename_metadata import OriginalFilenameMetadata
 from rapidata.api_client.models.original_filename_metadata_model import OriginalFilenameMetadataModel
@@ -201,6 +197,7 @@ from rapidata.api_client.models.probabilistic_attach_category_referee_config imp
 from rapidata.api_client.models.prompt_metadata import PromptMetadata
 from rapidata.api_client.models.prompt_metadata_input import PromptMetadataInput
 from rapidata.api_client.models.prompt_metadata_model import PromptMetadataModel
+from rapidata.api_client.models.public_order_model import PublicOrderModel
 from rapidata.api_client.models.public_text_metadata_input import PublicTextMetadataInput
 from rapidata.api_client.models.query_campaigns_model import QueryCampaignsModel
 from rapidata.api_client.models.query_model import QueryModel

{rapidata-1.3.0 → rapidata-1.4.0}/src/rapidata/api_client/models/create_client_model.py

@@ -27,7 +27,8 @@ class CreateClientModel(BaseModel):
     The model for creating a new client.
     """ # noqa: E501
     display_name: Optional[StrictStr] = Field(default=None, description="An optional display name for the client.", alias="displayName")
-    __properties: ClassVar[List[str]] = ["displayName"]
+    customer_id: Optional[StrictStr] = Field(default=None, alias="customerId")
+    __properties: ClassVar[List[str]] = ["displayName", "customerId"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -73,6 +74,11 @@ class CreateClientModel(BaseModel):
         if self.display_name is None and "display_name" in self.model_fields_set:
             _dict['displayName'] = None
 
+        # set to None if customer_id (nullable) is None
+        # and model_fields_set contains the field
+        if self.customer_id is None and "customer_id" in self.model_fields_set:
+            _dict['customerId'] = None
+
         return _dict
 
     @classmethod
@@ -85,7 +91,8 @@ class CreateClientModel(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
-            "displayName": obj.get("displayName")
+            "displayName": obj.get("displayName"),
+            "customerId": obj.get("customerId")
         })
         return _obj
 

{rapidata-1.3.0 → rapidata-1.4.0}/src/rapidata/api_client/models/create_order_result.py

@@ -26,7 +26,7 @@ class CreateOrderResult(BaseModel):
     """
     CreateOrderResult
     """ # noqa: E501
-    dataset_id: StrictStr = Field(alias="datasetId")
+    dataset_id: Optional[StrictStr] = Field(default=None, alias="datasetId")
     order_id: StrictStr = Field(alias="orderId")
     campaign_id: Optional[StrictStr] = Field(default=None, alias="campaignId")
     __properties: ClassVar[List[str]] = ["datasetId", "orderId", "campaignId"]
@@ -70,6 +70,11 @@ class CreateOrderResult(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
+        # set to None if dataset_id (nullable) is None
+        # and model_fields_set contains the field
+        if self.dataset_id is None and "dataset_id" in self.model_fields_set:
+            _dict['datasetId'] = None
+
         # set to None if campaign_id (nullable) is None
         # and model_fields_set contains the field
         if self.campaign_id is None and "campaign_id" in self.model_fields_set:

{rapidata-1.3.0 → rapidata-1.4.0}/src/rapidata/api_client/models/get_public_orders_result.py

@@ -19,7 +19,7 @@ import json
 
 from pydantic import BaseModel, ConfigDict
 from typing import Any, ClassVar, Dict, List
-from rapidata.api_client.models.order_model import OrderModel
+from rapidata.api_client.models.public_order_model import PublicOrderModel
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -27,7 +27,7 @@ class GetPublicOrdersResult(BaseModel):
     """
     GetPublicOrdersResult
     """ # noqa: E501
-    orders: List[OrderModel]
+    orders: List[PublicOrderModel]
     __properties: ClassVar[List[str]] = ["orders"]
 
     model_config = ConfigDict(
@@ -88,7 +88,7 @@ class GetPublicOrdersResult(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
-            "orders": [OrderModel.from_dict(_item) for _item in obj["orders"]] if obj.get("orders") is not None else None
+            "orders": [PublicOrderModel.from_dict(_item) for _item in obj["orders"]] if obj.get("orders") is not None else None
         })
         return _obj
 

rapidata-1.4.0/src/rapidata/api_client/models/order_model.py

@@ -0,0 +1,110 @@
+# 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 datetime import datetime
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr, field_validator
+from typing import Any, ClassVar, Dict, List, Optional
+from typing import Optional, Set
+from typing_extensions import Self
+
+class OrderModel(BaseModel):
+    """
+    OrderModel
+    """ # noqa: E501
+    id: StrictStr
+    order_date: Optional[datetime] = Field(alias="orderDate")
+    customer_mail: StrictStr = Field(alias="customerMail")
+    state: StrictStr
+    order_name: StrictStr = Field(alias="orderName")
+    is_public: StrictBool = Field(alias="isPublic")
+    __properties: ClassVar[List[str]] = ["id", "orderDate", "customerMail", "state", "orderName", "isPublic"]
+
+    @field_validator('state')
+    def state_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['Created', 'Submitted', 'ManualReview', 'Processing', 'Paused', 'Completed', 'Cancelled', 'Failed']):
+            raise ValueError("must be one of enum values ('Created', 'Submitted', 'ManualReview', 'Processing', 'Paused', 'Completed', 'Cancelled', 'Failed')")
+        return value
+
+    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 OrderModel 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.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # set to None if order_date (nullable) is None
+        # and model_fields_set contains the field
+        if self.order_date is None and "order_date" in self.model_fields_set:
+            _dict['orderDate'] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of OrderModel from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "id": obj.get("id"),
+            "orderDate": obj.get("orderDate"),
+            "customerMail": obj.get("customerMail"),
+            "state": obj.get("state"),
+            "orderName": obj.get("orderName"),
+            "isPublic": obj.get("isPublic")
+        })
+        return _obj
+
+

rapidata-1.4.0/src/rapidata/api_client/models/order_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.order_model import OrderModel
+from typing import Optional, Set
+from typing_extensions import Self
+
+class OrderModelPagedResult(BaseModel):
+    """
+    OrderModelPagedResult
+    """ # noqa: E501
+    total: StrictInt
+    page: StrictInt
+    page_size: StrictInt = Field(alias="pageSize")
+    items: List[OrderModel]
+    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 OrderModelPagedResult 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 OrderModelPagedResult 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": [OrderModel.from_dict(_item) for _item in obj["items"]] if obj.get("items") is not None else None,
+            "totalPages": obj.get("totalPages")
+        })
+        return _obj
+
+

rapidata-1.3.0/src/rapidata/api_client/models/order_model.py → rapidata-1.4.0/src/rapidata/api_client/models/public_order_model.py

@@ -22,9 +22,9 @@ from typing import Any, ClassVar, Dict, List
 from typing import Optional, Set
 from typing_extensions import Self
 
-class OrderModel(BaseModel):
+class PublicOrderModel(BaseModel):
     """
-    OrderModel
+    PublicOrderModel
     """ # noqa: E501
     id: StrictStr
     name: StrictStr
@@ -49,7 +49,7 @@ class OrderModel(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of OrderModel from a JSON string"""
+        """Create an instance of PublicOrderModel from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -74,7 +74,7 @@ class OrderModel(BaseModel):
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of OrderModel from a dict"""
+        """Create an instance of PublicOrderModel from a dict"""
         if obj is None:
             return None
 

{rapidata-1.3.0 → rapidata-1.4.0}/src/rapidata/api_client_README.md

@@ -159,8 +159,6 @@ Class | Method | HTTP request | Description
  - [AddValidationRapidModelTruth](rapidata/api_client/docs/AddValidationRapidModelTruth.md)
  - [AddValidationRapidResult](rapidata/api_client/docs/AddValidationRapidResult.md)
  - [AddValidationTextRapidModel](rapidata/api_client/docs/AddValidationTextRapidModel.md)
- - [AdminOrderModel](rapidata/api_client/docs/AdminOrderModel.md)
- - [AdminOrderModelPagedResult](rapidata/api_client/docs/AdminOrderModelPagedResult.md)
  - [AgeGroup](rapidata/api_client/docs/AgeGroup.md)
  - [AgeUserFilterModel](rapidata/api_client/docs/AgeUserFilterModel.md)
  - [AggregatorType](rapidata/api_client/docs/AggregatorType.md)
@@ -224,8 +222,6 @@ Class | Method | HTTP request | Description
  - [CreateSimplePipelineModelArtifactsInner](rapidata/api_client/docs/CreateSimplePipelineModelArtifactsInner.md)
  - [CreateSimplePipelineModelPipelineStepsInner](rapidata/api_client/docs/CreateSimplePipelineModelPipelineStepsInner.md)
  - [CreateUnsupportedOrderModel](rapidata/api_client/docs/CreateUnsupportedOrderModel.md)
- - [CustomerOrderModel](rapidata/api_client/docs/CustomerOrderModel.md)
- - [CustomerOrderModelPagedResult](rapidata/api_client/docs/CustomerOrderModelPagedResult.md)
  - [Datapoint](rapidata/api_client/docs/Datapoint.md)
  - [DatapointAsset](rapidata/api_client/docs/DatapointAsset.md)
  - [DatapointGetByIdGet200Response](rapidata/api_client/docs/DatapointGetByIdGet200Response.md)
@@ -325,7 +321,7 @@ Class | Method | HTTP request | Description
  - [NullAssetModel1](rapidata/api_client/docs/NullAssetModel1.md)
  - [NullAssetModel2](rapidata/api_client/docs/NullAssetModel2.md)
  - [OrderModel](rapidata/api_client/docs/OrderModel.md)
- - [OrderQueryGet200Response](rapidata/api_client/docs/OrderQueryGet200Response.md)
+ - [OrderModelPagedResult](rapidata/api_client/docs/OrderModelPagedResult.md)
  - [OrderState](rapidata/api_client/docs/OrderState.md)
  - [OriginalFilenameMetadata](rapidata/api_client/docs/OriginalFilenameMetadata.md)
  - [OriginalFilenameMetadataModel](rapidata/api_client/docs/OriginalFilenameMetadataModel.md)
@@ -339,6 +335,7 @@ Class | Method | HTTP request | Description
  - [PromptMetadata](rapidata/api_client/docs/PromptMetadata.md)
  - [PromptMetadataInput](rapidata/api_client/docs/PromptMetadataInput.md)
  - [PromptMetadataModel](rapidata/api_client/docs/PromptMetadataModel.md)
+ - [PublicOrderModel](rapidata/api_client/docs/PublicOrderModel.md)
  - [PublicTextMetadataInput](rapidata/api_client/docs/PublicTextMetadataInput.md)
  - [QueryCampaignsModel](rapidata/api_client/docs/QueryCampaignsModel.md)
  - [QueryModel](rapidata/api_client/docs/QueryModel.md)

{rapidata-1.3.0 → rapidata-1.4.0}/src/rapidata/rapidata_client/feature_flags/feature_flags.py

@@ -1,4 +1,10 @@
 from rapidata.api_client.models.feature_flag_model import FeatureFlagModel
+from enum import StrEnum
+
+class TranslationBehaviour(StrEnum):
+    BOTH = "both"
+    ONLY_ORIGINAL = "only original"
+    ONLY_TRANSLATED = "only translated"
 
 
 class FeatureFlags:
@@ -49,16 +55,22 @@ class FeatureFlags:
         self._flags["alert_on_fast_response"] = str(milliseconds)
         return self
 
-    def disable_translation(self, value: bool = True):
-        """Disable automatic translation of all texts in the UI.
+
+    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:
-            value (bool, optional): Whether to disable translation. Defaults to True.
+            behaviour (TranslationBehaviour): The translation behaviour. Defaults to TranslationBehaviour.BOTH.
 
         Returns:
             FeatureFlags: The current FeatureFlags instance for method chaining.
         """
-        self._flags["disable_translation"] = str(value)
+        self._flags["translation_behaviour"] = behaviour
         return self
 
     def free_text_minimum_characters(self, value: int):
@@ -108,4 +120,4 @@ class FeatureFlags:
             FeatureFlags: The current FeatureFlags instance for method chaining.
         """
         self._flags[key] = value
-        return self
+        return self

{rapidata-1.3.0 → rapidata-1.4.0}/src/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 cast
+from typing import Optional, cast
 from rapidata.api_client.models.workflow_artifact_model import WorkflowArtifactModel
 from tqdm import tqdm
 
@@ -22,12 +22,14 @@ class RapidataOrder:
     def __init__(
         self,
         order_id: str,
-        dataset: RapidataDataset,
+        dataset: Optional[RapidataDataset],
         openapi_service: OpenAPIService,
+        name: str,
     ):
         self.openapi_service = openapi_service
         self.order_id = order_id
         self._dataset = dataset
+        self.name = name
         self._workflow_id = None
 
     def submit(self):
@@ -67,16 +69,16 @@ class RapidataOrder:
                 if current_completed > completed_rapids:
                     pbar.update(current_completed - completed_rapids)
                     completed_rapids = current_completed
-                
+
                 if completed_rapids >= total_rapids:
                     break
-                
+
                 time.sleep(refresh_rate)
 
     def _get_workflow_id(self):
         if self._workflow_id:
             return self._workflow_id
-        
+
         for _ in range(2):
             try:
                 order_result = self.openapi_service.order_api.order_get_by_id_get(self.order_id)
@@ -88,7 +90,7 @@ class RapidataOrder:
         if not self._workflow_id:
             raise Exception("Order has not started yet. Please wait for a few seconds and try again.")
         return self._workflow_id
-    
+
     def _get_total_rapids(self):
         workflow_id = self._get_workflow_id()
         return self.openapi_service.workflow_api.workflow_get_progress_get(workflow_id).total