diff --git a/requirements.txt b/requirements.txt
index 999bd5fd..0a5e9a3d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -47,3 +47,4 @@ reportportal-client
 scikit-learn
 seaborn
 structlog
+google.geai