Onde encontro a biblioteca da Gemini para usar no ambiente do Colab? (Se houver bibloteca da Gemini)

Ei, João! Tudo bem?

Não há uma biblioteca Gemini específica que possamos importar como na aula mostra o import openai. Mas podemos usar o Gemini de outra forma dentro do Colab, que é usar a API do Gemini importando a biblioteca google-generativeai. E estes são os passos:

Instale a biblioteca

• Abra o seu notebook no Colab e digite o comando abaixo:

!pip install google-generativeai

Configuração da API Key

• Acesso o Google AI Studio e clique no botão azul escrito Criar chave de API;
• Na caixa de diálago, escolha a opção Gemini API, em seguida em Criar uma chave de API em um projeto atual.
• Aguarde uns segundos para que a chave seja gerada, copie e a guarde em um local de fácil acesso.

• Volte ao notebook no Google Colab.
• No menu à esquerda do Colab, clique no ícone de chave (Secrets).
• No campo Name, insira um nome como GEMINI_API_KEY.
• No campo Value, cole a chave de API que você copiou.
• Ative a opção "Notebook access" para permitir que o notebook acesse essa chave.

Adaptar o código da aula para usar neste contexto.

• Use a chave no código, colocando-a em userdata.get. Por exemplo:

from google.colab import userdata
import google.generativeai as genai


genai.configure(api_key=userdata.get('GEMINI_API_KEY'))


model = genai.GenerativeModel('gemini-2.0-flash')


prompt = "You are a helpful assistant. Respond to user queries concisely."
response = model.generate_content(prompt)


print(response.text)

Retorno: Okay, I will do my best to respond to your queries concisely and helpfully. Ask away!

• Depois execute o código com a interação que deseja, por exemplo:

from google.colab import userdata
import google.generativeai as genai

# Configura a API key
genai.configure(api_key=userdata.get('GEMINI_API_KEY'))

# Configura o modelo
model = genai.GenerativeModel('gemini-2.0-flash')

# Inicia uma sessão de chat com o contexto inicial (equivalente ao system role)
chat = model.start_chat(history=[
    {"role": "user", "parts": ["You are a helpful assistant."]},
    {"role": "model", "parts": ["Understood! How can I assist you?"]}])

# Envia uma mensagem
response = chat.send_message("Qual é a capital do Brasil?")
print(response.text)

Retorno: A capital do Brasil é Brasília.

Prontinho, poderá usar a API do Gemini dessa forma. Caso queira ver o código executado, este é o meu notebook usado para teste.

Espero ter ajudado e qualquer dúvida, compartilhe no fórum.

Até mais, João!