Connect to Cohere for chat completions, text embeddings, document reranking, text classification, tokenization, batch embed jobs, dataset management, and model fine-tuning.
| Action | Description | Parameters | Returns |
|---|---|---|---|
cancel_embed_job ⚠ | Cancel a running batch embed job | embed_job_id | bool |
chat | Generate a chat response | model, messages [temperature, max_tokens, tools, documents, response_format] | ChatResponse |
check_api_key | Check whether the API key is valid | — | ApiKeyCheck |
classify | Classify text into labelled categories | model, inputs, examples | ClassifyResponse |
create_dataset | Create a dataset from an uploaded file | name, type, data [filename, keep_original_file, skip_malformed_input, eval_data] | Dataset |
create_embed_job | Create a batch embed job | model, dataset_id, input_type [name, embedding_types, truncate] | EmbedJob |
create_finetuned_model | Create a fine-tuned model | name, settings | FinetunedModel |
delete_dataset ⚠ | Delete a dataset | dataset_id | bool |
delete_finetuned_model ⚠ | Delete a fine-tuned model | finetuned_model_id | bool |
detokenize | Convert token IDs back into text | model, tokens | DetokenizeResponse |
embed | Create text embeddings | model, texts, input_type [embedding_types, truncate] | EmbedResponse |
get_dataset | Get a dataset by ID | dataset_id | Dataset |
get_dataset_usage | Get organization dataset storage usage | — | DatasetUsage |
get_embed_job | Get a batch embed job by ID | embed_job_id | EmbedJob |
get_finetuned_model | Get a fine-tuned model by ID | finetuned_model_id | FinetunedModel |
get_model | Get a Cohere model by name | model | CohereModel |
list_datasets | List datasets | [dataset_type, limit, offset] | list |
list_embed_jobs | List batch embed jobs | — | list |
list_finetuned_models | List fine-tuned models | — | list |
list_models | List available Cohere models | [endpoint, default_only] | list |
rerank | Rerank documents by relevance to a query | model, query, documents [top_n, max_tokens_per_doc] | RerankResponse |
tokenize | Tokenize text into token IDs | model, text | TokenizeResponse |
Full schema explorer, code generation, and live API testing