chore:add GitPython && alembic requirement

This commit is contained in:
aries_ckt 2023-10-17 13:52:13 +08:00
parent 496696537d
commit da87e40163

View File

@ -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"]