From d64ce8c3b0ca89fbfac67bddf2bf66ad1bb1ddb2 Mon Sep 17 00:00:00 2001
From: Gordon Mickel <gordon.mickel@gmail.com>
Date: Tue, 6 Aug 2024 18:26:13 +0200
Subject: [PATCH] fix: model check for diff mode

---
 src/ai/task-workflow.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/ai/task-workflow.ts b/src/ai/task-workflow.ts
index 57faa3f..57ab926 100644
--- a/src/ai/task-workflow.ts
+++ b/src/ai/task-workflow.ts
@@ -44,7 +44,7 @@ export async function runAIAssistedTask(options: AiAssistedTaskOptions) {
 
     const modelKey = await selectModel(options);
 
-    if (options.diff && modelKey !== 'claude-3-sonnet-20240229') {
+    if (options.diff && modelKey !== 'claude-3-5-sonnet-20240620') {
       console.log(
         chalk.yellow(
           'Diff-based code modifications are currently only supported with Claude 3.5 Sonnet. Falling back to full-file code modifications.',