Skip to main content

Integrating with Vertex AI in Orkes Conductor

To effectively utilize AI and LLM tasks in Orkes Conductor, it's essential to integrate your Conductor cluster with the necessary AI and LLM models.

Vertex AI offers a range of models that can be incorporated into the Orkes Conductor cluster. The choice of model depends on your unique use case, the functionalities you require, and the specific natural language processing tasks you intend to tackle.

This guide will provide the steps for integrating the Vertex AI provider with Orkes Conductor.

Steps to integrate with Vertex AI


Vertex AI integration is compatible with GCP clusters only.

  1. Navigate to Integrations from the left menu on your Orkes Conductor console.
  2. Click +New integration button from the top-right of your window.
  3. Under the AI / LLM section, choose Vertex AI.
  4. Click +Add and provide the following parameters:

Create new Vertex AI Integration

Integration nameProvide a name for the integration.
LocationProvide the Google Cloud region of your GCP account.
Project IDProvide the project ID in GCP.
PublisherProvide the publisher name in GCP.
DescriptionProvide a description of your integration.
  1. You can toggle-on the Active button to activate the integration instantly.
  2. Click Save.

Adding Vertex AI models to the integration

Now, you have integrated your Conductor console with the Vertex AI provider. The next step is integrating with the specific Vertex AI models.

Vertex AI has different models, such as Bison, Gecko, etc. Each model is to be used for different use cases, such as text completion, generating embeddings, etc.

Depending on your use case, you must configure different models within your Vertex AI configuration.

To add a new model to the Vertex AI integration:

  1. Navigate to the integrations page and click the '+' button next to the integration created.

Create new model for Vertex AI Integration

  1. Click +New model.
  2. Provide the model name and an optional description for the model. You can get the complete list of models in Vertex AI here.

Creating new model for Vertex AI Integration

  1. Click the Enable checkbox to enable the model immediately.
  2. Click Save.

This ensures the integration model is saved for future use in LLM tasks within Orkes Conductor.

RBAC - Governance on who can use Integrations

Now, the integration with the required models is ready. Next, we should determine the access control to these models.

The permission can be granted to applications/groups within the Orkes Conductor console.

To provide explicit permission to Groups:

  1. Navigate to Access Control > Groups from the left menu on your Orkes Conductor console.
  2. Create a new group or choose an existing group.
  3. Under the Permissions section, click +Add Permission.
  4. Under the Integrations tab, select the required integrations with the required permissions.

Add Permissions for Vertex AI Integration

  1. Click Add Permissions. This ensures that all the group members can access these integration models in their workflows.

Similarly, you can also provide permissions to applications.