Gemini Pro API安全设置详解：如何有效避免内容阻断

def get_gemini_response(question, safety_settings=None):
    if safety_settings is None:
        safety_settings = {
            'SEXUALLY_EXPLICIT': 'block_none',
            'HATE_SPEECH': 'block_none',
            'HARASSMENT': 'block_none',
            'DANGEROUS_CONTENT': 'block_none'
        }
    # ... 调用模型 ...

BlockedPromptException: block_reason: SAFETY safety_ratings { category: HARM_CATEGORY_SEXUALLY_EXPLICIT probability: NEGLIGIBLE } safety_ratings { category: HARM_CATEGORY_HATE_SPEECH probability: HIGH } safety_ratings { category: HARM_CATEGORY_HARASSMENT probability: NEGLIGIBLE } safety_ratings { category: HARM_CATEGORY_DANGEROUS_CONTENT probability: NEGLIGIBLE }

from vertexai.preview.generative_models import (
    GenerativeModel,
    HarmCategory, 
    HarmBlockThreshold,
    GenerationResponse
)
from google.cloud.aiplatform_v1beta1.types.content import SafetySetting

from vertexai.preview.generative_models import (
    GenerativeModel,
    HarmCategory, 
    HarmBlockThreshold,
    GenerationResponse
)
from google.cloud.aiplatform_v1beta1.types.content import SafetySetting

def get_gemini_response_with_custom_safety(prompt_text: str, text: str) -> GenerationResponse:
    """
    使用自定义安全设置调用Gemini Pro模型。

    Args:
        prompt_text: 包含占位符的提示模板。
        text: 填充到提示模板中的实际文本。

    Returns:
        模型的生成响应。
    """
    model = GenerativeModel("gemini-pro")

    response: GenerationResponse = model.generate_content(
        prompt_text.format(text),
        generation_config={
            "max_output_tokens": 2048,
            "temperature": 0,
            "top_p": 1,
        },
        safety_settings=[
            SafetySetting(
                category=HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
                threshold=HarmBlockThreshold.BLOCK_NONE,
            ),
            SafetySetting(
                category=HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
                threshold=HarmBlockThreshold.BLOCK_NONE,
            ),
            SafetySetting(
                category=HarmCategory.HARM_CATEGORY_HATE_SPEECH,
                threshold=HarmBlockThreshold.BLOCK_NONE,
            ),
            SafetySetting(
                category=HarmCategory.HARM_CATEGORY_HARASSMENT,
                threshold=HarmBlockThreshold.BLOCK_NONE,
            ),
        ]
    )
    return response

# 示例用法
if __name__ == "__main__":
    example_prompt = "请描述以下文本内容：{}"
    # 假设这里有一个可能触发安全审查的文本
    example_text = "一个关于争议性事件的描述，可能包含敏感词汇。" 

    try:
        # 调用自定义安全设置的函数
        gemini_response = get_gemini_response_with_custom_safety(example_prompt, example_text)
        print("Gemini Pro 响应内容：")
        for part in gemini_response.candidates[0].content.parts:
            print(part.text)

        # 打印安全评分（即使设置为BLOCK_NONE，模型仍会返回评分）
        print("\n安全评分：")
        for rating in gemini_response.candidates[0].safety_ratings:
            print(f"  类别: {rating.category.name}, 概率: {rating.probability.name}")

    except Exception as e:
        print(f"调用Gemini Pro API时发生错误: {e}")