Migrar do Gemini 1.5 para o Gemini 2.0 com a API Gemini na Vertex AI

Este guia mostra como migrar dos modelos Gemini 1.0 e Gemini 1.5 (Flash e Pro) para os modelos Gemini 2.0.

Diferenças entre o Gemini 1.5 e o Gemini 2.0

Confira a seguir algumas diferenças entre o Gemini 2.0 e os modelos 1.0 e 1.5:

Configuração e migração

SDK da Gen AI

Recomendamos que você migre para o SDK Gen AI ao fazer upgrade para o Gemini 2.0.

Se você optar por usar o SDK da Gen AI, o processo de configuração será diferente do SDK da Vertex AI.

Para mais informações, acesse o SDK da IA generativa do Google.

Instalar

pip install --upgrade google-genai
Para saber mais, consulte a documentação de referência do SDK.

Defina variáveis de ambiente para usar o SDK da IA generativa com a Vertex AI:

# Replace the `GOOGLE_CLOUD_PROJECT` and `GOOGLE_CLOUD_LOCATION` values
# with appropriate values for your project.
export GOOGLE_CLOUD_PROJECT=GOOGLE_CLOUD_PROJECT
export GOOGLE_CLOUD_LOCATION=us-central1
export GOOGLE_GENAI_USE_VERTEXAI=True

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:
# ...

Substitua GOOGLE_CLOUD_PROJECT pelo ID do projeto Google Cloud e GOOGLE_CLOUD_LOCATION pelo local do projeto Google Cloud (por exemplo, us-central1).

SDK da Vertex AI

Se você reutilizar o SDK da Vertex AI, o processo de configuração será o mesmo para os modelos 1.0, 1.5 e 2.0. Para mais informações, consulte Introdução ao SDK da Vertex AI para Python.

Instale o SDK:

pip install --upgrade --quiet google-cloud-aiplatform

Confira a seguir um exemplo de código curto que usa o SDK da Vertex AI para Python:

import vertexai
from vertexai.generative_models import GenerativeModel

# TODO(developer): Update and un-comment below line
# PROJECT_ID = "your-project-id"
vertexai.init(project=PROJECT_ID, location="us-central1")

model = GenerativeModel("gemini-1.5-flash-002")

response = model.generate_content(
    "What's a good name for a flower shop that specializes in selling bouquets of dried flowers?"
)

print(response.text)
# Example response:
# **Emphasizing the Dried Aspect:**
# * Everlasting Blooms
# * Dried & Delightful
# * The Petal Preserve
# ...

Substitua PROJECT_ID pelo ID do projeto Google Cloud e LOCATION pelo local do projeto Google Cloud (por exemplo, us-central1). Em seguida, mude o ID do modelo de gemini-1.5-flash-002 para gemini-2.0-flash.