trinity.buffer.viewer module

class trinity.buffer.viewer.SQLExperienceViewer(config: StorageConfig)

Bases: object

__init__(config: StorageConfig) → None

get_experiences(offset: int, limit: int = 10, filters: Dict | None = None) → List[Experience]

total_experiences(filters: Dict | None = None) → int

static run_viewer(model_path: str, db_url: str, table_name: str, schema_type: str, port: int) → None

Start the Streamlit viewer.

Parameters:

• model_path (str) – Path to the tokenizer/model directory.

• db_url (str) – Database URL for the experience database.

• table_name (str) – Name of the experience table in the database.

• schema_type (str) – Schema type of the experience table.

• port (int) – Port number to run the Streamlit app on.

trinity.buffer.viewer.get_color_for_action_mask(action_mask_value: int) → str

trinity.buffer.viewer.render_token_detail_html(html: str) → None

trinity.buffer.viewer.render_experience(exp: Experience, tokenizer: Any) → None

Render a single experience in Streamlit.

trinity.buffer.viewer.parse_args()

trinity.buffer.viewer.get_viewer(db_url: str, table_name: str, schema_type: str) → SQLExperienceViewer

trinity.buffer.viewer.main()