Skip to content

Commit 464044f

Browse files
authored
fix: should be string and not enum (#325)
1 parent a09f976 commit 464044f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ai21/clients/studio/resources/studio_library.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ def create(
4949
)
5050

5151
if body.get("batch_id"):
52-
body["upload_mode"] = UploadMode.BATCH
52+
body["upload_mode"] = UploadMode.BATCH.value
5353

5454
raw_response = self._post(path=f"/{self._module_name}", files=files, body=body, response_cls=dict)
5555

@@ -127,7 +127,7 @@ async def create(
127127
)
128128

129129
if body.get("batch_id"):
130-
body["upload_mode"] = UploadMode.BATCH
130+
body["upload_mode"] = UploadMode.BATCH.value
131131

132132
raw_response = await self._post(path=f"/{self._module_name}", files=files, body=body, response_cls=dict)
133133

0 commit comments

Comments
 (0)