mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-05 19:11:52 +00:00
feat(model): support openai embedding model
This commit is contained in:
9
setup.py
9
setup.py
@@ -319,6 +319,14 @@ def all_datasource_requires():
|
||||
setup_spec.extras["datasource"] = ["pymssql", "pymysql"]
|
||||
|
||||
|
||||
def openai_requires():
|
||||
"""
|
||||
pip install "db-gpt[openai]"
|
||||
"""
|
||||
setup_spec.extras["openai"] = ["openai", "tiktoken"]
|
||||
llama_cpp_python_cuda_requires()
|
||||
|
||||
|
||||
def all_requires():
|
||||
requires = set()
|
||||
for _, pkgs in setup_spec.extras.items():
|
||||
@@ -339,6 +347,7 @@ llama_cpp_requires()
|
||||
quantization_requires()
|
||||
all_vector_store_requires()
|
||||
all_datasource_requires()
|
||||
openai_requires()
|
||||
|
||||
# must be last
|
||||
all_requires()
|
||||
|
Reference in New Issue
Block a user