github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-01 09:06:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore:add GitPython && alembic requirement

Browse Source
This commit is contained in:
aries_ckt
2023-10-17 13:52:13 +08:00
parent 496696537d
commit da87e40163
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
setup.py
View File

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