DriaAPIWrapper#

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

Dria API 的封装器。

此封装器便于与Dria的向量搜索和检索服务进行交互,包括创建知识库、插入数据和获取搜索结果。

Parameters:
  • api_key (str)

  • contract_id (str | None)

  • top_n (int)

api_key#

您用于访问Dria的API密钥。

contract_id#

要与之交互的知识库的合同ID。

top_n#

搜索时要获取的顶部结果数量。

方法

__init__(api_key[, contract_id, top_n])

create_knowledge_base(name, description, ...)

创建一个新的知识库。

insert_data(data)

将数据插入知识库。

query_with_vector(vector)

执行基于向量的查询。

run(query)

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

search(query)

执行基于文本的搜索。

__init__(api_key: str, contract_id: str | None = None, top_n: int = 10)[source]#
Parameters:
  • api_key (str)

  • contract_id (str | None)

  • top_n (int)

create_knowledge_base(name: str, description: str, category: str, embedding: str) str[source]#

创建一个新的知识库。

Parameters:
  • name (str)

  • 描述 (str)

  • category (str)

  • embedding (str)

Return type:

字符串

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

将数据插入知识库。

Parameters:

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

Return type:

字符串

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

执行基于向量的查询。

Parameters:

向量 (列表[浮点数])

Return type:

列表[字典[str, 任意]]

run(query: str | List[float]) List[Dict[str, Any]] | None[来源]#

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

Parameters:
  • query (str | List[float]) – 用于基于文本搜索的字符串或用于

  • query. (基于向量的)

Returns:

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

Return type:

列表[字典[str, 任意]] | 无

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

执行基于文本的搜索。

Parameters:

query (str)

Return type:

列表[字典[str, 任意]]