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

feat(model): Support claude proxy models (#2155)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-11-26 19:47:28 +08:00
committed by GitHub
parent 9d8673a02f
commit 61509dc5ea
20 changed files with 508 additions and 157 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
setup.py
View File

@@ -670,6 +670,13 @@ def openai_requires():
setup_spec.extras["openai"] += setup_spec.extras["rag"]
def proxy_requires():
"""
pip install "dbgpt[proxy]"
"""
setup_spec.extras["proxy"] = setup_spec.extras["openai"] + ["anthropic"]
def gpt4all_requires():
"""
pip install "dbgpt[gpt4all]"
@@ -727,6 +734,7 @@ def default_requires():
setup_spec.extras["default"] += setup_spec.extras["datasource"]
setup_spec.extras["default"] += setup_spec.extras["torch"]
setup_spec.extras["default"] += setup_spec.extras["cache"]
setup_spec.extras["default"] += setup_spec.extras["proxy"]
setup_spec.extras["default"] += setup_spec.extras["code"]
if INCLUDE_QUANTIZATION:
# Add quantization extra to default, default is True
@@ -763,6 +771,7 @@ cache_requires()
observability_requires()
openai_requires()
proxy_requires()
# must be last
default_requires()
all_requires()