Add ollama to setup.py

kagisearch · Mar 20, 2024 · ece2d88 · ece2d88
1 parent b0b6d8c
commit ece2d88
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -31,6 +31,7 @@
         "grpcio~=1.54.2",
         "google-generativeai",
         "mistralai",
+        "ollama",
     ],
     extras_require={
         "local": ["einops", "accelerate"]