From a8c9d67d49861cb78b3bdf137dae0989ce231bb6 Mon Sep 17 00:00:00 2001
From: "Joel Z. Leibo" <jzl@google.com>
Date: Fri, 9 Aug 2024 01:58:26 -0700
Subject: [PATCH] Add ollama to requirements

PiperOrigin-RevId: 661178571
Change-Id: Ia9e342256ec87b886c6a1d3c73eb6c7acf3b4392
---
 setup.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/setup.py b/setup.py
index 41a989ec..01faf7fd 100644
--- a/setup.py
+++ b/setup.py
@@ -77,6 +77,7 @@ def _remove_excluded(description: str) -> str:
         'matplotlib',
         'mistralai',
         'numpy',
+        'ollama',
         'openai>=1.3.0',
         'pandas<=2.0.3',
         'python-dateutil',