api_key

Secret

Default: {"type": "env_var", "env_vars": ["COHERE_API_KEY", "CO_API_KEY"], "strict": False}

The Cohere API key. Required.

model

String

embed-english-v3.0,
embed-english-light-v3.0,
embed-multilingual-v3.0,
embed-multilingual-light-v3.0,
embed-english-v2.0,
embed-english-light-v2.0,
embed-multilingual-v2.0
Default: embed-english-v2.0,

The name of the model to use. For a list of supported models, see Cohere documentation.
Required.

input_type

String

search_document,
search_query,
classification,
clustering
Default: search_query

Specifies the type of input for the model. Optional for models lower than v3, required for v3 and higher.

api_base_url

String

Default: https://api.cohere.com

The Cohere API base URL.
Optional.

truncate

String

NONE,
START,
END
Default: END

Truncate embeddings that are too long from the start or the end. Possible values:
START discards the start of the input until the remaining input is exactly the maximum input token length for the model.
END discards the end of the input until the remaining input is exactly the maximum input token length for the model.
NONEreturns an error if the input exceeds the maximum input token length.

Optional.

use_async_client

Boolean

True
False
Default: False

Flag to select the AsyncClient. Recommended for applications with many concurrent calls.
Optional.

timeout

Integer

Default: 120

Request timeout in seconds.
Optional.