Changed to gpt-4o-mini, Added organisation info

2025-02-17 21:40:08 +01:00
parent 65716ff842
commit 7151cd0c4d
2 changed files with 80 additions and 36 deletions
--- a/backend/accounts/tasks.py
+++ b/backend/accounts/tasks.py
@@ -34,7 +34,7 @@ def create_document_for_organization(confirmation_email):
    for risk in top_risks:
        controls_content += f"Risk: {risk.risk_id} - {risk.risk_name}\n"

-        selected_controls = get_controls_for_risk(risk)
+        selected_controls = get_controls_for_risk(risk ,organization=organization)
        
        for control_id, weight in selected_controls:
            control = Control.objects.filter(id=control_id).first()