mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-06 03:20:41 +00:00
refactor: The first refactored version for sdk release (#907)
Co-authored-by: chengfangyin2 <chengfangyin3@jd.com>
This commit is contained in:
17
setup.py
17
setup.py
@@ -292,7 +292,7 @@ def torch_requires(
|
||||
)
|
||||
torchvision_url = _build_wheels(
|
||||
"torchvision",
|
||||
torch_version,
|
||||
torchvision_version,
|
||||
base_url_func=lambda v, x, y: f"https://download.pytorch.org/whl/{x}",
|
||||
supported_cuda_versions=supported_versions,
|
||||
)
|
||||
@@ -375,10 +375,8 @@ def core_requires():
|
||||
"auto-gpt-plugin-template",
|
||||
"gTTS==2.3.1",
|
||||
"langchain>=0.0.286",
|
||||
# 从固定==2.0.22改动到可变版本是发现其余dependencies都是>=1.4,同时如pydoris是<2
|
||||
# change from fixed version 2.0.22 to variable version, because other dependencies are >=1.4, such as pydoris is <2
|
||||
"SQLAlchemy>=1.4,<3",
|
||||
# for doris
|
||||
"pydoris>=1.0.2,<2.0.0",
|
||||
"fastapi==0.98.0",
|
||||
"pymysql",
|
||||
"duckdb==0.8.1",
|
||||
@@ -529,7 +527,14 @@ def all_datasource_requires():
|
||||
pip install "db-gpt[datasource]"
|
||||
"""
|
||||
|
||||
setup_spec.extras["datasource"] = ["pymssql", "pymysql", "pyspark", "psycopg2"]
|
||||
setup_spec.extras["datasource"] = [
|
||||
"pymssql",
|
||||
"pymysql",
|
||||
"pyspark",
|
||||
"psycopg2",
|
||||
# for doris
|
||||
"pydoris>=1.0.2,<2.0.0",
|
||||
]
|
||||
|
||||
|
||||
def openai_requires():
|
||||
@@ -636,7 +641,7 @@ setuptools.setup(
|
||||
extras_require=setup_spec.extras,
|
||||
entry_points={
|
||||
"console_scripts": [
|
||||
"dbgpt=pilot.scripts.cli_scripts:main",
|
||||
"dbgpt=dbgpt.cli.cli_scripts:main",
|
||||
],
|
||||
},
|
||||
)
|
||||
|
Reference in New Issue
Block a user