From 2f41c13e19ce174ded2370a85b4edf3cb30138a3 Mon Sep 17 00:00:00 2001 From: linyq Date: Tue, 12 Nov 2024 09:42:35 +0800 Subject: [PATCH] =?UTF-8?q?feat(vision=5Fanalyzer):=20=E5=BC=80=E6=94=BE?= =?UTF-8?q?=20Gemini=20=E6=A8=A1=E5=9E=8B=E5=AE=89=E5=85=A8=E8=AE=BE?= =?UTF-8?q?=E7=BD=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 在配置 API 客户端时，添加了安全设置配置 - 设置所有危害类别的阻塞阈值为 BLOCK_NONE，以开放模型的限制 --- app/utils/vision_analyzer.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/app/utils/vision_analyzer.py b/app/utils/vision_analyzer.py index eb143ec..8024729 100644 --- a/app/utils/vision_analyzer.py +++ b/app/utils/vision_analyzer.py @@ -11,6 +11,7 @@ import google.generativeai as genai import PIL.Image import traceback + class VisionAnalyzer: """视觉分析器类""" @@ -28,7 +29,15 @@ class VisionAnalyzer: def _configure_client(self): """配置API客户端""" genai.configure(api_key=self.api_key) - self.model = genai.GenerativeModel(self.model_name) + # 开放 Gemini 模型安全设置 + from google.generativeai.types import HarmCategory, HarmBlockThreshold + safety_settings = { + HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE, + HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE, + HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE, + HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE, + } + self.model = genai.GenerativeModel(self.model_name, safety_settings=safety_settings) @retry( stop=stop_after_attempt(3),