Update search_utils.py

search_utils.py

@@ -43,4 +43,29 @@ def internet_search(query, search_type="web", num_results=20):
 @lru_cache(maxsize=100)
 def cached_internet_search(query: str):
     logger.info(f"Performing cached internet search for query: {query}")
-    return internet_search(query, search_type="news")
+    return internet_search(query, search_type="news")
+
+def analyze_data(query, data_type="news"):
+    logger.info(f"Analyzing {data_type} for query: {query}")
+    
+    if data_type == "news":
+        data = cached_internet_search(query)
+        prompt_generator = generate_news_prompt
+        system_prompt = NEWS_ASSISTANT_PROMPT
+    else:
+        data = internet_search(query, search_type="web")
+        prompt_generator = generate_search_prompt
+        system_prompt = SEARCH_ASSISTANT_PROMPT
+    
+    if not data:
+        logger.error(f"Failed to fetch {data_type} data")
+        return None
+
+    prompt = prompt_generator(query, data)
+    messages = [
+        {"role": "system", "content": system_prompt},
+        {"role": "user", "content": prompt}
+    ]
+
+    logger.info(f"{data_type.capitalize()} analysis completed")
+    return messages,data