Merge pull request #5405 from open-webui/main

refac: filter non chat completions models
2025-01-18 00:30:51 +00:00 · 2024-09-13 19:23:02 +01:00 · 2024-09-13 19:23:02 +01:00 · 8a928e5356
commit 8a928e5356
parent c1d3481c41 adf958559b
1 changed files with 13 additions and 0 deletions
--- a/backend/open_webui/apps/openai/main.py
+++ b/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",
+                        ]
+                    )
                ]
            )