Cohere integration

This will help you get started with Cohere completion models (LLMs) using LangChain. For detailed documentation on Cohere features and configuration options, please refer to the API reference.

Overview

Integration details

Class	Package	Local	Serializable	PY support	Downloads	Version
Cohere	@langchain/cohere	❌	✅	✅

Setup

To access Cohere models you'll need to create a Cohere account, get an API key, and install the @langchain/cohere integration package.

Credentials

Head to cohere.com to sign up to Cohere and generate an API key. Once you've done this set the COHERE_API_KEY environment variable:

export COHERE_API_KEY="your-api-key"

If you want to get automated tracing of your model calls you can also set your LangSmith API key by uncommenting below:

# export LANGSMITH_TRACING="true"
# export LANGSMITH_API_KEY="your-api-key"

Installation

The LangChain Cohere integration lives in the @langchain/cohere package:

npm install @langchain/cohere @langchain/core

yarn add @langchain/cohere @langchain/core

pnpm add @langchain/cohere @langchain/core

Instantiation

Now we can instantiate our model object and generate text completions:

import { Cohere } from "@langchain/cohere"

const llm = new Cohere({
  model: "command",
  temperature: 0,
  maxTokens: undefined,
  maxRetries: 2,
  // other params...
})

Custom client for Cohere on Azure, Cohere on AWS Bedrock, and standalone Cohere instance

We can instantiate a custom CohereClient and pass it to the Cohere constructor.

Note: If a custom client is provided both COHERE_API_KEY environment variable and apiKey parameter in the constructor will be ignored.

import { Cohere } from "@langchain/cohere";
import { CohereClient } from "cohere-ai";

const client = new CohereClient({
  token: "<your-api-key>",
  environment: "<your-cohere-deployment-url>", //optional
  // other params
});

const llmWithCustomClient = new Cohere({
  client,
  // other params...
});

Invocation

const inputText = "Cohere is an AI company that "

const completion = await llm.invoke(inputText)
completion

Cohere is a company that provides natural language processing models that help companies improve human-machine interactions. Cohere was founded in 2019 by Aidan Gomez, Ivan Zhang, and Nick Frosst.

API reference

For detailed documentation of all Cohere features and configurations head to the API reference.