add setupt

microsoft · Aug 3, 2023 · e4c29c8 · e4c29c8
1 parent 7e6b550
commit e4c29c8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -89,6 +89,7 @@
             "sympy",
             "wolframalpha",
             "joblib<1.3.0",  # temp solution for joblib 1.3.0 issue, no need once https://github.com/joblib/joblib-spark/pull/48 is merged
+            "tiktoken",
         ],
         "catboost": ["catboost>=0.26"],
         "blendsearch": [
@@ -139,7 +140,7 @@
         ],
         "benchmark": ["catboost>=0.26", "psutil==5.8.0", "xgboost==1.3.3", "pandas==1.1.4"],
         "openai": ["openai==0.27.8", "diskcache"],
-        "autogen": ["openai==0.27.8", "diskcache", "termcolor"],
+        "autogen": ["openai==0.27.8", "diskcache", "termcolor", "tiktoken"],
         "mathchat": ["openai==0.27.8", "diskcache", "termcolor", "sympy", "pydantic==1.10.9", "wolframalpha"],
         "synapse": [
             "joblibspark>=0.5.0",