Engine and router

Selection → Routing → Invocation → Refine-validated Expansion → Critic gate → (Recursion) → Synthesis

import dspy
from qlaw.signatures import LensRouter as LensRouterSignature
from qlaw.signatures import Termination

class LensRouter(dspy.Module):
    """Auto-select the lens from node type + context (model: deepseek-v4-flash).

    NOTE: the signature is imported ALIASED — the module class below shadows the
    bare name, so `dspy.Predict(LensRouter)` would pass the module class itself
    as a signature and break at compile time."""
    def __init__(self):
        super().__init__()
        self.predict = dspy.Predict(LensRouterSignature)

    def forward(self, node, **kwargs):
        return self.predict(node=node)  # Prediction: lens

def default_lens_for(node_type) -> str:
    """Deterministic fallback."""
    from qlaw.graph import NodeType
    return {
        NodeType.PROBLEM: "decompose", NodeType.COMPONENT: "decompose",
        NodeType.QUESTION: "ontology", NodeType.PLAN: "trajectories",
        NodeType.TRAJECTORY: "critique", NodeType.DATA: "enrich",
    }.get(node_type, "explain")

class Terminator(dspy.Module):
    """Decide whether to stop recursion at a node (model: deepseek-v4-flash)."""
    def __init__(self):
        super().__init__()
        self.predict = dspy.Predict(Termination)

    def forward(self, node, **kwargs):
        return self.predict(node=node)  # Prediction: stop (bool)