Google Gen AI SDK

Google Gen AI SDK 通过 Vertex AI 上的 Gemini Developer API 和 Gemini API 为 Gemini 2.0 提供统一的界面。除了少数例外情况,在一个平台上运行的代码将在两个平台上运行。这意味着,您可以使用 Developer API 为应用创建原型,然后将应用迁移到 Vertex AI,而无需重写代码。

Google Gen AI SDK 还支持 Gemini 1.5 模型。

Python

Google Gen AI SDK for Python 可在 PyPI 和 GitHub 上找到:

如需了解详情,请参阅 Python SDK 参考文档(在新标签页中打开)

快速入门

根据您是否是在快速模式下使用 Vertex AI,选择以下选项之一。

  • 使用 Vertex AI(包含所有 Google Cloud 功能和服务)
from google import genai from google.genai.types import HttpOptions  client = genai.Client(http_options=HttpOptions(api_version="v1")) response = client.models.generate_content(     model="gemini-2.0-flash-001",     contents="How does AI work?", ) print(response.text) # Example response: # Okay, let's break down how AI works. It's a broad field, so I'll focus on the ... # # Here's a simplified overview: # ...
  • 在快速模式下使用 Vertex AI
from google import genai  # TODO(developer): Update below line API_KEY = "YOUR_API_KEY"  client = genai.Client(vertexai=True, api_key=API_KEY)  response = client.models.generate_content(     model="gemini-2.0-flash-001",     contents="""Explain bubble sort to me.""", )  print(response.text) # Example response: # Bubble Sort is a simple sorting algorithm that repeatedly steps through the list

Go

适用于 Go 的 Google Gen AI SDK 可在 go.dev 和 GitHub 上找到:

快速入门

import ( 	"context" 	"fmt" 	"io"  	"google.golang.org/genai" )  // generateWithText shows how to generate text using a text prompt. func generateWithText(w io.Writer) error { 	ctx := context.Background()  	client, err := genai.NewClient(ctx, &genai.ClientConfig{ 		HTTPOptions: genai.HTTPOptions{APIVersion: "v1"}, 	}) 	if err != nil { 		return fmt.Errorf("failed to create genai client: %w", err) 	}  	resp, err := client.Models.GenerateContent(ctx, 		"gemini-2.0-flash-001", 		genai.Text("How does AI work?"), 		nil, 	) 	if err != nil { 		return fmt.Errorf("failed to generate content: %w", err) 	}  	respText, err := resp.Text() 	if err != nil { 		return fmt.Errorf("failed to convert model response to text: %w", err) 	} 	fmt.Fprintln(w, respText) 	// Example response: 	// That's a great question! Understanding how AI works can feel like ... 	// ... 	// **1. The Foundation: Data and Algorithms** 	// ...  	return nil } 

Java

您可以在 Maven Central 和 GitHub 上找到适用于 Java 的 Google Gen AI SDK: