phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

Merge pull request #4856 from msabramo/fix-test_prompt_factory-flake8-warning

Browse source 
Fix `test_prompt_factory` flake8 warning
This commit is contained in:
Krish Dholakia 2024-07-24 09:38:48 -07:00 • committed by GitHub
commit 035d03c7f2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_prompt_factory.py
View file

@ -212,7 +212,7 @@ def test_convert_url_to_img():
[
("data:image/jpeg;base64,1234", "image/jpeg"),
("data:application/pdf;base64,1234", "application/pdf"),
("data:image\/jpeg;base64,1234", "image/jpeg"),
(r"data:image\/jpeg;base64,1234", "image/jpeg"),
],
)
def test_base64_image_input(url, expected_media_type):