diff --git a/.circleci/config.yml b/.circleci/config.yml
index caf0e8396..d3d51a483 100644
--- a/.circleci/config.yml
+++ b/.circleci/config.yml
@@ -29,6 +29,7 @@ jobs:
             pip install pytest-asyncio
             pip install mypy
             pip install -q google-generativeai
+            pip install google-cloud-aiplatform
             pip install "boto3>=1.28.57"
             pip install appdirs
             pip install langchain