nat.plugins.redis.schema

Attributes

logger
INDEX_NAME
DEFAULT_DIM

Functions

create_schema([embedding_dim])	Create the Redis search schema for redis_memory.
ensure_index_exists(→ None)	Ensure the Redis search index exists, creating it if necessary.

Module Contents

logger

INDEX_NAME = 'memory_idx'

DEFAULT_DIM = 384

create_schema(embedding_dim: int = DEFAULT_DIM)

Create the Redis search schema for redis_memory.

Args:

embedding_dim (int): Dimension of the embedding vectors

Returns:

tuple: Schema definition for Redis search

async ensure_index_exists(

client: redis.asyncio.Redis,

key_prefix: str,

embedding_dim: int | None,

) → None

Ensure the Redis search index exists, creating it if necessary.

Args:

client (redis.Redis): Redis client instance key_prefix (str): Prefix for keys to be indexed embedding_dim (Optional[int]): Dimension of embedding vectors. If None, uses default.