Google Gen AI SDK

新版 Google Gen AI SDK 通过 Gemini Developer API 和 Gemini Enterprise API (Vertex AI) 提供了一个统一的 Gemini 2.0 接口。除了少数例外情况外，在一个平台上运行的代码在另一个平台上也能运行。Gen AI SDK 还支持 Gemini 1.5 模型。

Python

Python 版 Google Gen AI SDK 可在 PyPI 和 GitHub 上获取：

或者，试用“开始使用”记事本。

如需了解详情，请参阅 Python SDK 参考文档（在新标签页中打开）。

快速入门

1. 安装 SDK bash pip install google-genai

2. 导入库

from google import genai
from google.genai import types

3. 创建客户端

client = genai.Client(api_key='GEMINI_API_KEY')

4. 生成内容

response = client.models.generate_content(
    model='gemini-1.5-flash', contents='How does RLHF work?'
)
print(response.text)

Go

适用于 Go 的 Google Gen AI SDK 可在 go.dev 和 GitHub 上获取：

快速入门

1. 导入库

import "google.golang.org/genai"

2. 创建客户端

client, err := genai.NewClient(ctx, &genai.ClientConfig{
    APIKey:   apiKey,
    Backend:  genai.BackendGoogleAI,
})

3. 生成内容

// Call the GenerateContent method
  result, err := client.Models.GenerateContent(ctx, "gemini-2.0-flash-exp", genai.Text("Tell me about New York?"), nil)