ChatGPT自定义功能开发指南：从零构建你的第一个AI助手

1次阅读

共计 2533 个字符，预计需要花费 7 分钟才能阅读完成。

ChatGPT API 是 OpenAI 提供的一项强大工具，允许开发者通过编程方式与 ChatGPT 模型进行交互。它可以用于构建各种 AI 应用，如智能客服、内容生成助手、编程辅助工具等。通过 API，开发者可以定制对话流程、控制响应风格，并集成到自己的应用程序中。

对于初学者来说，ChatGPT API 提供了一个相对简单的入口点来体验 AI 技术的强大功能。它基于 RESTful 架构，支持多种编程语言调用，其中 Python 是最常用的选择之一。

在开始自定义功能开发时，开发者常会遇到以下几个问题：

上下文丢失：ChatGPT 本身是无状态的，如果不妥善管理对话历史，AI 会 ” 忘记 ” 之前的对话内容
响应延迟：API 调用需要网络往返，可能导致用户体验不佳
长对话处理：当对话内容超过模型的最大 token 限制时，会出现截断问题
错误处理：网络波动或 API 限制可能导致调用失败
成本控制：不当的使用方式可能导致意外的高额费用

有效的对话上下文管理是构建连贯对话体验的关键。以下是实现建议：

维护一个对话历史列表，存储所有消息
每条消息都应标记角色（”user” 或 ”assistant”）
定期清理过时的对话内容以避免 token 超限
重要信息可以显式地 ” 告诉 ”AI 记住

使用 Python 调用 ChatGPT API 时，建议遵循以下模式：

使用官方 OpenAI Python 库
合理设置超时参数
实现重试机制应对临时故障
添加速率限制避免触发 API 限制

处理长对话时可以考虑：

定期总结对话内容
优先保留最近的对话历史
对于超长内容，可以分段处理
使用更高级的模型（如 gpt-4-32k）处理超长上下文

以下是一个完整的自定义聊天机器人实现示例：

import openai
import logging
from typing import List, Dict

# 配置日志
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

class ChatBot:
    def __init__(self, api_key: str, model: str = "gpt-3.5-turbo"):
        """ 初始化聊天机器人

        Args:
            api_key: OpenAI API 密钥
            model: 使用的模型名称
        """
        openai.api_key = api_key
        self.model = model
        self.conversation_history: List[Dict] = []

    def add_message(self, role: str, content: str) -> None:
        """ 添加消息到对话历史

        Args:
            role: 消息角色（"user" 或 "assistant"）content: 消息内容
        """self.conversation_history.append({"role": role,"content": content})

    def get_response(self, user_input: str, max_tokens: int = 150) -> str:
        """ 获取 AI 响应

        Args:
            user_input: 用户输入
            max_tokens: 最大返回 token 数

        Returns:
            AI 生成的响应内容
        """self.add_message("user", user_input)

        try:
            response = openai.ChatCompletion.create(
                model=self.model,
                messages=self.conversation_history,
                max_tokens=max_tokens,
                temperature=0.7
            )

            ai_response = response.choices[0].message.content
            self.add_message("assistant", ai_response)

            # 简单的对话历史管理，防止过长
            if len(self.conversation_history) > 10:
                self.conversation_history = self.conversation_history[-6:]

            return ai_response

        except Exception as e:
            logger.error(f"API 调用失败: {str(e)}")
            return "抱歉，我遇到了一些问题，请稍后再试。"

# 使用示例
if __name__ == "__main__":
    bot = ChatBot(api_key="your-api-key-here")

    while True:
        user_input = input("你:")
        if user_input.lower() in ["退出", "exit", "quit"]:
            break

        response = bot.get_response(user_input)
        print(f"AI 助手: {response}")