diff --git a/tests/llm_translation/test_bedrock_embedding.py b/tests/llm_translation/test_bedrock_embedding.py index feda016bd9f..d0e89d4b430 100644 --- a/tests/llm_translation/test_bedrock_embedding.py +++ b/tests/llm_translation/test_bedrock_embedding.py @@ -464,6 +464,8 @@ def test_bedrock_embedding_region_bug_reproduction(): os.environ["AWS_REGION_NAME"] = original_region_name else: os.environ.pop("AWS_REGION_NAME", None) + + def test_bedrock_titan_g1_text_02_model_info(): """Test that amazon.titan-embed-g1-text-02 has correct pricing metadata""" model_info = litellm.get_model_info("amazon.titan-embed-g1-text-02")