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):