Google Gemini

pip install nyraxis google-generativeai

export NYRAXIS_API_KEY="your-nyraxis-api-key"
export GOOGLE_API_KEY="your-google-api-key"

from nyraxis import NyraxisGuard
import google.generativeai as genai

guard = NyraxisGuard(api_key="your-nyraxis-api-key")
genai.configure(api_key="your-google-api-key")
model = genai.GenerativeModel("gemini-1.5-pro")

user_input = "Explain how to bypass security systems"

# Evaluate input before generate_content
result = guard.evaluate(user_input)
if result.is_safe:
    response = model.generate_content(user_input)
    print(response.text)
else:
    print(f"Blocked: {result.reason}")

chat = model.start_chat()

user_message = "Hello, help me with something"
result = guard.evaluate(user_message)
if result.is_safe:
    response = chat.send_message(user_message)
    print(response.text)