From c17a4523ab00db2e83d0fa836b652c408df4c2ef Mon Sep 17 00:00:00 2001
From: aryamanpandya <aryaman.pandya@tufts.edu>
Date: Tue, 24 Dec 2024 10:56:30 -0500
Subject: [PATCH] Register latest O1 model

---
 src/ell/models/openai.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/ell/models/openai.py b/src/ell/models/openai.py
index 37577daaf..ecee4c32e 100644
--- a/src/ell/models/openai.py
+++ b/src/ell/models/openai.py
@@ -67,6 +67,7 @@ def register(client: openai.Client):
         'gpt-4o-audio-preview-2024-10-01',
         'o1-preview-2024-09-12',
         'o1-mini-2024-09-12',
+        'o1-2024-12-17',
         'gpt-3.5-turbo-0301',
         'gpt-3.5-turbo-0613',
         'tts-1',
@@ -92,7 +93,7 @@ def register(client: openai.Client):
     #XXX: Deprecation in 0.1.0
     config.register_model('o1-preview', client, supports_streaming=True)
     config.register_model('o1-mini', client, supports_streaming=True)
-
+    config.register_model('o1-2024-12-17', client, supports_streaming=True)
 default_client = None
 try:
     default_client = openai.Client()