fix: filter outlet

a9e5003c · Timothy J. Baek · 32a61ea1 · a9e5003c
Commit a9e5003c authored Jun 02, 2024 by Timothy J. Baek
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 4 deletions

backend/main.py backend/main.py +4 -4

No files found.
--- a/backend/main.py
+++ b/backend/main.py
@@ -498,10 +498,10 @@ async def chat_completed(form_data: dict, user=Depends(get_verified_user)):
    ]
    sorted_filters = sorted(filters, key=lambda x: x["pipeline"]["priority"])

-    model = app.state.MODELS[model_id]
-
-    if "pipeline" in model:
-        sorted_filters = [model] + sorted_filters
+    if model_id in app.state.MODELS:
+        model = app.state.MODELS[model_id]
+        if "pipeline" in model:
+            sorted_filters = [model] + sorted_filters

    for filter in sorted_filters:
        r = None