Improvements around reporting and work directory retention options

tests/test_config.py

@@ -18,6 +18,7 @@ from audita.config import (
     DEFAULT_NORMALIZE_MAX_SEGMENT_GAP,
     DEFAULT_NORMALIZE_MAX_SEGMENT_TOKENS,
     DEFAULT_WORK_DIR,
+    DEFAULT_WORK_DIR_RETENTION,
 )
 from audita.errors import AuditaConfigError
 
@@ -50,6 +51,8 @@ def test_config_uses_defaults_with_api_key():
     assert config.normalize_max_segment_tokens == DEFAULT_NORMALIZE_MAX_SEGMENT_TOKENS
     assert config.normalize_max_segment_tokens == 2048
     assert config.work_dir == Path(DEFAULT_WORK_DIR)
+    assert config.work_dir_retention == DEFAULT_WORK_DIR_RETENTION
+    assert config.work_dir_retention == "auto"
 
 
 def test_config_env_overrides_defaults():
@@ -70,6 +73,7 @@ def test_config_env_overrides_defaults():
             "AUDITA_NORMALIZE_MAX_SEGMENT_DURATION": "45.0",
             "AUDITA_NORMALIZE_MAX_SEGMENT_TOKENS": "512",
             "AUDITA_WORK_DIR": "/tmp/custom-audita",
+            "AUDITA_WORK_DIR_RETENTION": "always",
         }
     )
 
@@ -87,6 +91,7 @@ def test_config_env_overrides_defaults():
     assert config.normalize_max_segment_duration == 45.0
     assert config.normalize_max_segment_tokens == 512
     assert config.work_dir == Path("/tmp/custom-audita")
+    assert config.work_dir_retention == "always"
 
 
 def test_config_cli_overrides_env():
@@ -105,6 +110,7 @@ def test_config_cli_overrides_env():
             "AUDITA_NORMALIZE_MAX_SEGMENT_DURATION": "45.0",
             "AUDITA_NORMALIZE_MAX_SEGMENT_TOKENS": "512",
             "AUDITA_WORK_DIR": "/tmp/env-audita",
+            "AUDITA_WORK_DIR_RETENTION": "always",
         },
         overrides=ConfigOverrides(
             max_section_tokens=100,
@@ -121,6 +127,7 @@ def test_config_cli_overrides_env():
             normalize_max_segment_duration=30.0,
             normalize_max_segment_tokens=256,
             work_dir=Path("/tmp/cli-audita"),
+            work_dir_retention="never",
         ),
     )
 
@@ -138,6 +145,7 @@ def test_config_cli_overrides_env():
     assert config.normalize_max_segment_duration == 30.0
     assert config.normalize_max_segment_tokens == 256
     assert config.work_dir == Path("/tmp/cli-audita")
+    assert config.work_dir_retention == "never"
 
 
 def test_config_requires_api_key():
@@ -195,6 +203,13 @@ def test_config_rejects_invalid_grammar_validation_enabled():
         )
 
 
+def test_config_rejects_invalid_work_dir_retention():
+    with pytest.raises(AuditaConfigError):
+        AuditaConfig.from_sources(
+            env={"OPENROUTER_API_KEY": "key", "AUDITA_WORK_DIR_RETENTION": "sometimes"}
+        )
+
+
 def test_legacy_confidence_threshold_env_is_ignored():
     config = AuditaConfig.from_sources(
         env={"OPENROUTER_API_KEY": "key", "AUDITA_CONFIDENCE_THRESHOLD": "0.9"}