Create a Cohere inference endpoint

Auth

Create an inference endpoint to perform an inference task with the cohere service.

Required privileges

Cluster privileges manage_inference

Path parameters

task_type enum required

The type of the inference task that the model will perform.

Values: completion rerank text_embedding
cohere_inference_id string required

The unique identifier of the inference endpoint.

Query parameters

timeout string · default: 30s

Specifies the amount of time to wait for the inference endpoint to be created.

Request body

chunking_settings object

The chunking configuration object. Applies only to the text_embedding task type. Not applicable to the rerank or completion task type.

Reference
max_chunk_size number · default: 250

The maximum size of a chunk in words. This value cannot be lower than 20 (for sentence strategy) or 10 (for word strategy). This value should not exceed the window size for the associated model.

overlap number · default: 100

The number of overlapping words for chunks. It is applicable only to a word chunking strategy. This value cannot be higher than half the max_chunk_size value.

sentence_overlap number · default: 1

The number of overlapping sentences for chunks. It is applicable only for a sentence chunking strategy. It can be either 1 or 0.

separator_group string

Only applicable to the recursive strategy and required when using it.

Sets a predefined list of separators in the saved chunking settings based on the selected text type. Values can be markdown or plaintext.

Using this parameter is an alternative to manually specifying a custom separators list.

separators string[]

Only applicable to the recursive strategy and required when using it.

A list of strings used as possible split points when chunking text.

Each string can be a plain string or a regular expression (regex) pattern. The system tries each separator in order to split the text, starting from the first item in the list.

After splitting, it attempts to recombine smaller pieces into larger chunks that stay within the max_chunk_size limit, to reduce the total number of chunks generated.

strategy string · default: sentence

The chunking strategy: sentence, word, none or recursive.

  • If strategy is set to recursive, you must also specify:
  • max_chunk_size
  • either separators orseparator_group

Learn more about different chunking strategies in the linked documentation.

Reference
service enum required

The type of service supported for the specified task type. In this case, cohere.

Values: cohere
service_settings object required

Settings used to install the inference model. These settings are specific to the cohere service.

api_key string required

A valid API key for your Cohere account. You can find or create your Cohere API keys on the Cohere API key settings page.

Important

You need to provide the API key only once, during the inference model creation. The get inference endpoint API does not retrieve your API key.

External docs
embedding_type enum · default: float

For a text_embedding task, the types of embeddings you want to get back. Use binary for binary embeddings, which are encoded as bytes with signed int8 precision. Use bit for binary embeddings, which are encoded as bytes with signed int8 precision (this is a synonym of binary). Use byte for signed int8 embeddings (this is a synonym of int8). Use float for the default float embeddings. Use int8 for signed int8 embeddings.

Values: binary bit byte float int8
model_id string required

For a completion, rerank, or text_embedding task, the name of the model to use for the inference task.

rate_limit object

This setting helps to minimize the number of rate limit errors returned from Cohere. By default, the cohere service sets the number of requests allowed per minute to 10000.

requests_per_minute number

The number of requests allowed per minute. By default, the number of requests allowed per minute is set by each service as follows:

  • alibabacloud-ai-search service: 1000
  • amazonbedrock service: 240
  • anthropic service: 50
  • azureaistudio service: 240
  • azureopenai service and task type text_embedding: 1440
  • azureopenai service and task types completion or chat_completion: 120
  • cohere service: 10000
  • contextualai service: 1000
  • elastic service and task type chat_completion: 240
  • fireworksai service: 6000
  • googleaistudio service: 360
  • googlevertexai service: 30000
  • hugging_face service: 3000
  • jinaai service: 2000
  • llama service: 3000
  • mistral service: 240
  • openai service and task type text_embedding: 3000
  • openai service and task type completion: 500
  • openshift_ai service: 3000
  • voyageai service: 2000
  • watsonxai service: 120
similarity enum

The similarity measure. If the embedding_type is float, the default value is dot_product. If the embedding_type is int8 or byte, the default value is cosine.

Values: cosine dot_product l2_norm
task_settings object

Settings to configure the inference task. These settings are specific to the task type you specified.

input_type enum required

For a text_embedding task, the type of input passed to the model. Valid values are:

  • classification: Use it for embeddings passed through a text classifier.
  • clustering: Use it for the embeddings run through a clustering algorithm.
  • ingest: Use it for storing document embeddings in a vector database.
  • search: Use it for storing embeddings of search queries run against a vector database to find relevant documents.
Important

The input_type field is required when using embedding models v3 and higher.

Values: classification clustering ingest search
return_documents boolean

For a rerank task, return doc text within the results.

top_n number

For a rerank task, the number of most relevant documents to return. It defaults to the number of the documents. If this inference endpoint is used in a text_similarity_reranker retriever query and top_n is set, it must be greater than or equal to rank_window_size in the query.

truncate enum

For a text_embedding task, the method to handle inputs longer than the maximum token length. Valid values are:

  • END: When the input exceeds the maximum input token length, the end of the input is discarded.
  • NONE: When the input exceeds the maximum input token length, an error is returned.
  • START: When the input exceeds the maximum input token length, the start of the input is discarded.
Values: END NONE START

Response

application/json
chunking_settings object

The chunking configuration object. Applies only to the embedding, sparse_embedding and text_embedding task types. Not applicable to the rerank, completion, or chat_completion task types.

max_chunk_size number · default: 250

The maximum size of a chunk in words. This value cannot be lower than 20 (for sentence strategy) or 10 (for word strategy). This value should not exceed the window size for the associated model.

overlap number · default: 100

The number of overlapping words for chunks. It is applicable only to a word chunking strategy. This value cannot be higher than half the max_chunk_size value.

sentence_overlap number · default: 1

The number of overlapping sentences for chunks. It is applicable only for a sentence chunking strategy. It can be either 1 or 0.

separator_group string

Only applicable to the recursive strategy and required when using it.

Sets a predefined list of separators in the saved chunking settings based on the selected text type. Values can be markdown or plaintext.

Using this parameter is an alternative to manually specifying a custom separators list.

separators string[]

Only applicable to the recursive strategy and required when using it.

A list of strings used as possible split points when chunking text.

Each string can be a plain string or a regular expression (regex) pattern. The system tries each separator in order to split the text, starting from the first item in the list.

After splitting, it attempts to recombine smaller pieces into larger chunks that stay within the max_chunk_size limit, to reduce the total number of chunks generated.

strategy string · default: sentence

The chunking strategy: sentence, word, none or recursive.

  • If strategy is set to recursive, you must also specify:
  • max_chunk_size
  • either separators orseparator_group

Learn more about different chunking strategies in the linked documentation.

Reference
service string required

The service type

service_settings object required

Settings specific to the service

task_settings object

Task settings specific to the service and task type

inference_id string required

The inference Id

task_type enum required

The task type

Values: text_embedding rerank completion