isayahc commited on
Commit
6592db0
·
unverified ·
1 Parent(s): 08754e8

Now google sources include the session_id

Browse files
rag_app/structured_tools/structured_tools.py CHANGED
@@ -80,8 +80,13 @@ def google_search(query: str) -> str:
80
  search_results:dict = websearch.results(query, 3)
81
  print(search_results)
82
  if len(search_results)>1:
 
83
  cleaner_sources =format_search_results(search_results)
84
  parsed_csources = parse_list_to_dicts(cleaner_sources)
 
 
 
 
85
  db.add_many(parsed_csources)
86
  else:
87
  cleaner_sources = search_results
 
80
  search_results:dict = websearch.results(query, 3)
81
  print(search_results)
82
  if len(search_results)>1:
83
+ # add session id
84
  cleaner_sources =format_search_results(search_results)
85
  parsed_csources = parse_list_to_dicts(cleaner_sources)
86
+
87
+ # add the session id to each element in `parsed_csources`
88
+ [i.update({"session_id":db.session_id}) for i in parsed_csources]
89
+
90
  db.add_many(parsed_csources)
91
  else:
92
  cleaner_sources = search_results