max tokens

tools/indentify/audio_identifier.py

@@ -54,9 +54,9 @@ class AudioIdentifier:
         """确保Gemini已配置"""
         if not self._configured:
             # 与图片模块保持一致读取 GEMINI_API_KEY_1，若无则回退 GEMINI_API_KEY
-            self.api_key = os.getenv('GEMINI_API_KEY_1') or os.getenv('GEMINI_API_KEY')
+            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_1 或 GEMINI_API_KEY')
+                raise ValueError('请在环境变量中设置 GEMINI_API_KEY 或 GEMINI_API_KEY_1')
             genai.configure(api_key=self.api_key)
             # 使用通用多模态模型进行音频理解
             self.model = genai.GenerativeModel(
@@ -64,7 +64,7 @@ class AudioIdentifier:
                 generation_config=genai.GenerationConfig(
                     response_mime_type='text/plain',
                     temperature=0.2,
-                    max_output_tokens=40960
+                    max_output_tokens=409600
                 ),
                 safety_settings={
                     HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,

tools/indentify/video_identifier.py

@@ -79,7 +79,7 @@ class VideoIdentifier:
     def _ensure_configured(self):
         """确保Gemini已配置"""
         if not self._configured:
-            self.api_key = os.getenv('GEMINI_API_KEY')
+            self.api_key = os.getenv('GEMINI_API_KEY') or os.getenv('GEMINI_API_KEY_1')
             print(f"配置Gemini: {self.api_key}")
             genai.configure(api_key=self.api_key)
             self._configured = True
@@ -406,7 +406,7 @@ class VideoIdentifier:
                         generation_config=genai.GenerationConfig(
                             response_mime_type='application/json',
                             temperature=0.3,
-                            max_output_tokens=40960
+                            max_output_tokens=409600
                         ),
                         safety_settings={
                             HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,