Skip to content

Commit f79ea7c

Browse files
committed
fix: kwargs bug
1 parent f593b58 commit f79ea7c

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

src/memos/mem_reader/multi_model_struct.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,17 +48,18 @@ def _concat_multi_model_memories(
4848

4949
@timed
5050
def _process_multi_model_data(
51-
self, scene_data_info: MessagesType, info, **kwargs
51+
self, scene_data_info: MessagesType, info, mode: str = "fine", **kwargs
5252
) -> list[TextualMemoryItem]:
5353
"""
5454
Process multi-model data using MultiModelParser.
5555
5656
Args:
5757
scene_data_info: MessagesType input
5858
info: Dictionary containing user_id and session_id
59+
mode: mem-reader mode, fast for quick process while fine for
60+
better understanding via calling llm
5961
**kwargs: Additional parameters (mode, etc.)
6062
"""
61-
mode = kwargs.get("mode", "fine")
6263
# Pop custom_tags from info (same as simple_struct.py)
6364
# must pop here, avoid add to info, only used in sync fine mode
6465
custom_tags = info.pop("custom_tags", None) if isinstance(info, dict) else None

src/memos/mem_reader/read_multi_model/multi_model_parser.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -231,6 +231,9 @@ def process_transfer(
231231

232232
# Parse in fine mode (pass custom_tags to parse_fine)
233233
try:
234+
context_items = kwargs.pop("custom_tags", None)
235+
custom_tags = kwargs.pop("custom_tags", None)
236+
info = kwargs.pop("info", None)
234237
return parser.parse_fine(
235238
message, info, context_items=context_items, custom_tags=custom_tags, **kwargs
236239
)

0 commit comments

Comments
 (0)