diff --git a/ra_aid/tools/memory.py b/ra_aid/tools/memory.py index 960dec8..2152fe1 100644 --- a/ra_aid/tools/memory.py +++ b/ra_aid/tools/memory.py @@ -55,8 +55,6 @@ _global_memory: Dict[str, Any] = { "plans": [], "tasks": {}, # Dict[int, str] - ID to task mapping "task_id_counter": 1, # Counter for generating unique task IDs - "key_facts": {}, # Dict[int, str] - ID to fact mapping (deprecated, using DB now) - "key_fact_id_counter": 1, # Counter for generating unique fact IDs (deprecated, using DB now) "implementation_requested": False, "related_files": {}, # Dict[int, str] - ID to filepath mapping "related_file_id_counter": 1, # Counter for generating unique file IDs diff --git a/tests/ra_aid/tools/test_agent.py b/tests/ra_aid/tools/test_agent.py index 8087867..d07b6e0 100644 --- a/tests/ra_aid/tools/test_agent.py +++ b/tests/ra_aid/tools/test_agent.py @@ -15,8 +15,6 @@ from ra_aid.tools.memory import _global_memory @pytest.fixture def reset_memory(): """Reset global memory before each test""" - _global_memory["key_facts"] = {} - _global_memory["key_fact_id_counter"] = 0 _global_memory["research_notes"] = [] _global_memory["plans"] = [] _global_memory["tasks"] = {} @@ -26,8 +24,6 @@ def reset_memory(): _global_memory["work_log"] = [] yield # Clean up after test - _global_memory["key_facts"] = {} - _global_memory["key_fact_id_counter"] = 0 _global_memory["research_notes"] = [] _global_memory["plans"] = [] _global_memory["tasks"] = {} diff --git a/tests/ra_aid/tools/test_memory.py b/tests/ra_aid/tools/test_memory.py index 4c63fa1..24effe1 100644 --- a/tests/ra_aid/tools/test_memory.py +++ b/tests/ra_aid/tools/test_memory.py @@ -29,8 +29,6 @@ from ra_aid.database.models import KeyFact @pytest.fixture def reset_memory(): """Reset global memory before each test""" - _global_memory["key_facts"] = {} - _global_memory["key_fact_id_counter"] = 0 _global_memory["research_notes"] = [] _global_memory["plans"] = [] _global_memory["tasks"] = {} @@ -40,8 +38,6 @@ def reset_memory(): _global_memory["work_log"] = [] yield # Clean up after test - _global_memory["key_facts"] = {} - _global_memory["key_fact_id_counter"] = 0 _global_memory["research_notes"] = [] _global_memory["plans"] = [] _global_memory["tasks"] = {}