Update Src/AiCommitMessage/Services/GenerateCommitMessageService.cs

guibranco · Dec 13, 2024 · 7ea53ab · 7ea53ab
1 parent ce6d1df
commit 7ea53ab
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/Src/AiCommitMessage/Services/GenerateCommitMessageService.cs b/Src/AiCommitMessage/Services/GenerateCommitMessageService.cs
@@ -50,12 +50,24 @@ public string GenerateCommitMessage(GenerateCommitMessageOptions options)
             ? GitHelper.GetBranchName()
             : options.Branch;
         var diff = string.IsNullOrEmpty(options.Diff) ? GitHelper.GetGitDiff() : options.Diff;
+        if (Encoding.UTF8.GetByteCount(diff) > 10240) // 10 KB limit
+        {
+            throw new InvalidOperationException("🚫 The staged changes are too large to process. Please reduce the number of files or size of changes and try again.");
+        }
+
         var message = options.Message;
 
         if (IsMergeConflictResolution(message))
         {
             return message;
         }
+        try
+        {
+            // Existing code to call OpenAI API
+        }
+        catch (Exception ex) when (ex is HttpRequestException || ex is TaskCanceledException)
+
+            throw new InvalidOperationException("⚠️ OpenAI API is currently unavailable. Please try again later.");
 
         if (string.IsNullOrEmpty(branch) && string.IsNullOrEmpty(diff))
         {