diff --git a/scribe/lib/scribe/ai/data/network/ai_api.dart b/scribe/lib/scribe/ai/data/network/ai_api.dart
index d9845ff77..ecae0061c 100644
--- a/scribe/lib/scribe/ai/data/network/ai_api.dart
+++ b/scribe/lib/scribe/ai/data/network/ai_api.dart
@@ -1,6 +1,7 @@
 import 'package:dio/dio.dart';
 import 'package:scribe/scribe/ai/data/model/ai_api_response.dart';
 import 'package:scribe/scribe/ai/data/model/ai_api_request.dart';
+import 'package:scribe/scribe/ai/data/network/ai_api_exception.dart';
 
 class AIApi {
   final Dio _dio;
@@ -18,7 +19,7 @@ class AIApi {
     final url = _endpoint;
 
     if (url == null) {
-      throw Exception('AI API is not available');
+      throw AIApiNotAvailableException();
     }
 
     final response = await _dio.post(
@@ -28,11 +29,14 @@ class AIApi {
 
     if (response.statusCode == 200) {
       if (response.data == null || response.data.isEmpty) {
-        throw Exception('AI API returned empty response');
+        throw AIApiEmptyResponseException();
       }
       return AIApiResponse.fromJson(response.data);
     } else {
-      throw Exception('AI API returned status code: ${response.statusCode}');
+      throw AIApiException(
+        'AI API returned status code: ${response.statusCode}',
+        statusCode: response.statusCode,
+      );
     }
   }
 }
diff --git a/scribe/lib/scribe/ai/data/network/ai_api_exception.dart b/scribe/lib/scribe/ai/data/network/ai_api_exception.dart
new file mode 100644
index 000000000..70b9b5f47
--- /dev/null
+++ b/scribe/lib/scribe/ai/data/network/ai_api_exception.dart
@@ -0,0 +1,22 @@
+class AIApiException implements Exception {
+  final String message;
+  final int? statusCode;
+
+  AIApiException(this.message, {this.statusCode});
+
+  @override
+  String toString() {
+    if (statusCode != null) {
+      return 'AIApiException: $message (status code: $statusCode)';
+    }
+    return 'AIApiException: $message';
+  }
+}
+
+class AIApiNotAvailableException extends AIApiException {
+  AIApiNotAvailableException() : super('AI API is not available');
+}
+
+class AIApiEmptyResponseException extends AIApiException {
+  AIApiEmptyResponseException() : super('AI API returned empty response');
+}