add rich as runtime dependency and improve pytest coverage

Author: yacinemebarki

packages/agent-os/pyproject.toml

@@ -44,6 +44,7 @@ classifiers = [
 # Core dependencies (minimal - just what's needed for kernel)
 dependencies = [
     "pydantic>=2.4.0",
+    "rich>=13.0.0"
 ]
 
 [project.urls]

packages/agent-os/tests/test_cli_output.py

@@ -1,25 +1,30 @@
-# test_cli_run.py
-from agent_os.policies.cli import success, error, warn, policy_violation, passed_check
+from agent_os.policies.cli import success,error,warn,policy_violation,passed_check
 
-def main():
-    print("=== Testing CLI functions with colored output ===\n")
-
-    print("1️⃣  success():")
+def test_success(capsys):
     success("This is a success message")
+    captured = capsys.readouterr()
+    assert "success message" in captured.out.lower()
+
 
-    print("\n2️⃣  error():")
+def test_error(capsys):
     error("This is an error message")
+    captured = capsys.readouterr()
+    assert "error message" in captured.err.lower()
 
-    print("\n3️⃣  warn():")
+
+def test_warn(capsys):
     warn("This is a warning message")
+    captured = capsys.readouterr()
+    assert "warning message" in captured.out.lower()
 
-    print("\n4️⃣  policy_violation():")
-    policy_violation("Policy violation detected!")
 
-    print("\n5️⃣  passed_check():")
-    passed_check("Test passed successfully!")
+def test_policy_violation(capsys):
+    policy_violation("Policy violation detected!")
+    captured = capsys.readouterr()
+    assert "policy violation" in captured.out.lower()
 
-    print("\n=== Done testing CLI functions ===")
 
-if __name__ == "__main__":
-    main()
+def test_passed_check(capsys):
+    passed_check("Test passed successfully!")
+    captured = capsys.readouterr()
+    assert "test passed" in captured.out.lower()