aquin.session_mode Namespace Reference

Functions
bool	tool_requires_model (str tool_name)
str	resolve_legacy_verb (str verb)
dict[str, str]	cli_verbs_for_mode (SessionMode|None mode)
bool	verb_known (str verb)
SessionMode	normalize_mode (str|None raw)
frozenset[str]	tools_for_mode (SessionMode mode)
bool	is_tool_allowed (str tool_name, SessionMode mode)
dict[str, str]	filter_cli_tool_map (dict[str, str] full_map, SessionMode mode)
list[dict]	get_filtered_tool_schemas (SessionMode mode)
str	mode_label (SessionMode mode)
SessionMode	mode_for_model_id (str model_id)
SessionMode|None	mode_for_active_model ()

Variables
	SessionMode = Literal["llm"]
frozenset	SHARED_TOOLS
frozenset	MODEL_FREE_TOOLS
frozenset	LLM_TOOLS
frozenset	LLM_CLI_ONLY = frozenset({"trace", "feature logit", "feature neighbor"})
dict	SHARED_CLI_VERBS
dict	LLM_CLI_VERBS

Function Documentation

cli_verbs_for_mode()

dict[str, str] cli_verbs_for_mode

(

SessionMode | None

mode

)

CLI verb map for the active session mode.

Definition at line 115 of file session_mode.py.

filter_cli_tool_map()

dict[str, str] filter_cli_tool_map	(	dict[str, str]	full_map,
		SessionMode	mode )

Definition at line 137 of file session_mode.py.

References tools_for_mode().

get_filtered_tool_schemas()

list[dict] get_filtered_tool_schemas

(

SessionMode

mode

)

Definition at line 142 of file session_mode.py.

References tools_for_mode().

is_tool_allowed()

bool is_tool_allowed	(	str	tool_name,
		SessionMode	mode )

Definition at line 133 of file session_mode.py.

References tools_for_mode().

mode_for_active_model()

SessionMode | None mode_for_active_model

(

)

Infer mode from the model last loaded via `aquin load` (~/.aquin/active_model.txt).

Definition at line 161 of file session_mode.py.

mode_for_model_id()

SessionMode mode_for_model_id

(

str

model_id

)

Validate model id and return LLM session mode.

Definition at line 153 of file session_mode.py.

mode_label()

str mode_label

(

SessionMode

mode

)

Definition at line 149 of file session_mode.py.

normalize_mode()

SessionMode normalize_mode

(

str | None

raw

)

Definition at line 125 of file session_mode.py.

resolve_legacy_verb()

str resolve_legacy_verb

(

str

verb

)

Definition at line 111 of file session_mode.py.

Referenced by verb_known().

tool_requires_model()

bool tool_requires_model

(

str

tool_name

)

Definition at line 36 of file session_mode.py.

tools_for_mode()

frozenset[str] tools_for_mode

(

SessionMode

mode

)

Definition at line 129 of file session_mode.py.

Referenced by filter_cli_tool_map(), get_filtered_tool_schemas(), and is_tool_allowed().

verb_known()

bool verb_known

(

str

verb

)

Definition at line 120 of file session_mode.py.

References resolve_legacy_verb().

Variable Documentation

LLM_CLI_ONLY

frozenset aquin.session_mode.LLM_CLI_ONLY = frozenset({"trace", "feature logit", "feature neighbor"})

Definition at line 72 of file session_mode.py.

LLM_CLI_VERBS

dict aquin.session_mode.LLM_CLI_VERBS

Initial value:

=  {
    "benchmark": "run_benchmarks_on_top_feature",
    "benchmarks": "run_benchmarks_on_top_feature",
    "consistency-eval": "run_consistency_eval",
    "suppression-eval": "run_suppression_eval",
    "boundary-eval": "run_boundary_eval",
    "steer": "run_steer_and_show",
    "multi-steer": "run_multi_steer",
    "layer-analysis": "run_layer_analysis",
    "perturbation": "run_perturbation_sensitivity",
    "attention": "run_attention_routing",
    "check-weights": "check_weights",
    "simulate": "run_simulation",
    "red-team": "run_red_team",
}

Definition at line 94 of file session_mode.py.

LLM_TOOLS

frozenset aquin.session_mode.LLM_TOOLS

Definition at line 39 of file session_mode.py.

MODEL_FREE_TOOLS

frozenset aquin.session_mode.MODEL_FREE_TOOLS

Initial value:

=  frozenset({
    "read_session_memory",
    "write_session_memory",
    "list_simulation_runs",
    "load_simulation_run",
    "compare_simulations",
})

Definition at line 27 of file session_mode.py.

SessionMode

aquin.session_mode.SessionMode = Literal["llm"]

Definition at line 15 of file session_mode.py.

SHARED_CLI_VERBS

dict aquin.session_mode.SHARED_CLI_VERBS

Initial value:

=  {
    "feature locate": "run_find_feature",
    "feature logit": "get_feature_logits",
    "feature neighbor": "get_feature_neighbors",
    "sae-stats": "run_sae_stats",
    "diff weight": "run_merge_analysis",
    "check trajectory": "run_trajectory_analysis",
    "diff residue": "run_residual_drift",
    "confidence-analysis": "run_confidence_analysis",
    "eval": "run_custom_eval",
    "list simulation": "list_simulation_runs",
    "replay simulation": "load_simulation_run",
    "compare simulation": "compare_simulations",
    "list-runs": "list_simulation_runs",
    "load-run": "load_simulation_run",
    "compare-runs": "compare_simulations",
}

Definition at line 76 of file session_mode.py.

SHARED_TOOLS

frozenset aquin.session_mode.SHARED_TOOLS

Initial value:

=  frozenset({
    "write_session_memory",
    "read_session_memory",
    "run_custom_eval",
    "run_sae_stats",
})

Definition at line 19 of file session_mode.py.