Spaces:
Runtime error
Runtime error
Update article_generator.py
Browse files- 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(
|
| 247 |
-
response_dict[query] =
|
| 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 |
|