diff --git a/deepsearcher/agent/deep_search.py b/deepsearcher/agent/deep_search.py index a804109..f2e48e3 100644 --- a/deepsearcher/agent/deep_search.py +++ b/deepsearcher/agent/deep_search.py @@ -158,7 +158,7 @@ class DeepSearch(BaseAgent): self.max_iter = max_iter self.route_collection = route_collection self.text_window_splitter = text_window_splitter - self.web_search = WebSearch() if web_search else None + self.web_search = False def invoke(self, query: str, dim: int, **kwargs) -> list[str]: """ @@ -223,7 +223,7 @@ class DeepSearch(BaseAgent): content = self.llm.remove_think(content) return self.llm.literal_eval(content) - def _search_chunks(self, query: str) -> list[RetrievalResult]: + def _search_chunks(self, query: str, **kwargs) -> list[RetrievalResult]: results = [] # 本地向量搜索 @@ -248,6 +248,7 @@ class DeepSearch(BaseAgent): send_info(f"本地向量搜索找到 {len(vector_results)} 个结果") # 网页搜索 + self.web_search = WebSearch() if kwargs.get('web_search', False) else None if self.web_search: web_results = self.web_search.search_with_retry(query, size=2) if web_results: @@ -362,7 +363,7 @@ class DeepSearch(BaseAgent): # Execute all search tasks sequentially for query in sub_queries: - results = self._search_chunks(query) + results = self._search_chunks(query, **kwargs) all_search_results.extend(results) diff --git a/deepsearcher/online_query.py b/deepsearcher/online_query.py index 48203a5..3ca20af 100644 --- a/deepsearcher/online_query.py +++ b/deepsearcher/online_query.py @@ -21,7 +21,8 @@ def query(original_query: str, **kwargs) -> tuple[str, list[RetrievalResult]]: """ default_searcher = configuration.default_searcher max_iter = kwargs.get("max_iter", 3) - return default_searcher.query(original_query, max_iter=max_iter) + web_search = kwargs.get("web_search", False) + return default_searcher.query(original_query, max_iter=max_iter, web_search=web_search) def retrieve(original_query: str, max_iter: int | None = None) -> tuple[list[RetrievalResult], list[str]]: diff --git a/deepsearcher/templates/html/index.html b/deepsearcher/templates/html/index.html index 64323c8..418fbd8 100644 --- a/deepsearcher/templates/html/index.html +++ b/deepsearcher/templates/html/index.html @@ -86,6 +86,18 @@ value="3" /> +