mirror of
https://github.com/linyqh/NarratoAI.git
synced 2025-12-12 11:22:51 +00:00
feat(vision_analyzer): 开放 Gemini 模型安全设置
- 在配置 API 客户端时,添加了安全设置配置 - 设置所有危害类别的阻塞阈值为 BLOCK_NONE,以开放模型的限制
This commit is contained in:
parent
eaa8ceb7e3
commit
2f41c13e19
@ -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),
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user