From 03f738eff6e2056b2d34979e0729a7ee464ae8fe Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Wed, 5 Feb 2025 19:01:00 -0800
Subject: [PATCH] fix test_models_by_provider

---
 litellm/__init__.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/litellm/__init__.py b/litellm/__init__.py
index 144a1932e24..c49b3214b9b 100644
--- a/litellm/__init__.py
+++ b/litellm/__init__.py
@@ -411,6 +411,7 @@ anyscale_models: List = []
 cerebras_models: List = []
 galadriel_models: List = []
 sambanova_models: List = []
+assemblyai_models: List = []
 
 
 def is_bedrock_pricing_only_model(key: str) -> bool:
@@ -560,6 +561,8 @@ def add_known_models():
             galadriel_models.append(key)
         elif value.get("litellm_provider") == "sambanova_models":
             sambanova_models.append(key)
+        elif value.get("litellm_provider") == "assemblyai":
+            assemblyai_models.append(key)
 
 
 add_known_models()
@@ -631,6 +634,7 @@ model_list = (
     + galadriel_models
     + sambanova_models
     + azure_text_models
+    + assemblyai_models
 )
 
 model_list_set = set(model_list)
@@ -684,6 +688,7 @@ models_by_provider: dict = {
     "cerebras": cerebras_models,
     "galadriel": galadriel_models,
     "sambanova": sambanova_models,
+    "assemblyai": assemblyai_models,
 }
 
 # mapping for those models which have larger equivalents