瀏覽代碼

fix: filter outlet

Timothy J. Baek 11 月之前
父節點
當前提交
a9e5003c4f
共有 1 個文件被更改,包括 4 次插入4 次删除
  1. 4 4
      backend/main.py

+ 4 - 4
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