logical_reasoner.py

class logical_reasoner(dspy.Module):
    def __init__(self):
        super().__init__()
        self.logical_reasoner = dspy.Predict("text -> premises, conclusions")
        self.checker = dspy.ChainOfThought(check_logic)

    def forward(self,text):
        prediction = self.logical_reasoner(text=text)
        result = self.checker(premises=prediction.premises, conclusions=prediction.conclusions)
        return result

text="If it is raining, then the grass is wet. The grass is wet. Therefore, it is not raining."

lr = logical_reasoner()

print(lr(text=text))