|
@@ -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
|
|
|
|