让AI对话更顺畅——文心一言Python循环调用教程

后端

2023-11-13 11:32:59

文心一言对话顺畅攻略：Python循环调用与自定义询问终极教程

前言

文心一言作为百度推出的强大对话式AI，赋予开发者解锁顺畅自然的人机对话的无限可能。然而，在使用过程中，困扰不少开发者的是，每次交互都需要重新输入问题，导致对话支离破碎，严重影响用户体验。

别再烦恼！本教程将带你深入探索文心一言循环调用 和自定义询问 的秘密，让你轻松实现流畅无碍的AI对话，大幅提升交互体验。

文心一言API准备

在开启我们的探索之旅前，你需要准备一个文心一言账号，并获取API Key 和Secret Key 。详细获取流程可参考百度官方文档。

Python代码环境

接下来，让我们搭建Python代码环境。首先，导入必要的库：

import requests
import json

API请求发送函数

为了方便与文心一言进行交互，我们定义了一个发送API请求的函数：

def send_request(prompt, API_Key, Secret_Key):
  """
  发送API请求

  Args:
    prompt: 要发送的提示
    API_Key: API Key
    Secret_Key: Secret Key

  Returns:
    API返回的JSON格式结果
  """

  # 构造请求头
  headers = {
      "Content-Type": "application/json",
      "apiKey": API_Key,
      "secretKey": Secret_Key,
  }

  # 构造请求体
  data = {
      "prompt": prompt,
  }

  # 发送请求
  response = requests.post(
      "https://api.wenxin.baidu.com/generate",
      headers=headers,
      data=json.dumps(data),
  )

  # 返回结果
  return response.json()

循环调用

循环调用允许我们在每次交互中将上一次的回复作为新的提示，从而实现连续的对话。代码如下：

prompt = "你好，我是你的AI助手。"

while True:
  # 发送请求
  response = send_request(prompt, API_Key, Secret_Key)

  # 获取回复
  reply = response["candidates"][0]["output"]

  # 打印回复
  print("AI:")
  print(reply)

  # 更新提示
  prompt = "我:" + input()

自定义询问

除了循环调用，我们还可以通过prompt参数自定义要询问的问题：

prompt = "今天的天气怎么样？"

# 发送请求
response = send_request(prompt, API_Key, Secret_Key)

# 获取回复
reply = response["candidates"][0]["output"]

# 打印回复
print("AI:")
print(reply)