updated get top risks prompt

backend/accounts/tests/test_tasks.py

@@ -16,7 +16,6 @@ class CeleryTaskTests(TestCase):
             compliance_frameworks=["Ab", "Ba"],
             industry_sector="Technology",
             it_dependency=8,
-            data_sensitivity="High",
             network_infrastructure="Cloud-based",
             remote_workforce_percentage="50%",
             third_party_vendor_access="10-20",
@@ -36,7 +35,11 @@ class CeleryTaskTests(TestCase):
     @patch("backend.accounts.tasks.get_controls_for_risk")
     @patch("backend.accounts.tasks.send_payment_email")
     def test_create_document_for_organization(self, mock_send_payment_email, mock_get_controls_for_risk, mock_get_top_risk):
-        mock_get_top_risk.return_value = [self.risk.risk_id] 
+        mock_get_top_risk.return_value = [{
+            "risk_id": int(self.risk.risk_id),
+            "risk_name": self.risk.risk_name,
+            "explanation": "Test explanation"
+        }]
         mock_get_controls_for_risk.return_value = [(self.control.id, 5, 7)]  
         create_document_for_organization(self.organization.email)