diff --git a/pyproject.toml b/pyproject.toml
index f9bdd0c3..5ecb9df7 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -79,7 +79,7 @@ orjson = "*"
 schedule = "*"
 uvloop = {version = "*", markers = "sys_platform == 'linux' or sys_platform == 'darwin'"}
 winloop = {version = "*", markers = "sys_platform == 'win32'"}
-
+google-generativeai = "*"
 [tool.poetry.scripts]
 swarms = "swarms.cli.main:main"