diff --git a/litellm/provider_endpoints_support_backup.json b/litellm/provider_endpoints_support_backup.json index 9c734ea62e2..55e72cf1ca7 100644 --- a/litellm/provider_endpoints_support_backup.json +++ b/litellm/provider_endpoints_support_backup.json @@ -3349,16 +3349,25 @@ "url": "https://docs.litellm.ai/docs/providers/cheapaimodel", "endpoints": { "chat_completions": true, - "messages": false, - "responses": false, + "responses": true, + "text_completion": true, "embeddings": false, "image_generations": false, + "image_edits": false, "audio_transcriptions": false, "audio_speech": false, - "moderations": false, - "batches": false, "rerank": false, - "a2a": false + "ocr": false, + "search": false, + "moderations": false, + "realtime": false, + "video_generations": false, + "batches": false, + "messages": true, + "interactions": true, + "systemone": false, + "a2a": true, + "generateContent": true } }, "recraft": { diff --git a/provider_endpoints_support.json b/provider_endpoints_support.json index 9c734ea62e2..55e72cf1ca7 100644 --- a/provider_endpoints_support.json +++ b/provider_endpoints_support.json @@ -3349,16 +3349,25 @@ "url": "https://docs.litellm.ai/docs/providers/cheapaimodel", "endpoints": { "chat_completions": true, - "messages": false, - "responses": false, + "responses": true, + "text_completion": true, "embeddings": false, "image_generations": false, + "image_edits": false, "audio_transcriptions": false, "audio_speech": false, - "moderations": false, - "batches": false, "rerank": false, - "a2a": false + "ocr": false, + "search": false, + "moderations": false, + "realtime": false, + "video_generations": false, + "batches": false, + "messages": true, + "interactions": true, + "systemone": false, + "a2a": true, + "generateContent": true } }, "recraft": { diff --git a/tests/unit/llms/openai_like/test_cheapaimodel_provider.py b/tests/unit/llms/openai_like/test_cheapaimodel_provider.py index e3e97c4abf6..29e16167b04 100644 --- a/tests/unit/llms/openai_like/test_cheapaimodel_provider.py +++ b/tests/unit/llms/openai_like/test_cheapaimodel_provider.py @@ -123,8 +123,12 @@ def test_cheapaimodel_supported_endpoints(): matrix_path = Path(litellm.__file__).parent / "provider_endpoints_support_backup.json" endpoints = json.loads(matrix_path.read_text())["providers"]["cheapaimodel"]["endpoints"] + # The matrix is generated by scripts/generate_provider_endpoints_support.py, which + # also marks the endpoints LiteLLM bridges onto chat completions. cheapaimodel itself + # serves no embeddings, images, audio or rerank. assert endpoints["chat_completions"] is True - assert [name for name, supported in endpoints.items() if supported] == ["chat_completions"] + for name in ("embeddings", "image_generations", "audio_transcriptions", "audio_speech", "rerank"): + assert endpoints[name] is False, name def test_cheapaimodel_chat_completion_request():