jihuaqiang před 2 týdny
rodič
revize
f0c6315e3d
1 změnil soubory, kde provedl 7 přidání a 8 odebrání
  1. 7 8
      gemini.py

+ 7 - 8
gemini.py

@@ -19,20 +19,19 @@ class GeminiProcessor:
         # 加载环境变量
         load_dotenv()
         
-        # 获取API密钥,支持多密钥配置
-        self.api_key = os.getenv('GEMINI_API_KEY')
-        if not self.api_key:
-            # 如果没有基础密钥,尝试使用第一个多密钥
-            self.api_key = os.getenv('GEMINI_API_KEY_1')
-            if not self.api_key:
-                raise ValueError("未找到GEMINI_API_KEY或GEMINI_API_KEY_1环境变量")
-        
         # 延迟配置Gemini,在真正使用时再设置
         self._configured = False
 
     def _ensure_configured(self):
         """确保Gemini已配置"""
         if not self._configured:
+            # 获取API密钥,支持多密钥配置
+            self.api_key = os.getenv('GEMINI_API_KEY')
+            if not self.api_key:
+                # 如果没有基础密钥,尝试使用第一个多密钥
+                self.api_key = os.getenv('GEMINI_API_KEY_1')
+                if not self.api_key:
+                    raise ValueError("未找到GEMINI_API_KEY或GEMINI_API_KEY_1环境变量")
             genai.configure(api_key=self.api_key)
             self._configured = True