trinity.buffer.storage.sql module

SQL database storage for experience and task buffers.

Primary classes (async, used by Ray actors and production hot paths):

SQLExperienceStorage, SQLTaskStorage

Factory:

SQLStorage.get_wrapper(config) — returns Ray actor handle.

class trinity.buffer.storage.sql.SQLExperienceStorage(config: StorageConfig)

Bases: object

Primary async SQL storage for experiences. Used directly as Ray actor.

__init__(config: StorageConfig) → None

async prepare() → None

Initialize async engine and create tables.

async write(data: List[Experience]) → None

async read(batch_size: int | None = None, **kwargs) → List[Experience]

async count(filters: Dict | None = None) → int

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

async classmethod load_from_dataset(dataset: Dataset, config: StorageConfig) → SQLExperienceStorage

acquire() → int

release() → int

class trinity.buffer.storage.sql.SQLTaskStorage(config: StorageConfig)

Bases: object

Primary async SQL storage for tasks. Used directly as Ray actor.

__init__(config: StorageConfig) → None

async prepare() → None

Initialize async engine and create tables.

async write(data: List[Dict]) → None

async read(batch_size: int | None = None) → List[Task]

async classmethod load_from_dataset(dataset: Dataset, config: StorageConfig) → SQLTaskStorage

acquire() → int

release() → int

class trinity.buffer.storage.sql.SQLStorage

Bases: object

Factory for creating SQL storage Ray actors.

classmethod get_wrapper(config: StorageConfig)