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)[source]¶

Wrapper around Dria API.

This wrapper facilitates interactions with Dria’s vector search and retrieval services, including creating knowledge bases, inserting data, and fetching search results.

Parameters

  • api_key (str) –

  • contract_id (Optional[str]) –

  • top_n (int) –

api_key¶

Your API key for accessing Dria.

contract_id¶

The contract ID of the knowledge base to interact with.

top_n¶

Number of top results to fetch for a search.

Methods

__init__(api_key[, contract_id, top_n])

create_knowledge_base(name, description, ...)

Create a new knowledge base.

insert_data(data)

Insert data into the knowledge base.

query_with_vector(vector)

Perform a vector-based query.

run(query)

Method to handle both text-based searches and vector-based queries.

search(query)

Perform a text-based search.
__init__(api_key: str, contract_id: Optional[str] = None, top_n: int = 10)[source]¶
Parameters

  • api_key (str) –

  • contract_id (Optional[str]) –

  • top_n (int) –

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

Create a new knowledge base.

Parameters

  • name (str) –

  • description (str) –

  • category (str) –

  • embedding (str) –

Return type

str

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

Insert data into the knowledge base.

Parameters

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

Return type

str

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

Perform a vector-based query.

Parameters

vector (List[float]) –

Return type

List[Dict[str, Any]]

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

Method to handle both text-based searches and vector-based queries.

Parameters

  • query (Union[str, List[float]]) – A string for text-based search or a list of floats for

  • query. (vector-based) –

Returns

The search or query results from Dria.

Return type

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

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

Perform a text-based search.

Parameters

query (str) –

Return type

List[Dict[str, Any]]