Function side_effect

tests/agent/test_conversational_agent.py:29–55

(*args, **kwargs)

def side_effect(*args, **kwargs):
    message = kwargs["messages"][0]["content"]

    if "good" in message:
        return {
            "choices": [
                {
                    "message": {
                        "role": "assistant",
                        "content": "yes, question is resolved",
                    }
                }
            ],
            "usage": 10,
        }
    else:
        return {
            "choices": [
                {
                    "message": {
                        "role": "assistant",
                        "content": "no, question is not resolved",
                    }
                }
            ],
            "usage": 10,
        }


@pytest.fixture