Normalize web search results and handle Tavily results dict; ensure search uses TAVILY_API_KEY

app.py

@@ -34,17 +34,29 @@ def chat(user_message, chat_history, chat_pairs, sources, plan):
         try:
             # Perform web search using the entire user message as the query
             results = run_web_search(user_message, num_results=5, domain_filter="")
+            # Normalize results:
+            # Tavily may return a dictionary with a "results" key containing
+            # the list of search results. If so, extract that list. If it's a
+            # list already, use it directly. Otherwise, default to an empty list.
+            if isinstance(results, dict):
+                normalized_results = results.get("results", [])
+            elif isinstance(results, list):
+                normalized_results = results
+            else:
+                normalized_results = []
             # Ensure the sources list is initialised
             if sources is None:
                 sources = []
-            sources.extend(results)
+            sources.extend(normalized_results)
             # Summarise results into a simple string with title and URL
             summary_lines = []
-            for r in results:
-                title = r.get("title", "")
-                url = r.get("url", "")
-                if title or url:
-                    summary_lines.append(f"{title} - {url}")
+            for r in normalized_results:
+                # Defensive: ensure r is a dict
+                if isinstance(r, dict):
+                    title = r.get("title", "")
+                    url = r.get("url", "")
+                    if title or url:
+                        summary_lines.append(f"{title} - {url}")
             if summary_lines:
                 assistant_reply = "Here are some resources I found:\n" + "\n".join(summary_lines)
             else:
@@ -133,15 +145,23 @@ def run_search(query, chat_history, chat_pairs, sources, plan, num_results=5, do
             f"(Error: {e})"
         )
         return summary, chat_history, chat_pairs, sources or [], plan
+    # Normalize the results: Tavily may return a dict with 'results'
+    if isinstance(results, dict):
+        normalized_results = results.get("results", [])
+    elif isinstance(results, list):
+        normalized_results = results
+    else:
+        normalized_results = []
     if sources is None:
         sources = []
-    sources.extend(results)
+    sources.extend(normalized_results)
     # summarise results into a string for display
     summary_lines = []
-    for r in results:
-        title = r.get("title", "")
-        url = r.get("url", "")
-        summary_lines.append(f"{title} - {url}")
+    for r in normalized_results:
+        if isinstance(r, dict):
+            title = r.get("title", "")
+            url = r.get("url", "")
+            summary_lines.append(f"{title} - {url}")
     summary = "\n".join(summary_lines)
     return summary, chat_history, chat_pairs, sources, plan