Merge pull request #5406 from open-webui/main

refac
2025-02-20 12:00:22 +00:00 · 2024-09-13 19:26:51 +01:00 · 2024-09-13 19:26:51 +01:00 · d75d638b9a
commit d75d638b9a
parent 8a928e5356 b64c9d966a
1 changed files with 17 additions and 3 deletions
--- a/backend/open_webui/apps/openai/main.py
+++ b/backend/open_webui/apps/openai/main.py
@ -333,10 +333,24 @@ async def get_models(url_idx: Optional[int] = None, user=Depends(get_verified_us
            r.raise_for_status()

            response_data = r.json()
+
            if "api.openai.com" in url:
-                response_data["data"] = list(
-                    filter(lambda model: "gpt" in model["id"], response_data["data"])
-                )
+                # Filter the response data
+                response_data["data"] = [
+                    model
+                    for model in response_data["data"]
+                    if not any(
+                        name in model["id"]
+                        for name in [
+                            "babbage",
+                            "dall-e",
+                            "davinci",
+                            "embedding",
+                            "tts",
+                            "whisper",
+                        ]
+                    )
+                ]

            return response_data
        except Exception as e: