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

test: fix test

Browse source
This commit is contained in:
Krrish Dholakia 2025-03-03 13:33:39 -08:00
parent 1911146909
commit ce5fc9dbde
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/local_testing/test_get_model_info.py
View file

@ -340,6 +340,8 @@ def test_get_model_info_bedrock_models():
base_model = BedrockModelInfo.get_base_model(k) base_model = BedrockModelInfo.get_base_model(k)
base_model_info = litellm.model_cost[base_model] base_model_info = litellm.model_cost[base_model]
for base_model_key, base_model_value in base_model_info.items(): for base_model_key, base_model_value in base_model_info.items():
if "invoke/" in k:
continue
if base_model_key.startswith("supports_"): if base_model_key.startswith("supports_"):
assert ( assert (
base_model_key in v base_model_key in v