From deb27f8f27c4fc11254cca3b4d28b9744d0725dd Mon Sep 17 00:00:00 2001
From: Sven Grunewaldt <strayer@olle-orks.org>
Date: Thu, 10 Oct 2024 11:56:00 +0200
Subject: [PATCH] feat: Add new model settings for azure/o1-preview

Fixes litellm exceptions and incorrect model settings when using
o1-preview on Azure OpenAI.
---
 aider/models.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/aider/models.py b/aider/models.py
index 99bdd093d..f94d83b44 100644
--- a/aider/models.py
+++ b/aider/models.py
@@ -513,6 +513,18 @@ MODEL_SETTINGS = [
         use_temperature=False,
         streaming=False,
     ),
+    ModelSettings(
+        "azure/o1-mini",
+        "whole",
+        weak_model_name="azure/gpt-4o-mini",
+        editor_model_name="azure/gpt-4o",
+        editor_edit_format="editor-diff",
+        use_repo_map=True,
+        reminder="user",
+        use_system_prompt=False,
+        use_temperature=False,
+        streaming=False,
+    ),
     ModelSettings(
         "o1-mini",
         "whole",
@@ -537,6 +549,18 @@ MODEL_SETTINGS = [
         use_temperature=False,
         streaming=False,
     ),
+    ModelSettings(
+        "azure/o1-preview",
+        "diff",
+        weak_model_name="azure/gpt-4o-mini",
+        editor_model_name="azure/gpt-4o",
+        editor_edit_format="editor-diff",
+        use_repo_map=True,
+        reminder="user",
+        use_system_prompt=False,
+        use_temperature=False,
+        streaming=False,
+    ),
     ModelSettings(
         "o1-preview",
         "architect",