如何使用python调用百度AI“文心一言”接口自动写文？

现在AI越来越流行，前期16给大家分享过用python对接Openai实现批量写作的方法，今天给大家分享一个不需要魔法也不需要代理，就能实现AI批量写作的方法--使用文心一言接口。

文心一言是百度的一个AI写作程序，我们现在还可以免费使用3.5模型，但是如果要批量写文章就需要用程序来对接官方的接口，然后实现批量且自动的写作和保存本地，或者说发布的这么一个过程！

我们用到的开发语言：

python

编程思路：

1.在百度文心一言官网申请或购买他的key，后面我们需要填写在代码中一个是“API_KEY”，一个是“SECRET_KEY”
2.调用接口
3.返回结果的处理（后续我们可以保存可以对文章进行二次处理等等操作）

python代码部分

def chat_with_ai(user_message):
    access_token = get_access_token()

    if access_token:
        url = f"https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions?access_token={access_token}"
        
        payload = json.dumps({
            "messages": [
                {
                    "role": "user",
                    "content": user_message
                },
            ]
        })
        headers = {
            'Content-Type': 'application/json'
        }
        
        response = requests.post(url, headers=headers, data=payload)

        if response.status_code == 200:
            data_obj = json.loads(response.text)
            result = data_obj['result']
            print(result)
            return result
        else:
            return ""
    else:
        return ""

def get_access_token():
    url = "https://aip.baidubce.com/oauth/2.0/token"
    params = {"grant_type": "client_credentials", "client_id": API_KEY, "client_secret": SECRET_KEY}
    response = requests.post(url, params=params)
    access_token = response.json().get("access_token")
    return access_token

代码讲解

这里打包好两个函数，get_access_token()大概意思可以理解成，使用自己的密匙链接百度文心一言，chat_with_ai(user_message)这个函数里判断第一个函数，如果链接成功就进行投喂数据，也可以理解成对文心一言进行提问，传进去的user_message这个参数就是我们的问题描述。
这样理解起来简单吧，都封装好的函数了，直接在python里就可以正常调用！前提是，前面也提到了要去申请KEY,有两个值需要你在前面定义API_KEY和SECRET_KEY，这个是官网接口对接提供的。
最后返回的result就是文心一言返回给你的文章，后续自己想怎么处理就在于你个人了！
16原创文，请勿转载！