scriptoriumadmin/ReMe
1
0
Fork 0
mirror of https://github.com/agentscope-ai/ReMe.git synced 2026-09-06 08:16:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

es update without reduntant embedding

Browse source
This commit is contained in:
xianzhe.xxz 2024-07-23 15:50:43 +08:00
parent 46ce7d8cb7
commit 7d11369f5a
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
memory_scope/storage/llama_index_es_memory_store.py
View file

@ -124,9 +124,10 @@ class LlamaIndexEsMemoryStore(BaseMemoryStore):
Returns:
TextNode: The converted TextNode with content and metadata from the MemoryNode.
"""
"""
return TextNode(id_=memory_node.memory_id,
text=memory_node.content,
embedding=memory_node.vector,
metadata=memory_node.model_dump(exclude={"content"}))
@staticmethod
@ -140,4 +141,4 @@ class LlamaIndexEsMemoryStore(BaseMemoryStore):
Returns:
MemoryNode: The converted MemoryNode with text and metadata from the NodeWithScore.
"""
return MemoryNode(content=text_node.text, **text_node.metadata)
return MemoryNode(content=text_node.text, vector=text_node.embedding, **text_node.metadata)