diff --git a/modules/models/base_model.py b/modules/models/base_model.py index 00db9243..da265f45 100644 --- a/modules/models/base_model.py +++ b/modules/models/base_model.py @@ -465,10 +465,19 @@ def prepare_inputs( ) elif use_websearch: search_results = [] - with DDGS() as ddgs: - ddgs_gen = ddgs.text(fake_inputs, backend="lite") - for r in islice(ddgs_gen, 10): - search_results.append(r) + with retrieve_proxy() as proxy: + if proxy[0] or proxy[1]: + proxies = {} + if proxy[0]: + proxies["http"] = proxy[0] + if proxy[1]: + proxies["https"] = proxy[1] + else: + proxies = None + with DDGS(proxies=proxies) as ddgs: + ddgs_gen = ddgs.text(fake_inputs, backend="lite") + for r in islice(ddgs_gen, 10): + search_results.append(r) reference_results = [] for idx, result in enumerate(search_results): logging.debug(f"搜索结果{idx + 1}:{result}") diff --git a/requirements.txt b/requirements.txt index e33971a2..06a54de4 100644 --- a/requirements.txt +++ b/requirements.txt @@ -19,7 +19,7 @@ openpyxl pandoc wolframalpha faiss-cpu==1.7.4 -duckduckgo-search>=3.9.5 +duckduckgo-search>=4.1.1 arxiv wikipedia google.generativeai