Skip to content

Commit 9615282

Browse files
author
黑布林
committed
turn off graph recall
1 parent 390ba29 commit 9615282

File tree

1 file changed

+4
-1
lines changed
  • src/memos/memories/textual/tree_text_memory/retrieve

1 file changed

+4
-1
lines changed

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

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,9 @@ def __init__(
5959
# Create internet retriever from config if provided
6060
self.internet_retriever = internet_retriever
6161
self.moscube = moscube
62+
self.vec_cot = (
63+
search_strategy.get("vec_cot", "false") == "true" if search_strategy else False
64+
)
6265
self.use_fast_graph = (
6366
search_strategy.get("fast_graph", "false") == "true" if search_strategy else False
6467
)
@@ -373,7 +376,7 @@ def _retrieve_from_long_term_and_user(
373376

374377
# chain of thinking
375378
cot_embeddings = []
376-
if self.search_strategy["vec_cot"]:
379+
if self.vec_cot:
377380
queries = self._cot_query(query)
378381
if len(queries) > 1:
379382
cot_embeddings = self.embedder.embed(queries)

0 commit comments

Comments
 (0)