added action response

apps/spotlight/service.py

@@ -1,3 +1,4 @@
+from dataclasses import dataclass
 from typing import Callable, Dict
 
 import requests
@@ -14,6 +15,12 @@
 }
 
 
+@dataclass
+class ActionResponse:
+    message: str = None
+    is_success: bool = None
+
+
 class HelpService:
     @classmethod
     def extract_citations(cls, *, citations: list) -> list:
@@ -85,8 +92,11 @@ def trigger_export(cls, *, workspace_id: int):
         }
         url = f'http://localhost:8000/api/workspaces/{workspace_id}/trigger_export/'
         action_response = requests.post(url, json={}, headers=headers)
+        if action_response.status_code == 200:
+            return ActionResponse(message="Export triggered successfully", is_success=True)
+        return ActionResponse(message="Export triggered failed", is_success=False)
 
     @classmethod
-    def action(cls, *, code: str, workspace_id: str):
+    def action(cls, *, code: str, workspace_id: str) -> ActionResponse:
         action_function = cls._get_action_function_from_code(code=code)
-        action_function(workspace_id=workspace_id)
+        return action_function(workspace_id=workspace_id)

apps/spotlight/views.py

@@ -93,8 +93,11 @@ def post(self, request, *args, **kwargs):
         code = payload["code"]
 
         try:
-            ActionService.action(code=code, workspace_id=workspace_id)
-            return JsonResponse(data={"message": "Action triggered successfully"}, status=200)
+            action_response = ActionService.action(code=code, workspace_id=workspace_id)
+            if action_response.is_success is True:
+                return JsonResponse(data={"message": action_response.message}, status=200)
+            else:
+                return JsonResponse(data={"message": action_response.message}, status=500)
         except Exception as e:
             print(e)
             return JsonResponse(data={"message": "Action failed"}, status=500)