From 0566ab2c31d4aca0350dc5d8d79717df84416b17 Mon Sep 17 00:00:00 2001
From: Manohar Paturi <186662190+ManoharPaturi@users.noreply.github.com>
Date: Tue, 20 Oct 2026 09:34:19 +0530
Subject: [PATCH] FIX: Add YAML and JSONL extension mappings to get_mime_type

Signed-off-by: Manohar Paturi <186662190+ManoharPaturi@users.noreply.github.com>
---
 pyrit/common/mime_type.py           | 3 +++
 tests/unit/common/test_mime_type.py | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/pyrit/common/mime_type.py b/pyrit/common/mime_type.py
index 6298b3cff7..66b7d723ce 100644
--- a/pyrit/common/mime_type.py
+++ b/pyrit/common/mime_type.py
@@ -11,6 +11,9 @@
     ".csv": "text/csv",
     ".md": "text/markdown",
     ".json": "application/json",
+    ".jsonl": "application/jsonlines",
+    ".yaml": "application/yaml",
+    ".yml": "application/yaml",
     ".xml": "application/xml",
     ".png": "image/png",
     ".jpg": "image/jpeg",
diff --git a/tests/unit/common/test_mime_type.py b/tests/unit/common/test_mime_type.py
index 61c91d7c8b..6fd92f7677 100644
--- a/tests/unit/common/test_mime_type.py
+++ b/tests/unit/common/test_mime_type.py
@@ -18,6 +18,9 @@
         ("video.avi", "video/x-msvideo"),
         ("document.rtf", "application/rtf"),
         ("archive.zip", "application/zip"),
+        ("config.yaml", "application/yaml"),
+        ("prompt.yml", "application/yaml"),
+        ("dataset.jsonl", "application/jsonlines"),
     ],
 )
 def test_get_mime_type_explicit_mapping_takes_precedence(file_path: str, expected_mime_type: str):