From bbddb98aca3bbe8f544b0e0faeb55f69b6691104 Mon Sep 17 00:00:00 2001 From: Anusha Yella <136127167+techcontributor@users.noreply.github.com> Date: Mon, 9 Sep 2024 22:56:04 +0530 Subject: [PATCH] Update docstring (#1837) Co-authored-by: Anu --- mem0/memory/graph_memory.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/mem0/memory/graph_memory.py b/mem0/memory/graph_memory.py index 138c5119..7cdeb025 100644 --- a/mem0/memory/graph_memory.py +++ b/mem0/memory/graph_memory.py @@ -44,10 +44,7 @@ class MemoryGraph: Args: data (str): The data to add to the graph. - stored_memories (list): A list of stored memories. - - Returns: - dict: A dictionary containing the entities added to the graph. + filters (dict): A dictionary containing filters to be applied during the addition. """ # retrieve the search results @@ -217,6 +214,7 @@ class MemoryGraph: Args: query (str): Query to search for. + filters (dict): A dictionary containing filters to be applied during the search. Returns: dict: A dictionary containing: @@ -262,7 +260,7 @@ class MemoryGraph: Retrieves all nodes and relationships from the graph database based on optional filtering criteria. Args: - all_memories (list): A list of dictionaries, each containing: + filters (dict): A dictionary containing filters to be applied during the retrieval. Returns: list: A list of dictionaries, each containing: - 'contexts': The base data store response for each memory. @@ -297,6 +295,7 @@ class MemoryGraph: source (str): The name of the source node. target (str): The name of the target node. relationship (str): The type of the relationship. + filters (dict): A dictionary containing filters to be applied during the update. Raises: Exception: If the operation fails.