Преглед на файлове

refac: filter non chat completions models

Timothy J. Baek преди 8 месеца
родител
ревизия
adf958559b
променени са 1 файла, в които са добавени 13 реда и са изтрити 0 реда
  1. 13 0
      backend/open_webui/apps/openai/main.py

+ 13 - 0
backend/open_webui/apps/openai/main.py

@@ -218,6 +218,19 @@ def merge_models_lists(model_lists):
                         "urlIdx": idx,
                     }
                     for model in models
+                    if "api.openai.com"
+                    not in app.state.config.OPENAI_API_BASE_URLS[idx]
+                    or not any(
+                        name in model["id"]
+                        for name in [
+                            "babbage",
+                            "dall-e",
+                            "davinci",
+                            "embedding",
+                            "tts",
+                            "whisper",
+                        ]
+                    )
                 ]
             )