From fe10c1ab82ae64099b495f0ce9c84d7974d839ed Mon Sep 17 00:00:00 2001
From: clavedeluna <danalitovsky+git@gmail.com>
Date: Mon, 3 Mar 2025 18:03:23 -0300
Subject: [PATCH] add o1 models to registry

---
 src/codemodder/llm.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

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"