Skip to content

Commit bf59156

Browse files
authored
add o1 models to registry (#1013)
1 parent 96628ed commit bf59156

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

src/codemodder/llm.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,13 @@
3535
"log_token_usage",
3636
]
3737

38-
models = ["gpt-4-turbo-2024-04-09", "gpt-4o-2024-05-13", "gpt-35-turbo-0125"]
38+
models = [
39+
"gpt-4-turbo-2024-04-09",
40+
"gpt-4o-2024-05-13",
41+
"gpt-35-turbo-0125",
42+
"o1-mini",
43+
"o1",
44+
]
3945
DEFAULT_AZURE_OPENAI_API_VERSION = "2024-02-01"
4046

4147

0 commit comments

Comments
 (0)