El nuevo SDK de Google Gen AI proporciona una interfaz unificada a Gemini 2.0 a través de la API para desarrolladores de Gemini y la API de Gemini en Vertex AI. Con algunas excepciones, el código que se ejecuta en una plataforma se ejecutará en ambas. Esto significa que puedes crear prototipos de una aplicación con la API para desarrolladores y, luego, migrar la aplicación a Vertex AI sin reescribir el código.
El SDK de Gen AI también admite los modelos Gemini 1.5.
Python
El SDK de Google Gen AI para Python está disponible en PyPI y GitHub:
Para obtener más información, consulta la referencia del SDK de Python (se abre en una pestaña nueva).
Guía de inicio rápido
1. Importa la biblioteca
from google import genai
2. Crea un cliente
Elige una de las siguientes opciones, según si usas Vertex AI en el modo exprés o no.
Crea un cliente para Vertex AI (con todas las Google Cloud funciones y servicios)
Especifica el ID y la ubicación del proyecto cuando crees el cliente.
client = genai.Client( vertexai=True, project='your-project-id', location='us-central1' )
Crea un cliente para Vertex AI en modo exprés
Si usas Vertex AI en modo exprés, primero debes crear una clave de API. Especifica tu clave de API cuando crees el cliente.
client = genai.Client( vertexai=True, api_key='your-api-key' )
3. Genera contenido
response = client.models.generate_content(
model='gemini-1.5-pro-002', contents='What is your name?'
)
print(response.text)
Go
El SDK de IA generativa de Google para Go está disponible en go.dev y GitHub:
Guía de inicio rápido
1. Importa bibliotecas
import "google.golang.org/genai"
2. Crea un cliente
client, err := genai.NewClient(ctx, &genai.ClientConfig{
Project: project,
Location: location,
Backend: genai.BackendVertexAI,
})
3. Genera contenido
// Call the GenerateContent method
result, err := client.Models.GenerateContent(ctx, "gemini-2.0-flash-exp", genai.Text("Tell me about New York?"), nil)