import json

from tavily import TavilyClient


def tavily_ai(prompt):
    """
    :param prompt:
    :return:
    """
    tavily = TavilyClient(api_key="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")
    response = tavily.search(
        query="{}, 通过这个标题生成一遍文章".format(prompt),
        search_depth="advanced",
        include_images=True,
        include_answer=True,
        include_raw_content=True

    )
    return response


title = "易中天对退休人的忠告，看了才没白活"

res = tavily_ai(title)
print(json.dumps(res, ensure_ascii=False, indent=4))