scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 109

add test

Browse source
This commit is contained in:
Ishaan Jaff 2025-08-05 13:49:58 -07:00
parent 645ac875aa
commit e8d65cad57
1 changed files with 14 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
tests/test_litellm/litellm_core_utils/prompt_templates/test_litellm_core_utils_prompt_templates_factory.py
View file

@ -139,6 +139,20 @@ def test_bedrock_validate_format_image_or_video():
result = BedrockImageProcessor._validate_format(f"video/{format}", format)
assert result == format, f"Expected {format}, got {result}"
# Test valid document formats
valid_document_formats = {
"application/pdf": "pdf",
"text/csv": "csv",
"application/vnd.openxmlformats-officedocument.wordprocessingml.document": "docx",
"application/vnd.openxmlformats-officedocument.spreadsheetml.sheet": "xlsx",
}
for mime, expected in valid_document_formats.items():
print("testing mime", mime, "expected", expected)
result = BedrockImageProcessor._validate_format(
mime, mime.split("/")[1]
)
assert result == expected, f"Expected {expected}, got {result}"
# def test_ollama_pt_consecutive_system_messages():
# """Test handling consecutive system messages"""