当前位置：首页 > 文章列表 > 文章 > python教程 > 如何使用 OpenAI 在 Python 中构建简单的聊天机器人 [分步指南]

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

来源：dev.to 2024-09-06 09:27:56 0浏览收藏

一分耕耘，一分收获！既然都打开这篇《如何使用 OpenAI 在 Python 中构建简单的聊天机器人 [分步指南]》，就坚持看下去，学下去吧！本文主要会给大家讲到等等知识点，如果大家对本文有好的建议或者看到有不足之处，非常欢迎大家积极提出！在后续文章我会继续更新文章相关的内容，希望对大家都有所帮助！

创建聊天机器人从未如此简单！借助 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()