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

feat(model): Support siliconflow models (#2157)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-11-27 19:41:37 +08:00
committed by GitHub
parent 61509dc5ea
commit e5ec47145f
5 changed files with 129 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
dbgpt/_private/config.py
View File

@@ -140,6 +140,17 @@ class Config(metaclass=Singleton):
os.environ["claude_proxyllm_api_base"] = os.getenv(
"ANTHROPIC_BASE_URL", "https://api.anthropic.com"
)
self.silicon_flow_proxy_api_key = os.getenv("SILICON_FLOW_API_KEY")
if self.silicon_flow_proxy_api_key:
os.environ[
"silicon_flow_proxyllm_proxy_api_key"
] = self.silicon_flow_proxy_api_key
os.environ["silicon_flow_proxyllm_proxyllm_backend"] = os.getenv(
"SILICON_FLOW_MODEL_VERSION", "Qwen/Qwen2.5-Coder-32B-Instruct"
)
os.environ["silicon_flow_proxyllm_api_base"] = os.getenv(
"SILICON_FLOW_API_BASE", "https://api.siliconflow.cn/v1"
)
self.proxy_server_url = os.getenv("PROXY_SERVER_URL")