OpenWebUI/pipelines
1
0
Fork 0
mirror of https://github.com/open-webui/pipelines synced 2025-05-13 17:10:45 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #510 from lmtr0/main

Browse Source 
Support for more openai models
This commit is contained in:
Tim Jaeryang Baek 2025-04-18 18:08:44 -07:00 committed by GitHub
commit 9901e3de77
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
examples/pipelines/providers/openai_manifold_pipeline.py
View File

@ -59,6 +59,13 @@ class Pipeline:
f"{self.valves.OPENAI_API_BASE_URL}/models", headers=headers f"{self.valves.OPENAI_API_BASE_URL}/models", headers=headers
) )
allowed_models = [
"gpt",
"o1",
"o3",
"o4",
]
models = r.json() models = r.json()
return [ return [
{ {
@ -66,7 +73,7 @@ class Pipeline:
"name": model["name"] if "name" in model else model["id"], "name": model["name"] if "name" in model else model["id"],
} }
for model in models["data"] for model in models["data"]
if "gpt" in model["id"] or "o1" in model["id"] or "o3" in model["id"] if any(substring in model["id"] for substring in allowed_models)
] ]
except Exception as e: except Exception as e: