oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2026-01-20 15:40:23 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Properly filter out failed web search downloads from attachments

Browse source
This commit is contained in:
oobabooga 2025-06-08 19:25:23 -07:00
parent f3388c2ab4
commit f9a007c6a8
1 changed files with 4 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
modules/web_search.py
View file

@ -38,7 +38,7 @@ def download_web_page(url, timeout=5):
return text
except Exception as e:
logger.error(f"Error downloading {url}: {e}")
return f"[Error downloading content from {url}: {str(e)}]"
return ""
def perform_web_search(query, num_pages=3, max_workers=5):
@ -74,9 +74,7 @@ def perform_web_search(query, num_pages=3, max_workers=5):
'url': url,
'content': content
}
except Exception as e:
logger.error(f"Error downloading {url}: {e}")
# Include failed downloads with empty content
except Exception:
search_results[index] = {
'title': title,
'url': url,
@ -108,7 +106,7 @@ def add_web_search_attachments(history, row_idx, user_message, search_query, sta
return
# Filter out failed downloads before adding attachments
successful_results = [result for result in search_results if result['content'] and result['content'].strip()]
successful_results = [result for result in search_results if result['content'].strip()]
if not successful_results:
logger.warning("No successful downloads to add as attachments")
@ -130,7 +128,7 @@ def add_web_search_attachments(history, row_idx, user_message, search_query, sta
}
history['metadata'][key]["attachments"].append(attachment)
logger.info(f"Added {len(successful_results)} successful web search results as attachments")
logger.info(f"Added {len(successful_results)} successful web search results as attachments.")
except Exception as e:
logger.error(f"Error in web search: {e}")