DB-GPT/packages/dbgpt-ext/pyproject.toml at main · eosphoros-ai/DB-GPT · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
[project]
name = "dbgpt-ext"
version = "0.7.5"
description = "Add your description here"
authors = [
    { name = "csunny", email = "cfqcsunny@gmail.com" }
]
license = "MIT"
readme = "README.md"
requires-python = ">= 3.10"

dependencies = [
    "dbgpt>=0.7.0",
    "pymysql",
]

[project.urls]
Homepage = "https://github.com/eosphoros-ai/DB-GPT"
Documentation = "http://docs.dbgpt.cn/docs/overview"
Repository = "https://github.com/eosphoros-ai/DB-GPT.git"
Issues = "https://github.com/eosphoros-ai/DB-GPT/issues"

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[project.optional-dependencies]
rag = [
    "spacy==3.7",
    "markdown",
    "bs4",
    "python-pptx",
    "python-docx",
    "olefile",
    "pypdf",
    "pdfplumber",
]
graph_rag = [
    # For visualization in code
    "networkx",
    "dbgpt-tugraph-plugins>=0.1.1",
    "neo4j"
]
datasource_mysql = [
    # mysqlclient 2.2.x have pkg-config issue on 3.10+
    # If you want to install psycopg2 and mysqlclient in ubuntu, you should install
    # libpq-dev and libmysqlclient-dev first.
    "mysqlclient==2.1.0",
]
datasource_oracle = [
    "oracledb==3.1.0",  # use python-oracledb，new driver for Oracle
]

datasource_postgres = [
    # "psycopg2", # In production, you can install psycopg2 instead of psycopg2-binary
    "psycopg2-binary",
]
datasource_clickhouse = [
    "clickhouse-connect",
]
datasource_spark = ["pyspark"]
datasource_mssql = ["pymssql"]
datasource_hive = [
    "pyhive",
    "thrift",
    "thrift_sasl",
]
datasource_vertica = ["vertica-python"]
datasource_duckdb = [
    "duckdb",
    "duckdb-engine==0.9.1",
]
# datasource_doris = ["pydoris>=1.0.2,<2.0.0"]
storage_milvus = ["pymilvus"]
storage_weaviate = ["weaviate-client"]
storage_chromadb = [
    "onnxruntime>=1.14.1,<=1.18.1",
    "chromadb>=0.4.22"
    ]
storage_elasticsearch = ["elasticsearch"]
storage_obvector = ["pyobvector"]

file_oss = [
    "oss2" # Aliyun OSS
]
file_s3 = [
    "boto3"
]

[tool.uv]
managed = true
dev-dependencies = [
    "pytest>=8.3.4",
]

[tool.hatch.build.targets.wheel]
packages = ["src/dbgpt_ext"]
exclude = [
    "src/dbgpt_ext/**/tests",
    "src/dbgpt_ext/**/tests/*",
    "src/dbgpt_ext/tests",
    "src/dbgpt_ext/tests/*",
    "src/dbgpt_ext/**/examples",
    "src/dbgpt_ext/**/examples/*"
]