帮我给它写一个测试函数

好的，以下是一个简单的测试函数：

from chitchat import ChitGPT

def test_chitchat():
    # 创建 ChitGPT 的实例
    chitchat_model = ChitGPT()

    # 开始与 AI 进行对话
    while True:
        # 获取用户输入
        user_input = input("You: ")

        # 生成 AI 的回复
        response = chitchat_model.generate_response(user_input)

        # 输出 AI 的回复
        print("AI: " + response)

        # 如果用户输入 'exit'，则退出对话
        if user_input.lower() == "exit":
            break

您可以通过调用 `generate_response` 方法来获取 AI 的回复。在这个测试函数中，我们创建了一个 ChitGPT 的实例，然后开始与 AI 进行对话。当用户输入 'exit'时，对话结束。