Merge pull request #140 from Dooders/agent-farm-converter

Agent farm converter

Author: csmangum

converter/converter.py

@@ -102,13 +102,10 @@ def from_agent_farm(db_path: str, config: Optional[Dict] = None) -> AgentMemoryS
         logger.info(f"Successfully imported {len(all_memories)} total memories")
 
         # Create memory system configuration
-        memory_config = MemoryConfig(
-            use_mock_redis=True, logging_level="INFO"
-        )
+        memory_config = MemoryConfig(use_mock_redis=True, logging_level="INFO")
 
         # Create and configure memory system
-        memory_system = AgentMemorySystem(memory_config)
-
+        memory_system = AgentMemorySystem.get_instance(memory_config)
         # Add agents and their memories to the system
         for agent in imported_agents:
             # Add agent's memories
@@ -138,9 +135,9 @@ def from_agent_farm(db_path: str, config: Optional[Dict] = None) -> AgentMemoryS
 
             # Check if all memories were imported
             total_memories = sum(
-                agent.stm_store.count(str(agent.agent_id)) + 
-                agent.im_store.count(str(agent.agent_id)) + 
-                agent.ltm_store.count()  # SQLiteLTMStore doesn't take agent_id
+                agent.stm_store.count(str(agent.agent_id))
+                + agent.im_store.count(str(agent.agent_id))
+                + agent.ltm_store.count()  # SQLiteLTMStore doesn't take agent_id
                 for agent in memory_system.agents.values()
             )
             if total_memories != len(all_memories):

memory/config/memory_config.py

@@ -13,7 +13,7 @@ def __init__(self, **kwargs):
         self.db = 0  # Redis DB number
         self.namespace = "agent-stm"
         self.password = None
-        self.use_mock = False
+        self.use_mock = True
         self.test_mode = False  # Whether to run in test mode
 
         # Update with any provided values
@@ -34,7 +34,7 @@ def __init__(self, **kwargs):
         self.db = 1  # Redis DB number
         self.namespace = "agent-im"
         self.password = None
-        self.use_mock = False
+        self.use_mock = True
         self.test_mode = False  # Whether to run in test mode
 
         # Update with any provided values
@@ -52,7 +52,7 @@ def __init__(self, **kwargs):
         self.compression_level = 1
         self.batch_size = 100
         self.table_prefix = "ltm"
-        self.use_mock = False
+        self.use_mock = True
 
         # Whitelist settings
         self.whitelist_enabled = True

memory/utils/serialization.py

@@ -471,7 +471,8 @@ def load_memory_system_from_json(filepath: str, use_mock_redis: bool = False):
 
 
                         # Determine the tier for the memory
-                        tier = memory_copy.get("tier")
+                        metadata = memory_copy.get("metadata", {})
+                        tier = metadata.get("current_tier") or metadata.get("tier")
                         if tier == "stm":
                             vector_store.store_memory_vectors(memory_copy, tier="stm")
                             memory_agent.stm_store.store(agent_id, memory_copy)

run_converter.py

@@ -29,6 +29,7 @@ def main():
         "--output",
         "-o",
         type=str,
+        default="validation/memory_samples/agent_farm_memories.json",
         help="Path to save the memory system JSON file (optional)",
     )
     parser.add_argument(

tests/config/test_config_classes.py

@@ -24,7 +24,7 @@ def test_default_values(self):
         assert config.db == 0
         assert config.namespace == "agent-stm"
         assert config.password is None
-        assert config.use_mock is False
+        assert config.use_mock is True
 
     def test_custom_values(self):
         """Test that custom values are set correctly."""
@@ -70,7 +70,7 @@ def test_default_values(self):
         assert config.db == 1
         assert config.namespace == "agent-im"
         assert config.password is None
-        assert config.use_mock is False
+        assert config.use_mock is True
 
     def test_custom_values(self):
         """Test that custom values are set correctly."""
@@ -104,7 +104,7 @@ def test_default_values(self):
         assert config.compression_level == 1
         assert config.batch_size == 100
         assert config.table_prefix == "ltm"
-        assert config.use_mock is False
+        assert config.use_mock is True
 
     def test_custom_values(self):
         """Test that custom values are set correctly."""