litellm/tests/image_gen_tests
Ishaan Jaff 73c7b73aa0
(feat) Add cost tracking for Azure Dall-e-3 Image Generation + use base class to ensure basic image generation tests pass (#6716)
* add BaseImageGenTest

* use 1 class for unit testing

* add debugging to BaseImageGenTest

* TestAzureOpenAIDalle3

* fix response_cost_calculator

* test_basic_image_generation

* fix img gen basic test

* fix _select_model_name_for_cost_calc

* fix test_aimage_generation_bedrock_with_optional_params

* fix undo changes cost tracking

* fix response_cost_calculator

* fix test_cost_azure_gpt_35
2024-11-12 20:02:16 -08:00
..
base_image_generation_test.py (feat) Add cost tracking for Azure Dall-e-3 Image Generation + use base class to ensure basic image generation tests pass (#6716) 2024-11-12 20:02:16 -08:00
test_bedrock_image_gen_unit_tests.py (feat) add cost tracking stable diffusion 3 on Bedrock (#6676) 2024-11-11 20:21:44 -08:00
test_image_generation.py (feat) Add cost tracking for Azure Dall-e-3 Image Generation + use base class to ensure basic image generation tests pass (#6716) 2024-11-12 20:02:16 -08:00
vertex_key.json (feat) Add Bedrock Stability.ai Stable Diffusion 3 Image Generation models (#6673) 2024-11-08 19:26:03 -08:00