fix: added _sub_module_name

Author: shirbiton

ai21/clients/studio/resources/studio_library.py

@@ -16,15 +16,16 @@
 
 
 class StudioLibrary(StudioResource):
-    _module_name = "library"
+    _module_name = "library/files"
 
     def __init__(self, client: AI21HTTPClient):
         super().__init__(client)
         self.files = LibraryFiles(client)
 
 
 class LibraryFiles(StudioResource):
-    _module_name = "library"
+    _module_name = "library/files"
+    _sub_module_name = "library"
 
     def create(
         self,
@@ -50,15 +51,15 @@ def create(
         if body.get("batch_id"):
             body["upload_mode"] = UploadMode.BATCH
 
-        raw_response = self._post(path=f"/{self._module_name}/files", files=files, body=body, response_cls=dict)
+        raw_response = self._post(path=f"/{self._module_name}", files=files, body=body, response_cls=dict)
 
         return raw_response["fileId"]
 
     def get(self, file_id: str) -> FileResponse:
-        return self._get(path=f"/{self._module_name}/files/{file_id}", response_cls=FileResponse)
+        return self._get(path=f"/{self._module_name}/{file_id}", response_cls=FileResponse)
 
     def get_batch_status(self, batch_id: str) -> BatchStatusResponse:
-        return self._get(path=f"/{self._module_name}/batches/{batch_id}/status", response_cls=BatchStatusResponse)
+        return self._get(path=f"/{self._sub_module_name}/batches/{batch_id}/status", response_cls=BatchStatusResponse)
 
     def list(
         self,
@@ -69,7 +70,7 @@ def list(
     ) -> List[FileResponse]:
         params = remove_not_given({"offset": offset, "limit": limit})
 
-        return self._get(path=f"/{self._module_name}/files", params=params, response_cls=List[FileResponse])
+        return self._get(path=f"/{self._module_name}", params=params, response_cls=List[FileResponse])
 
     def update(
         self,
@@ -86,22 +87,23 @@ def update(
                 **kwargs,
             }
         )
-        self._put(path=f"/{self._module_name}/files/{file_id}", body=body)
+        self._put(path=f"/{self._module_name}/{file_id}", body=body)
 
     def delete(self, file_id: str) -> None:
-        self._delete(path=f"/{self._module_name}/files/{file_id}")
+        self._delete(path=f"/{self._module_name}/{file_id}")
 
 
 class AsyncStudioLibrary(AsyncStudioResource):
-    _module_name = "library"
+    _module_name = "library/files"
 
     def __init__(self, client: AsyncAI21HTTPClient):
         super().__init__(client)
         self.files = AsyncLibraryFiles(client)
 
 
 class AsyncLibraryFiles(AsyncStudioResource):
-    _module_name = "library"
+    _module_name = "library/files"
+    _sub_module_name = "library"
 
     async def create(
         self,
@@ -127,15 +129,17 @@ async def create(
         if body.get("batch_id"):
             body["upload_mode"] = UploadMode.BATCH
 
-        raw_response = await self._post(path=f"/{self._module_name}/files", files=files, body=body, response_cls=dict)
+        raw_response = await self._post(path=f"/{self._module_name}", files=files, body=body, response_cls=dict)
 
         return raw_response["fileId"]
 
     async def get(self, file_id: str) -> FileResponse:
-        return await self._get(path=f"/{self._module_name}/files/{file_id}", response_cls=FileResponse)
+        return await self._get(path=f"/{self._module_name}/{file_id}", response_cls=FileResponse)
 
     async def get_batch_status(self, batch_id: str) -> BatchStatusResponse:
-        return await self._get(path=f"/{self._module_name}/batches/{batch_id}/status", response_cls=BatchStatusResponse)
+        return await self._get(
+            path=f"/{self._sub_module_name}/batches/{batch_id}/status", response_cls=BatchStatusResponse
+        )
 
     async def list(
         self,
@@ -146,7 +150,7 @@ async def list(
     ) -> List[FileResponse]:
         params = remove_not_given({"offset": offset, "limit": limit})
 
-        return await self._get(path=f"/{self._module_name}/files", params=params, response_cls=List[FileResponse])
+        return await self._get(path=f"/{self._module_name}", params=params, response_cls=List[FileResponse])
 
     async def update(
         self,
@@ -163,7 +167,7 @@ async def update(
                 **kwargs,
             }
         )
-        await self._put(path=f"/{self._module_name}/files/{file_id}", body=body)
+        await self._put(path=f"/{self._module_name}/{file_id}", body=body)
 
     async def delete(self, file_id: str) -> None:
-        await self._delete(path=f"/{self._module_name}/files/{file_id}")
+        await self._delete(path=f"/{self._module_name}/{file_id}")