谷歌ChatGPT API接入实战：从认证到对话接口调用全解析

2次阅读

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

在实际开发中，接入谷歌 ChatGPT API 往往会遇到几个典型问题：

OAuth 2.0 认证流程复杂 ：需要处理多种 grant_type，管理访问令牌(access_token) 和刷新令牌 (refresh_token) 的生命周期
流式响应处理困难：原生 API 返回的流式数据需要特殊处理才能实现类似 ChatGPT 的逐字输出效果
权限管理繁琐：IAM 配置不当可能导致服务账户权限过大，违反最小权限原则(Principle of Least Privilege)

登录 Google Cloud Platform 控制台
创建新项目或选择现有项目
在 API 库中搜索并启用 ”Generative Language API”
在 ” 凭据 ” 页面创建服务账户(Service Account)
为服务账户生成 JSON 密钥文件

from google.oauth2 import service_account
from google.auth.transport.requests import Request
import google.auth

# 自动刷新令牌的认证流程
credentials = service_account.Credentials.from_service_account_file(
    'service-account.json',
    scopes=['https://www.googleapis.com/auth/cloud-platform']
)

if credentials.expired:
    credentials.refresh(Request())

const {GoogleAuth} = require('google-auth-library');

async function getAccessToken() {
  const auth = new GoogleAuth({
    keyFile: 'service-account.json',
    scopes: ['https://www.googleapis.com/auth/cloud-platform']
  });

  const client = await auth.getClient();
  const accessToken = await client.getAccessToken();
  return accessToken;
}

import requests

url = "https://generativelanguage.googleapis.com/v1beta/models/chat-bison-001:generateMessage"

headers = {"Authorization": f"Bearer {access_token}",
    "Content-Type": "application/json"
}

params = {"key": "your-api-key"}

data = {"prompt": {"messages": [{"content": "Hello!"}]},
    "temperature": 0.7,
    "candidateCount": 1
}

response = requests.post(
    url, 
    headers=headers, 
    params=params, 
    json=data,
    stream=True,
    timeout=30
)

for chunk in response.iter_content(chunk_size=1024):
    print(chunk.decode('utf-8'), end='', flush=True)