Skip to content

Commit 47d2ad6

Browse files
committed
fix: mem-reader bug
1 parent 4f49ee6 commit 47d2ad6

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

src/memos/mem_reader/simple_struct.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import concurrent.futures
22
import copy
33
import json
4+
import os
45
import re
56

67
from abc import ABC
@@ -19,11 +20,12 @@
1920
SIMPLE_STRUCT_DOC_READER_PROMPT,
2021
SIMPLE_STRUCT_DOC_READER_PROMPT_ZH,
2122
SIMPLE_STRUCT_MEM_READER_EXAMPLE,
22-
SIMPLE_STRUCT_MEM_READER_PROMPT_ZH,
23-
SIMPLE_STRUCT_MEM_READER_PROMPT,
2423
SIMPLE_STRUCT_MEM_READER_EXAMPLE_ZH,
24+
SIMPLE_STRUCT_MEM_READER_PROMPT,
25+
SIMPLE_STRUCT_MEM_READER_PROMPT_ZH,
2526
)
2627

28+
2729
logger = log.get_logger(__name__)
2830
PROMPT_DICT = {
2931
"chat": {
@@ -207,7 +209,7 @@ def get_scene_data_info(self, scene_data: list, type: str) -> list[str]:
207209
elif type == "doc":
208210
for item in scene_data:
209211
try:
210-
if not isinstance(item, str):
212+
if os.path.exists(item):
211213
parsed_text = parser.parse(item)
212214
results.append({"file": "pure_text", "text": parsed_text})
213215
else:

0 commit comments

Comments
 (0)