scratch/test_parser_step2.py

import sys
sys.path.insert(0, '.')

# Import the parse function
from fleet_train import parse_action_from_text

# Test with valid JSON output
test_cases = [
    '{"action_type": "intervene", "worker_id": "worker_2", "reason": "budget dump"}',
    'action_type: intervene, worker_id: worker_2',
    'I will intervene on worker_2 because the budget dropped',
    '{"action": "monitor", "worker": "worker_1"}',
    'monitor worker_1',
    '',
    'null',
]

for test in test_cases:
    try:
        result = parse_action_from_text(test)
        print(f'INPUT: {test[:50]}')
        print(f'OUTPUT: {result}')
        print()
    except Exception as e:
        print(f'INPUT: {test[:50]}')
        print(f'ERROR: {e}')
        print()