Update feasibility_agent.py

feasibility_agent.py

@@ -86,8 +86,8 @@ def keyword_generation(report):
 def dataset_search(output_list):
   for usecase_dict in output_list:
     query = usecase_dict['keyword']
-    query_format = 'kaggle OR github OR huggingface AND ({query})'
-    links = keyword_search.invoke({'query': query_format})
+    query_format = f'{query} AND dataset OR implementation guide site:(kaggle.com OR github.com OR huggingface.co OR paperswithcode.com OR arxiv.org OR research.google.com)'
+    links = keyword_search.invoke({'query': query_format.format(query)})
     usecase_dict['links'] = links
   return output_list