rag_system.auto_search module

RAG Auto-Search Manager.

Manages per-channel auto-search configuration and provides context injection for automatic RAG searches on user messages.

Adapted for the v3 multi-platform architecture: channel keys use platform:channel_id composite format.

class rag_system.auto_search.RAGAutoSearchManager(redis_client)

Bases: object

Per-channel auto-search configuration backed by Redis.

Parameters:

redis_client (aioredis.Redis)

__init__(redis_client)

Initialize the instance.

Parameters:

redis_client (Redis) – Redis connection client.

Return type:

None

async set_channel_config(channel_key, store_names, enabled=True, n_results=3, min_score=0.5)

Set auto-search configuration for a channel.

Parameters:

• channel_key (str) – Composite key "platform:channel_id".

• store_names (List[str])

• enabled (bool)

• n_results (int)

• min_score (float)

Return type:

Dict[str, Any]

async get_channel_config(channel_key)

Retrieve the channel config.

Parameters:

channel_key (str) – The channel key value.

Returns:

The result.

Return type:

Optional[Dict[str, Any]]

async disable_channel(channel_key)

Disable channel.

Parameters:

channel_key (str) – The channel key value.

Returns:

True on success, False otherwise.

Return type:

bool

async remove_channel_config(channel_key)

Delete the specified channel config.

Parameters:

channel_key (str) – The channel key value.

Returns:

True on success, False otherwise.

Return type:

bool

async list_configured_channels()

List configured channels.

Returns:

The result.

Return type:

List[Dict[str, Any]]

async search_for_message(channel_key, message_content, chunk_size=10000, query_embedding=None, user_id='')

Perform auto-search if the channel is configured.

Parameters:

• query_embedding (list[float] | None) – Pre-computed embedding for message_content. When provided it is forwarded to ChromaDB to skip its internal embedding call.

• user_id (str) – The message author. Used to enforce access control on cloud_usr_ stores.

• string (Returns XML-formatted RAG context)

• None. (or)

• channel_key (str)

• message_content (str)

• chunk_size (int)

Return type:

Optional[str]