gemini-omega

Google Gemini Integration with OmegaEngine Governance

✨ Features

• 🤖 Drop-in Replacement — Same API as Google GenAI SDK
• 🛡️ Multi-modal Governance — Text, images, audio, video
• 📊 Response Filtering — Policy-based output control
• 📈 Full Audit Trail — Log all Gemini interactions
• ⚡ Streaming Support — Governed streaming
• 🔧 Function Calling — Validate tool executions

📦 Installation

pip install gemini-omega

🚀 Quick Start

from gemini_omega import GenerativeModel

model = GenerativeModel(
    model_name="gemini-1.5-pro",
    google_api_key="your-google-key",
    omega_api_key="your-omega-key",
    policy_id="gemini_policy",
)

response = model.generate_content("Explain black holes")

print(response.text)
print(response.omega_decision)  # "ALLOW"
print(response.omega_audit_id)  # "aud_xyz789"

🖼️ Multi-modal Governance

import PIL.Image

image = PIL.Image.open("document.jpg")

response = model.generate_content([
    "What's in this image?",
    image
])

# Image content is also governed
print(response.omega_decision)

📄 License

Licensed under the Apache License 2.0.