diff --git a/docs/docs/modules/agents/agent_types/tool_calling.ipynb b/docs/docs/modules/agents/agent_types/tool_calling.ipynb
index 46ba2eda5c..e9fcae5d6c 100644
--- a/docs/docs/modules/agents/agent_types/tool_calling.ipynb
+++ b/docs/docs/modules/agents/agent_types/tool_calling.ipynb
@@ -36,7 +36,10 @@
"```{=mdx}\n",
"import ChatModelTabs from \"@theme/ChatModelTabs\";\n",
"\n",
- "\n",
+ "\n",
"```"
]
},
diff --git a/docs/src/theme/ChatModelTabs.js b/docs/src/theme/ChatModelTabs.js
index 073171c99d..60e3d18dc4 100644
--- a/docs/src/theme/ChatModelTabs.js
+++ b/docs/src/theme/ChatModelTabs.js
@@ -23,12 +23,14 @@ os.environ["${apiKeyName}"] = getpass.getpass()`;
* @typedef {Object} ChatModelTabsProps - Component props.
* @property {string} [openaiParams] - Parameters for OpenAI chat model. Defaults to `model="gpt-3.5-turbo-0125"`
* @property {string} [anthropicParams] - Parameters for Anthropic chat model. Defaults to `model="claude-3-sonnet-20240229"`
+ * @property {string} [cohereParams] - Parameters for Cohere chat model. Defaults to `model="command-r"`
* @property {string} [fireworksParams] - Parameters for Fireworks chat model. Defaults to `model="accounts/fireworks/models/mixtral-8x7b-instruct"`
* @property {string} [mistralParams] - Parameters for Mistral chat model. Defaults to `model="mistral-large-latest"`
* @property {string} [googleParams] - Parameters for Google chat model. Defaults to `model="gemini-pro"`
* @property {string} [togetherParams] - Parameters for Together chat model. Defaults to `model="mistralai/Mixtral-8x7B-Instruct-v0.1"`
* @property {boolean} [hideOpenai] - Whether or not to hide OpenAI chat model.
* @property {boolean} [hideAnthropic] - Whether or not to hide Anthropic chat model.
+ * @property {boolean} [hideCohere] - Whether or not to hide Cohere chat model.
* @property {boolean} [hideFireworks] - Whether or not to hide Fireworks chat model.
* @property {boolean} [hideMistral] - Whether or not to hide Mistral chat model.
* @property {boolean} [hideGoogle] - Whether or not to hide Google VertexAI chat model.
@@ -43,12 +45,14 @@ export default function ChatModelTabs(props) {
const {
openaiParams,
anthropicParams,
+ cohereParams,
fireworksParams,
mistralParams,
googleParams,
togetherParams,
hideOpenai,
hideAnthropic,
+ hideCohere,
hideFireworks,
hideMistral,
hideGoogle,
@@ -59,6 +63,7 @@ export default function ChatModelTabs(props) {
const openAIParamsOrDefault = openaiParams ?? `model="gpt-3.5-turbo-0125"`;
const anthropicParamsOrDefault =
anthropicParams ?? `model="claude-3-sonnet-20240229"`;
+ const cohereParamsOrDefault = cohereParams ?? `model="command-r"`;
const fireworksParamsOrDefault =
fireworksParams ??
`model="accounts/fireworks/models/mixtral-8x7b-instruct"`;
@@ -90,6 +95,24 @@ export default function ChatModelTabs(props) {
default: false,
shouldHide: hideAnthropic,
},
+ {
+ value: "Google",
+ label: "Google",
+ text: `from langchain_google_vertexai import ChatVertexAI\n\n${llmVarName} = ChatVertexAI(${googleParamsOrDefault})`,
+ apiKeyName: "GOOGLE_API_KEY",
+ packageName: "langchain-google-vertexai",
+ default: false,
+ shouldHide: hideGoogle,
+ },
+ {
+ value: "Cohere",
+ label: "Cohere",
+ text: `from langchain_cohere import ChatCohere\n\n${llmVarName} = ChatCohere(${cohereParamsOrDefault})`,
+ apiKeyName: "COHERE_API_KEY",
+ packageName: "langchain-cohere",
+ default: false,
+ shouldHide: hideCohere,
+ },
{
value: "FireworksAI",
label: "FireworksAI",
@@ -108,15 +131,6 @@ export default function ChatModelTabs(props) {
default: false,
shouldHide: hideMistral,
},
- {
- value: "Google",
- label: "Google",
- text: `from langchain_google_vertexai import ChatVertexAI\n\n${llmVarName} = ChatVertexAI(${googleParamsOrDefault})`,
- apiKeyName: "GOOGLE_API_KEY",
- packageName: "langchain-google-vertexai",
- default: false,
- shouldHide: hideGoogle,
- },
{
value: "TogetherAI",
label: "TogetherAI",