reframr/__init__.py

import sys
from pathlib import Path

_VENDOR_ROOT = Path(__file__).resolve().parent.parent / ".vendor"
for _vendor_path in (_VENDOR_ROOT / "python", _VENDOR_ROOT / "sitepkgs"):
    if _vendor_path.exists():
        vendor_text = str(_vendor_path)
        if vendor_text not in sys.path:
            sys.path.insert(0, vendor_text)

from .checkpoint import inspect_checkpoint, read_safetensor_file
from .config import ReframrConfig
from .embeddings import EmbeddingModel, fit_ppmi_embedding
from .hippo import AnalyticalMemoryUnit, hippo_legs_matrix
from .model import ReframrModel
from .reasoning import REASONING_CONTROL_TOKENS, REASONING_PROFILES, TOKENIZER_NAME
from .tokenizer import NativeTokenizer

__all__ = [
    "AnalyticalMemoryUnit",
    "EmbeddingModel",
    "NativeTokenizer",
    "REASONING_CONTROL_TOKENS",
    "REASONING_PROFILES",
    "ReframrConfig",
    "ReframrModel",
    "TOKENIZER_NAME",
    "fit_ppmi_embedding",
    "hippo_legs_matrix",
    "inspect_checkpoint",
    "read_safetensor_file",
]