Retrieval Augmented Generation (RAG)

1
# Retrieve the documents
2
documents = [
3
    {
4
        "data": {
5
            "title": "Tall penguins",
6
            "snippet": "Emperor penguins are the tallest.",
7
        }
8
    },
9
    {
10
        "data": {
11
            "title": "Penguin habitats",
12
            "snippet": "Emperor penguins only live in Antarctica.",
13
        }
14
    },
15
    {
16
        "data": {
17
            "title": "What are animals?",
18
            "snippet": "Animals are different from plants.",
19
        }
20
    },
21
]
22
23
# Add the user message
24
message = "Where do the tallest penguins live?"
25
26
messages = [{"role": "user", "content": message}]
27
28
response = co.chat(
29
    model="command-a-03-2025",
30
    messages=messages,
31
    documents=documents,
32
)
33
34
print(response.message.content[0].text)
35
36
print(response.message.citations)

1
message = "Who is more popular: Nsync or Backstreet Boys?"
2
3
# Define the query generation tool
4
query_gen_tool = [
5
    {
6
        "type": "function",
7
        "function": {
8
            "name": "internet_search",
9
            "description": "Returns a list of relevant document snippets for a textual query retrieved from the internet",
10
            "parameters": {
11
                "type": "object",
12
                "properties": {
13
                    "queries": {
14
                        "type": "array",
15
                        "items": {"type": "string"},
16
                        "description": "a list of queries to search the internet with.",
17
                    }
18
                },
19
                "required": ["queries"],
20
            },
21
        },
22
    }
23
]
24
25
# Define a system message to optimize search query generation
26
instructions = "Write a search query that will find helpful information for answering the user's question accurately. If you need more than one search query, write a list of search queries. If you decide that a search is very unlikely to find information that would be useful in constructing a response to the user, you should instead directly answer."
27
28
# Generate search queries (if any)
29
import json
30
31
search_queries = []
32
33
res = co.chat(
34
    model="command-a-03-2025",
35
    messages=[
36
        {"role": "system", "content": instructions},
37
        {"role": "user", "content": message},
38
    ],
39
    tools=query_gen_tool,
40
)
41
42
if res.message.tool_calls:
43
    for tc in res.message.tool_calls:
44
        queries = json.loads(tc.function.arguments)["queries"]
45
        search_queries.extend(queries)
46
47
print(search_queries)

1
documents = [
2
    {
3
        "data": {
4
            "title": "CSPC: Backstreet Boys Popularity Analysis - ChartMasters",
5
            "snippet": "↓ Skip to Main Content\n\nMusic industry – One step closer to being accurate\n\nCSPC: Backstreet Boys Popularity Analysis\n\nHernán Lopez Posted on February 9, 2017 Posted in CSPC 72 Comments Tagged with Backstreet Boys, Boy band\n\nAt one point, Backstreet Boys defined success: massive albums sales across the globe, great singles sales, plenty of chart topping releases, hugely hyped tours and tremendous media coverage.\n\nIt is true that they benefited from extraordinarily good market conditions in all markets. After all, the all-time record year for the music business, as far as revenues in billion dollars are concerned, was actually 1999. That is, back when this five men group was at its peak.",
6
        }
7
    },
8
    {
9
        "data": {
10
            "title": "CSPC: NSYNC Popularity Analysis - ChartMasters",
11
            "snippet": "↓ Skip to Main Content\n\nMusic industry – One step closer to being accurate\n\nCSPC: NSYNC Popularity Analysis\n\nMJD Posted on February 9, 2018 Posted in CSPC 27 Comments Tagged with Boy band, N'Sync\n\nAt the turn of the millennium three teen acts were huge in the US, the Backstreet Boys, Britney Spears and NSYNC. The latter is the only one we haven’t study so far. It took 15 years and Adele to break their record of 2,4 million units sold of No Strings Attached in its first week alone.\n\nIt wasn’t a fluke, as the second fastest selling album of the Soundscan era prior 2015, was also theirs since Celebrity debuted with 1,88 million units sold.",
12
        }
13
    },
14
    {
15
        "data": {
16
            "title": "CSPC: Backstreet Boys Popularity Analysis - ChartMasters",
17
            "snippet": " 1997, 1998, 2000 and 2001 also rank amongst some of the very best years.\n\nYet the way many music consumers – especially teenagers and young women’s – embraced their output deserves its own chapter. If Jonas Brothers and more recently One Direction reached a great level of popularity during the past decade, the type of success achieved by Backstreet Boys is in a completely different level as they really dominated the business for a few years all over the world, including in some countries that were traditionally hard to penetrate for Western artists.\n\nWe will try to analyze the extent of that hegemony with this new article with final results which will more than surprise many readers.",
18
        }
19
    },
20
    {
21
        "data": {
22
            "title": "CSPC: NSYNC Popularity Analysis - ChartMasters",
23
            "snippet": " Was the teen group led by Justin Timberlake really that big? Was it only in the US where they found success? Or were they a global phenomenon?\n\nAs usual, I’ll be using the Commensurate Sales to Popularity Concept in order to relevantly gauge their results. This concept will not only bring you sales information for all NSYNC‘s albums, physical and download singles, as well as audio and video streaming, but it will also determine their true popularity. If you are not yet familiar with the CSPC method, the next page explains it with a short video. I fully recommend watching the video before getting into the sales figures.",
24
        }
25
    },
26
]
27
28
# Add the user message
29
message = "Who is more popular: Nsync or Backstreet Boys?"
30
messages = [{"role": "user", "content": message}]
31
32
response = co.chat(
33
    model="command-a-03-2025",
34
    messages=messages,
35
    documents=documents,
36
)
37
38
print(response.message.content[0].text)

1
print(response.message.citations)