justyns / llm-vertex

Plugin for LLM adding support for Google Cloud Vertex AI

llm-vertex

Plugin for LLM adding support for Google Cloud Vertex AI.

Please note that this plugin is for Vertex AI specifically, not Google AI Studio.

For Gemini support using AI Studio, please see llm-gemini instead.

Supported models:

• gemini-1.5-pro-preview-0409
• gemini-1.0-pro-vision-001
• gemini-experimental

Installation

pip install llm-vertex

Use

First, authenticate using gcloud:

gcloud auth application-default login

Export two environment variables for the GCP Project and location you want to use:

export VERTEX_PROJECT_ID=gcp-project-id VERTEX_LOCATION=us-east1

Run llm and specify one of the provided models:

❯ llm -m gemini-1.5-pro-preview-0409 "What's one clever name for a pet pelican?"
"Gulliver" would be a clever name for a pet pelican, referencing both its large gullet and its potential for long journeys! 🦜