SDKs de Google Gen AI

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)