如何使用 OpenAI 在 Python 中构建简单的聊天机器人 [分步指南]-Python教程-PHP中文网

如何使用 OpenAI 在 Python 中构建简单的聊天机器人 [分步指南]

WBOY

发布： 2024-08-31 06:02:33

原创

539 人浏览过

How to Build a Simple Chatbot in Python Using OpenAI [Step-by-Step Guide]

创建聊天机器人从未如此简单！借助 OpenAI 强大的 API，您只需几个步骤即可使用 Python 构建一个简单而有效的聊天机器人。本指南将引导您完成整个过程，非常适合初学者和开发人员。让我们深入了解吧！？

？你将学到什么

在本教程中，您将学习如何：

安装 OpenAI Python 库
设置您的 OpenAI API 密钥
编写 Python 代码与 OpenAI API 交互
为您的聊天机器人构建持续的对话循环

最后，您将拥有一个功能齐全的聊天机器人，您可以自定义和扩展它。准备好开始了吗？我们走吧！

？先决条件

在我们开始之前，请确保您已经：

Python 3.7+ 安装在您的计算机上吗？
OpenAI API 密钥？（您可以通过在 OpenAI 注册获得一个）

️第1步：安装OpenAI Python库

为了与 OpenAI 的 API 交互，我们需要安装 openai Python 包。打开终端并运行：

pip install openai

登录后复制

这将安装最新版本的 OpenAI Python 客户端库。

？第 2 步：设置您的 OpenAI API 密钥

安装库后，下一步是在 Python 脚本中设置 OpenAI API 密钥。您可以将其设置为环境变量，也可以直接在代码中设置（请注意，不建议在生产环境中直接包含它）。

以下是如何在 Python 代码中包含 API 密钥：

import openai

# Set up your OpenAI API key
openai.api_key = "your-api-key-here"

登录后复制

⚠️ 重要： 将“your-api-key-here”替换为您来自 OpenAI 的实际 API 密钥。

？第 3 步：编写聊天机器人功能

接下来，我们将创建一个 Python 函数，将用户的输入发送到 OpenAI API 并返回聊天机器人的响应。

def chat_with_openai(user_input):
    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",  # Use the GPT-3.5 model
        messages=[
            {"role": "system", "content": "You are a helpful assistant."},  # System message
            {"role": "user", "content": user_input},  # User input
        ]
    )

    # Return the chatbot's reply
    return response['choices'][0]['message']['content']

登录后复制

？第 4 步：构建持续的对话循环

为了使聊天机器人具有交互性，我们需要构建一个允许持续对话的循环。

def start_chatbot():
    print("? Welcome! I'm your chatbot. Type 'exit' to end the chat.\n")

    while True:
        user_input = input("You: ")

        if user_input.lower() == 'exit':
            print("Goodbye! ?")
            break

        response = chat_with_openai(user_input)
        print(f"Bot: {response}\n")

登录后复制

？第 5 步：运行您的聊天机器人

现在，您所要做的就是运行 start_chatbot() 函数即可开始与您的机器人聊天！

if __name__ == "__main__":
    start_chatbot()

登录后复制

？恭喜！您已经构建了您的聊天机器人

就是这样！您现在已经有了一个使用 Python 和 OpenAI 构建的简单聊天机器人。您可以扩展此机器人来处理更复杂的对话，添加上下文感知等功能，或将其集成到 Web 应用程序中。

？聊天机器人的完整 Python 代码

这是聊天机器人的完整 Python 代码：

import openai

# Set up your OpenAI API key
openai.api_key = "your-api-key-here"

# Function to interact with OpenAI
def chat_with_openai(user_input):
    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=[
            {"role": "system", "content": "You are a helpful assistant."},
            {"role": "user", "content": user_input},
        ]
    )
    return response['choices'][0]['message']['content']

# Function to start the chatbot
def start_chatbot():
    print("? Welcome! I'm your chatbot. Type 'exit' to end the chat.\n")

    while True:
        user_input = input("You: ")
        if user_input.lower() == 'exit':
            print("Goodbye! ?")
            break
        response = chat_with_openai(user_input)
        print(f"Bot: {response}\n")

# Start the chatbot
if __name__ == "__main__":
    start_chatbot()

登录后复制