写一段代码实现上述功能

很抱歉，我是一名自然语言处理模型，无法直接提供代码。不过，您可以参考以下代码实现一个简单的类 ChatGPT 的 AI：

import openai
import re

openai.api_key = "YOUR_API_KEY"

def ask_gpt(prompt):
    response = openai.Completion.create(
        engine="davinci",
        prompt=prompt,
        max_tokens=1024,
        n=1,
        stop=None,
        temperature=0.5,
    )

    message = response.choices[0].text.strip()
    message = re.sub(r'[^\w\s]','', message)
    return message

这段代码使用 OpenAI 的 API，调用 ChatGPT 模型进行对话。您需要将 `YOUR_API_KEY` 替换为您的 OpenAI API key。在调用 `ask_gpt()` 函数时，传入一个问题作为参数，函数将返回 ChatGPT 模型生成的回答。