label-studio-sdk 1.0.8__py3-none-any.whl → 1.0.11__py3-none-any.whl

label_studio_sdk/types/refined_prompt_response.py

@@ -1,64 +1,54 @@
 # 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 .prompt_version import PromptVersion
+import pydantic
 from .refined_prompt_response_refinement_status import RefinedPromptResponseRefinementStatus
+from .prompt_version import PromptVersion
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class RefinedPromptResponse(pydantic_v1.BaseModel):
-    title: typing.Optional[str] = pydantic_v1.Field(default=None)
+class RefinedPromptResponse(UniversalBaseModel):
+    title: typing.Optional[str] = pydantic.Field(default=None)
     """
     Title of the refined prompt
     """
 
-    reasoning: typing.Optional[str] = pydantic_v1.Field(default=None)
+    reasoning: typing.Optional[str] = pydantic.Field(default=None)
     """
     Reasoning behind the refinement
     """
 
-    prompt: str = pydantic_v1.Field()
+    prompt: str = pydantic.Field()
     """
     The refined prompt text
     """
 
-    refinement_job_id: typing.Optional[str] = pydantic_v1.Field(default=None)
+    refinement_job_id: typing.Optional[str] = pydantic.Field(default=None)
     """
     Unique identifier for the refinement job
     """
 
-    refinement_status: typing.Optional[RefinedPromptResponseRefinementStatus] = pydantic_v1.Field(default=None)
+    refinement_status: typing.Optional[RefinedPromptResponseRefinementStatus] = pydantic.Field(default=None)
     """
     Status of the refinement job
     """
 
-    total_cost: typing.Optional[str] = pydantic_v1.Field(default=None)
+    total_cost: typing.Optional[str] = pydantic.Field(default=None)
     """
     Total cost of the refinement job (in USD)
     """
 
-    previous_version: typing.Optional[PromptVersion] = pydantic_v1.Field(default=None)
+    previous_version: typing.Optional[PromptVersion] = pydantic.Field(default=None)
     """
     Previous version of the prompt
     """
 
-    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/s3export_storage.py

@@ -1,134 +1,127 @@
 # 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 datetime as dt
+import pydantic
 from .s3export_storage_status import S3ExportStorageStatus
+import typing_extensions
+from ..core.serialization import FieldMetadata
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class S3ExportStorage(pydantic_v1.BaseModel):
+class S3ExportStorage(UniversalBaseModel):
     id: typing.Optional[int] = None
     type: typing.Optional[str] = None
     synchronizable: typing.Optional[bool] = None
-    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[S3ExportStorageStatus] = 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
     """
 
-    bucket: typing.Optional[str] = pydantic_v1.Field(default=None)
+    bucket: typing.Optional[str] = pydantic.Field(default=None)
     """
     S3 bucket name
     """
 
-    prefix: typing.Optional[str] = pydantic_v1.Field(default=None)
+    prefix: typing.Optional[str] = pydantic.Field(default=None)
     """
     S3 bucket prefix
     """
 
-    regex_filter: typing.Optional[str] = pydantic_v1.Field(default=None)
+    regex_filter: typing.Optional[str] = pydantic.Field(default=None)
     """
     Cloud storage 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
     """
 
-    aws_access_key_id: typing.Optional[str] = pydantic_v1.Field(default=None)
+    aws_access_key_id: typing.Optional[str] = pydantic.Field(default=None)
     """
     AWS_ACCESS_KEY_ID
     """
 
-    aws_secret_access_key: typing.Optional[str] = pydantic_v1.Field(default=None)
+    aws_secret_access_key: typing.Optional[str] = pydantic.Field(default=None)
     """
     AWS_SECRET_ACCESS_KEY
     """
 
-    aws_session_token: typing.Optional[str] = pydantic_v1.Field(default=None)
+    aws_session_token: typing.Optional[str] = pydantic.Field(default=None)
     """
     AWS_SESSION_TOKEN
     """
 
-    aws_sse_kms_key_id: typing.Optional[str] = pydantic_v1.Field(default=None)
+    aws_sse_kms_key_id: typing.Optional[str] = pydantic.Field(default=None)
     """
     AWS SSE KMS Key ID
     """
 
-    region_name: typing.Optional[str] = pydantic_v1.Field(default=None)
+    region_name: typing.Optional[str] = pydantic.Field(default=None)
     """
     AWS Region
     """
 
-    s3endpoint: typing.Optional[str] = pydantic_v1.Field(alias="s3_endpoint", default=None)
+    s3endpoint: typing_extensions.Annotated[typing.Optional[str], FieldMetadata(alias="s3_endpoint")] = pydantic.Field(
+        default=None
+    )
     """
     S3 Endpoint
     """
 
-    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
-        allow_population_by_field_name = True
-        populate_by_name = 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/s3import_storage.py

@@ -1,140 +1,133 @@
 # 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 datetime as dt
+import pydantic
 from .s3import_storage_status import S3ImportStorageStatus
+import typing_extensions
+from ..core.serialization import FieldMetadata
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class S3ImportStorage(pydantic_v1.BaseModel):
+class S3ImportStorage(UniversalBaseModel):
     id: typing.Optional[int] = None
     type: typing.Optional[str] = None
     synchronizable: typing.Optional[bool] = None
     presign: typing.Optional[bool] = None
-    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[S3ImportStorageStatus] = 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
     """
 
-    bucket: typing.Optional[str] = pydantic_v1.Field(default=None)
+    bucket: typing.Optional[str] = pydantic.Field(default=None)
     """
     S3 bucket name
     """
 
-    prefix: typing.Optional[str] = pydantic_v1.Field(default=None)
+    prefix: typing.Optional[str] = pydantic.Field(default=None)
     """
     S3 bucket prefix
     """
 
-    regex_filter: typing.Optional[str] = pydantic_v1.Field(default=None)
+    regex_filter: typing.Optional[str] = pydantic.Field(default=None)
     """
     Cloud storage 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
     """
 
-    aws_access_key_id: typing.Optional[str] = pydantic_v1.Field(default=None)
+    aws_access_key_id: typing.Optional[str] = pydantic.Field(default=None)
     """
     AWS_ACCESS_KEY_ID
     """
 
-    aws_secret_access_key: typing.Optional[str] = pydantic_v1.Field(default=None)
+    aws_secret_access_key: typing.Optional[str] = pydantic.Field(default=None)
     """
     AWS_SECRET_ACCESS_KEY
     """
 
-    aws_session_token: typing.Optional[str] = pydantic_v1.Field(default=None)
+    aws_session_token: typing.Optional[str] = pydantic.Field(default=None)
     """
     AWS_SESSION_TOKEN
     """
 
-    aws_sse_kms_key_id: typing.Optional[str] = pydantic_v1.Field(default=None)
+    aws_sse_kms_key_id: typing.Optional[str] = pydantic.Field(default=None)
     """
     AWS SSE KMS Key ID
     """
 
-    region_name: typing.Optional[str] = pydantic_v1.Field(default=None)
+    region_name: typing.Optional[str] = pydantic.Field(default=None)
     """
     AWS Region
     """
 
-    s3endpoint: typing.Optional[str] = pydantic_v1.Field(alias="s3_endpoint", default=None)
+    s3endpoint: typing_extensions.Annotated[typing.Optional[str], FieldMetadata(alias="s3_endpoint")] = pydantic.Field(
+        default=None
+    )
     """
     S3 Endpoint
     """
 
-    presign_ttl: typing.Optional[int] = pydantic_v1.Field(default=None)
+    presign_ttl: typing.Optional[int] = pydantic.Field(default=None)
     """
     Presigned URLs TTL (in minutes)
     """
 
-    recursive_scan: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    recursive_scan: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Perform recursive scan over the bucket content
     """
 
-    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
-        allow_population_by_field_name = True
-        populate_by_name = 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/s3s_export_storage.py

@@ -1,80 +1,73 @@
 # 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
+import typing_extensions
+from ..core.serialization import FieldMetadata
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class S3SExportStorage(pydantic_v1.BaseModel):
+class S3SExportStorage(UniversalBaseModel):
     id: typing.Optional[int] = None
-    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
     """
 
-    bucket: typing.Optional[str] = pydantic_v1.Field(default=None)
+    bucket: typing.Optional[str] = pydantic.Field(default=None)
     """
     S3 bucket name
     """
 
-    prefix: typing.Optional[str] = pydantic_v1.Field(default=None)
+    prefix: typing.Optional[str] = pydantic.Field(default=None)
     """
     S3 bucket prefix
     """
 
-    external_id: typing.Optional[str] = pydantic_v1.Field(default=None)
+    external_id: typing.Optional[str] = pydantic.Field(default=None)
     """
     AWS External ID
     """
 
-    role_arn: typing.Optional[str] = pydantic_v1.Field(default=None)
+    role_arn: typing.Optional[str] = pydantic.Field(default=None)
     """
     AWS Role ARN
     """
 
-    region_name: typing.Optional[str] = pydantic_v1.Field(default=None)
+    region_name: typing.Optional[str] = pydantic.Field(default=None)
     """
     AWS Region
     """
 
-    s3endpoint: typing.Optional[str] = pydantic_v1.Field(alias="s3_endpoint", default=None)
+    s3endpoint: typing_extensions.Annotated[typing.Optional[str], FieldMetadata(alias="s3_endpoint")] = pydantic.Field(
+        default=None
+    )
     """
     S3 Endpoint
     """
 
-    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
-        allow_population_by_field_name = True
-        populate_by_name = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow