diff --git a/src/DiffEngine/AiCliDetector.cs b/src/DiffEngine/AiCliDetector.cs
index 516f17d0..39f4d50a 100644
--- a/src/DiffEngine/AiCliDetector.cs
+++ b/src/DiffEngine/AiCliDetector.cs
@@ -8,7 +8,11 @@ static AiCliDetector()
 
         // GitHub Copilot
         // https://docs.github.com/en/copilot/using-github-copilot/using-github-copilot-in-the-command-line
-        IsCopilot = variables.Contains("GITHUB_COPILOT_RUNTIME");
+        // Copilot CLI sets AI_AGENT=github-copilot / github-copilot-cli, VS Code agent mode sets github_copilot_vscode_agent
+        var aiAgent = Environment.GetEnvironmentVariable("AI_AGENT");
+        IsCopilot = variables.Contains("GITHUB_COPILOT_RUNTIME") ||
+                    variables.Contains("COPILOT_AGENT_SESSION_ID") ||
+                    (aiAgent != null && aiAgent.StartsWith("github", StringComparison.OrdinalIgnoreCase));
 
         // Aider
         // https://aider.chat/docs/config/dotenv.html