Merge pull request #510 from lmtr0/main

tjbck · web-flow · commit 9901e3de77a3 · 2025-04-18T18:08:44.000-07:00
Support for more openai models
diff --git a/examples/pipelines/providers/openai_manifold_pipeline.py b/examples/pipelines/providers/openai_manifold_pipeline.py
@@ -59,14 +59,21 @@ def get_openai_models(self):
                     f"{self.valves.OPENAI_API_BASE_URL}/models", headers=headers
                 )
 
+                allowed_models = [
+                    "gpt",
+                    "o1",
+                    "o3",
+                    "o4",
+                ]
+
                 models = r.json()
                 return [
                     {
                         "id": model["id"],
                         "name": model["name"] if "name" in model else model["id"],
                     }
                     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:

Original file line number	Diff line number	Diff line change
`@@ -59,14 +59,21 @@ def get_openai_models(self):`
`59`	`59`	`f"{self.valves.OPENAI_API_BASE_URL}/models", headers=headers`
`60`	`60`	`)`
`61`	`61`
	`62`	`+ allowed_models = [`
	`63`	`+ "gpt",`
	`64`	`+ "o1",`
	`65`	`+ "o3",`
	`66`	`+ "o4",`
	`67`	`+ ]`
	`68`	`+`
`62`	`69`	`models = r.json()`
`63`	`70`	`return [`
`64`	`71`	`{`
`65`	`72`	`"id": model["id"],`
`66`	`73`	`"name": model["name"] if "name" in model else model["id"],`
`67`	`74`	`}`
`68`	`75`	`for model in models["data"]`
`69`		`- if "gpt" in model["id"] or "o1" in model["id"] or "o3" in model["id"]`
	`76`	`+ if any(substring in model["id"] for substring in allowed_models)`
`70`	`77`	`]`
`71`	`78`
`72`	`79`	`except Exception as e:`