Yasu777 commited on
Commit
2d5bef1
·
verified ·
1 Parent(s): 8f80ca3

Update article_generator.py

Browse files
Files changed (1) hide show
  1. article_generator.py +2 -6
article_generator.py CHANGED
@@ -220,7 +220,6 @@ def generate_text_with_gpt4(prompt):
220
  )
221
  return response.choices[0]["message"]["content"].strip()
222
 
223
- # 初期データをTavily検索で収集する関数
224
  # 初期データをTavily検索で収集する関数
225
  def perform_initial_tavily_search(h2_texts, h3_texts):
226
  tavily_search_tool = EnhancedTavilySearchTool()
@@ -238,13 +237,10 @@ def perform_initial_tavily_search(h2_texts, h3_texts):
238
  print("Performing Tavily search with queries:", queries)
239
  responses = tavily_search_tool.search(queries)
240
 
241
- # フィルタリングを適用
242
- filtered_responses = filter_responses_by_domain(responses, allowed_domains)
243
-
244
  response_dict = {}
245
  for i, query in enumerate(queries):
246
- if i < len(filtered_responses): # 応答リストの範囲内にあることを確認
247
- response_dict[query] = filtered_responses[i]
248
  else:
249
  response_dict[query] = "No response received"
250
 
 
220
  )
221
  return response.choices[0]["message"]["content"].strip()
222
 
 
223
  # 初期データをTavily検索で収集する関数
224
  def perform_initial_tavily_search(h2_texts, h3_texts):
225
  tavily_search_tool = EnhancedTavilySearchTool()
 
237
  print("Performing Tavily search with queries:", queries)
238
  responses = tavily_search_tool.search(queries)
239
 
 
 
 
240
  response_dict = {}
241
  for i, query in enumerate(queries):
242
+ if i < len(responses): # 応答リストの範囲内にあることを確認
243
+ response_dict[query] = responses[i]
244
  else:
245
  response_dict[query] = "No response received"
246