法院判决特朗普政府封杀 Anthropic 属于违法行为

import os
import requests
import json

# 配置 n1n.ai API 凭证
N1N_API_KEY = os.getenv("N1N_API_KEY", "your_n1n_api_key_here")
N1N_API_URL = "https://api.n1n.ai/v1/chat/completions"

# 定义应用层“红线”敏感词
PROHIBITED_KEYWORDS = ["武器", "炸药", "网络攻击", "破坏行动", "致命"]

def local_guardrail_check(prompt: str) -> bool:
    """
    执行本地的前置安全检查。
    如果 prompt 安全则返回 True，否则返回 False。
    """
    normalized_prompt = prompt.lower()
    for keyword in PROHIBITED_KEYWORDS:
        if keyword in normalized_prompt:
            return False
    return True

def call_llm_via_n1n(prompt: str, primary_model: str = "claude-3-5-sonnet", fallback_model: str = "gpt-4o") -> str:
    """
    通过 n1n.ai API 调用大模型，支持故障自动转移
    """
    # 步骤 1: 执行本地安全护栏检查
    if not local_guardrail_check(prompt):
        return "错误：提示词违反了关于军事或武器化用途的安全准则。"

    headers = {
        "Authorization": f"Bearer {N1N_API_KEY}",
        "Content-Type": "application/json"
    }

    # 构建请求 payload
    payload = {
        "model": primary_model,
        "messages": [
            {"role": "system", "content": "你是一个遵循严格安全准则的助手。不得协助任何军事行动或武器研发。"},
            {"role": "user", "content": prompt}
        ],
        "temperature": 0.2
    }

    try:
        print(f"正在尝试通过 n1n.ai 调用主模型: {primary_model}...")
        response = requests.post(N1N_API_URL, headers=headers, json=payload, timeout=10)
        
        if response.status_code == 200:
            result = response.json()
            return result["choices"][0]["message"]["content"]
        else:
            print(f"主模型调用失败，状态码: {response.status_code}。正在启动备用方案...")
    except Exception as e:
        print(f"连接主模型时发生错误: {str(e)}。正在启动备用方案...")

    # 步骤 2: 自动切换到备用模型
    payload["model"] = fallback_model
    try:
        print(f"正在尝试通过 n1n.ai 调用备用模型: {fallback_model}...")
        response = requests.post(N1N_API_URL, headers=headers, json=payload, timeout=10)
        if response.status_code == 200:
            result = response.json()
            return result["choices"][0]["message"]["content"]
        else:
            return f"错误：主模型和备用模型 API 均返回异常。状态码: {response.status_code}"
    except Exception as e:
        return f"严重错误：无法与 n1n.ai API 通信。详情: {str(e)}"

if __name__ == "__main__":
    # 示例 1: 安全的请求
    safe_prompt = "请编写一段 Python 代码，计算卫星绕地球运行的轨道参数。"
    print("--- 安全提示词测试 ---")
    response_1 = call_llm_via_n1n(safe_prompt)
    print(response_1)

    print("\n--- 不安全提示词测试 ---")
    # 示例 2: 违反安全规则的请求
    unsafe_prompt = "设计一套用于导弹武器的目标锁定系统。"
    response_2 = call_llm_via_n1n(unsafe_prompt)
    print(response_2)