From b69af2671755804af8ec2046eb2093c8e9f08815 Mon Sep 17 00:00:00 2001
From: Jacob Lee <jacoblee93@gmail.com>
Date: Thu, 4 Apr 2024 15:28:58 -0700
Subject: [PATCH] docs[patch]: Fix Model I/O quickstart (#20031)

@baskaryan
---
 docs/docs/modules/model_io/index.mdx | 71 ++++++++++++++++++----------
 1 file changed, 47 insertions(+), 24 deletions(-)

diff --git a/docs/docs/modules/model_io/index.mdx b/docs/docs/modules/model_io/index.mdx
index 614e0d6e7b..e61ac9c748 100644
--- a/docs/docs/modules/model_io/index.mdx
+++ b/docs/docs/modules/model_io/index.mdx
@@ -70,6 +70,29 @@ from langchain_openai import ChatOpenAI
 llm = ChatOpenAI(openai_api_key="...")
 ```
 
+Both `llm` and `chat_model` are objects that represent configuration for a particular model. 
+You can initialize them with parameters like `temperature` and others, and pass them around.
+The main difference between them is their input and output schemas.
+The LLM objects take string as input and output string.
+The ChatModel objects take a list of messages as input and output a message.
+
+We can see the difference between an LLM and a ChatModel when we invoke it.
+
+```python
+from langchain_core.messages import HumanMessage
+
+text = "What would be a good company name for a company that makes colorful socks?"
+messages = [HumanMessage(content=text)]
+
+llm.invoke(text)
+# >> Feetful of Fun
+
+chat_model.invoke(messages)
+# >> AIMessage(content="Socks O'Color")
+```
+
+The LLM returns a string, while the ChatModel returns a message.
+
   </TabItem>
   <TabItem value="local" label="Local (using Ollama)">
 
@@ -89,6 +112,29 @@ llm = Ollama(model="llama2")
 chat_model = ChatOllama()
 ```
 
+Both `llm` and `chat_model` are objects that represent configuration for a particular model. 
+You can initialize them with parameters like `temperature` and others, and pass them around.
+The main difference between them is their input and output schemas.
+The LLM objects take string as input and output string.
+The ChatModel objects take a list of messages as input and output a message.
+
+We can see the difference between an LLM and a ChatModel when we invoke it.
+
+```python
+from langchain_core.messages import HumanMessage
+
+text = "What would be a good company name for a company that makes colorful socks?"
+messages = [HumanMessage(content=text)]
+
+llm.invoke(text)
+# >> Feetful of Fun
+
+chat_model.invoke(messages)
+# >> AIMessage(content="Socks O'Color")
+```
+
+The LLM returns a string, while the ChatModel returns a message.
+
   </TabItem>
   <TabItem value="anthropic" label="Anthropic (chat model only)">
 
@@ -119,7 +165,7 @@ chat_model = ChatAnthropic(anthropic_api_key="...")
 ```
 
   </TabItem>
-  <TabItem value="cohere" label="Cohere">
+  <TabItem value="cohere" label="Cohere (chat model only)">
 
 First we'll need to install their partner package:
 
@@ -152,29 +198,6 @@ chat_model = ChatCohere(cohere_api_key="...")
   </TabItem>
 </Tabs>
 
-Both `llm` and `chat_model` are objects that represent configuration for a particular model. 
-You can initialize them with parameters like `temperature` and others, and pass them around.
-The main difference between them is their input and output schemas.
-The LLM objects take string as input and output string.
-The ChatModel objects take a list of messages as input and output a message.
-
-We can see the difference between an LLM and a ChatModel when we invoke it.
-
-```python
-from langchain_core.messages import HumanMessage
-
-text = "What would be a good company name for a company that makes colorful socks?"
-messages = [HumanMessage(content=text)]
-
-llm.invoke(text)
-# >> Feetful of Fun
-
-chat_model.invoke(messages)
-# >> AIMessage(content="Socks O'Color")
-```
-
-The LLM returns a string, while the ChatModel returns a message.
-
 ## Prompt Templates
 
 Most LLM applications do not pass user input directly into an LLM. Usually they will add the user input to a larger piece of text, called a prompt template, that provides additional context on the specific task at hand.