Asaad Almutareb commited on
Commit
5bf49b7
·
1 Parent(s): 5604c54

fixed missing wikipedia sources from sqlite

Browse files
innovation_pathfinder_ai/structured_tools/structured_tools.py CHANGED
@@ -169,7 +169,11 @@ def wikipedia_search(query: str) -> str:
169
  api_wrapper = WikipediaAPIWrapper()
170
  wikipedia_search = WikipediaQueryRun(api_wrapper=api_wrapper)
171
  wikipedia_results = wikipedia_search.run(query)
172
- all_sources += create_wikipedia_urls_from_text(wikipedia_results)
 
 
 
 
173
  return wikipedia_results
174
 
175
  @tool
 
169
  api_wrapper = WikipediaAPIWrapper()
170
  wikipedia_search = WikipediaQueryRun(api_wrapper=api_wrapper)
171
  wikipedia_results = wikipedia_search.run(query)
172
+ formatted_summaries = format_wiki_summaries(wikipedia_results)
173
+ all_sources += formatted_summaries
174
+ parsed_summaries = parse_list_to_dicts(formatted_summaries)
175
+ add_many(parsed_summaries)
176
+ #all_sources += create_wikipedia_urls_from_text(wikipedia_results)
177
  return wikipedia_results
178
 
179
  @tool