add ai extra setup (#588)

holoviz · Jul 3, 2024 · 5594067 · 5594067
1 parent 5021d78
commit 5594067
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -28,6 +28,15 @@ def get_setup_version(reponame):
 ]
 
 extras_require = {
+    'ai': [
+        'nbformat',
+        'duckdb',
+        'pyarrow',
+        'openai',
+        'instructor',
+        'pydantic',
+        'datashader',   # for big datasets
+    ],
     'sql': [
         'duckdb',
         'intake-sql',
@@ -54,7 +63,8 @@ def get_setup_version(reponame):
         'nbsite >=0.8.2',
     ]
 }
-
+extras_require["ai_local"] = extras_require["ai"] + ["huggingface_hub"]
+extras_require["ai_llama"] = extras_require["ai_local"] + ["llama-cpp-python"]
 extras_require['all'] = sorted(set(sum(extras_require.values(), [])))
 
 ########## metadata for setuptools ##########