Spaces:
Sleeping
Sleeping
Tuchuanhuhuhu
commited on
Commit
·
bbf38ab
1
Parent(s):
8c04739
bugfix: 修复从 PaLM 切换到 GPT 模型时 API Key 出错的问题
Browse files- modules/models/models.py +2 -1
modules/models/models.py
CHANGED
|
@@ -562,6 +562,7 @@ def get_model(
|
|
| 562 |
try:
|
| 563 |
if model_type == ModelType.OpenAI:
|
| 564 |
logging.info(f"正在加载OpenAI模型: {model_name}")
|
|
|
|
| 565 |
model = OpenAIClient(
|
| 566 |
model_name=model_name,
|
| 567 |
api_key=access_key,
|
|
@@ -614,7 +615,7 @@ def get_model(
|
|
| 614 |
model = ChuanhuAgent_Client(model_name, access_key, user_name=user_name)
|
| 615 |
elif model_type == ModelType.GooglePaLM:
|
| 616 |
from .Google_PaLM import Google_PaLM_Client
|
| 617 |
-
access_key = os.environ.get("GOOGLE_PALM_API_KEY")
|
| 618 |
model = Google_PaLM_Client(model_name, access_key, user_name=user_name)
|
| 619 |
elif model_type == ModelType.LangchainChat:
|
| 620 |
from .azure import Azure_OpenAI_Client
|
|
|
|
| 562 |
try:
|
| 563 |
if model_type == ModelType.OpenAI:
|
| 564 |
logging.info(f"正在加载OpenAI模型: {model_name}")
|
| 565 |
+
access_key = os.environ.get("OPENAI_API_KEY", access_key)
|
| 566 |
model = OpenAIClient(
|
| 567 |
model_name=model_name,
|
| 568 |
api_key=access_key,
|
|
|
|
| 615 |
model = ChuanhuAgent_Client(model_name, access_key, user_name=user_name)
|
| 616 |
elif model_type == ModelType.GooglePaLM:
|
| 617 |
from .Google_PaLM import Google_PaLM_Client
|
| 618 |
+
access_key = os.environ.get("GOOGLE_PALM_API_KEY", access_key)
|
| 619 |
model = Google_PaLM_Client(model_name, access_key, user_name=user_name)
|
| 620 |
elif model_type == ModelType.LangchainChat:
|
| 621 |
from .azure import Azure_OpenAI_Client
|