label-studio-sdk 1.0.7__py3-none-any.whl → 1.0.10__py3-none-any.whl

label_studio_sdk/types/inference_run.py

@@ -1,20 +1,20 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-from .inference_run_created_by import InferenceRunCreatedBy
 from .inference_run_organization import InferenceRunOrganization
+from .inference_run_created_by import InferenceRunCreatedBy
 from .inference_run_project_subset import InferenceRunProjectSubset
 from .inference_run_status import InferenceRunStatus
+import datetime as dt
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class InferenceRun(pydantic_v1.BaseModel):
+class InferenceRun(UniversalBaseModel):
     organization: typing.Optional[InferenceRunOrganization] = None
     project: int
-    model_version: typing.Optional[str] = None
+    model_version: typing.Optional[int] = None
     created_by: typing.Optional[InferenceRunCreatedBy] = None
     project_subset: InferenceRunProjectSubset
     status: typing.Optional[InferenceRunStatus] = None
@@ -24,20 +24,11 @@ class InferenceRun(pydantic_v1.BaseModel):
     predictions_updated_at: typing.Optional[dt.datetime] = None
     completed_at: typing.Optional[dt.datetime] = None
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

label_studio_sdk/types/inference_run_cost_estimate.py

@@ -0,0 +1,47 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
+
+class InferenceRunCostEstimate(UniversalBaseModel):
+    prompt_cost_usd: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Cost of the prompt (in USD)
+    """
+
+    completion_cost_usd: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Cost of the completion (in USD)
+    """
+
+    total_cost_usd: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Total cost of the inference (in USD)
+    """
+
+    is_error: typing.Optional[bool] = pydantic.Field(default=None)
+    """
+    Whether an error occurred or not
+    """
+
+    error_type: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Type of error (e.g. "Timeout", "Rate Limit", etc)
+    """
+
+    error_message: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Error message details
+    """
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

label_studio_sdk/types/inference_run_created_by.py

@@ -2,4 +2,4 @@
 
 import typing
 
-InferenceRunCreatedBy = typing.Union[int, typing.Dict[str, typing.Any]]
+InferenceRunCreatedBy = typing.Union[int, typing.Dict[str, typing.Optional[typing.Any]]]

label_studio_sdk/types/inference_run_organization.py

@@ -2,4 +2,4 @@
 
 import typing
 
-InferenceRunOrganization = typing.Union[int, typing.Dict[str, typing.Any]]
+InferenceRunOrganization = typing.Union[int, typing.Dict[str, typing.Optional[typing.Any]]]

label_studio_sdk/types/key_indicator_value.py

@@ -1,30 +1,20 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
 import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 
-
-class KeyIndicatorValue(pydantic_v1.BaseModel):
+class KeyIndicatorValue(UniversalBaseModel):
     title: typing.Optional[str] = None
-    values: typing.Optional[typing.Dict[str, typing.Any]] = None
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
+    values: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
 
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

label_studio_sdk/types/key_indicators.py

@@ -1,7 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-
 from .key_indicators_item import KeyIndicatorsItem
 
 KeyIndicators = typing.List[KeyIndicatorsItem]

label_studio_sdk/types/key_indicators_item.py

@@ -1,51 +1,41 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
+import pydantic
 import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 from .key_indicators_item_additional_kpis_item import KeyIndicatorsItemAdditionalKpisItem
 from .key_indicators_item_extra_kpis_item import KeyIndicatorsItemExtraKpisItem
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class KeyIndicatorsItem(pydantic_v1.BaseModel):
-    key: str = pydantic_v1.Field()
+class KeyIndicatorsItem(UniversalBaseModel):
+    key: str = pydantic.Field()
     """
     The key for this KPI, where you can find the value from inside main_kpi
     """
 
-    title: str = pydantic_v1.Field()
+    title: str = pydantic.Field()
     """
     The title for this metric, to be displayed to the user
     """
 
     main_kpi: str
     secondary_kpi: typing.Optional[str] = None
-    additional_kpis: typing.Optional[typing.List[KeyIndicatorsItemAdditionalKpisItem]] = pydantic_v1.Field(default=None)
+    additional_kpis: typing.Optional[typing.List[KeyIndicatorsItemAdditionalKpisItem]] = pydantic.Field(default=None)
     """
     Additional KPIs to be displayed at the bottom of the box
     """
 
-    extra_kpis: typing.Optional[typing.List[KeyIndicatorsItemExtraKpisItem]] = pydantic_v1.Field(default=None)
+    extra_kpis: typing.Optional[typing.List[KeyIndicatorsItemExtraKpisItem]] = pydantic.Field(default=None)
     """
     Extra KPIs to be displayed in the hover-tootip for that indicator
     """
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

label_studio_sdk/types/key_indicators_item_additional_kpis_item.py

@@ -1,37 +1,27 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
 import typing
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 
-
-class KeyIndicatorsItemAdditionalKpisItem(pydantic_v1.BaseModel):
-    key: typing.Optional[str] = pydantic_v1.Field(default=None)
+class KeyIndicatorsItemAdditionalKpisItem(UniversalBaseModel):
+    key: typing.Optional[str] = pydantic.Field(default=None)
     """
     The key for this KPI, where you can find the value from inside main_kpi
     """
 
-    label: typing.Optional[str] = pydantic_v1.Field(default=None)
+    label: typing.Optional[str] = pydantic.Field(default=None)
     """
     The label for this KPI, to be displayed to the user
     """
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

label_studio_sdk/types/key_indicators_item_extra_kpis_item.py

@@ -1,37 +1,27 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
 import typing
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 
-
-class KeyIndicatorsItemExtraKpisItem(pydantic_v1.BaseModel):
-    key: typing.Optional[str] = pydantic_v1.Field(default=None)
+class KeyIndicatorsItemExtraKpisItem(UniversalBaseModel):
+    key: typing.Optional[str] = pydantic.Field(default=None)
     """
     The key for this KPI, where you can find the value from inside main_kpi
     """
 
-    label: typing.Optional[str] = pydantic_v1.Field(default=None)
+    label: typing.Optional[str] = pydantic.Field(default=None)
     """
     The label for this KPI, to be displayed to the user
     """
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

label_studio_sdk/types/local_files_export_storage.py

@@ -1,97 +1,88 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+import pydantic
+import datetime as dt
 from .local_files_export_storage_status import LocalFilesExportStorageStatus
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class LocalFilesExportStorage(pydantic_v1.BaseModel):
+class LocalFilesExportStorage(UniversalBaseModel):
     id: typing.Optional[int] = None
     type: typing.Optional[str] = None
     synchronizable: typing.Optional[bool] = None
-    path: typing.Optional[str] = pydantic_v1.Field(default=None)
+    path: typing.Optional[str] = pydantic.Field(default=None)
     """
     Local path
     """
 
-    regex_filter: typing.Optional[str] = pydantic_v1.Field(default=None)
+    regex_filter: typing.Optional[str] = pydantic.Field(default=None)
     """
     Regex for filtering objects
     """
 
-    use_blob_urls: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    use_blob_urls: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Interpret objects as BLOBs and generate URLs
     """
 
-    last_sync: typing.Optional[dt.datetime] = pydantic_v1.Field(default=None)
+    last_sync: typing.Optional[dt.datetime] = pydantic.Field(default=None)
     """
     Last sync finished time
     """
 
-    last_sync_count: typing.Optional[int] = pydantic_v1.Field(default=None)
+    last_sync_count: typing.Optional[int] = pydantic.Field(default=None)
     """
     Count of tasks synced last time
     """
 
-    last_sync_job: typing.Optional[str] = pydantic_v1.Field(default=None)
+    last_sync_job: typing.Optional[str] = pydantic.Field(default=None)
     """
     Last sync job ID
     """
 
     status: typing.Optional[LocalFilesExportStorageStatus] = None
-    traceback: typing.Optional[str] = pydantic_v1.Field(default=None)
+    traceback: typing.Optional[str] = pydantic.Field(default=None)
     """
     Traceback report for the last failed sync
     """
 
-    meta: typing.Optional[typing.Dict[str, typing.Any]] = pydantic_v1.Field(default=None)
+    meta: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
     """
     Meta and debug information about storage processes
     """
 
-    title: typing.Optional[str] = pydantic_v1.Field(default=None)
+    title: typing.Optional[str] = pydantic.Field(default=None)
     """
     Cloud storage title
     """
 
-    description: typing.Optional[str] = pydantic_v1.Field(default=None)
+    description: typing.Optional[str] = pydantic.Field(default=None)
     """
     Cloud storage description
     """
 
-    created_at: typing.Optional[dt.datetime] = pydantic_v1.Field(default=None)
+    created_at: typing.Optional[dt.datetime] = pydantic.Field(default=None)
     """
     Creation time
     """
 
-    can_delete_objects: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    can_delete_objects: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Deletion from storage enabled
     """
 
-    project: int = pydantic_v1.Field()
+    project: int = pydantic.Field()
     """
     A unique integer value identifying this project.
     """
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

label_studio_sdk/types/local_files_import_storage.py

@@ -1,92 +1,83 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+import pydantic
+import datetime as dt
 from .local_files_import_storage_status import LocalFilesImportStorageStatus
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class LocalFilesImportStorage(pydantic_v1.BaseModel):
+class LocalFilesImportStorage(UniversalBaseModel):
     id: typing.Optional[int] = None
     type: typing.Optional[str] = None
     synchronizable: typing.Optional[bool] = None
-    path: typing.Optional[str] = pydantic_v1.Field(default=None)
+    path: typing.Optional[str] = pydantic.Field(default=None)
     """
     Local path
     """
 
-    regex_filter: typing.Optional[str] = pydantic_v1.Field(default=None)
+    regex_filter: typing.Optional[str] = pydantic.Field(default=None)
     """
     Regex for filtering objects
     """
 
-    use_blob_urls: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    use_blob_urls: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Interpret objects as BLOBs and generate URLs
     """
 
-    last_sync: typing.Optional[dt.datetime] = pydantic_v1.Field(default=None)
+    last_sync: typing.Optional[dt.datetime] = pydantic.Field(default=None)
     """
     Last sync finished time
     """
 
-    last_sync_count: typing.Optional[int] = pydantic_v1.Field(default=None)
+    last_sync_count: typing.Optional[int] = pydantic.Field(default=None)
     """
     Count of tasks synced last time
     """
 
-    last_sync_job: typing.Optional[str] = pydantic_v1.Field(default=None)
+    last_sync_job: typing.Optional[str] = pydantic.Field(default=None)
     """
     Last sync job ID
     """
 
     status: typing.Optional[LocalFilesImportStorageStatus] = None
-    traceback: typing.Optional[str] = pydantic_v1.Field(default=None)
+    traceback: typing.Optional[str] = pydantic.Field(default=None)
     """
     Traceback report for the last failed sync
     """
 
-    meta: typing.Optional[typing.Dict[str, typing.Any]] = pydantic_v1.Field(default=None)
+    meta: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
     """
     Meta and debug information about storage processes
     """
 
-    title: typing.Optional[str] = pydantic_v1.Field(default=None)
+    title: typing.Optional[str] = pydantic.Field(default=None)
     """
     Cloud storage title
     """
 
-    description: typing.Optional[str] = pydantic_v1.Field(default=None)
+    description: typing.Optional[str] = pydantic.Field(default=None)
     """
     Cloud storage description
     """
 
-    created_at: typing.Optional[dt.datetime] = pydantic_v1.Field(default=None)
+    created_at: typing.Optional[dt.datetime] = pydantic.Field(default=None)
     """
     Creation time
     """
 
-    project: int = pydantic_v1.Field()
+    project: int = pydantic.Field()
     """
     A unique integer value identifying this project.
     """
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
 
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow