Using prompto
with Vertex AI
For prompts to Vertex AI API, you can simply add a line in your experiment (.jsonl
) file where you specify the api
to be vertexai
. See the models doc for some details on the environment variables you need to set.
Note that the Vertex AI API is different to the Gemini API. For Gemini API, see the gemini example.
We provide an example experiment file in data/input/vertexai-example.jsonl. You can run it with the following command (assuming that your working directory is the current directory of this notebook, i.e. examples/vertexai
):
Multimodal prompting
Multimodal prompting is available with the Vertex AI API. We provide an example notebook in the Multimodal prompting with Vertex AI notebook and example experiment file in data/input/vertexai-multimodal-example.jsonl. You can run it with the following command:
Environment variables
To run the experiment, you will need to set the following environment variables first:
export VERTEXAI_PROJECT_ID=<YOUR-VERTEXAI-PROJECT-ID>
export VERTEXAI_LOCATION_ID=<YOUR-VERTEXAI-LOCATION-ID>
When using Vertex AI, you need to set up the gcloud CLI and authenticate with your Google Cloud account. In particular, you will have ran
After this, you can set your default project ID with
By doing this, you can optionally choose to not set the VERTEXAI_PROJECT_ID
environment variable and the vertexai
library will use the default project ID set by the gcloud
CLI.
You can also use an .env
file to save these environment variables without needing to export them globally in the terminal:
By default, the prompto_run_experiment
command will look for an .env
file in the current directory. If you want to use a different .env
file, you can specify it with the --env
flag.
Also see the vertexai.ipynb notebook for a more detailed walkthrough on the how to set the environment variables and run the experiment and the different types of prompts you can run.
Do note that when you run the experiment, the input file (data/input/vertexai-example.jsonl) will be moved to the output directory (timestamped for when you run the experiment).