autogen/test/autogen/test_responsive_agent.py

import sys
from io import StringIO
import pytest
from flaml.autogen.agent import ResponsiveAgent


def test_responsive_agent(monkeypatch):
    dummy_agent_1 = ResponsiveAgent(name="dummy_agent_1", human_input_mode="ALWAYS")
    dummy_agent_2 = ResponsiveAgent(name="dummy_agent_2", human_input_mode="TERMINATE")

    monkeypatch.setattr(sys, "stdin", StringIO("exit"))
    dummy_agent_1.receive("hello", dummy_agent_2)  # receive a str
    monkeypatch.setattr(sys, "stdin", StringIO("TERMINATE\n\n"))
    dummy_agent_1.receive(
        {
            "content": "hello",
        },
        dummy_agent_2,
    )  # receive a dict

    # receive dict without openai fields to be printed, such as "content", 'function_call'. There should be no error raised.
    pre_len = len(dummy_agent_1.oai_conversations["dummy_agent_2"])
    dummy_agent_1.receive({"message": "hello"}, dummy_agent_2)
    assert pre_len == len(
        dummy_agent_1.oai_conversations["dummy_agent_2"]
    ), "When the message is not an valid openai message, it should not be appended to the oai conversation."

    monkeypatch.setattr(sys, "stdin", StringIO("exit"))
    dummy_agent_1.send("TERMINATE", dummy_agent_2)  # send a str
    monkeypatch.setattr(sys, "stdin", StringIO("exit"))
    dummy_agent_1.send(
        {
            "content": "TERMINATE",
        },
        dummy_agent_2,
    )  # send a dict

    # receive dict with no openai fields
    pre_len = len(dummy_agent_1.oai_conversations["dummy_agent_2"])
    dummy_agent_1.send({"message": "hello"}, dummy_agent_2)  # send dict with wrong field

    assert pre_len == len(
        dummy_agent_1.oai_conversations["dummy_agent_2"]
    ), "When the message is not a valid openai message, it should not be appended to the oai conversation."


if __name__ == "__main__":
    test_responsive_agent(pytest.monkeypatch)