Update article_generator.py

article_generator.py

@@ -247,16 +247,22 @@ def perform_initial_tavily_search(h2_texts, h3_texts):
     for idx, h2_text in enumerate(h2_texts):
         h3_for_this_h2 = [h3 for h3 in h3_texts if h3.startswith(f"{idx+1}-")]
         if not h3_for_this_h2 and h2_text.strip() != "まとめ":
-            print(f"{h2_text}に対応するh3が見つかりませんでした: インデックス {idx+1}")
+            print(f"No matching h3 elements found for h2: {h2_text} at index {idx+1}")
             continue
         query = f"{h2_text} {' '.join(h3_for_this_h2)}"
         queries.append(query)
-    
-    print("Tavily検索をクエリで実行中:", queries)
+
+    print("Tavily search executing with queries:", queries)
     responses = tavily_search_tool.search(queries)
-    # レスポンスのチェックとフィルタリング
+
+    # レスポンスの形式を確認
+    if not isinstance(responses, list):
+        print("Error: Response is not a list.")
+        return {}
+
+    # レスポンスをフィルタリング
     filtered_responses = filter_responses_by_domain(responses, allowed_domains)
-    response_dict = {query: (filtered_responses[i] if i < len(filtered_responses) else "レスポンスが受信されませんでした") for i, query in enumerate(queries)}
+    response_dict = {query: (filtered_responses[i] if i < len(filtered_responses) else "No response received") for i, query in enumerate(queries)}
     return response_dict
 
 def save_preloaded_tavily_data(data):