|
|
@@ -53,10 +53,10 @@ class ImageIdentifier:
|
|
|
def _ensure_configured(self):
|
|
|
"""确保Gemini已配置"""
|
|
|
if not self._configured:
|
|
|
- self.api_key = os.getenv('GEMINI_API_KEY_1')
|
|
|
+ self.api_key = os.getenv('GEMINI_API_KEY') or os.getenv('GEMINI_API_KEY_1')
|
|
|
|
|
|
if not self.api_key:
|
|
|
- raise ValueError("请在环境变量中设置 GEMINI_API_KEY")
|
|
|
+ raise ValueError("请在环境变量中设置 GEMINI_API_KEY 或 GEMINI_API_KEY_1")
|
|
|
genai.configure(api_key=self.api_key)
|
|
|
|
|
|
# 创建模型时设置安全设置，避免内容被过滤
|
