klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[graph_improvement]: Unique Id removed from update prompt (#2020)

Browse Source
This commit is contained in:
Mayank
2024-11-09 03:13:24 +05:30
committed by GitHub
parent eaf295756e
commit 6d535951df
2 changed files with 21 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
mem0/memory/graph_memory.py
View File

@@ -1,5 +1,7 @@
import logging
from mem0.memory.utils import format_entities
try:
from langchain_community.graphs import Neo4jGraph
except ImportError:
@@ -94,8 +96,8 @@ class MemoryGraph:
extracted_entities = []
logger.debug(f"Extracted entities: {extracted_entities}")
update_memory_prompt = get_update_memory_messages(search_output, extracted_entities)
search_output_string = format_entities(search_output)
update_memory_prompt = get_update_memory_messages(search_output_string, extracted_entities)
_tools = [UPDATE_MEMORY_TOOL_GRAPH, ADD_MEMORY_TOOL_GRAPH, NOOP_TOOL]
if self.llm_provider in ["azure_openai_structured", "openai_structured"]: