4 commits

Author	SHA1	Message	Date
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
Ishaan Jaff	25bae4cc23	(feat) add cost tracking stable diffusion 3 on Bedrock (#6676) ... * add cost tracking for sd3 * test_image_generation_bedrock * fix get model info for image cost * add cost_calculator for stability 1 models * add unit testing for bedrock image cost calc * test_cost_calculator_with_no_optional_params * add test_cost_calculator_basic * correctly allow size Optional * fix cost_calculator * sd3 unit tests cost calc	2024-11-11 20:21:44 -08:00
Ishaan Jaff	979dfe8ab2	(feat) Add Bedrock Stability.ai Stable Diffusion 3 Image Generation models (#6673) ... * add bedrock image gen async support * added async support for bedrock image gen * move image gen testing * add AmazonStability3Config * add AmazonStability3Config config * update AmazonStabilityConfig * update get_optional_params_image_gen * use 1 helper for _get_request_body * add transform_response_dict_to_openai_response for stability3 * test sd3-large-v1:0 * unit testing for bedrock image gen * fix load_vertex_ai_credentials * fix test_aimage_generation_vertex_ai * add stability.sd3-large-v1:0 to model cost map * add stability.stability.sd3-large-v1:0 to docs	2024-11-08 19:26:03 -08:00
Ishaan Jaff	086e1ac5bc	move image gen testing	2024-11-08 14:10:08 -08:00

Renamed from tests/local_testing/test_image_generation.py (Browse further)