feat: add manual internet

Author: fridayL

src/memos/api/handlers/chat_handler.py

@@ -213,7 +213,7 @@ def generate_chat_response() -> Generator[str, None, None]:
                         query=chat_req.query,
                         top_k=20,
                         session_id=chat_req.session_id,
-                        mode=SearchMode.FINE if chat_req.internet_search else SearchMode.FAST,
+                        mode=SearchMode.FAST,
                         internet_search=chat_req.internet_search,  # TODO this param is not worked at fine mode
                         moscube=chat_req.moscube,
                         chat_history=chat_req.history,

src/memos/memories/textual/tree.py

@@ -161,7 +161,7 @@ def search(
         info=None,
         mode: str = "fast",
         memory_type: str = "All",
-        manual_close_internet: bool = False,
+        manual_close_internet: bool = True,
         moscube: bool = False,
         search_filter: dict | None = None,
         user_name: str | None = None,
@@ -189,9 +189,6 @@ def search(
             list[TextualMemoryItem]: List of matching memories.
         """
         if (self.internet_retriever is not None) and manual_close_internet:
-            logger.warning(
-                "Internet retriever is init by config , but  this search set manual_close_internet is True  and will close it"
-            )
             searcher = Searcher(
                 self.dispatcher_llm,
                 self.graph_store,
@@ -201,6 +198,7 @@ def search(
                 internet_retriever=None,
                 moscube=moscube,
                 search_strategy=self.search_strategy,
+                manual_close_internet=manual_close_internet
             )
         else:
             searcher = Searcher(
@@ -212,6 +210,7 @@ def search(
                 internet_retriever=self.internet_retriever,
                 moscube=moscube,
                 search_strategy=self.search_strategy,
+                manual_close_internet=manual_close_internet
             )
         return searcher.search(
             query, top_k, info, mode, memory_type, search_filter, user_name=user_name

src/memos/memories/textual/tree_text_memory/retrieve/searcher.py

@@ -43,6 +43,7 @@ def __init__(
         internet_retriever: None = None,
         moscube: bool = False,
         search_strategy: dict | None = None,
+        manual_close_internet: bool = True,
     ):
         self.graph_store = graph_store
         self.embedder = embedder
@@ -58,7 +59,7 @@ def __init__(
         self.moscube = moscube
         self.vec_cot = search_strategy.get("cot", False) if search_strategy else False
         self.use_fast_graph = search_strategy.get("fast_graph", False) if search_strategy else False
-
+        self.manual_close_internet = manual_close_internet
         self._usage_executor = ContextThreadPoolExecutor(max_workers=4, thread_name_prefix="usage")
 
     @timed
@@ -458,7 +459,7 @@ def _retrieve_from_internet(
         user_id: str | None = None,
     ):
         """Retrieve and rerank from Internet source"""
-        if not self.internet_retriever or mode == "fast":
+        if not self.internet_retriever or self.manual_close_internet:
             logger.info(f"[PATH-C] '{query}' Skipped (no retriever, fast mode)")
             return []
         if memory_type not in ["All"]: