今天我将为大家分享一段简单易用的Python代码，快速实现一个python对话框简易聊天机器人，有了这个基础以后，后续可以实现各种其他机器人，例如微信机器人，公众号机器人。

Python代码

直接上代码，直接运行即可
import requests  # pip install requests
import json

class ChatBot:
    def __init__(self, api_key, model="gpt-4", url="https://openai.note-card.com/v1/chat/completions"):
        self.api_key = api_key
        self.model = model
        self.url = url
        self.headers = {
            'Accept': 'application/json',
            'Content-Type': 'application/json',
            'Authorization': f'Bearer {self.api_key}'
        }

        self.messages = [
            {
                "role": "system",
                "content": "你是一个聊天机器人"
            }
        ]

    def chat(self, user_message):
        self.messages.append({"role": "user", "content": user_message})

        payload = json.dumps({
            "model": self.model,
            "messages": self.messages
        })

        response = requests.post(self.url, headers=self.headers, data=payload)

        if response.status_code == 200:
            response_data = response.json()
            bot_reply = response_data['choices'][0]['message']['content']
            self.messages.append({"role": "assistant", "content": bot_reply})
            return bot_reply
        else:
            return f"Error: {response.status_code}, {response.text}"

def main():
    api_key = "你的API密钥"  
    # 这个 key 有 1美元的额度，如果用完了，可以来这里购买 https://openai.note-card.com/register?aff=BB5Y
    # 现在的价格是 0.53 元 = 1 美元
    bot = ChatBot(api_key)
    print("你好! 你可以开始与机器人对话了. 输入 'exit' 来结束对话.")

    while True:
        user_input = input("你: ")
        if user_input.lower() == 'exit':
            break

        reply = bot.chat(user_input)
        print(f"机器人: {reply}")

if __name__ == "__main__":
    main()

安装依赖

导入必要的库

import requests  # pip install requests
import json

我们使用 requests 库来发送HTTP请求，json 库用于处理JSON数据。

注意事项

  • API密钥保护：在实际使用中，请确保不要将你的API密钥暴露在公开的代码仓库或公共平台上。可以使用环境变量或配置文件来管理密钥，提高安全性。
  • 费用说明：代码注释中提到的API密钥有1美元的额度，使用完毕后可以通过提供的链接购买更多额度。目前价格为0.53元/美元。

推荐使用我们的API平台

如果你对API的使用有更多需求，或者希望获得更高的额度和更稳定的服务，欢迎访问我们的API平台：https://openai.note-card.com/register?aff=Spq3

为什么选择我们的API平台？

  • 高可靠性：经过笔记卡片微信机器人的压力测试，稳定性可靠。
  • 灵活计费：用多少花多少。
  • 优质支持：可以添加反馈群，随时问问题。
如果代码有任何问题，添加微信号：note_card

点击阅读全文可以更方便的复制代码