Add download by filename endpoint to Python

lib/python-beta/src/bailo/__init__.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-__version__ = "0.4.1"
+__version__ = "0.4.3"
 
 from bailo.core.agent import Agent, PkiAgent, TokenAgent
 from bailo.core.client import Client

lib/python-beta/src/bailo/core/client.py

@@ -297,7 +297,7 @@ def get_download_file(
         file_id: str,
         buffer: BytesIO,
     ):
-        """Download a specific file.
+        """Download a specific file by it's id.
 
         :param model_id: Unique model ID
         :param file_id: Unique file ID
@@ -314,6 +314,35 @@ def get_download_file(
         shutil.copyfileobj(req.raw, buffer)
         return file_id
 
+    def get_download_by_filename(
+        self,
+        model_id: str,
+        semver: str,
+        filename: str,
+        buffer: BytesIO,
+    ):
+        """Download a specific file.
+
+        :param model_id: Unique model ID
+        :param semver: Semver of the release
+        :param filename: The filename trying to download from
+        :param buffer: BytesIO object for bailo to write to
+        :return: The filename
+        """
+        if isinstance(self.agent, TokenAgent):
+            req = self.agent.get(
+                f"{self.url}/v2/token/model/{model_id}/release/{semver}/file/{filename}/download",
+                stream=True,
+                timeout=10_000,
+            )
+        else:
+            req = self.agent.get(
+                f"{self.url}/v2/model/{model_id}/release/{semver}/file/{filename}/download", stream=True, timeout=10_000
+            )
+
+        shutil.copyfileobj(req.raw, buffer)
+        return filename
+
     def simple_upload(self, model_id: str, name: str, buffer: BytesIO):
         """Create a simple file upload.
 

lib/python-beta/src/bailo/helper/release.py

@@ -132,16 +132,16 @@ def from_version(cls, client: Client, model_id: str, version: Version | str) ->
             draft,
         )
 
-    def download(self, file_id: str, buffer: BytesIO) -> str:
+    def download(self, filename: str, buffer: BytesIO) -> str:
         """Give returns a Reading object given the file id.
 
-        :param file_name: The name of the file to retrieve
+        :param filename: The name of the file to retrieve
         :param buffer: A BytesIO object
         :return: A JSON response object
         """
-        return self.client.get_download_file(self.model_id, file_id, buffer)
+        return self.client.get_download_by_filename(self.model_id, str(self.version), filename, buffer)
 
-    def upload(self, name: str, file: BytesIO) -> Any:
+    def upload(self, name: str, file: BytesIO) -> str:
         """Upload files in a given directory to the release.
 
         :param name: The name of the file to upload to bailo

lib/python-beta/tests/test_files.py

@@ -3,7 +3,7 @@
 from io import BytesIO
 
 import pytest
-from bailo.core.exceptions import ResponseException
+from bailo.core.exceptions import BailoException
 
 
 @pytest.mark.integration
@@ -14,10 +14,10 @@ def test_file_upload(example_model):
     example_release = example_model.create_release("0.1.0", "test")
 
     with file as file:
-        file_id = example_release.upload("test", file)
+        example_release.upload("test", file)
 
     download_file = BytesIO()
-    example_release.download(file_id, download_file)
+    example_release.download("test", download_file)
 
     # Check that file uploaded has the same contents as the one downloaded
     download_file.seek(0)
@@ -28,5 +28,5 @@ def test_file_upload(example_model):
 def test_source_target_doesnt_exist(example_model):
     download_file = BytesIO()
     example_release = example_model.create_release("0.1.0", "test")
-    with pytest.raises(ResponseException):
+    with pytest.raises(BailoException):
         example_release.download("non_existant_model", download_file)