diff --git a/enterprise_catalog/apps/ai_curation/utils/generate_curation_utils.py b/enterprise_catalog/apps/ai_curation/utils/generate_curation_utils.py index c2269fad8..c561c7109 100644 --- a/enterprise_catalog/apps/ai_curation/utils/generate_curation_utils.py +++ b/enterprise_catalog/apps/ai_curation/utils/generate_curation_utils.py @@ -18,6 +18,7 @@ CACHE_KEY = '{task_id}_{content_type}' +CACHE_TIMEOUT = 1200 def get_cache_key(task_id: str, content_type: str) -> str: @@ -219,9 +220,21 @@ def generate_curation(query: str, catalog_name: str, task_id: str): ) # Cache data for tweaking the filter - cache.set(get_cache_key(task_id=task_id, content_type='ocm_courses'), partially_filtered_ocm_courses) - cache.set(get_cache_key(task_id=task_id, content_type='exec_ed_courses'), partially_filtered_exec_ed_courses) - cache.set(get_cache_key(task_id=task_id, content_type='programs'), programs) + cache.set( + get_cache_key(task_id=task_id, content_type='ocm_courses'), + partially_filtered_ocm_courses, + CACHE_TIMEOUT + ) + cache.set( + get_cache_key(task_id=task_id, content_type='exec_ed_courses'), + partially_filtered_exec_ed_courses, + CACHE_TIMEOUT + ) + cache.set( + get_cache_key(task_id=task_id, content_type='programs'), + programs, + CACHE_TIMEOUT + ) # filter programs based on the filtered courses filtered_programs = apply_programs_filter(filtered_ocm_courses + filtered_exec_ed_courses, programs)