DSPy

pip install dspy-ai nyraxis

export NYRAXIS_API_KEY="your-api-key"
export NYRAXIS_PROJECT_ID="your-project-id"

import dspy
from nyraxis import Nyraxis

nx = Nyraxis()

lm = dspy.LM("openai/gpt-4")
dspy.configure(lm=lm)

class GuardedQA(dspy.Module):
    def __init__(self):
        self.generate = dspy.ChainOfThought("question -> answer")

    def forward(self, question):
        result = nx.evaluate(question)
        if result.blocked:
            return dspy.Prediction(answer="Request blocked by policy.")

        prediction = self.generate(question=question)

        output_result = nx.evaluate(prediction.answer, direction="output")
        if output_result.blocked:
            return dspy.Prediction(answer="Response blocked by policy.")

        return prediction

qa = GuardedQA()
response = qa(question="What is machine learning?")
print(response.answer)