dspy-omega

DSPy Integration for OmegaEngine

✨ Features

• 🧠 Programmatic Prompting — Govern DSPy modules
• 📊 Signature Validation — Validate input/output signatures
• 🔄 Optimizer Governance — Control optimization runs
• 📈 Trace Logging — Full audit of DSPy pipelines
• 🛡️ Module Wrappers — Governance for any DSPy module

📦 Installation

pip install dspy-omega

🚀 Quick Start

import dspy
from dspy_omega import OmegaModule, configure_omega

# Configure governance
configure_omega(
    api_key="your-omega-key",
    policy_id="dspy_policy",
)

class RAG(dspy.Module):
    def __init__(self):
        self.retrieve = dspy.Retrieve(k=3)
        self.generate = dspy.ChainOfThought("context, question -> answer")

    def forward(self, question):
        context = self.retrieve(question).passages
        return self.generate(context=context, question=question)

# Wrap with governance
governed_rag = OmegaModule(RAG())

# All module calls are governed
result = governed_rag(question="What is AI safety?")
print(result.answer)
print(result.omega_audit_id)

🔄 Optimizer Governance

from dspy_omega import governed_optimizer

# Control optimization runs
optimizer = governed_optimizer(
    dspy.MIPROv2(),
    omega_api_key="...",
    max_unsafe_retries=3,
)

📄 License

Licensed under the Apache License 2.0.