LLM Workflow Engine (LWE) Chat Vertex AI Provider plugin

Chat Vertex AI Provider plugin for LLM Workflow Engine

Access to Google Vertex AI chat models.

Installation

You must configure access to the Vertex AI API in Google Cloud by either:

• Having credentials configured for your environment (gcloud, workload identity, etc...)
• Storing the path to a service account JSON file as the GOOGLE_APPLICATION_CREDENTIALS environment variable

From packages

Install the latest version of this software directly from github with pip:

pip install git+https://github.com/llm-workflow-engine/lwe-plugin-provider-chat-vertexai

From source (recommended for development)

Install the latest version of this software directly from git:

git clone https://github.com/llm-workflow-engine/lwe-plugin-provider-chat-vertexai.git

Install the development package:

cd lwe-plugin-provider-chat-vertexai
pip install -e .

Configuration

Add the following to config.yaml in your profile:

plugins:
  enabled:
    - provider_chat_vertexai
    # Any other plugins you want enabled...

Usage

From a running LWE shell:

/provider chat_vertexai
/model model_name chat-bison