Google Gemini API 全栈调用指南：从 Edge 代理、流式传输（SSE）到结构化输出

客户端浏览器 (React / Next.js)
  │
  ▼ POST /api/chat/stream (避免客户端直连暴露 API Key，解决 CORS 跨域)
Hono 边缘网关 (部署于 Cloudflare / Vercel Edge)
  │
  ├── 1. 业务鉴权与速率限制 (Rate Limit)
  ├── 2. 注入服务端环境变量 GEMINI_API_KEY
  │
  ▼ Google Generative AI SDK (流式调用)
Google Gemini API (googleapis.com)
  │
  ▼ SSE (Server-Sent Events) 逐字推回
客户端实时打字机渲染

pnpm add @google/generative-ai hono

// src/routes/gemini.ts
import { Hono } from 'hono';
import { streamText } from 'hono/streaming';
import { GoogleGenerativeAI } from '@google/generative-ai';
 
const geminiRoute = new Hono();
 
geminiRoute.post('/chat/stream', async (c) => {
  const { prompt, history = [] } = await c.req.json();
  const apiKey = c.env?.GEMINI_API_KEY || process.env.GEMINI_API_KEY;
 
  if (!apiKey) {
    return c.json({ error: 'API Key 未配置' }, 500);
  }
 
  const genAI = new GoogleGenerativeAI(apiKey);
  // 使用高性价比的 Gemini 1.5 Flash 或最新的 2.0 Flash
  const model = genAI.getGenerativeModel({ model: 'gemini-1.5-flash' });
 
  // 开启流式 SSE 响应
  return streamText(c, async (stream) => {
    try {
      const chat = model.startChat({
        history: history.map((h: any) => ({
          role: h.role === 'user' ? 'user' : 'model',
          parts: [{ text: h.content }],
        })),
      });
 
      const result = await chat.sendMessageStream(prompt);
 
      for await (const chunk of result.stream) {
        const chunkText = chunk.text();
        await stream.write(chunkText);
      }
    } catch (err: any) {
      await stream.write(`\n[Error: ${err.message}]`);
    }
  });
});
 
export default geminiRoute;

import { GoogleGenerativeAI, SchemaType } from '@google/generative-ai';
 
const genAI = new GoogleGenerativeAI(process.env.GEMINI_API_KEY!);
 
const model = genAI.getGenerativeModel({
  model: 'gemini-1.5-flash',
  generationConfig: {
    responseMimeType: 'application/json',
    responseSchema: {
      type: SchemaType.OBJECT,
      properties: {
        sentiment: {
          type: SchemaType.STRING,
          enum: ['positive', 'neutral', 'negative'],
          description: '情感倾向分析结果',
        },
        confidence: {
          type: SchemaType.NUMBER,
          description: '置信度得分 (0-1)',
        },
        tags: {
          type: SchemaType.ARRAY,
          items: { type: SchemaType.STRING },
          description: '提取出的核心关键词',
        },
      },
      required: ['sentiment', 'confidence', 'tags'],
    },
  },
});
 
async function analyzeFeedback(text: string) {
  const result = await model.generateContent(`请分析以下用户评论：${text}`);
  const jsonResponse = JSON.parse(result.response.text());
  return jsonResponse;
}

// components/GeminiChat.tsx
'use client';
 
import { useState } from 'react';
 
export function GeminiChat() {
  const [input, setInput] = useState('');
  const [reply, setReply] = useState('');
  const [isGenerating, setIsGenerating] = useState(false);
 
  const handleSend = async () => {
    if (!input.trim() || isGenerating) return;
    
    setIsGenerating(true);
    setReply('');
 
    try {
      const response = await fetch('/api/chat/stream', {
        method: 'POST',
        headers: { 'Content-Type': 'application/json' },
        body: JSON.stringify({ prompt: input }),
      });
 
      if (!response.body) throw new Error('ReadableStream 不可用');
 
      const reader = response.body.getReader();
      const decoder = new TextDecoder();
 
      while (true) {
        const { done, value } = await reader.read();
        if (done) break;
        const textChunk = decoder.decode(value, { stream: true });
        setReply((prev) => prev + textChunk);
      }
    } catch (err: any) {
      setReply(`发生错误: ${err.message}`);
    } finally {
      setIsGenerating(false);
    }
  };
 
  return (
    <div className="space-y-4 max-w-xl">
      <div className="p-4 bg-gray-50 border rounded-lg min-h-[120px] whitespace-pre-wrap">
        {reply || (isGenerating ? 'Gemini 正在思考中...' : '等待输入提问...')}
      </div>
      <div className="flex gap-2">
        <input
          value={input}
          onChange={(e) => setInput(e.target.value)}
          placeholder="问问 Gemini..."
          className="flex-1 px-3 py-2 border rounded-md"
        />
        <button
          onClick={handleSend}
          disabled={isGenerating}
          className="px-4 py-2 bg-blue-600 text-white rounded-md disabled:bg-gray-400"
        >
          发送
        </button>
      </div>
    </div>
  );
}