diff --git a/setup.py b/setup.py index 3785a277d..32ba11d73 100644 --- a/setup.py +++ b/setup.py @@ -315,6 +315,8 @@ def core_requires(): "jsonschema", # TODO move transformers to default "transformers>=4.31.0", + "GitPython", + "alembic", ] @@ -403,11 +405,13 @@ def vllm_requires(): """ setup_spec.extras["vllm"] = ["vllm"] + # def chat_scene(): # setup_spec.extras["chat"] = [ # "" # ] + def default_requires(): """ pip install "db-gpt[default]" @@ -419,7 +423,7 @@ def default_requires(): "protobuf==3.20.3", "zhipuai", "dashscope", - "chardet" + "chardet", ] setup_spec.extras["default"] += setup_spec.extras["framework"] setup_spec.extras["default"] += setup_spec.extras["knowledge"]