ahvn.utils.vdb.vdb_utils module

Vector database utilities.

ahvn.utils.vdb.vdb_utils.parse_encoder_embedder(encoder=None, embedder=None)

Parse encoder and embedder parameters into standardized tuples and detect dimensions.

Parameters:

• encoder (Union[Callable[[Any], str], Tuple[Callable[[Any], str], Callable[[Any], str]]]) – Required encoder function or tuple (k_encoder, q_encoder).

• embedder (Union[str, Callable[[str], List[float]], Tuple[Callable[[str], List[float]], Callable[[str], List[float]]], LLM, None]) – Optional embedder function, tuple (k_embedder, q_embedder), or LLM instance.

Returns:

• encoder_tuple: (k_encoder, q_encoder) functions

• embedder_tuple: (k_embedder, q_embedder) functions or LLM instances

• k_dim: dimension of key embedder

• q_dim: dimension of query embedder

Return type:

Tuple of (encoder_tuple, embedder_tuple, k_dim, q_dim) where

Raises:

ValueError – If encoder is None or dimensions cannot be determined.

ahvn.utils.vdb.vdb_utils.resolve_vdb_config(collection=None, provider=None, **kwargs)

Resolve vector database configuration from parameters and environment variables.

Parameters:

• collection (str) – Optional collection name.

• provider (str) – Optional vector database provider name.

• **kwargs – Additional configuration parameters.

Returns:

Resolved configuration dictionary.