From db366489b3d2ffbca8c0cb4bd89b4da36b0aea8f Mon Sep 17 00:00:00 2001 From: csunny Date: Wed, 3 Jan 2024 09:55:33 +0800 Subject: [PATCH] fix: baichuan proxy modelset --- dbgpt/_private/config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dbgpt/_private/config.py b/dbgpt/_private/config.py index 2b6ea35a7..63811e17f 100644 --- a/dbgpt/_private/config.py +++ b/dbgpt/_private/config.py @@ -89,7 +89,7 @@ def __init__(self) -> None: # baichuan proxy self.bc_proxy_api_key = os.getenv("BAICHUAN_PROXY_API_KEY") self.bc_model_name = os.getenv("BAICHUN_MODEL_NAME", "Baichuan2-Turbo-192k") - if self.bc_proxy_api_key and self.bc_proxy_api_secret: + if self.bc_proxy_api_key and self.bc_model_name: os.environ["bc_proxyllm_proxy_api_key"] = self.bc_proxy_api_key os.environ["bc_proxyllm_proxyllm_backend"] = self.bc_model_name