diff --git a/src/memos/memories/textual/tree_text_memory/retrieve/searcher.py b/src/memos/memories/textual/tree_text_memory/retrieve/searcher.py index 8c30d74f3..7aa12bb60 100644 --- a/src/memos/memories/textual/tree_text_memory/retrieve/searcher.py +++ b/src/memos/memories/textual/tree_text_memory/retrieve/searcher.py @@ -288,6 +288,9 @@ def _parse_task( if parsed_goal.memories: query_embedding = self.embedder.embed(list({query, *parsed_goal.memories})) + if query_embedding is None: + query_embedding = self.embedder.embed([query]) + return parsed_goal, query_embedding, context, query @timed