Summarizing Text

1
import cohere
2
co = cohere.ClientV2(api_key="<YOUR API KEY>")
3
4
document = """Equipment rental in North America is predicted to “normalize” going into 2024,
5
according to Josh Nickell, vice president of equipment rental for the American Rental
6
Association (ARA).
7
“Rental is going back to ‘normal,’ but normal means that strategy matters again -
8
geography matters, fleet mix matters, customer type matters,” Nickell said. “In
9
late 2020 to 2022, you just showed up with equipment and you made money.
10
“Everybody was breaking records, from the national rental chains to the smallest
11
rental companies; everybody was having record years, and everybody was raising
12
prices. The conversation was, ‘How much are you up?’ And now, the conversation
13
is changing to ‘What’s my market like?’”
14
Nickell stressed this shouldn’t be taken as a pessimistic viewpoint. It’s simply
15
coming back down to Earth from unprecedented circumstances during the time of Covid.
16
Rental companies are still seeing growth, but at a more moderate level."""
17
18
message = f"Generate a concise summary of this text\n{document}"
19
20
response = co.chat(
21
    model="command-r-plus-08-2024",
22
    messages=[{"role": "user", "content": message}]
23
)
24
25
26
print(response.message.content[0].text)

1
message = f"Summarize this text in one sentence\n{document}"
2
3
response = co.chat(
4
    model="command-r-plus-08-2024",
5
    messages=[{"role": "user", "content": message}]
6
)
7
8
print(response.message.content[0].text)

1
message = f"Summarize this text in less than 10 words\n{document}"
2
3
response = co.chat(
4
    model="command-r-plus-08-2024",
5
    messages=[{"role": "user", "content": message}]
6
)
7
8
print(response.message.content[0].text)

1
message = f"Generate a concise summary of this text as bullet points\n{document}"
2
3
response = co.chat(
4
    model="command-r-plus-08-2024",
5
    messages=[{"role": "user", "content": message}]
6
)
7
8
print(response.message.content[0].text)

1
document_chunked = [
2
    {
3
        "data": {
4
            "text": "Equipment rental in North America is predicted to “normalize” going into 2024, according to Josh Nickell, vice president of equipment rental for the American Rental Association (ARA)."
5
        }
6
    },
7
    {
8
        "data": {
9
            "text": "“Rental is going back to ‘normal,’ but normal means that strategy matters again - geography matters, fleet mix matters, customer type matters,” Nickell said. “In late 2020 to 2022, you just showed up with equipment and you made money."
10
        }
11
    },
12
    {
13
        "data": {
14
            "text": "“Everybody was breaking records, from the national rental chains to the smallest rental companies; everybody was having record years, and everybody was raising prices. The conversation was, ‘How much are you up?’ And now, the conversation is changing to ‘What’s my market like?’”"
15
        }
16
    },
17
]

1
system_message = """## Task and Context
2
You will receive a series of text fragments from a document that are presented in chronological order. As the assistant, you must generate responses to user's requests based on the information given in the fragments. Ensure that your responses are accurate and truthful, and that you reference your sources where appropriate to answer the queries, regardless of their complexity."""

1
message = f"Summarize this text in one sentence."
2
3
response = co.chat(
4
    model="command-r-plus-08-2024",
5
    documents=document_chunked,
6
    messages=[
7
        {"role": "system", "content": system_message},
8
        {"role": "user", "content": message},
9
    ],
10
)
11
12
print(response.message.content[0].text)
13
14
if response.message.citations:
15
    print("\nCITATIONS:")
16
    for citation in response.message.citations:
17
        print(
18
            f"Start: {citation.start} | End: {citation.end} | Text: '{citation.text}'",
19
            end="",
20
        )
21
        if citation.sources:
22
            for source in citation.sources:
23
                print(f"| {source.id}")

1
# Before
2
3
co.summarize(
4
    format="bullets",
5
    length="short",
6
    extractiveness="low",
7
    text="""Equipment rental in North America is predicted to “normalize” going into 2024, according
8
  to Josh Nickell, vice president of equipment rental for the American Rental Association (ARA).
9
  “Rental is going back to ‘normal,’ but normal means that strategy matters again - geography
10
  matters, fleet mix matters, customer type matters,” Nickell said. “In late 2020 to 2022, you
11
  just showed up with equipment and you made money.
12
  “Everybody was breaking records, from the national rental chains to the smallest rental companies;
13
  everybody was having record years, and everybody was raising prices. The conversation was, ‘How
14
  much are you up?’ And now, the conversation is changing to ‘What’s my market like?’”
15
  Nickell stressed this shouldn’t be taken as a pessimistic viewpoint. It’s simply coming back
16
  down to Earth from unprecedented circumstances during the time of Covid. Rental companies are
17
  still seeing growth, but at a more moderate level.
18
  """,
19
)
20
21
# After
22
co.summarize(
23
    format="bullets",
24
    length="short",
25
    extractiveness="low",
26
    text="""Equipment rental in North America is predicted to “normalize” going into 2024, according
27
  to Josh Nickell, vice president of equipment rental for the American Rental Association (ARA).
28
  “Rental is going back to ‘normal,’ but normal means that strategy matters again - geography
29
  matters, fleet mix matters, customer type matters,” Nickell said. “In late 2020 to 2022, you
30
  just showed up with equipment and you made money.
31
  “Everybody was breaking records, from the national rental chains to the smallest rental companies;
32
  everybody was having record years, and everybody was raising prices. The conversation was, ‘How
33
  much are you up?’ And now, the conversation is changing to ‘What’s my market like?’”
34
  Nickell stressed this shouldn’t be taken as a pessimistic viewpoint. It’s simply coming back
35
  down to Earth from unprecedented circumstances during the time of Covid. Rental companies are
36
  still seeing growth, but at a more moderate level.
37
  """,
38
)