Ubuntu系统下ChatGPT集成实战：从命令行到API调用的完整指南

11次阅读

没有评论

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

在 Ubuntu 系统中集成 ChatGPT 可以为开发者带来诸多便利，如自动化代码生成、技术问题解答、文档摘要等。然而，实际操作中常遇到以下困难：

API 调用流程复杂，需要处理认证、请求格式和响应解析
Python 环境配置容易出现包冲突
网络请求性能受限于 OpenAI 的速率限制
生产环境部署需要考虑安全性和稳定性

本文将带领开发者一步步解决这些问题，实现 ChatGPT 在 Ubuntu 系统中的无缝集成。

在开始之前，我们需要确保系统满足以下要求：

Ubuntu 18.04 或更高版本
Python 3.8+ 环境
有效的 OpenAI API 密钥

sudo apt update
sudo apt install python3-pip python3-venv -y

python3 -m venv chatgpt_env
source chatgpt_env/bin/activate

pip install openai requests python-dotenv

下面是一个完整的 Python 示例，展示如何通过 OpenAI API 与 ChatGPT 交互：

import openai
import os
from dotenv import load_dotenv

# 加载环境变量
load_dotenv()

# 设置 API 密钥
openai.api_key = os.getenv('OPENAI_API_KEY')

# 定义 ChatGPT 交互函数
def chat_with_gpt(prompt, model="gpt-3.5-turbo", max_tokens=150):
    try:
        response = openai.ChatCompletion.create(
            model=model,
            messages=[{"role": "user", "content": prompt}],
            max_tokens=max_tokens,
            temperature=0.7,
        )
        return response.choices[0].message.content
    except openai.error.OpenAIError as e:
        print(f"OpenAI API 错误: {str(e)}")
        return None
    except Exception as e:
        print(f"未知错误: {str(e)}")
        return None

# 示例调用
if __name__ == "__main__":
    user_input = input("请输入您的问题:")
    response = chat_with_gpt(user_input)
    if response:
        print("ChatGPT 回复:", response)

使用 python-dotenv 管理敏感信息，避免 API 密钥硬编码
实现了基本的错误处理机制，捕获 OpenAI API 特定错误
提供了可配置的模型选择和最大 token 数
设置了适当的 temperature 值以获得平衡的创造性响应

我们可以将上述代码扩展为一个命令行工具：

#!/usr/bin/env python3
import argparse

# 添加参数解析
parser = argparse.ArgumentParser(description='ChatGPT 命令行接口')
parser.add_argument('prompt', type=str, help='要发送给 ChatGPT 的提示')
parser.add_argument('--model', type=str, default="gpt-3.5-turbo", 
                    help='使用的模型名称')
parser.add_argument('--max-tokens', type=int, default=150,
                    help='响应最大 token 数')

args = parser.parse_args()

response = chat_with_gpt(args.prompt, args.model, args.max_tokens)
print(response)

保存为 chatgpt_cli.py 并添加可执行权限后，即可通过命令行调用：

chmod +x chatgpt_cli.py
./chatgpt_cli.py "如何优化 Python 代码性能?" --max-tokens 200

在生产环境中使用 ChatGPT API 时，需要考虑以下因素：

OpenAI API 有严格的速率限制，建议：

实现请求队列和重试机制
监控 API 使用情况
考虑使用 time.sleep() 在请求之间添加延迟

完善错误处理应包括：

try:
    response = chat_with_gpt(prompt)
except openai.error.RateLimitError:
    print("达到速率限制，请稍后重试")
except openai.error.APIConnectionError:
    print("网络连接问题")
except openai.error.InvalidRequestError:
    print("无效请求，请检查参数")