Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
29 commits
Select commit Hold shift + click to select a range
2189062
fix: reorganizer bug (#64)
CaralHsi Jul 11, 2025
c90f241
feat: add ark/volcengine embedder support (#57)
ioo0s Jul 12, 2025
2012dd1
fix: GeneralTextMemory._embed_one_sentence Argument Error (#76)
J1awei-Yang Jul 15, 2025
339c756
feat: add neo4j share-db example (#59)
CaralHsi Jul 15, 2025
244e748
feat: add universal api embedder (#81)
CaralHsi Jul 15, 2025
86c999a
feat: Add embedder consistency validation with warning and tests (#69)
Dustyposa Jul 16, 2025
b7636eb
Feat: add demo api and scheduler vllm etc (#83)
fridayL Jul 16, 2025
a6b1574
feat: add download_examples command (#87)
Ki-Seki Jul 16, 2025
29b19a0
feat: add stale issue and PR management workflow (#89)
Ki-Seki Jul 16, 2025
a9ffe4f
feat: update issue and pr templates (#98)
Ki-Seki Jul 16, 2025
ba74bf1
Add More backend (#99)
Gaojianli Jul 16, 2025
90fbbfa
feat: add multi language (#92)
CaralHsi Jul 16, 2025
2387f6d
feat(eval): add longmemeval evaluation pipeline (#104)
Duguce Jul 17, 2025
bfc32d8
Feat:change deafult config for product (#108)
fridayL Jul 17, 2025
a0990be
Support customized system prompt base (#102)
newmeta-ben Jul 17, 2025
b4d5a88
feat: recursively cluster nodes to max_cluster_size (#105)
Nyakult Jul 17, 2025
9cbed6d
feat: return ids when add memory (#119)
CaralHsi Jul 17, 2025
8eb0738
Feat: add simple memos for user could easy start (#118)
fridayL Jul 18, 2025
4572d30
Feat: More Llm API (#113)
CaralHsi Jul 19, 2025
aa4f1f9
feat: update doc mem reader (#123)
Nyakult Jul 19, 2025
9dc520d
build: optimize dependencies management (#120)
Ki-Seki Jul 19, 2025
f00483a
Fix None issue in reorganizer queue (#125)
J1awei-Yang Jul 19, 2025
a3dfeae
feat: update PR template to include documentation issue/PR link (#126)
Ki-Seki Jul 19, 2025
56d6fd5
feat: add mcp serve support and client demo (#127)
fridayL Jul 20, 2025
956a9d7
Feat: neo4j community (#121)
CaralHsi Jul 21, 2025
01797eb
feat(example): add MOSProduct multi-user locomo example with shared N…
CaralHsi Jul 21, 2025
04e7dda
fix: update Python version constraints in project files (#129)
Ki-Seki Jul 21, 2025
c42bf96
chore: change version number to v0.2.1
fridayL Jul 21, 2025
727dc18
feat: one-click deployment with docker (#133)
tianxing02 Jul 21, 2025
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
fix: GeneralTextMemory._embed_one_sentence Argument Error (#76)
* fix: GeneralTextMemory._embed_one_sentence Argument Error

* fix: fix unittest
  • Loading branch information
J1awei-Yang authored Jul 15, 2025
commit 2012dd1578ace9392b1aad89e69e539fb7febea0
2 changes: 1 addition & 1 deletion src/memos/memories/textual/general.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand Down
6 changes: 3 additions & 3 deletions tests/memories/textual/test_general.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down Expand Up @@ -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
Expand Down Expand Up @@ -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)
Expand Down