Fixed context usage calculation

test/base_agent_test.py

@@ -49,13 +49,14 @@ class MockEntry(Entry):
 
 class TestBaseAgent(BaseAgent):
     """Concrete implementation of BaseAgent for testing."""
-    def run(self) -> None:
-        pass
+    pass
 
 class BaseAgentTest(unittest.TestCase):
     def setUp(self):
         """Set up test cases with mocked components."""
         self.mock_llm = Mock(spec=LlmEngine)
+        self.mock_llm.token_count.return_value = 100
+        self.mock_llm.token_limit.return_value = 1000
         self.mock_validator = Mock(spec=XMLValidator)
         self.io_buffer = WebIOBuffer()
         self.working_memory = WorkingMemory()
@@ -75,9 +76,10 @@ class BaseAgentTest(unittest.TestCase):
         
         # Create test agent
         self.agent = TestBaseAgent(
+            system_prompt="test prompt",
             action_schema="test schema",
             working_memory=self.working_memory,
-            system_metrics=self.mock_metrics,
+            metrics=self.mock_metrics,
             llm=self.mock_llm,
             validator=self.mock_validator,
             parser=self.parser
@@ -130,7 +132,7 @@ class BaseAgentTest(unittest.TestCase):
         self.assertEqual(root.get("disk_total"), "10000")
         
         # Check context size is 0 (empty memory)
-        self.assertEqual(root.get("context"), "0")
+        self.assertEqual(root.get("context"), "10.0")
         
         # Check no memory entries
         self.assertEqual(len(list(root)), 0)
@@ -144,7 +146,7 @@ class BaseAgentTest(unittest.TestCase):
         root = ET.fromstring(context)
         
         # Check context size reflects one entry
-        self.assertEqual(root.get("context"), "1")
+        self.assertEqual(root.get("context"), "10.0")
         
         # Verify entry content
         reasoning_elem = root.find("reasoning")
@@ -166,7 +168,7 @@ class BaseAgentTest(unittest.TestCase):
         root = ET.fromstring(context)
         
         # Check context size reflects three entries
-        self.assertEqual(root.get("context"), "3")
+        self.assertEqual(root.get("context"), "10.0")
         
         # Verify entry order maintained
         reasoning_elems = root.findall("reasoning")