Improved the system prompt

2024-11-04 14:10:49 +01:00
parent b4c70cd5b8
commit 5da6dca5ec
7 changed files with 150 additions and 214 deletions
--- a/test/system_metrics_test.py
+++ b/test/system_metrics_test.py
@@ -121,7 +121,7 @@ class SystemMetricsTest(unittest.TestCase):
        self.validate_usage_values([
            ("CPU", int(context.get("cpu"))),
            ("GPU", int(context.get("gpu"))),
-            ("Context", int(context.get("context")))
+            ("Context", int(context.get("context_usage")))
        ])
        
        # Validate stdin buffer size
@@ -169,7 +169,7 @@ def test_cpu_usage_detection(self):
        for value in test_values:
            context = self.metrics.generate_context(value)
            self.assertEqual(
-                int(context.get("context")),
+                int(context.get("context_usage")),
                round(value * 100)
            )