phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test fixes - azure deprecated dall-e-2

Browse source
This commit is contained in:
Ishaan Jaff 2025-04-02 20:56:20 -07:00
parent 74550df197
commit c3341a1e18
2 changed files with 5 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/local_testing/test_custom_callback_input.py
View file

@ -1133,10 +1133,10 @@ def test_image_generation_openai():
response = litellm.image_generation( response = litellm.image_generation(
prompt="A cute baby sea otter", prompt="A cute baby sea otter",
model="azure/", model="azure/dall-e-3-test",
api_base=os.getenv("AZURE_API_BASE"), api_version="2023-12-01-preview",
api_key=os.getenv("AZURE_API_KEY"), api_base=os.getenv("AZURE_SWEDEN_API_BASE"),
api_version="2023-06-01-preview", api_key=os.getenv("AZURE_SWEDEN_API_KEY"),
) )
print(f"response: {response}") print(f"response: {response}")

18
tests/local_testing/test_router.py
View file

@ -1136,16 +1136,7 @@ async def test_aimg_gen_on_router():
"api_base": os.getenv("AZURE_SWEDEN_API_BASE"), "api_base": os.getenv("AZURE_SWEDEN_API_BASE"),
"api_key": os.getenv("AZURE_SWEDEN_API_KEY"), "api_key": os.getenv("AZURE_SWEDEN_API_KEY"),
}, },
}, }
{
"model_name": "dall-e-2",
"litellm_params": {
"model": "azure/",
"api_version": "2023-06-01-preview",
"api_base": os.getenv("AZURE_API_BASE"),
"api_key": os.getenv("AZURE_API_KEY"),
},
},
] ]
router = Router(model_list=model_list, num_retries=3) router = Router(model_list=model_list, num_retries=3)
response = await router.aimage_generation( response = await router.aimage_generation(
@ -1153,13 +1144,6 @@ async def test_aimg_gen_on_router():
) )
print(response) print(response)
assert len(response.data) > 0 assert len(response.data) > 0
response = await router.aimage_generation(
model="dall-e-2", prompt="A cute baby sea otter"
)
print(response)
assert len(response.data) > 0
router.reset() router.reset()
except litellm.InternalServerError as e: except litellm.InternalServerError as e:
pass pass