fix: GeneralTextMemory._embed_one_sentence Argument Error (#76)

* fix: GeneralTextMemory._embed_one_sentence Argument Error

* fix: fix unittest

Author: J1awei-Yang

src/memos/memories/textual/general.py

@@ -202,7 +202,7 @@ def drop(
 
     def _embed_one_sentence(self, sentence: str) -> list[float]:
         """Embed a single sentence."""
-        return self.embedder.embed(sentence)[0]
+        return self.embedder.embed([sentence])[0]
 
 
 EXTRACTION_PROMPT_PART_1 = f"""You are a memory extractor. Your task is to extract memories from the given messages.

tests/memories/textual/test_general.py

@@ -98,7 +98,7 @@ def test_embed_one_sentence(self):
 
         embedding = self.memory._embed_one_sentence(sentence)
 
-        self.mock_embedder.embed.assert_called_once_with(sentence)
+        self.mock_embedder.embed.assert_called_once_with([sentence])
         self.assertEqual(embedding, expected_embedding)
 
     def test_extract(self):
@@ -206,7 +206,7 @@ def test_update_memory(self):
         self.memory.update(memory_id_to_update, new_memory_dict)
 
         self.mock_embedder.embed.assert_called_once_with(
-            "This is the updated memory content via dict."
+            ["This is the updated memory content via dict."]
         )
 
         args, _ = self.mock_vector_db.update.call_args
@@ -271,7 +271,7 @@ def test_search_memories(self):
 
         search_results = self.memory.search(query, top_k)
 
-        self.mock_embedder.embed.assert_called_once_with(query)
+        self.mock_embedder.embed.assert_called_once_with([query])
         self.mock_vector_db.search.assert_called_once_with(query_embedding, top_k)
 
         self.assertEqual(len(search_results), top_k)