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 RAG auto-search configuration and query fan-out, backed by Redis.

Stores one config record per channel under the stargazer:v3:rag:auto_search: Redis key prefix and, when a channel is enabled, automatically runs a semantic search across its configured stores for every inbound user message so the result can be injected into the LLM context. This is what makes a channel “RAG-aware” without the user having to invoke a tool. It owns no other state: all persistence is plain Redis GET / SET / SCAN / DELETE on string keys (config) and SISMEMBER on stargazer:v3:cloud_rag:shared: sets (cloud-store access control), while the actual vector search is delegated to rag_system.file_rag_manager.FileRAGManager instances resolved via rag_system.file_rag_manager.get_rag_store().

Constructed with a live redis.asyncio client by the RAG tool layer (tools/rag.py, tools/cloud_rag.py), the web config API (web/rag_config_api.py), and the inference message pipeline, whose search_for_message() is called per message from message_processor.generate_and_send.

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)

Write (create or overwrite) the auto-search config for one channel.

Builds the canonical config dict, clamps n_results to the 1-10 range and min_score to 0.0-1.0 so callers cannot persist out-of-band values, stamps a fresh updated_at timestamp, and JSON-serializes it into the single Redis key stargazer:v3:rag:auto_search:<channel_key> via SET. Once written, search_for_message() will start auto-searching that channel on the next message. An info log line records the change.

Called by the RAG admin tools in tools/rag.py and tools/cloud_rag.py (enable/share handlers) and by the web config API in web/rag_config_api.py to persist user edits.

Parameters:

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

• store_names (List[str]) – RAG store names to search for this channel.

• enabled (bool) – Whether auto-search is active for the channel.

• n_results (int) – Number of chunks to inject; clamped to 1-10.

• min_score (float) – Minimum similarity to keep a chunk; clamped to 0.0-1.0.

Returns:

The persisted config dict (post-clamping, with the new updated_at value).

Return type:

Dict[str, Any]

async get_channel_config(channel_key)

Load and decode the persisted auto-search config for a channel.

Reads the stargazer:v3:rag:auto_search:<channel_key> Redis key with a single GET and JSON-decodes it, returning None when the channel was never configured. This is the canonical read used both to render current settings and to gate whether a message should be auto-searched at all.

Called internally by search_for_message() (the enablement check) and disable_channel(), by the web config API in web/rag_config_api.py, by the cloud-RAG share/unshare tools in tools/cloud_rag.py, and by the RAG status tool in tools/rag.py.

Parameters:

channel_key (str) – Composite "platform:channel_id" identifier.

Returns:

The decoded config dict, or None if the channel has no stored config.

Return type:

Optional[Dict[str, Any]]

async disable_channel(channel_key)

Turn off auto-search for a channel without discarding its store list.

Loads the existing config via get_channel_config(), flips enabled to False, refreshes updated_at, and writes the record back with a Redis SET. This is a soft toggle: the configured store_names are preserved so the channel can be re-enabled later without re-selecting stores. Returns False (a no-op) when the channel was never configured.

Called by the RAG admin tool in tools/rag.py (the disable action).

Parameters:

channel_key (str) – Composite "platform:channel_id" identifier.

Returns:

True if a config existed and was updated, False if there was nothing to disable.

Return type:

bool

async remove_channel_config(channel_key)

Permanently delete a channel’s auto-search config from Redis.

Issues a single DEL on the stargazer:v3:rag:auto_search:<channel_key> key. Unlike disable_channel(), this discards the stored store_names entirely, so the channel reverts to having no RAG configuration at all.

Called by the web config API in web/rag_config_api.py, by the RAG admin tool in tools/rag.py, and by the cloud-RAG unshare flow in tools/cloud_rag.py when the last shared store is removed.

Parameters:

channel_key (str) – Composite "platform:channel_id" identifier.

Returns:

True if a key was actually deleted, False if no config existed for the channel.

Return type:

bool

async list_configured_channels()

Enumerate every channel that currently has an auto-search config.

Walks all stargazer:v3:rag:auto_search:* keys with a non-blocking SCAN iterator and GET-decodes each into its config dict. Used to render an admin overview of which channels have RAG enabled and against which stores; the scan plus per-key fetch makes this O(number of configured channels) rather than a single bulk read.

Called by the web config API in web/rag_config_api.py and by the RAG status/listing tool in tools/rag.py.

Returns:

One decoded config dict per configured channel (order follows the Redis scan, i.e. unspecified).

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 3072-d embedding for message_content. When provided it is forwarded to the pgvector store as the KNN query vector, skipping a redundant 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]