从零构建ChatGPT API调用：技术选型与实战避坑指南

2次阅读

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

开发者在集成 ChatGPT API 时，常会遇到以下几个典型问题：

认证头处理：需要正确生成和管理 JWT 令牌，处理过期和刷新逻辑
流式响应解析 ：对 SSE(Server-Sent Events) 数据流的处理不够熟悉
token 计数策略：估算提示词和响应的 token 消耗，避免超出限额

在通信协议选择上，REST 相比 gRPC 有几个优势：

更简单的调试和测试工具支持
更好的语言和框架兼容性
对浏览器环境更友好

import aiohttp
import json

async def chat_completion(prompt):
    headers = {'Authorization': f'Bearer {API_KEY}',
        'Content-Type': 'application/json'
    }

    payload = {
        'model': 'gpt-3.5-turbo',
        'messages': [{'role': 'user', 'content': prompt}],
        'stream': True  # 启用流式响应
    }

    try:
        async with aiohttp.ClientSession() as session:
            async with session.post(
                'https://api.openai.com/v1/chat/completions',
                headers=headers,
                json=payload
            ) as resp:
                async for line in resp.content:
                    if line.startswith(b'data:'):
                        chunk = line[6:].strip()
                        if chunk != b'[DONE]':
                            yield json.loads(chunk)
    except Exception as e:
        logging.error(f'API 调用失败: {str(e)}')
        raise

const axios = require('axios');
const {createInterface} = require('readline');

async function streamChat(prompt) {
  const response = await axios({
    method: 'post',
    url: 'https://api.openai.com/v1/chat/completions',
    headers: {'Authorization': `Bearer ${process.env.OPENAI_KEY}`,
      'Content-Type': 'application/json'
    },
    data: {
      model: 'gpt-3.5-turbo',
      messages: [{role: 'user', content: prompt}],
      stream: true
    },
    responseType: 'stream'
  });

  const rl = createInterface({
    input: response.data,
    crlfDelay: Infinity
  });

  rl.on('line', (line) => {if (line.startsWith('data:')) {const data = line.substring(6);
      if (data !== '[DONE]') {process.stdout.write(JSON.parse(data).choices[0]?.delta?.content || '');
      }
    }
  });
}