cohere-omega

Cohere Integration with OmegaEngine Governance

✨ Features

• 🔍 RAG Governance — Secure retrieval-augmented generation
• 📊 Rerank Validation — Govern document reranking
• 🛡️ Embed Safety — Validate embedding inputs
• 📈 Full Audit — Log all Cohere API calls
• 🏢 Enterprise Ready — SSO, RBAC, and audit logging

📦 Installation

pip install cohere-omega

🚀 Quick Start

from cohere_omega import Client

co = Client(
    cohere_api_key="your-cohere-key",
    omega_api_key="your-omega-key",
    policy_id="cohere_policy",
)

# Governed chat
response = co.chat(
    message="What are the key findings?",
    documents=[...],
)

print(response.text)
print(response.omega_decision)

🔍 Governed RAG

# Rerank with governance
results = co.rerank(
    query="quarterly earnings",
    documents=[...],
    omega_validate_docs=True,  # Validate documents
)

for doc in results.results:
    if doc.omega_approved:
        use_document(doc)

📄 License

Licensed under the Apache License 2.0.