AI代码生成工具如何调用ChatGPT：从API集成到实战避坑指南

1次阅读

没有评论

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

在开发过程中，手动编写重复性代码不仅效率低下，还容易引入错误。虽然 ChatGPT API 能显著提升代码生成效率，但开发者常遇到以下挑战：

API 认证流程复杂，需要处理密钥管理和请求签名
响应限流导致服务不稳定
不同编程语言下的实现差异大
生产环境中突发错误难以处理

优点：实现简单，兼容性广
缺点：每次请求都需要建立新连接
适用场景：低频调用或简单集成

优点：长连接减少握手开销
缺点：实现复杂度高
适用场景：实时交互式应用

优点：封装完善，错误处理机制健全
缺点：语言支持有限
适用场景：快速原型开发

import requests
import os

# 从环境变量获取 API 密钥
API_KEY = os.getenv('OPENAI_API_KEY')
HEADERS = {'Authorization': f'Bearer {API_KEY}',
    'Content-Type': 'application/json'
}

# 构造请求参数
payload = {
    'model': 'gpt-3.5-turbo',
    'messages': [{'role': 'system', 'content': '你是一个专业的代码助手'},
        {'role': 'user', 'content': '用 Python 实现快速排序'}
    ],
    'temperature': 0.7
}

# 发送请求并处理响应
try:
    response = requests.post(
        'https://api.openai.com/v1/chat/completions',
        headers=HEADERS,
        json=payload,
        timeout=30
    )
    response.raise_for_status()
    generated_code = response.json()['choices'][0]['message']['content']
    print(generated_code)
except requests.exceptions.RequestException as e:
    print(f'API 请求失败: {e}')

const axios = require('axios');
require('dotenv').config();

async function generateCode(prompt) {
  const maxRetries = 3;
  let retryCount = 0;

  while (retryCount < maxRetries) {
    try {
      const response = await axios.post(
        'https://api.openai.com/v1/chat/completions',
        {
          model: 'gpt-3.5-turbo',
          messages: [{ role: 'system', content: '你是一个专业的代码助手'},
            {role: 'user', content: prompt}
          ]
        },
        {
          headers: {'Authorization': `Bearer ${process.env.OPENAI_API_KEY}`,
            'Content-Type': 'application/json'
          },
          timeout: 20000
        }
      );

      return response.data.choices[0].message.content;
    } catch (error) {if (error.response?.status === 429) {const delay = Math.pow(2, retryCount) * 1000;
        await new Promise(resolve => setTimeout(resolve, delay));
        retryCount++;
      } else {throw error;}
    }
  }
  throw new Error('Max retries exceeded');
}

// 使用示例
generateCode('用 JavaScript 实现数组去重')
  .then(code => console.log(code))
  .catch(err => console.error(err));