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

label_studio_sdk/ml/types/ml_create_response.py

@@ -1,78 +1,68 @@
 # 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
 from .ml_create_response_auth_method import MlCreateResponseAuthMethod
+from ...core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class MlCreateResponse(pydantic_v1.BaseModel):
-    url: typing.Optional[str] = pydantic_v1.Field(default=None)
+class MlCreateResponse(UniversalBaseModel):
+    url: typing.Optional[str] = pydantic.Field(default=None)
     """
     ML backend URL
     """
 
-    project: typing.Optional[int] = pydantic_v1.Field(default=None)
+    project: typing.Optional[int] = pydantic.Field(default=None)
     """
     Project ID
     """
 
-    is_interactive: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    is_interactive: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Is interactive
     """
 
-    title: typing.Optional[str] = pydantic_v1.Field(default=None)
+    title: typing.Optional[str] = pydantic.Field(default=None)
     """
     Title
     """
 
-    description: typing.Optional[str] = pydantic_v1.Field(default=None)
+    description: typing.Optional[str] = pydantic.Field(default=None)
     """
     Description
     """
 
-    auth_method: typing.Optional[MlCreateResponseAuthMethod] = pydantic_v1.Field(default=None)
+    auth_method: typing.Optional[MlCreateResponseAuthMethod] = pydantic.Field(default=None)
     """
     Auth method
     """
 
-    basic_auth_user: typing.Optional[str] = pydantic_v1.Field(default=None)
+    basic_auth_user: typing.Optional[str] = pydantic.Field(default=None)
     """
     Basic auth user
     """
 
-    basic_auth_pass: typing.Optional[str] = pydantic_v1.Field(default=None)
+    basic_auth_pass: typing.Optional[str] = pydantic.Field(default=None)
     """
     Basic auth password
     """
 
-    extra_params: typing.Optional[typing.Dict[str, typing.Any]] = pydantic_v1.Field(default=None)
+    extra_params: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
     """
     Extra parameters
     """
 
-    timeout: typing.Optional[int] = pydantic_v1.Field(default=None)
+    timeout: typing.Optional[int] = pydantic.Field(default=None)
     """
     Response model timeout
     """
 
-    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/ml/types/ml_update_response.py

@@ -1,78 +1,68 @@
 # 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
 from .ml_update_response_auth_method import MlUpdateResponseAuthMethod
+from ...core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class MlUpdateResponse(pydantic_v1.BaseModel):
-    url: typing.Optional[str] = pydantic_v1.Field(default=None)
+class MlUpdateResponse(UniversalBaseModel):
+    url: typing.Optional[str] = pydantic.Field(default=None)
     """
     ML backend URL
     """
 
-    project: typing.Optional[int] = pydantic_v1.Field(default=None)
+    project: typing.Optional[int] = pydantic.Field(default=None)
     """
     Project ID
     """
 
-    is_interactive: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    is_interactive: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Is interactive
     """
 
-    title: typing.Optional[str] = pydantic_v1.Field(default=None)
+    title: typing.Optional[str] = pydantic.Field(default=None)
     """
     Title
     """
 
-    description: typing.Optional[str] = pydantic_v1.Field(default=None)
+    description: typing.Optional[str] = pydantic.Field(default=None)
     """
     Description
     """
 
-    auth_method: typing.Optional[MlUpdateResponseAuthMethod] = pydantic_v1.Field(default=None)
+    auth_method: typing.Optional[MlUpdateResponseAuthMethod] = pydantic.Field(default=None)
     """
     Auth method
     """
 
-    basic_auth_user: typing.Optional[str] = pydantic_v1.Field(default=None)
+    basic_auth_user: typing.Optional[str] = pydantic.Field(default=None)
     """
     Basic auth user
     """
 
-    basic_auth_pass: typing.Optional[str] = pydantic_v1.Field(default=None)
+    basic_auth_pass: typing.Optional[str] = pydantic.Field(default=None)
     """
     Basic auth password
     """
 
-    extra_params: typing.Optional[typing.Dict[str, typing.Any]] = pydantic_v1.Field(default=None)
+    extra_params: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
     """
     Extra parameters
     """
 
-    timeout: typing.Optional[int] = pydantic_v1.Field(default=None)
+    timeout: typing.Optional[int] = pydantic.Field(default=None)
     """
     Response model timeout
     """
 
-    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/model_providers/client.py

@@ -1,20 +1,21 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
 import typing
-from json.decoder import JSONDecodeError
-
-from ..core.api_error import ApiError
-from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
-from ..core.jsonable_encoder import jsonable_encoder
-from ..core.pydantic_utilities import pydantic_v1
+from ..core.client_wrapper import SyncClientWrapper
 from ..core.request_options import RequestOptions
 from ..types.model_provider_connection import ModelProviderConnection
-from ..types.model_provider_connection_budget_reset_period import ModelProviderConnectionBudgetResetPeriod
-from ..types.model_provider_connection_created_by import ModelProviderConnectionCreatedBy
-from ..types.model_provider_connection_organization import ModelProviderConnectionOrganization
+from ..core.pydantic_utilities import parse_obj_as
+from json.decoder import JSONDecodeError
+from ..core.api_error import ApiError
 from ..types.model_provider_connection_provider import ModelProviderConnectionProvider
 from ..types.model_provider_connection_scope import ModelProviderConnectionScope
+from ..types.model_provider_connection_organization import ModelProviderConnectionOrganization
+from ..types.model_provider_connection_created_by import ModelProviderConnectionCreatedBy
+import datetime as dt
+from ..types.model_provider_connection_budget_reset_period import ModelProviderConnectionBudgetResetPeriod
+from ..core.serialization import convert_and_respect_annotation_metadata
+from ..core.jsonable_encoder import jsonable_encoder
+from ..core.client_wrapper import AsyncClientWrapper
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -40,7 +41,7 @@ class ModelProvidersClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -48,11 +49,19 @@ class ModelProvidersClient:
         client.model_providers.list()
         """
         _response = self._client_wrapper.httpx_client.request(
-            "api/model-provider-connections/", method="GET", request_options=request_options
+            "api/model-provider-connections/",
+            method="GET",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(typing.List[ModelProviderConnection], _response.json())  # type: ignore
+                return typing.cast(
+                    typing.List[ModelProviderConnection],
+                    parse_obj_as(
+                        type_=typing.List[ModelProviderConnection],  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -129,7 +138,7 @@ class ModelProvidersClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -147,8 +156,12 @@ class ModelProvidersClient:
                 "deployment_name": deployment_name,
                 "endpoint": endpoint,
                 "scope": scope,
-                "organization": organization,
-                "created_by": created_by,
+                "organization": convert_and_respect_annotation_metadata(
+                    object_=organization, annotation=ModelProviderConnectionOrganization, direction="write"
+                ),
+                "created_by": convert_and_respect_annotation_metadata(
+                    object_=created_by, annotation=ModelProviderConnectionCreatedBy, direction="write"
+                ),
                 "created_at": created_at,
                 "updated_at": updated_at,
                 "is_internal": is_internal,
@@ -163,7 +176,13 @@ class ModelProvidersClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(ModelProviderConnection, _response.json())  # type: ignore
+                return typing.cast(
+                    ModelProviderConnection,
+                    parse_obj_as(
+                        type_=ModelProviderConnection,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -188,7 +207,7 @@ class ModelProvidersClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -198,11 +217,19 @@ class ModelProvidersClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"api/model-provider-connections/{jsonable_encoder(pk)}", method="GET", request_options=request_options
+            f"api/model-provider-connections/{jsonable_encoder(pk)}",
+            method="GET",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(ModelProviderConnection, _response.json())  # type: ignore
+                return typing.cast(
+                    ModelProviderConnection,
+                    parse_obj_as(
+                        type_=ModelProviderConnection,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -226,7 +253,7 @@ class ModelProvidersClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -236,7 +263,9 @@ class ModelProvidersClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"api/model-provider-connections/{jsonable_encoder(pk)}", method="DELETE", request_options=request_options
+            f"api/model-provider-connections/{jsonable_encoder(pk)}",
+            method="DELETE",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
@@ -321,7 +350,7 @@ class ModelProvidersClient:
 
         Examples
         --------
-        from label_studio_sdk.client import LabelStudio
+        from label_studio_sdk import LabelStudio
 
         client = LabelStudio(
             api_key="YOUR_API_KEY",
@@ -340,8 +369,12 @@ class ModelProvidersClient:
                 "deployment_name": deployment_name,
                 "endpoint": endpoint,
                 "scope": scope,
-                "organization": organization,
-                "created_by": created_by,
+                "organization": convert_and_respect_annotation_metadata(
+                    object_=organization, annotation=ModelProviderConnectionOrganization, direction="write"
+                ),
+                "created_by": convert_and_respect_annotation_metadata(
+                    object_=created_by, annotation=ModelProviderConnectionCreatedBy, direction="write"
+                ),
                 "created_at": created_at,
                 "updated_at": updated_at,
                 "is_internal": is_internal,
@@ -356,7 +389,13 @@ class ModelProvidersClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(ModelProviderConnection, _response.json())  # type: ignore
+                return typing.cast(
+                    ModelProviderConnection,
+                    parse_obj_as(
+                        type_=ModelProviderConnection,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -385,19 +424,35 @@ class AsyncModelProvidersClient:
 
         Examples
         --------
-        from label_studio_sdk.client import AsyncLabelStudio
+        import asyncio
+
+        from label_studio_sdk import AsyncLabelStudio
 
         client = AsyncLabelStudio(
             api_key="YOUR_API_KEY",
         )
-        await client.model_providers.list()
+
+
+        async def main() -> None:
+            await client.model_providers.list()
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "api/model-provider-connections/", method="GET", request_options=request_options
+            "api/model-provider-connections/",
+            method="GET",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(typing.List[ModelProviderConnection], _response.json())  # type: ignore
+                return typing.cast(
+                    typing.List[ModelProviderConnection],
+                    parse_obj_as(
+                        type_=typing.List[ModelProviderConnection],  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -474,14 +529,22 @@ class AsyncModelProvidersClient:
 
         Examples
         --------
-        from label_studio_sdk.client import AsyncLabelStudio
+        import asyncio
+
+        from label_studio_sdk import AsyncLabelStudio
 
         client = AsyncLabelStudio(
             api_key="YOUR_API_KEY",
         )
-        await client.model_providers.create(
-            provider="OpenAI",
-        )
+
+
+        async def main() -> None:
+            await client.model_providers.create(
+                provider="OpenAI",
+            )
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
             "api/model-provider-connections/",
@@ -492,8 +555,12 @@ class AsyncModelProvidersClient:
                 "deployment_name": deployment_name,
                 "endpoint": endpoint,
                 "scope": scope,
-                "organization": organization,
-                "created_by": created_by,
+                "organization": convert_and_respect_annotation_metadata(
+                    object_=organization, annotation=ModelProviderConnectionOrganization, direction="write"
+                ),
+                "created_by": convert_and_respect_annotation_metadata(
+                    object_=created_by, annotation=ModelProviderConnectionCreatedBy, direction="write"
+                ),
                 "created_at": created_at,
                 "updated_at": updated_at,
                 "is_internal": is_internal,
@@ -508,7 +575,13 @@ class AsyncModelProvidersClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(ModelProviderConnection, _response.json())  # type: ignore
+                return typing.cast(
+                    ModelProviderConnection,
+                    parse_obj_as(
+                        type_=ModelProviderConnection,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -533,21 +606,37 @@ class AsyncModelProvidersClient:
 
         Examples
         --------
-        from label_studio_sdk.client import AsyncLabelStudio
+        import asyncio
+
+        from label_studio_sdk import AsyncLabelStudio
 
         client = AsyncLabelStudio(
             api_key="YOUR_API_KEY",
         )
-        await client.model_providers.get(
-            pk=1,
-        )
+
+
+        async def main() -> None:
+            await client.model_providers.get(
+                pk=1,
+            )
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"api/model-provider-connections/{jsonable_encoder(pk)}", method="GET", request_options=request_options
+            f"api/model-provider-connections/{jsonable_encoder(pk)}",
+            method="GET",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(ModelProviderConnection, _response.json())  # type: ignore
+                return typing.cast(
+                    ModelProviderConnection,
+                    parse_obj_as(
+                        type_=ModelProviderConnection,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -571,17 +660,27 @@ class AsyncModelProvidersClient:
 
         Examples
         --------
-        from label_studio_sdk.client import AsyncLabelStudio
+        import asyncio
+
+        from label_studio_sdk import AsyncLabelStudio
 
         client = AsyncLabelStudio(
             api_key="YOUR_API_KEY",
         )
-        await client.model_providers.delete(
-            pk=1,
-        )
+
+
+        async def main() -> None:
+            await client.model_providers.delete(
+                pk=1,
+            )
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"api/model-provider-connections/{jsonable_encoder(pk)}", method="DELETE", request_options=request_options
+            f"api/model-provider-connections/{jsonable_encoder(pk)}",
+            method="DELETE",
+            request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
@@ -666,15 +765,23 @@ class AsyncModelProvidersClient:
 
         Examples
         --------
-        from label_studio_sdk.client import AsyncLabelStudio
+        import asyncio
+
+        from label_studio_sdk import AsyncLabelStudio
 
         client = AsyncLabelStudio(
             api_key="YOUR_API_KEY",
         )
-        await client.model_providers.update(
-            pk=1,
-            provider="OpenAI",
-        )
+
+
+        async def main() -> None:
+            await client.model_providers.update(
+                pk=1,
+                provider="OpenAI",
+            )
+
+
+        asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
             f"api/model-provider-connections/{jsonable_encoder(pk)}",
@@ -685,8 +792,12 @@ class AsyncModelProvidersClient:
                 "deployment_name": deployment_name,
                 "endpoint": endpoint,
                 "scope": scope,
-                "organization": organization,
-                "created_by": created_by,
+                "organization": convert_and_respect_annotation_metadata(
+                    object_=organization, annotation=ModelProviderConnectionOrganization, direction="write"
+                ),
+                "created_by": convert_and_respect_annotation_metadata(
+                    object_=created_by, annotation=ModelProviderConnectionCreatedBy, direction="write"
+                ),
                 "created_at": created_at,
                 "updated_at": updated_at,
                 "is_internal": is_internal,
@@ -701,7 +812,13 @@ class AsyncModelProvidersClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(ModelProviderConnection, _response.json())  # type: ignore
+                return typing.cast(
+                    ModelProviderConnection,
+                    parse_obj_as(
+                        type_=ModelProviderConnection,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)