tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-28 10:20:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

update test

Browse Source 
Signed-off-by: omrishiv <327609+omrishiv@users.noreply.github.com>
This commit is contained in:
omrishiv
2025-03-10 20:34:52 -07:00
parent 338722beba
commit d25693b96c
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/litellm/llms/chat/test_converse_handler.py
+7 -6
View File
@@ -8,12 +8,13 @@ sys.path.insert(
) # Adds the parent directory to the system path
import litellm
def test_encode_model_id_with_inference_profile():
"""
Tests to make sure model name is being escaped correctly when used with an inference profile
:return:
Test instance profile is properly encoded when used as a model
"""
test = "bedrock/converse/arn:aws:bedrock:us-east-1:12345678910:application-inference-profile/ujdtmcirjhevpi"
expected = "bedrock/converse/arn%3Aaws%3Abedrock%3Aus-east-1%3A12345678910%3Aapplication-inference-profile%2Fujdtmcirjhevpi"
got = BedrockConverseLLM.encode_model_id(test)
assert expected == got
test_model = "arn:aws:bedrock:us-east-1:12345678910:application-inference-profile/ujdtmcirjhevpi"
expected_model = "arn%3Aaws%3Abedrock%3Aus-east-1%3A12345678910%3Aapplication-inference-profile%2Fujdtmcirjhevpi"
bedrock_converse_llm = BedrockConverseLLM()
returned_model = bedrock_converse_llm.encode_model_id(test_model)
assert expected_model == returned_model