fix

chat-apropo · Nov 23, 2024 · 7a1a2a8 · 7a1a2a8
1 parent d5000fb
commit 7a1a2a8
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/backend/routes.py b/backend/routes.py
@@ -29,6 +29,11 @@ def add_routers(app: FastAPI) -> None:
     app.include_router(router_ui)
 
 
+MODEL_BLACKLIST = [
+    "TextGenerations",
+    "ImageGenerations",
+]
+
 BEST_MODELS_ORDERED = [
     # "gpt-4o",
     # "gpt-4o-mini",
@@ -38,7 +43,7 @@ def add_routers(app: FastAPI) -> None:
 BEST_MODELS_ORDERED += [
     model_name
     for model_name in provider_and_models.all_model_names
-    if model_name not in BEST_MODELS_ORDERED
+    if model_name not in BEST_MODELS_ORDERED and model_name not in MODEL_BLACKLIST
 ]