O novo SDK Google Gen AI oferece uma interface unificada para o Gemini 2.0 pela API Gemini para desenvolvedores e pela API Gemini na Vertex AI. Com poucas exceções, o código executado em uma plataforma é executado nas duas. Isso significa que você pode criar protótipos de um aplicativo usando a API Developer e migrar o aplicativo para a Vertex AI sem reescrever o código.
O SDK de IA generativa também oferece suporte aos modelos Gemini 1.5.
Python
O SDK da Google Gen AI para Python está disponível no PyPI e no GitHub:
- google-genai no PyPI (link em inglês)
- python-genai no GitHub
Para saber mais, consulte a referência do SDK para Python (abre em uma nova guia).
Guia de início rápido
1. Importar a biblioteca
from google import genai
2. Criar um cliente
Escolha uma das opções a seguir, dependendo se você está usando a Vertex AI no modo expresso ou não.
Criar um cliente para a Vertex AI (com todos os recursos e serviços Google Cloud )
Especifique o ID e o local do projeto ao criar o cliente.
client = genai.Client( vertexai=True, project='your-project-id', location='us-central1' )
Criar um cliente para a Vertex AI no modo expresso
Se você estiver usando a Vertex AI no modo expresso, primeiro crie uma chave de API. Especifique sua chave de API ao criar o cliente.
client = genai.Client( vertexai=True, api_key='your-api-key' )
3. Gerar conteúdo
response = client.models.generate_content(
model='gemini-1.5-pro-002', contents='What is your name?'
)
print(response.text)
Go
O SDK do Google Gen AI para Go está disponível em go.dev e GitHub:
Guia de início rápido
1. Importar bibliotecas
import "google.golang.org/genai"
2. Criar um cliente
client, err := genai.NewClient(ctx, &genai.ClientConfig{
Project: project,
Location: location,
Backend: genai.BackendVertexAI,
})
3. Gerar conteúdo
// Call the GenerateContent method
result, err := client.Models.GenerateContent(ctx, "gemini-2.0-flash-exp", genai.Text("Tell me about New York?"), nil)