diff --git a/src/codemodder/llm.py b/src/codemodder/llm.py index b0de58d5..f9d66a28 100644 --- a/src/codemodder/llm.py +++ b/src/codemodder/llm.py @@ -35,7 +35,13 @@ "log_token_usage", ] -models = ["gpt-4-turbo-2024-04-09", "gpt-4o-2024-05-13", "gpt-35-turbo-0125"] +models = [ + "gpt-4-turbo-2024-04-09", + "gpt-4o-2024-05-13", + "gpt-35-turbo-0125", + "o1-mini", + "o1", +] DEFAULT_AZURE_OPENAI_API_VERSION = "2024-02-01"