From 8a0678f465e4df73aad7a0c1612888e39e422ae8 Mon Sep 17 00:00:00 2001
From: DavdGao <gaodawei.gdw@alibaba-inc.com>
Date: Tue, 12 Mar 2024 17:40:15 +0800
Subject: [PATCH] add package ollama

---
 setup.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 2a9e6a99f..31e94abc6 100644
--- a/setup.py
+++ b/setup.py
@@ -44,12 +44,14 @@
     "tiktoken",
     "Pillow",
     "requests",
-    "openai>=1.3.0",
     "numpy",
     "Flask==3.0.0",
     "Flask-Cors==4.0.0",
     "Flask-SocketIO==5.3.6",
+    # TODO: move into other requires
     "dashscope==1.14.1",
+    "openai>=1.3.0",
+    "ollama>=0.1.7"
 ]
 
 distribute_requires = minimal_requires + rpc_requires