File tree Expand file tree Collapse file tree 1 file changed +4
-1
lines changed
src/memos/memories/textual/tree_text_memory/retrieve Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Original file line number Diff line number Diff 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 )
You can’t perform that action at this time.
0 commit comments