diff --git a/ai21/clients/studio/resources/studio_library.py b/ai21/clients/studio/resources/studio_library.py index 31cade4e..684f356f 100644 --- a/ai21/clients/studio/resources/studio_library.py +++ b/ai21/clients/studio/resources/studio_library.py @@ -49,7 +49,7 @@ def create( ) if body.get("batch_id"): - body["upload_mode"] = UploadMode.BATCH + body["upload_mode"] = UploadMode.BATCH.value raw_response = self._post(path=f"/{self._module_name}", files=files, body=body, response_cls=dict) @@ -127,7 +127,7 @@ async def create( ) if body.get("batch_id"): - body["upload_mode"] = UploadMode.BATCH + body["upload_mode"] = UploadMode.BATCH.value raw_response = await self._post(path=f"/{self._module_name}", files=files, body=body, response_cls=dict)