Fixed context usage calculation

test/web_socket_manager_test.py

@@ -31,7 +31,9 @@ class WebSocketManagerTest(AioHTTPTestCase):
             "memory_used": 1000,
             "memory_total": 2000,
             "disk_used": 5000,
-            "disk_total": 10000
+            "disk_total": 10000,
+            "token_count": 100,
+            "token_limit": 1000
         }
         
         # Create minimal mocks for other components
@@ -41,6 +43,8 @@ class WebSocketManagerTest(AioHTTPTestCase):
         self.mock_llm.infer.side_effect = mock_infer
         self.mock_validator = Mock(spec=XMLValidator)
         self.mock_validator.validate.return_value = None
+        self.mock_llm.token_count.return_value = 100
+        self.mock_llm.token_limit.return_value = 1000
         
         # Create parser with real IO buffer
         self.parser = ResponseParser(self.io_buffer)
@@ -50,7 +54,7 @@ class WebSocketManagerTest(AioHTTPTestCase):
             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