Skip to main content

Setup

export GOOGLE_APPLICATION_CREDENTIALS=/path/to/service-account.json
export VERTEXAI_PROJECT=your-project-id
export VERTEXAI_LOCATION=us-central1

Usage

from praisonaiagents import ImageAgent

agent = ImageAgent(llm="vertex_ai/imagegeneration@006")
result = agent.generate("A sunset over mountains")
print(result.data[0].url)

Models

ModelDescription
vertex_ai/imagegeneration@006Imagen 3
vertex_ai/imagen-3.0-generate-001Imagen 3 Fast
vertex_ai/imagen-3.0-capability-001Imagen 3 Edit