From 6bd325e4322923338ca1745a663975cfb7266cbf Mon Sep 17 00:00:00 2001
From: Dustin Miller <1342542+spdustin@users.noreply.github.com>
Date: Fri, 21 Jun 2024 11:31:16 -0500
Subject: [PATCH] Add support for vertex_ai/claude_* models

---
 aider/models.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/aider/models.py b/aider/models.py
index ac150d546..a16e70340 100644
--- a/aider/models.py
+++ b/aider/models.py
@@ -196,6 +196,20 @@ MODEL_SETTINGS = [
         weak_model_name="openrouter/anthropic/claude-3-haiku-20240307",
         use_repo_map=True,
     ),
+    # Vertex AI Claude models
+    ModelSettings(
+        "vertex_ai/claude-3-5-sonnet@20240620",
+        "diff",
+        weak_model_name="vertex_ai/claude-3-haiku@20240307",
+        use_repo_map=True,
+    ),
+    ModelSettings(
+        "vertex_ai/claude-3-opus@20240229",
+        "diff",
+        weak_model_name="vertex_ai/claude-3-haiku@20240307",
+        use_repo_map=True,
+        send_undo_reply=True,
+    ),
     # Cohere
     ModelSettings(
         "command-r-plus",