Reframr-RFM-v1-Base / reframr /reasoning.py
OkeyMeta's picture
Release Reframr-RFM-v1-Base public checkpoint
2147ce8 verified
TOKENIZER_NAME = "FrameToken"
REASONING_CONTROL_TOKENS: tuple[str, ...] = (
"<reason>",
"<plan>",
"<reflect>",
"<answer>",
"<memory>",
"<retrieve>",
"<focus>",
"<verify>",
"<tool>",
)
REASONING_PROFILES: dict[str, tuple[str, ...]] = {
"none": (),
"deep": ("<reason>",),
"memory": ("<memory>", "<retrieve>", "<focus>"),
"tool": ("<tool>", "<reason>", "<verify>"),
}
def reasoning_prefix(mode: str) -> list[str]:
if mode not in REASONING_PROFILES:
raise ValueError(f"Unknown reasoning mode: {mode}")
return list(REASONING_PROFILES[mode])