Skip to content

Commit 445a43b

Browse files
committed
fix: GeneralTextMemory._embed_one_sentence Argument Error
1 parent c90f241 commit 445a43b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/memos/memories/textual/general.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -202,7 +202,7 @@ def drop(
202202

203203
def _embed_one_sentence(self, sentence: str) -> list[float]:
204204
"""Embed a single sentence."""
205-
return self.embedder.embed(sentence)[0]
205+
return self.embedder.embed([sentence])[0]
206206

207207

208208
EXTRACTION_PROMPT_PART_1 = f"""You are a memory extractor. Your task is to extract memories from the given messages.

0 commit comments

Comments
 (0)