Spaces:
Sleeping
Sleeping
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
|