SDK do Google Gen AI

O SDK da IA generativa do Google 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 para desenvolvedores e migrar o aplicativo para a Vertex AI sem reescrever o código.

O SDK da IA generativa do Google 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:

Para saber mais, consulte a referência do SDK para Python (abre em uma nova guia).

Guia de início rápido

Escolha uma das opções a seguir, dependendo se você está usando a Vertex AI no modo expresso ou não.

  • Usar a Vertex AI (com todos os recursos e serviços 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:
# ...
  • Usar a Vertex AI no modo expresso
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

O SDK do Google Gen AI para Go está disponível em go.dev e GitHub:

Guia de início rápido

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

O SDK do Google Gen AI para Java está disponível no Maven Central e no GitHub: