Update openai_manifold_pipeline.py

lmtr0 · web-flow · commit 3cb201d2d640 · 2025-02-19T09:32:27.000-03:00
added support for o1 and o3 models
diff --git a/examples/pipelines/providers/openai_manifold_pipeline.py b/examples/pipelines/providers/openai_manifold_pipeline.py
@@ -66,7 +66,7 @@ def get_openai_models(self):
                         "name": model["name"] if "name" in model else model["id"],
                     }
                     for model in models["data"]
-                    if "gpt" in model["id"]
+                    if "gpt" in model["id"] or "o1" in model["id"] or "o3" in model["id"]
                 ]
 
             except Exception as e:

Original file line number	Diff line number	Diff line change
`@@ -66,7 +66,7 @@ def get_openai_models(self):`
`66`	`66`	`"name": model["name"] if "name" in model else model["id"],`
`67`	`67`	`}`
`68`	`68`	`for model in models["data"]`
`69`		`- if "gpt" in model["id"]`
	`69`	`+ if "gpt" in model["id"] or "o1" in model["id"] or "o3" in model["id"]`
`70`	`70`	`]`
`71`	`71`
`72`	`72`	`except Exception as e:`