diff --git a/litellm/__init__.py b/litellm/__init__.py index 3c5aecb82c2..04463266947 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -475,6 +475,7 @@ nlp_cloud_models: Set = set() aleph_alpha_models: Set = set() bedrock_models: Set = set() bedrock_converse_models: Set = set(BEDROCK_CONVERSE_MODELS) +fal_ai_models: Set = set() fireworks_ai_models: Set = set() fireworks_ai_embedding_models: Set = set() deepinfra_models: Set = set() @@ -663,6 +664,8 @@ def add_known_models(): text_completion_codestral_models.add(key) elif value.get("litellm_provider") == "xai": xai_models.add(key) + elif value.get("litellm_provider") == "fal_ai": + fal_ai_models.add(key) elif value.get("litellm_provider") == "deepseek": deepseek_models.add(key) elif value.get("litellm_provider") == "meta_llama": @@ -813,6 +816,7 @@ model_list = list( | gemini_models | text_completion_codestral_models | xai_models + | fal_ai_models | deepseek_models | azure_ai_models | voyage_models @@ -895,6 +899,7 @@ models_by_provider: dict = { "aleph_alpha": aleph_alpha_models, "text-completion-codestral": text_completion_codestral_models, "xai": xai_models, + "fal_ai": fal_ai_models, "deepseek": deepseek_models, "mistral": mistral_chat_models, "azure_ai": azure_ai_models,