From 0ec58fc7c6aa36e7e2f4176d1238c7c1a6f67996 Mon Sep 17 00:00:00 2001 From: "ajaysi (aider)" Date: Sat, 14 Sep 2024 19:21:25 +0530 Subject: [PATCH] fix: handle empty data in get_related_topics_and_save_csv to prevent list index out of range error --- lib/ai_web_researcher/google_trends_researcher.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/lib/ai_web_researcher/google_trends_researcher.py b/lib/ai_web_researcher/google_trends_researcher.py index aa6ce6b6..0bb86371 100644 --- a/lib/ai_web_researcher/google_trends_researcher.py +++ b/lib/ai_web_researcher/google_trends_researcher.py @@ -339,6 +339,11 @@ def get_related_topics_and_save_csv(search_keywords): logger.error(f"Failed to get pytrends realted topics: {err}") return None + # Check if data contains expected keys + if not data or 'top' not in list(data.values())[0] or 'rising' not in list(data.values())[0]: + logger.error("No related topics found.") + return pd.DataFrame() + # Extract data from the result top_topics = list(data.values())[0]['top'] rising_topics = list(data.values())[0]['rising']