怎么调用文心一言的api接口生成一个简单的聊天机器人（python代码）

这篇文章距离上次修改已过207天，其中的内容可能已经有所变动。

要调用文心一言的API接口生成一个简单的聊天机器人，你需要使用Python的requests库来发送HTTP请求。以下是一个简单的例子：

首先，安装requests库（如果你还没有安装的话）：

pip install requests

然后，使用以下Python代码创建一个简单的聊天机器人：

import requests

def send_message(message):
    # 文心一言的API接口地址
    api_url = "https://openapi.baidu.com/oauth/2.0/token"
    # 替换为你的API Key和Secret Key
    api_key = "YOUR_API_KEY"
    secret_key = "YOUR_SECRET_KEY"

    # 获取access_token
    response = requests.post(api_url, data={
        'grant_type': 'client_credentials',
        'client_id': api_key,
        'client_secret': secret_key
    })
    access_token = response.json()['access_token']

    # 设置文心一言的对话API接口
    text_generate_url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/chat/completions"

    # 发送消息
    response = requests.post(text_generate_url, data={
        "session_id": "chatbot",  # 可以自定义，表示会话ID
        "log_id": "123",  # 可以自定义，表示日志ID
        "request": {
            "query": message,  # 用户输入的消息
            "user_id": "test_user"  # 用户ID
        }
    }, headers={
        'Content-Type': 'application/json',
        'Authorization': 'Bearer ' + access_token
    })

    # 解析返回的消息
    response_json = response.json()
    if 'results' in response_json and len(response_json['results']) > 0:
        return response_json['results'][0]['values']['text']
    else:
        return "对不起，我无法理解你的问题。"

# 用户与机器人交互的示例
while True:
    message = input("你: ")
    if message.strip() != '':
        reply = send_message(message)
        print("机器人: ", reply)

在使用这段代码之前，请确保你已经从百度AI开放平台申请了文心一言的API Key和Secret Key，并且替换了代码中的YOUR_API_KEY和YOUR_SECRET_KEY。

这个简易的聊天机器人会源源不断地接收用户输入的消息，并返回文心一言预测的回复。你可以根据需要扩展这个简单的聊天机器人，比如添加更复杂的会话处理逻辑、上下文管理、多轮对话等功能。

怎么调用文心一言的api接口生成一个简单的聊天机器人（python代码）

评论已关闭

推荐阅读