fix: baichuan proxy modelset

eosphoros-ai · Jan 3, 2024 · db36648 · db36648
1 parent ef5570f
commit db36648
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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