fix(sdk): removing deprecating param from our sdk and docs changes with it (#4740)

tests/memory/test_apache_age_e2e.py

@@ -16,10 +16,10 @@ Run:
 
 import json
 import os
-import pytest
 from unittest.mock import MagicMock, patch
 
 import age
+import pytest
 
 from mem0.memory.apache_age_memory import MemoryGraph  # noqa: E402
 
@@ -386,7 +386,7 @@ class TestPublicAPICRUD:
             )
         self.mg.ag.commit()
 
-        results = self.mg.get_all({"user_id": "u1"}, limit=3)
+        results = self.mg.get_all({"user_id": "u1"}, top_k=3)
         assert len(results) == 3
 
     def test_delete_all_removes_user_data(self):

tests/memory/test_apache_age_memory.py

@@ -175,7 +175,7 @@ class TestGetAll:
             {"source": "alice", "relationship": "KNOWS", "target": "bob"},
             {"source": "alice", "relationship": "LIKES", "target": "hiking"},
         ])
-        results = instance.get_all({"user_id": "u1"}, limit=10)
+        results = instance.get_all({"user_id": "u1"}, top_k=10)
         assert len(results) == 2
         assert results[0]["source"] == "alice"
         assert results[0]["relationship"] == "KNOWS"
@@ -187,7 +187,7 @@ class TestGetAll:
         instance._exec_cypher = MagicMock(return_value=[
             {"source": "n0", "relationship": "R", "target": "m0"},
         ])
-        instance.get_all({"user_id": "u1"}, limit=3)
+        instance.get_all({"user_id": "u1"}, top_k=3)
         # Verify limit was passed as a parameter to the query
         cypher_stmt = instance._exec_cypher.call_args[0][0]
         assert "LIMIT %s" in cypher_stmt

tests/memory/test_graph_memory_soft_delete.py

@@ -88,7 +88,7 @@ class TestSearchExcludesSoftDeleted:
 
     def test_get_all_filters_soft_deleted(self):
         mg = _create_graph_memory()
-        mg.get_all(filters={"user_id": "user1"}, limit=10)
+        mg.get_all(filters={"user_id": "user1"}, top_k=10)
 
         cypher = mg.graph.query.call_args[0][0]
         assert "r.valid IS NULL OR r.valid = true" in cypher
@@ -271,7 +271,7 @@ class TestSoftDeleteWithFilters:
 
     def test_get_all_with_agent_id_filters_soft_deleted(self):
         mg = _create_graph_memory()
-        mg.get_all(filters={"user_id": "user1", "agent_id": "agent1"}, limit=10)
+        mg.get_all(filters={"user_id": "user1", "agent_id": "agent1"}, top_k=10)
 
         cypher = mg.graph.query.call_args[0][0]
         assert "r.valid IS NULL OR r.valid = true" in cypher

tests/memory/test_main.py

@@ -35,7 +35,7 @@ class TestAddToVectorStoreErrors:
 
         memory = Memory()
         memory.config = mocker.MagicMock()
-        memory.config.custom_fact_extraction_prompt = None
+        memory.config.custom_instructions = None
         memory.config.custom_update_memory_prompt = None
         memory.api_version = "v1.1"
 
@@ -139,7 +139,7 @@ class TestAsyncAddToVectorStoreErrors:
 
         memory = AsyncMemory()
         memory.config = mocker.MagicMock()
-        memory.config.custom_fact_extraction_prompt = None
+        memory.config.custom_instructions = None
         memory.config.custom_update_memory_prompt = None
         memory.api_version = "v1.1"
 
@@ -187,7 +187,7 @@ def _build_memory_instance(mocker, memory_cls):
     mocker.patch("mem0.memory.main.MEM0_TELEMETRY", False)
     memory = memory_cls()
     memory.config = mocker.MagicMock()
-    memory.config.custom_fact_extraction_prompt = None
+    memory.config.custom_instructions = None
     memory.config.custom_update_memory_prompt = None
     memory.api_version = "v1.1"
     memory.vector_store = mocker.MagicMock()
@@ -309,8 +309,8 @@ def test_create_then_search_and_get_all_return_same_timestamps(mocker):
     memory.vector_store.list.return_value = [[mem_result]]
 
     # Step 3: Call search and get_all, compare timestamps
-    search_results = memory._search_vector_store("pizza", filters={"user_id": "alice"}, limit=10, threshold=None)
-    get_all_results = memory._get_all_from_vector_store(filters={"user_id": "alice"}, limit=100)
+    search_results = memory._search_vector_store("pizza", filters={"user_id": "alice"}, top_k=10, threshold=None)
+    get_all_results = memory._get_all_from_vector_store(filters={"user_id": "alice"}, top_k=100)
 
     search_item = search_results[0]
     get_all_item = get_all_results[0]
@@ -376,8 +376,8 @@ def test_search_and_get_all_consistent_after_update(mocker):
     memory.vector_store.search.return_value = [mem_result]
     memory.vector_store.list.return_value = [[mem_result]]
 
-    search_results = memory._search_vector_store("pizza", filters={"user_id": "alice"}, limit=10, threshold=None)
-    get_all_results = memory._get_all_from_vector_store(filters={"user_id": "alice"}, limit=100)
+    search_results = memory._search_vector_store("pizza", filters={"user_id": "alice"}, top_k=10, threshold=None)
+    get_all_results = memory._get_all_from_vector_store(filters={"user_id": "alice"}, top_k=100)
 
     assert search_results[0]["created_at"] == get_all_results[0]["created_at"]
     assert search_results[0]["updated_at"] == get_all_results[0]["updated_at"]

tests/memory/test_neptune_analytics_memory.py

@@ -1,8 +1,10 @@
 import unittest
 from unittest.mock import MagicMock, patch
+
 import pytest
-from mem0.graphs.neptune.neptunegraph import MemoryGraph
+
 from mem0.graphs.neptune.base import NeptuneBase
+from mem0.graphs.neptune.neptunegraph import MemoryGraph
 
 
 class TestNeptuneMemory(unittest.TestCase):
@@ -134,7 +136,7 @@ class TestNeptuneMemory(unittest.TestCase):
             mock_bm25_instance.get_top_n.return_value = reranked_results
 
             # Call the search method
-            result = self.memory_graph.search("Find Alice", self.test_filters, limit=5)
+            result = self.memory_graph.search("Find Alice", self.test_filters, top_k=5)
 
             # Verify the method calls
             self.memory_graph._retrieve_nodes_from_data.assert_called_once_with("Find Alice", self.test_filters)
@@ -162,7 +164,7 @@ class TestNeptuneMemory(unittest.TestCase):
         self.mock_graph.query.return_value = mock_query_result
 
         # Call the get_all method
-        result = self.memory_graph.get_all(self.test_filters, limit=10)
+        result = self.memory_graph.get_all(self.test_filters, top_k=10)
 
         # Verify the method calls
         self.memory_graph._get_all_cypher.assert_called_once_with(self.test_filters, 10)
@@ -256,7 +258,7 @@ class TestNeptuneMemory(unittest.TestCase):
         self.mock_graph.query.side_effect = [mock_query_result1, mock_query_result2]
 
         # Call the _search_graph_db method
-        result = self.memory_graph._search_graph_db(node_list, self.test_filters, limit=10)
+        result = self.memory_graph._search_graph_db(node_list, self.test_filters, top_k=10)
 
         # Verify the method calls
         self.assertEqual(self.mock_embedding_model.embed.call_count, 2)

tests/memory/test_neptune_memory.py

@@ -190,7 +190,7 @@ class TestNeptuneMemory(unittest.TestCase):
             mock_bm25_instance.get_top_n.return_value = reranked_results
 
             # Call the search method
-            result = self.memory_graph.search("Find Alice", self.test_filters, limit=5)
+            result = self.memory_graph.search("Find Alice", self.test_filters, top_k=5)
 
             # Verify the method calls
             self.memory_graph._retrieve_nodes_from_data.assert_called_once_with("Find Alice", self.test_filters)
@@ -218,7 +218,7 @@ class TestNeptuneMemory(unittest.TestCase):
         self.mock_graph.query.return_value = mock_query_result
 
         # Call the get_all method
-        result = self.memory_graph.get_all(self.test_filters, limit=10)
+        result = self.memory_graph.get_all(self.test_filters, top_k=10)
 
         # Verify the method calls
         self.memory_graph._get_all_cypher.assert_called_once_with(self.test_filters, 10)
@@ -331,7 +331,7 @@ class TestNeptuneMemory(unittest.TestCase):
         self.mock_graph.query.side_effect = [mock_query_result1, mock_query_result2]
 
         # Call the _search_graph_db method
-        result = self.memory_graph._search_graph_db(node_list, self.test_filters, limit=10)
+        result = self.memory_graph._search_graph_db(node_list, self.test_filters, top_k=10)
 
         # Verify the method calls
         self.assertEqual(self.mock_embedding_model.embed.call_count, 2)