AQIT 0.1.0
Loading...
Searching...
No Matches
aquin.attention_check Namespace Reference

Functions

tuple[Path, Path] write_attention_check (dict[str, Any] result, *, str|None tool_name, str|Path cwd)
None _plot_attention_check (dict[str, Any] result, str|None tool_name, Path png_path)
None _plot_llm_attention (dict[str, Any] result, Path png_path, Any plt, Any np)
str _short_token (str tok, int limit=10)

Variables

str _JSON_NAME = "attention-check.json"
str _PNG_NAME = "attention-check.png"

Function Documentation

◆ _plot_attention_check()

None _plot_attention_check ( dict[str, Any] result,
str | None tool_name,
Path png_path )
protected

Definition at line 40 of file attention_check.py.

References _plot_llm_attention().

Referenced by write_attention_check().

◆ _plot_llm_attention()

None _plot_llm_attention ( dict[str, Any] result,
Path png_path,
Any plt,
Any np )
protected

Definition at line 63 of file attention_check.py.

Referenced by _plot_attention_check().

◆ _short_token()

str _short_token ( str tok,
int limit = 10 )
protected

Definition at line 98 of file attention_check.py.

◆ write_attention_check()

tuple[Path, Path] write_attention_check ( dict[str, Any] result,
* ,
str | None tool_name,
str | Path cwd )

Definition at line 19 of file attention_check.py.

References _plot_attention_check().

Variable Documentation

◆ _JSON_NAME

str aquin.attention_check._JSON_NAME = "attention-check.json"
protected

Definition at line 15 of file attention_check.py.

◆ _PNG_NAME

str aquin.attention_check._PNG_NAME = "attention-check.png"
protected

Definition at line 16 of file attention_check.py.