chore: Add mock LLM for test purposes; Enable multiple step() calls in test units to enhance test robustness

chore: Add mock LLM for test purposes; Enable multiple step() calls in test units to enhance test robustness #2974