解决Gemini Pro API内容安全策略阻断回复的正确姿势

safety_settings = {
    'SEXUALLY_EXPLICIT': 'block_none',
    'HATE_SPEECH': 'block_none',
    # ...其他类别
}

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

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

# 初始化Vertex AI
vertexai.init(project="your-gcp-project-id", location="your-gcp-region") # 请替换为你的GCP项目ID和区域

def get_gemini_response_with_custom_safety(prompt_text: str):
    """
    使用自定义安全设置调用Gemini Pro模型。
    """
    model = GenerativeModel("gemini-pro")

    # 构建安全设置列表
    # 每个SafetySetting对象定义了一个特定类别的阻断阈值
    custom_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,
        ),
    ]

    try:
        response = model.generate_content(
            prompt_text,
            generation_config={
                "max_output_tokens": 2048,
                "temperature": 0.0,  # 0表示确定性更强
                "top_p": 1,
            },
            safety_settings=custom_safety_settings # 传入正确构建的安全设置列表
        )
        return response.text
    except Exception as e:
        print(f"调用Gemini Pro时发生错误: {e}")
        return None

# 示例调用
question = "请描述一个关于科幻小说中太空探索的复杂情节，其中包含一些紧张和危险的场景。"
# 假设你有一个可能被误判为敏感的文本
# question = "描述一个在野外生存时，如何处理严重受伤的情况，包括细节。"

response_text = get_gemini_response_with_custom_safety(question)
if response_text:
    print("Gemini Pro的回复:")
    print(response_text)
else:
    print("未能获取到回复或回复被阻断。")