fix: content-type and accept parsing

More flexibility than an exact string match since there can be some additional params

Signed-off-by: Raphael Glon <[email protected]>

Author: oOraph

src/huggingface_inference_toolkit/serialization/base.py

@@ -4,7 +4,6 @@
 
 content_type_mapping = {
     "application/json": Jsoner,
-    "application/json; charset=utf-8": Jsoner,
     "text/csv": None,
     "text/plain": None,
     # image types
@@ -39,9 +38,11 @@
 
 class ContentType:
     @staticmethod
-    def get_deserializer(content_type):
-        if content_type in content_type_mapping:
-            return content_type_mapping[content_type]
+    def get_deserializer(content_type: str):
+        # Extract media type from content type
+        extracted = content_type.split(";")[0]
+        if extracted in content_type_mapping:
+            return content_type_mapping[extracted]
         else:
             message = f"""
                 Content type "{content_type}" not supported.
@@ -51,13 +52,15 @@ def get_deserializer(content_type):
             raise Exception(message)
 
     @staticmethod
-    def get_serializer(accept):
-        if accept in content_type_mapping:
-            return content_type_mapping[accept]
-        else:
-            message = f"""
-                Accept type "{accept}" not supported.
-                Supported accept types are:
-                {", ".join(list(content_type_mapping.keys()))}
-            """
-            raise Exception(message)
+    def get_serializer(accept: str):
+        extracts = accept.split(",")
+        for extract in extracts:
+            extracted = extract.split(";")[0]
+            if extracted in content_type_mapping:
+                return content_type_mapping[extracted]
+        message = f"""
+            Accept type "{accept}" not supported.
+            Supported accept types are:
+            {", ".join(list(content_type_mapping.keys()))}
+        """
+        raise Exception(message)