From 6a1a9748ac268dba00f98d212793ebe09b567250 Mon Sep 17 00:00:00 2001
From: Beingpax <prakashjoshipaxtek@gmail.com>
Date: Fri, 25 Jul 2025 17:54:07 +0545
Subject: [PATCH] refactor: Unify and simplify error handling for all AI
 providers

---
 VoiceInk/Services/AIEnhancementService.swift | 13 ++-----------
 1 file changed, 2 insertions(+), 11 deletions(-)

diff --git a/VoiceInk/Services/AIEnhancementService.swift b/VoiceInk/Services/AIEnhancementService.swift
index f008e2b..25f1853 100644
--- a/VoiceInk/Services/AIEnhancementService.swift
+++ b/VoiceInk/Services/AIEnhancementService.swift
@@ -182,17 +182,8 @@ class AIEnhancementService: ObservableObject {
                 let result = try await aiService.enhanceWithOllama(text: formattedText, systemPrompt: systemMessage)
                 let filteredResult = AIEnhancementOutputFilter.filter(result)
                 return filteredResult
-            } catch let error as LocalAIError {
-                switch error {
-                case .serviceUnavailable:
-                    throw EnhancementError.notConfigured
-                case .modelNotFound:
-                    throw EnhancementError.enhancementFailed
-                case .serverError:
-                    throw EnhancementError.serverError
-                default:
-                    throw EnhancementError.enhancementFailed
-                }
+            } catch {
+                throw EnhancementError.customError(error.localizedDescription)
             }
         }