Update src/ZhipuClient.py

src/ZhipuClient.py

@@ -3,18 +3,18 @@ import os
 
 class ZhipuClient:
     def __init__(self, api_key_file_path = None):
-        if api_key_file_path is None:
-            cands = ['./datas/zhipu_key.txt', '../datas/zhipu_key.txt']
-            flag = False
-            for cand in cands:
-                if os.path.exists(cand):
-                    api_key_file_path = cand
-                    flag = True
-                    break
-            if not flag:
-                raise ValueError("No valid api key file found.")
+        # if api_key_file_path is None:
+        #     cands = ['./datas/zhipu_key.txt', '../datas/zhipu_key.txt']
+        #     flag = False
+        #     for cand in cands:
+        #         if os.path.exists(cand):
+        #             api_key_file_path = cand
+        #             flag = True
+        #             break
+        #     if not flag:
+        #         raise ValueError("No valid api key file found.")
 
-        self.api_key = self._load_access_token(api_key_file_path)
+        self.api_key = os.getenv("ZHIPU_4_API")
         self.client = ZhipuAI(api_key=self.api_key)
 
     def _load_access_token(self, file_path):