refactor: improve error handling and message

Author: ArslanSaleem

backend/app/api/v1/chat.py

@@ -3,6 +3,7 @@
 
 from app.config import settings
 from app.database import get_db
+from app.exceptions import CreditLimitExceededException
 from app.logger import Logger
 from app.models.asset_content import AssetProcessingStatus
 from app.repositories import (
@@ -256,11 +257,17 @@ def draft_with_ai(draft_request: DraftRequest, db: Session = Depends(get_db)):
         if not api_key:
             raise HTTPException(status_code=404, detail="API Key not found!")
 
-        response = request_draft_with_ai(api_key.key, draft_request.model_dump_json())
+        try:
+            response = request_draft_with_ai(api_key.key, draft_request.model_dump_json())
+
+        except CreditLimitExceededException:
+                raise HTTPException(
+                    status_code=402, detail="Credit limit Reached, Wait next month or upgrade your Plan!"
+                )
 
         return {
             "status": "success",
-            "message": "Draft sucessfully generated!",
+            "message": "Draft successfully generated!",
             "data": {"response": response["response"]},
         }
 

backend/app/requests/__init__.py

@@ -235,6 +235,12 @@ def request_draft_with_ai(api_token: str, draft_request: dict) -> dict:
 
     try:
         if response.status_code not in [200, 201]:
+
+            if response.status_code == 402:
+                raise CreditLimitExceededException(
+                    response.json().get("detail", "Credit limit exceeded!")
+                )
+
             logger.error(
                 f"Failed to draft with AI. It returned {response.status_code} code: {response.text}"
             )

frontend/src/services/chat.tsx

@@ -63,6 +63,14 @@ export const draft_with_ai = async (data: ChatDraftRequest) => {
     );
     return response.data.data;
   } catch (error) {
-    throw error;
+    if (axios.isAxiosError(error)) {
+      if (error.response?.data) {
+        throw new Error(error.response.data.detail);
+      } else {
+        throw new Error("Failed to generate draft with AI. Please try again.");
+      }
+    } else {
+      throw new Error("Failed to generate draft with AI. Please try again.");
+    }
   }
 };