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

feat: add gemini support (#953)

Browse Source 
Signed-off-by: yihong0618 <zouzou0208@gmail.com>
Signed-off-by: Fangyin Cheng <staneyffer@gmail.com>
Co-authored-by: Fangyin Cheng <staneyffer@gmail.com>
This commit is contained in:
yihong
2023-12-23 11:10:42 +08:00
committed by GitHub
parent e1ace141f7
commit 12234ae258
8 changed files with 243 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
dbgpt/_private/config.py
View File

@@ -61,7 +61,7 @@ class Config(metaclass=Singleton):
if self.zhipu_proxy_api_key:
os.environ["zhipu_proxyllm_proxy_api_key"] = self.zhipu_proxy_api_key
os.environ["zhipu_proxyllm_proxyllm_backend"] = os.getenv(
"ZHIPU_MODEL_VERSION", "chatglm_pro"
"ZHIPU_MODEL_VERSION"
)
# wenxin
@@ -95,6 +95,14 @@ class Config(metaclass=Singleton):
os.environ["bc_proxyllm_proxy_api_secret"] = self.bc_proxy_api_secret
os.environ["bc_proxyllm_proxyllm_backend"] = self.bc_model_version
# gemini proxy
self.gemini_proxy_api_key = os.getenv("GEMINI_PROXY_API_KEY")
if self.gemini_proxy_api_key:
os.environ["gemini_proxyllm_proxy_api_key"] = self.gemini_proxy_api_key
os.environ["gemini_proxyllm_proxyllm_backend"] = os.getenv(
"GEMINI_MODEL_VERSION", "gemini-pro"
)
self.proxy_server_url = os.getenv("PROXY_SERVER_URL")
self.elevenlabs_api_key = os.getenv("ELEVENLABS_API_KEY")