向量数据库

VectorDBToolSpec

基类：BaseToolSpec

向量数据库工具规范。

源代码位于 llama-index-integrations/tools/llama-index-tools-vector-db/llama_index/tools/vector_db/base.py

class VectorDBToolSpec(BaseToolSpec):
    """Vector DB tool spec."""

    spec_functions = ["auto_retrieve_fn"]

    def __init__(
        self,
        index: BaseIndex,  # TODO typing
    ) -> None:
        """Initialize with parameters."""
        self._index = index

    def auto_retrieve_fn(
        self,
        query: str,
        top_k: int,
        filter_key_list: List[str],
        filter_value_list: List[str],
    ) -> str:
        """
        Auto retrieval function.

        Performs auto-retrieval from a vector database, and then applies a set of filters.

        Args:
            query (str): The query to search
            top_k (int): The number of results to retrieve
            filter_key_list (List[str]): The list of filter keys
            filter_value_list (List[str]): The list of filter values

        """
        exact_match_filters = [
            ExactMatchFilter(key=k, value=v)
            for k, v in zip(filter_key_list, filter_value_list)
        ]
        retriever = VectorIndexRetriever(
            self._index,
            filters=MetadataFilters(filters=exact_match_filters),
            top_k=top_k,
        )
        query_engine = RetrieverQueryEngine.from_args(retriever)

        response = query_engine.query(query)
        return str(response)

auto_retrieve_fn

auto_retrieve_fn(query: str, top_k: int, filter_key_list: List[str], filter_value_list: List[str]) -> str

自动检索函数。

从向量数据库执行自动检索，然后应用一组过滤器。

参数

名称	类型	描述	默认值
query	str	要搜索的查询	必填
top_k	int	要检索的结果数量	必填
filter_key_list	List[str]	过滤器键列表	必填
filter_value_list	List[str]	过滤器值列表	必填

源代码位于 llama-index-integrations/tools/llama-index-tools-vector-db/llama_index/tools/vector_db/base.py

def auto_retrieve_fn(
    self,
    query: str,
    top_k: int,
    filter_key_list: List[str],
    filter_value_list: List[str],
) -> str:
    """
    Auto retrieval function.

    Performs auto-retrieval from a vector database, and then applies a set of filters.

    Args:
        query (str): The query to search
        top_k (int): The number of results to retrieve
        filter_key_list (List[str]): The list of filter keys
        filter_value_list (List[str]): The list of filter values

    """
    exact_match_filters = [
        ExactMatchFilter(key=k, value=v)
        for k, v in zip(filter_key_list, filter_value_list)
    ]
    retriever = VectorIndexRetriever(
        self._index,
        filters=MetadataFilters(filters=exact_match_filters),
        top_k=top_k,
    )
    query_engine = RetrieverQueryEngine.from_args(retriever)

    response = query_engine.query(query)
    return str(response)