feature_cli.py

Go to the documentation of this file.

# Copyright (c) 2025-present Aquin Labs Private Limited. All Rights Reserved.
"""aquin feature locate | logit | neighbor"""
 
import sys
 
 
def _print_help() -> None:
    print("SAE feature tools.")
    print("")
    print("Usage: aquin feature <locate|logit|neighbor> ...")
    print("")
    print("  locate   Rank features on labeled probes (e.g. honest vs deceptive)")
    print("           aquin feature locate [--prompts <json|jsonl>] [--layer N]")
    print("")
    print("  logit    Tokens boosted/suppressed by one feature")
    print("           aquin feature logit --feature <idx> [--layer N] [--topk N]")
    print("")
    print("  neighbor Nearest decoder neighbors of one feature")
    print("           aquin feature neighbor --feature <idx> [--layer N] [--topk N]")
    print("")
    print("Docs: https://aquin.app/docs/inspection-sae/llm")
 
 
def cmd_feature(args: list[str]) -> None:
    if not args or args[0] in ("-h", "--help", "help"):
        _print_help()
        return
 
    sub = args[0]
    rest = args[1:]
 
    if sub == "locate":
        from aquin.find_feature_cli import cmd_find_feature
 
        cmd_find_feature(rest)
        return
 
    if sub in ("logit", "logits"):
        from aquin.cli import cmd_feature_logits
 
        cmd_feature_logits(rest)
        return
 
    if sub in ("neighbor", "neighbors"):
        from aquin.cli import cmd_feature_neighbors
 
        cmd_feature_neighbors(rest)
        return
 
    print(f"Unknown feature subcommand: {sub}")
    print("Run: aquin feature --help")
    sys.exit(1)