diff --git a/metagpt/llm.py b/metagpt/llm.py index b653df946..4edcd7a83 100644 --- a/metagpt/llm.py +++ b/metagpt/llm.py @@ -24,7 +24,7 @@ def LLM() -> "BaseGPTAPI": llm = Claude() elif CONFIG.spark_api_key: llm = SparkAPI() - elif CONFIG.zhipuai_api_key and CONFIG.zhipuai_api_key != "YOUR_API_KEY": + elif CONFIG.zhipuai_api_key: llm = ZhiPuAIGPTAPI() else: raise RuntimeError("You should config a LLM configuration first") diff --git a/metagpt/provider/zhipuai_api.py b/metagpt/provider/zhipuai_api.py index 064ec35ba..3161c0e88 100644 --- a/metagpt/provider/zhipuai_api.py +++ b/metagpt/provider/zhipuai_api.py @@ -13,6 +13,7 @@ from tenacity import ( ) from requests import ConnectionError +import openai import zhipuai from metagpt.config import CONFIG @@ -46,6 +47,7 @@ class ZhiPuAIGPTAPI(BaseGPTAPI): def __init_zhipuai(self, config: CONFIG): assert config.zhipuai_api_key zhipuai.api_key = config.zhipuai_api_key + openai.api_key = zhipuai.api_key # due to use openai sdk, set the api_key but it will't be used. def _const_kwargs(self, messages: list[dict]) -> dict: kwargs = {