diff --git a/src/memos/multi_mem_cube/single_cube.py b/src/memos/multi_mem_cube/single_cube.py index 6df410c19..6a91f436f 100644 --- a/src/memos/multi_mem_cube/single_cube.py +++ b/src/memos/multi_mem_cube/single_cube.py @@ -617,7 +617,7 @@ def add_before_search( # 3. Call LLM try: - raw = self.mem_reader.llm.generate([{"role": "user", "content": prompt}]) + raw = self.mem_reader.general_llm.generate([{"role": "user", "content": prompt}]) success, parsed_result = parse_keep_filter_response(raw) if not success: