added auto-detection for the output type on the client side to be able to download binary files such as videos

samples/models/dummy/config.json

@@ -3,6 +3,8 @@
   "tags": ["dummy"],
   "file": "model.py",
 
+  "output_type": "video/mp4",
+
   "inputs": {
     "file": {"type": "file"}
   }

source/manager/ModelManager.py

@@ -1,3 +1,4 @@
+import asyncio
 import json
 import os
 import typing
@@ -10,6 +11,11 @@ from source import model, api
 
 
 class ModelManager:
+    """
+    The model manager
+    Load the list of models available, ensure that only one model is loaded at the same time.
+    """
+
     def __init__(self, application: api.Application, model_library: os.PathLike | str):
         self.application: api.Application = application
         self.model_library: Path = Path(model_library)
@@ -20,7 +26,9 @@ class ModelManager:
         self.models: dict[str, model.base.BaseModel] = {}
 
         # the currently loaded model
-        # TODO(Faraphel): load more than one model at a time ? require a way more complex manager to handle memory issue
+        # TODO(Faraphel): load more than one model at a time ?
+        #  would require a way more complex manager to handle memory issue
+        #  having two calculations at the same time might not be worth it either
         self.current_loaded_model: typing.Optional[model.base.BaseModel] = None
 
         # lock to avoid concurrent inference and concurrent model loading and unloading

source/model/PythonModel.py

@@ -77,6 +77,12 @@ class PythonModel(base.BaseModel):
             infer_api,
             methods=["POST"],
             tags=self.tags,
+            # summary=...,
+            # description=...,
+            response_class=fastapi.responses.StreamingResponse,
+            responses={
+                200: {"content": {self.output_type: {}}}
+            },
         )
 
     def _load(self) -> None:

source/utils/__init__.py

@@ -1 +1,2 @@
 from . import parameters
+from . import mimetypes

source/utils/mimetypes.py

@@ -0,0 +1,21 @@
+def is_textlike(mimetype: str) -> bool:
+    """
+    Determinate if a mimetype is considered as holding text
+    :param mimetype: the mimetype to check
+    :return: True if the mimetype represent text, False otherwise
+    """
+
+    # check the family of the mimetype
+    if mimetype.startswith("text/"):
+        return True
+
+    # check applications formats that are text formatted
+    if mimetype in [
+        "application/xml",
+        "application/json",
+        "application/javascript"
+    ]:
+        return True
+
+    # otherwise consider the file as non-text
+    return False