Fix failing tests

This commit is contained in:
Sameer Kankute 2026-04-24 17:08:26 +05:30
parent 91c355dfd2
commit 5234931d46
No known key found for this signature in database
3 changed files with 8 additions and 2 deletions

View file

@ -215,6 +215,7 @@ jobs:
tests/proxy_unit_tests/test_google_gemini_proxy_request.py
tests/proxy_unit_tests/test_get_favicon.py
tests/proxy_unit_tests/test_get_image.py
tests/proxy_unit_tests/test_reducto_ocr_route.py
tests/proxy_unit_tests/test_ui_path_detection.py
tests/proxy_unit_tests/test_prompt_test_endpoint.py
tests/proxy_unit_tests/test_check_batch_cost.py

View file

@ -617,6 +617,7 @@ minimax_models: Set = set()
aws_polly_models: Set = set()
gigachat_models: Set = set()
llamagate_models: Set = set()
reducto_models: Set = set()
bedrock_mantle_models: Set = set()
@ -882,6 +883,8 @@ def add_known_models(model_cost_map: Optional[Dict] = None):
gigachat_models.add(key)
elif value.get("litellm_provider") == "llamagate":
llamagate_models.add(key)
elif value.get("litellm_provider") == "reducto":
reducto_models.add(key)
elif value.get("litellm_provider") == "bedrock_mantle":
bedrock_mantle_models.add(key)
@ -992,6 +995,7 @@ model_list = list(
| ovhcloud_models
| lemonade_models
| docker_model_runner_models
| reducto_models
| bedrock_mantle_models
| set(clarifai_models)
)
@ -1097,6 +1101,7 @@ models_by_provider: dict = {
"aws_polly": aws_polly_models,
"gigachat": gigachat_models,
"llamagate": llamagate_models,
"reducto": reducto_models,
"bedrock_mantle": bedrock_mantle_models,
}

4
uv.lock generated
View file

@ -9,7 +9,7 @@ resolution-markers = [
]
[options]
exclude-newer = "2026-04-20T17:15:10.783024Z"
exclude-newer = "2026-04-21T11:36:47.108684Z"
exclude-newer-span = "P3D"
[manifest]
@ -3085,7 +3085,7 @@ wheels = [
[[package]]
name = "litellm"
version = "1.83.13"
version = "1.83.10"
source = { editable = "." }
dependencies = [
{ name = "aiohttp" },