mirror of
https://github.com/linyqh/NarratoAI.git
synced 2025-12-12 03:02:48 +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 PIL.Image
|
||||||
import traceback
|
import traceback
|
||||||
|
|
||||||
|
|
||||||
class VisionAnalyzer:
|
class VisionAnalyzer:
|
||||||
"""视觉分析器类"""
|
"""视觉分析器类"""
|
||||||
|
|
||||||
@ -28,7 +29,15 @@ class VisionAnalyzer:
|
|||||||
def _configure_client(self):
|
def _configure_client(self):
|
||||||
"""配置API客户端"""
|
"""配置API客户端"""
|
||||||
genai.configure(api_key=self.api_key)
|
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(
|
@retry(
|
||||||
stop=stop_after_attempt(3),
|
stop=stop_after_attempt(3),
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user