From 322aef6f1d24e86e40718b702a4cb11660061822 Mon Sep 17 00:00:00 2001
From: better629 <webdesign_xmu2012@163.com>
Date: Tue, 21 Nov 2023 19:24:57 +0800
Subject: [PATCH] fix zhipuapi's key to init openai api_key when using openai
 sdk

---
 metagpt/llm.py                  | 2 +-
 metagpt/provider/zhipuai_api.py | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

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 = {