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

TestAzureOpenAIDalle3

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-12 16:18:49 -08:00
parent a6ce4254f0
commit 2415e7d0a6
1 changed files with 10 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

9
tests/image_gen_tests/test_image_generation.py
View file

@ -23,6 +23,10 @@ import litellm
import json import json
import tempfile import tempfile
from base_image_generation_test import BaseImageGenTest from base_image_generation_test import BaseImageGenTest
import logging
from litellm._logging import verbose_logger
verbose_logger.setLevel(logging.DEBUG)
def get_vertex_ai_creds_json() -> dict: def get_vertex_ai_creds_json() -> dict:
@ -114,11 +118,16 @@ class TestOpenAIDalle3(BaseImageGenTest):
class TestAzureOpenAIDalle3(BaseImageGenTest): class TestAzureOpenAIDalle3(BaseImageGenTest):
def get_base_image_generation_call_args(self) -> dict: def get_base_image_generation_call_args(self) -> dict:
litellm.set_verbose = True
return { return {
"model": "azure/dall-e-3-test", "model": "azure/dall-e-3-test",
"api_version": "2023-09-01-preview", "api_version": "2023-09-01-preview",
"metadata": {
"model_info": {
"base_model": "dall-e-3", "base_model": "dall-e-3",
} }
},
}
@pytest.mark.parametrize( @pytest.mark.parametrize(