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.

参数:

• 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.

返回:

• 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

返回类型:

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

抛出:

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.

参数:

• collection (str) -- Optional collection name.

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

• **kwargs -- Additional configuration parameters.

返回:

Resolved configuration dictionary.