From cb5a515d70acd6bf3d0592e4c2eb006a47e48929 Mon Sep 17 00:00:00 2001
From: Beingpax <prakashjoshipaxtek@gmail.com>
Date: Tue, 24 Jun 2025 16:34:40 +0545
Subject: [PATCH] Refine enhancement retry logic

---
 VoiceInk/Services/AIEnhancementService.swift | 2 +-
 VoiceInk/Services/AIService.swift            | 2 --
 2 files changed, 1 insertion(+), 3 deletions(-)

diff --git a/VoiceInk/Services/AIEnhancementService.swift b/VoiceInk/Services/AIEnhancementService.swift
index adfd1e6..c807bf9 100644
--- a/VoiceInk/Services/AIEnhancementService.swift
+++ b/VoiceInk/Services/AIEnhancementService.swift
@@ -455,7 +455,7 @@ class AIEnhancementService: ObservableObject {
         guard retryCount < maxRetries - 1 else { return false }
         
         switch error {
-        case .rateLimitExceeded, .serverError, .networkError:
+        case .rateLimitExceeded, .serverError:
             return true
         default:
             return false
diff --git a/VoiceInk/Services/AIService.swift b/VoiceInk/Services/AIService.swift
index 5d22126..0ca586a 100644
--- a/VoiceInk/Services/AIService.swift
+++ b/VoiceInk/Services/AIService.swift
@@ -535,5 +535,3 @@ class AIService: ObservableObject {
 extension Notification.Name {
     static let aiProviderKeyChanged = Notification.Name("aiProviderKeyChanged")
 } 
-
-