diff --git a/litellm/__init__.py b/litellm/__init__.py index 40f90c4743a..83a97c6fd0e 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -644,6 +644,7 @@ gemini_models: Set = set() xai_models: Set = set() zai_models: Set = set() deepseek_models: Set = set() +clinepass_models: Set = set() tencent_models: Set = set() runwayml_models: Set = set() azure_ai_models: Set = set() @@ -866,6 +867,8 @@ def _populate_provider_model_sets(model_cost_map: Dict) -> None: fal_ai_models.add(key) elif value.get("litellm_provider") == "deepseek": deepseek_models.add(key) + elif value.get("litellm_provider") == "clinepass": + clinepass_models.add(key) elif value.get("litellm_provider") == "tencent": tencent_models.add(key) elif value.get("litellm_provider") == "runwayml": @@ -1078,6 +1081,7 @@ model_list = list( | zai_models | fal_ai_models | deepseek_models + | clinepass_models | modelscope_models | azure_ai_models | voyage_models @@ -1183,6 +1187,7 @@ def _build_models_by_provider() -> dict: "zai": zai_models, "fal_ai": fal_ai_models, "deepseek": deepseek_models, + "clinepass": clinepass_models, "tencent": tencent_models, "runwayml": runwayml_models, "mistral": mistral_chat_models, diff --git a/tests/unit/litellm_core_utils/test_exception_mapping_utils.py b/tests/unit/litellm_core_utils/test_exception_mapping_utils.py index 9de768ea47b..8a68ffd7d9b 100644 --- a/tests/unit/litellm_core_utils/test_exception_mapping_utils.py +++ b/tests/unit/litellm_core_utils/test_exception_mapping_utils.py @@ -812,6 +812,7 @@ PROVIDERS_WITH_A_HANDLER = ( "azure", "azure_ai", "bedrock", + "clinepass", "cloudflare", "cohere", "databricks",