diff --git a/setup.py b/setup.py
index 1cb48f4..2a282f4 100644
--- a/setup.py
+++ b/setup.py
@@ -31,6 +31,7 @@
         "grpcio~=1.54.2",
         "google-generativeai",
         "mistralai",
+        "ollama",
     ],
     extras_require={
         "local": ["einops", "accelerate"]