langchain_community.utilities.dria_index.DriaAPIWrapper

class langchain_community.utilities.dria_index.DriaAPIWrapper(api_key: str, contract_id: Optional[str] = None, top_n: int = 10)

封装了Dria API。

该封装简化了与Dria的向量搜索和检索服务的交互，包括创建知识库、插入数据和获取搜索结果。

属性:

api_key: 用于访问Dria的API密钥。 contract_id: 要交互的知识库的合同ID。 top_n: 搜索时要获取的前N个结果的数量。

Methods

__init__(api_key[, contract_id, top_n])
create_knowledge_base(name, description, ...)	创建一个新的知识库。
insert_data(data)	将数据插入知识库。
query_with_vector(vector)	执行基于向量的查询。
run(query)	处理基于文本搜索和基于向量查询的方法。
search(query)	执行基于文本的搜索。

Parameters

• api_key (str) –

• contract_id (Optional[str]) –

• top_n (int) –

__init__(api_key: str, contract_id: Optional[str] = None, top_n: int = 10)

Parameters

• api_key (str) –

• contract_id (Optional[str]) –

• top_n (int) –

create_knowledge_base(name: str, description: str, category: str, embedding: str) → str

创建一个新的知识库。

Parameters

• name (str) –

• description (str) –

• category (str) –

• embedding (str) –

Return type

str

insert_data(data: List[Dict[str, Any]]) → str

将数据插入知识库。

Parameters

data (List[Dict[str, Any]]) –

Return type

str

query_with_vector(vector: List[float]) → List[Dict[str, Any]]

执行基于向量的查询。

Parameters

vector (List[float]) –

Return type

List[Dict[str, Any]]

run(query: Union[str, List[float]]) → Optional[List[Dict[str, Any]]]

处理基于文本搜索和基于向量查询的方法。

参数：

query：用于文本搜索的字符串或用于向量查询的浮点数列表。

返回：

来自Dria的搜索或查询结果。

Parameters

query (Union[str, List[float]]) –

Return type

Optional[List[Dict[str, Any]]]

search(query: str) → List[Dict[str, Any]]

执行基于文本的搜索。

Parameters

query (str) –

Return type

List[Dict[str, Any]]