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

feat(model): Support DeepSeek proxy LLM (#1491)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-05-07 17:40:16 +08:00
committed by GitHub
parent 863b5404dd
commit b38305b72b
7 changed files with 152 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
dbgpt/_private/config.py
View File

@@ -128,6 +128,16 @@ class Config(metaclass=Singleton):
os.environ["moonshot_proxyllm_api_base"] = os.getenv(
"MOONSHOT_API_BASE", "https://api.moonshot.cn/v1"
)
# Deepseek proxy
self.deepseek_proxy_api_key = os.getenv("DEEPSEEK_API_KEY")
if self.deepseek_proxy_api_key:
os.environ["deepseek_proxyllm_proxy_api_key"] = self.deepseek_proxy_api_key
os.environ["deepseek_proxyllm_proxyllm_backend"] = os.getenv(
"DEEPSEEK_MODEL_VERSION", "deepseek-chat"
)
os.environ["deepseek_proxyllm_api_base"] = os.getenv(
"DEEPSEEK_API_BASE", "https://api.deepseek.com/v1"
)
self.proxy_server_url = os.getenv("PROXY_SERVER_URL")