Skip to content

Commit 87170e3

Browse files
feat(memcube): single cube use general_llm (#1253)
* feat: add backend for openai * feat: change memreader for scheaduer * feat: change singlecube llm --------- Co-authored-by: harvey_xiang <harvey_xiang@163.com>
1 parent b82af21 commit 87170e3

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/memos/multi_mem_cube/single_cube.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -617,7 +617,7 @@ def add_before_search(
617617

618618
# 3. Call LLM
619619
try:
620-
raw = self.mem_reader.llm.generate([{"role": "user", "content": prompt}])
620+
raw = self.mem_reader.general_llm.generate([{"role": "user", "content": prompt}])
621621
success, parsed_result = parse_keep_filter_response(raw)
622622

623623
if not success:

0 commit comments

Comments
 (0)