Hello World for Google Gemini Pro

hello_world_gemini_pro.py

import os
import google.generativeai as genai


# GOOGLE API KEY
# 1. Create one at https://aistudio.google.com/app/apikey
# 2. In your terminal, do `export GOOGLE_API_KEY = 'YOUR_KEY'`
GOOGLE_API_KEY = os.getenv('GOOGLE_API_KEY')

# List all models available
for m in genai.list_models():
  if 'generateContent' in m.supported_generation_methods:
    print(m.name)

# Example using `gemini-pro`
model = genai.GenerativeModel('gemini-pro')
response = model.generate_content("")
print(response.text)