Method test_replay_with_chatmessage

tests/test_monitoring.py:190–201

Regression test for dict(message) to message.dict() fix

(self)

class ReplayTester(unittest.TestCase):
    def test_replay_with_chatmessage(self):
        """Regression test for dict(message) to message.dict() fix"""
        logger = AgentLogger()
        memory = AgentMemory(system_prompt="test")
        step = ActionStep(step_number=1, timing=0)
        step.model_input_messages = [ChatMessage(role=MessageRole.USER, content="Hello")]
        memory.steps.append(step)

        try:
            memory.replay(logger, detailed=True)
        except TypeError as e:
            self.fail(f"Replay raised an error: {e}")


class AgentLoggerLogTaskTester(unittest.TestCase):