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

feat(model): Support moonshot proxy LLM (#1404)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-04-10 23:41:50 +08:00
committed by GitHub
parent 1c6a897137
commit 7d6dfd9ea8
9 changed files with 186 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
dbgpt/_private/config.py
View File

@@ -118,6 +118,16 @@ class Config(metaclass=Singleton):
os.environ["yi_proxyllm_proxy_api_base"] = os.getenv(
"YI_API_BASE", "https://api.lingyiwanwu.com/v1"
)
# Moonshot proxy
self.moonshot_proxy_api_key = os.getenv("MOONSHOT_API_KEY")
if self.moonshot_proxy_api_key:
os.environ["moonshot_proxyllm_proxy_api_key"] = self.moonshot_proxy_api_key
os.environ["moonshot_proxyllm_proxyllm_backend"] = os.getenv(
"MOONSHOT_MODEL_VERSION", "moonshot-v1-8k"
)
os.environ["moonshot_proxyllm_api_base"] = os.getenv(
"MOONSHOT_API_BASE", "https://api.moonshot.cn/v1"
)
self.proxy_server_url = os.getenv("PROXY_SERVER_URL")