[fix] use media_type in messages

Author: Daggx

edenai_apis/apis/anthropic/anthropic_api.py

@@ -156,7 +156,9 @@ def __format_anthropic_messages(
                                     "type": "image",
                                     "source": {
                                         "type": "base64",
-                                        "media_type": "image/jpeg",
+                                        "media_type": content_item["content"][
+                                            "media_type"
+                                        ],
                                         "data": media_data,
                                     },
                                 }
@@ -167,7 +169,7 @@ def __format_anthropic_messages(
                                 "type": "image",
                                 "source": {
                                     "type": "base64",
-                                    "media_type": "image/jpeg",
+                                    "media_type": content_item["content"]["media_type"],
                                     "data": content_item["content"]["media_base64"],
                                 },
                             }

edenai_apis/apis/google/google_multimodal_api.py

@@ -61,11 +61,12 @@ def __format_google_messages(
                         parts.append({"text": content["content"]["text"]})
                     elif content["type"] == "media_url":
                         media_url = content["content"]["media_url"]
+                        media_type = content["content"]["media_type"]
                         response = requests.get(media_url)
 
                         data = base64.b64encode(response.content).decode("utf-8")
                         parts.append(
-                            {"inlineData": {"data": data, "mimeType": "image/jpeg"}}
+                            {"inlineData": {"data": data, "mimeType": media_type}}
                         )
                     elif content["type"] == "media_base64":
                         media_base64 = content["content"]["media_base64"]

edenai_apis/features/multimodal/chat/chat_args.py

@@ -11,7 +11,8 @@ def chat_arguments(provider_name: str):
                     {
                         "type": "media_url",
                         "content": {
-                            "media_url": "https://upload.wikimedia.org/wikipedia/commons/thumb/d/dd/Gfp-wisconsin-madison-the-nature-boardwalk.jpg/2560px-Gfp-wisconsin-madison-the-nature-boardwalk.jpg"
+                            "media_url": "https://upload.wikimedia.org/wikipedia/commons/thumb/d/dd/Gfp-wisconsin-madison-the-nature-boardwalk.jpg/2560px-Gfp-wisconsin-madison-the-nature-boardwalk.jpg",
+                            "media_type": "image/jpeg",
                         },
                     },
                 ],