Add regenerate summary action in admin

Author: susilnem

per/admin.py

@@ -4,12 +4,14 @@
 
 from django.contrib import admin
 from django.http import HttpResponse
+from django.utils.translation import get_language as django_get_language
 from reversion_compare.admin import CompareVersionAdmin
 
 import per.models as models
 from api.models import Appeal
 from lang.admin import TranslationAdmin, TranslationInlineModelAdmin
 from per.admin_classes import GotoNextModelAdmin, RegionRestrictedAdmin
+from per.task import generate_summary
 
 
 class FormDataInline(admin.TabularInline, TranslationInlineModelAdmin):
@@ -324,6 +326,23 @@ class OpsLearningCacheResponseAdmin(TranslationAdmin):
         "exported_file",
         "exported_at",
     )
+    actions = ["regenerate_summary"]
+
+    def regenerate_summary(self, request, queryset):
+        """
+        Regenerate the summary of the selected OpsLearningCacheResponse objects.
+        """
+        requested_lang = django_get_language()
+        for obj in queryset:
+            generate_summary.delay(
+                ops_learning_summary_id=obj.id,
+                filter_data=obj.used_filters,
+                translation_lazy=requested_lang != "en",
+                # NOTE: Regenerating the summary will overwrite the cache
+                overwrite_prompt_cache=True,
+            )
+
+    regenerate_summary.short_description = "Regenerate Summary for selected Ops Learning Cache Response"
 
 
 class OpsLearningPromptResponseCacheAdmin(admin.ModelAdmin):

per/ops_learning_summary.py

@@ -831,13 +831,21 @@ def _modify_summary(summary: dict) -> dict:
         return processed_summary
 
     @classmethod
-    def _get_or_create_summary(cls, prompt: str, prompt_hash: str, type: OpsLearningPromptResponseCache.PromptType) -> dict:
+    def _get_or_create_summary(
+        cls, prompt: str, prompt_hash: str, type: OpsLearningPromptResponseCache.PromptType, overwrite_prompt_cache: bool = False
+    ) -> dict:
         instance, created = OpsLearningPromptResponseCache.objects.get_or_create(
             prompt_hash=prompt_hash,
             type=type,
             defaults={"prompt": prompt},
         )
-        if created or not bool(instance.response):
+        """
+        NOTE:
+        1. If the prompt response is not found in the cache, it regenerates the summary
+        2. If overwrite_prompt_cache is True, it regenerates the summary
+        3. If new obj is created, it generates the summary
+        """
+        if overwrite_prompt_cache or created or not bool(instance.response):
             summary = cls.generate_summary(prompt, type)
             instance.response = summary
             instance.save(update_fields=["response"])
@@ -926,6 +934,7 @@ def get_or_create_primary_summary(
         cls,
         ops_learning_summary_instance: OpsLearningCacheResponse,
         primary_learning_prompt: str,
+        overwrite_prompt_cache: bool = False,
     ):
         """Retrieves or Generates the primary summary based on the provided prompt."""
         logger.info("Retrieving or generating primary summary.")
@@ -938,6 +947,7 @@ def get_or_create_primary_summary(
             prompt=primary_learning_prompt,
             prompt_hash=primary_prompt_hash,
             type=OpsLearningPromptResponseCache.PromptType.PRIMARY,
+            overwrite_prompt_cache=overwrite_prompt_cache,
         )
 
         # Saving into the database
@@ -959,6 +969,7 @@ def get_or_create_secondary_summary(
         cls,
         ops_learning_summary_instance: OpsLearningCacheResponse,
         secondary_learning_prompt: str,
+        overwrite_prompt_cache: bool = False,
     ):
         """Retrieves or Generates the summary based on the provided prompts."""
         logger.info("Retrieving or generating secondary summary.")
@@ -971,6 +982,7 @@ def get_or_create_secondary_summary(
             prompt=secondary_learning_prompt,
             prompt_hash=secondary_prompt_hash,
             type=OpsLearningPromptResponseCache.PromptType.SECONDARY,
+            overwrite_prompt_cache=overwrite_prompt_cache,
         )
 
         # Saving into the database

per/task.py

@@ -7,7 +7,7 @@
 from per.ops_learning_summary import OpsLearningSummaryTask
 
 
-def generate_ops_learning_summary(ops_learning_summary_id: int, filter_data: dict):
+def generate_ops_learning_summary(ops_learning_summary_id: int, filter_data: dict, overwrite_prompt_cache: bool = False):
     ops_learning_summary_instance = OpsLearningCacheResponse.objects.filter(id=ops_learning_summary_id).first()
     if not ops_learning_summary_instance:
         logger.error("Ops learning summary not found", exc_info=True)
@@ -44,6 +44,7 @@ def generate_ops_learning_summary(ops_learning_summary_id: int, filter_data: dic
             OpsLearningSummaryTask.get_or_create_primary_summary(
                 ops_learning_summary_instance=ops_learning_summary_instance,
                 primary_learning_prompt=primary_learning_prompt,
+                overwrite_prompt_cache=overwrite_prompt_cache,
             )
 
             # Prioritize excerpts for secondary insights
@@ -54,6 +55,7 @@ def generate_ops_learning_summary(ops_learning_summary_id: int, filter_data: dic
             OpsLearningSummaryTask.get_or_create_secondary_summary(
                 ops_learning_summary_instance=ops_learning_summary_instance,
                 secondary_learning_prompt=secondary_learning_prompt,
+                overwrite_prompt_cache=overwrite_prompt_cache,
             )
 
             # Change Ops Learning Summary Status to SUCCESS
@@ -79,10 +81,19 @@ def generate_ops_learning_summary(ops_learning_summary_id: int, filter_data: dic
 
 
 @shared_task
-def generate_summary(ops_learning_summary_id: int, filter_data: dict, translation_lazy: bool = True):
+def generate_summary(
+    ops_learning_summary_id: int,
+    filter_data: dict,
+    translation_lazy: bool = True,
+    overwrite_prompt_cache: bool = False,
+):
     with redis_lock(key=RedisLockKey.OPERATION_LEARNING_SUMMARY, id=ops_learning_summary_id) as acquired:
         if not acquired:
             logger.warning("Ops learning summary generation is already in progress")
             return False
         with override_settings(CELERY_TASK_ALWAYS_EAGER=not translation_lazy):
-            return generate_ops_learning_summary(ops_learning_summary_id, filter_data)
+            return generate_ops_learning_summary(
+                ops_learning_summary_id=ops_learning_summary_id,
+                filter_data=filter_data,
+                overwrite_prompt_cache=overwrite_prompt_cache,
+            )