semantic_router.encoders.cohere

​

CohereEncoder Objects

class CohereEncoder(DenseEncoder, AsymmetricDenseMixin)

Dense encoder that uses Cohere API to embed documents. Supports text only. Requires a Cohere API key from https://dashboard.cohere.com/api-keys.

​

__init__

def __init__(name: Optional[str] = None,
             cohere_api_key: Optional[str] = None,
             score_threshold: float = 0.3)

Initialize the Cohere encoder.

Arguments:

• name (str): The name of the embedding model to use.
• cohere_api_key (str): The API key for the Cohere client, can also be set via the COHERE_API_KEY environment variable.
• score_threshold (float): The threshold for the score of the embedding.
• input_type (str): The type of input to embed.

​

__call__

def __call__(docs: List[str]) -> List[List[float]]

Embed a list of documents. Supports text only.

Arguments:

• docs (List[str]): The documents to embed.

Returns:

List[List[float]]: The vector embeddings of the documents.