Building an Agent with Cohere

1
# pip install cohere numpy
2
3
import numpy as np
4
import cohere
5
6
co = cohere.Client("COHERE_API_KEY") # Get your API key: https://dashboard.cohere.com/api-keys

1
# Create the tools
2
def search_faqs(query):
3
    faqs = [
4
        {"text": "Reimbursing Travel Expenses: Easily manage your travel expenses by submitting them through our finance tool. Approvals are prompt and straightforward."},
5
        {"text": "Working from Abroad: Working remotely from another country is possible. Simply coordinate with your manager and ensure your availability during core hours."}
6
    ]
7
    return  {"faqs" : faqs}
8
9
def search_emails(query):
10
    emails = [
11
        {"from": "it@co1t.com", "to": "david@co1t.com", "date": "2024-06-24", "subject": "Setting Up Your IT Needs", "text": "Greetings! To ensure a seamless start, please refer to the attached comprehensive guide, which will assist you in setting up all your work accounts."},
12
        {"from": "john@co1t.com", "to": "david@co1t.com", "date": "2024-06-24", "subject": "First Week Check-In", "text": "Hello! I hope you're settling in well. Let's connect briefly tomorrow to discuss how your first week has been going. Also, make sure to join us for a welcoming lunch this Thursday at noon—it's a great opportunity to get to know your colleagues!"}
13
    ]
14
    return  {"emails" : emails}
15
    
16
def create_calendar_event(date: str, time: str, duration: int):
17
    # You can implement any logic here
18
    return {"is_success": True,
19
            "message": f"Created a {duration} hour long event at {time} on {date}"}
20
    
21
functions_map = {
22
    "search_faqs": search_faqs,
23
    "search_emails": search_emails,
24
    "create_calendar_event": create_calendar_event
25
}

1
# Define the tools
2
tools = [
3
    {
4
      "name": "search_faqs",
5
      "description": "Given a user query, searches a company's frequently asked questions (FAQs) list and returns the most relevant matches to the query.",
6
      "parameter_definitions": {
7
        "query": {
8
          "description": "The query from the user",
9
          "type": "str",
10
          "required": True
11
        }
12
      }
13
    }, 
14
    {
15
      "name": "search_emails",
16
      "description": "Given a user query, searches a person's emails and returns the most relevant matches to the query.",
17
      "parameter_definitions": {
18
        "query": {
19
          "description": "The query from the user",
20
          "type": "str",
21
          "required": True
22
        }
23
      }
24
    }, 
25
    {
26
      "name": "create_calendar_event",
27
      "description": "Creates a new calendar event of the specified duration at the specified time and date. A new event cannot be created on the same time as an existing event.",
28
      "parameter_definitions": {
29
        "date": {
30
          "description": "the date on which the event starts, formatted as mm/dd/yy",
31
          "type": "str",
32
          "required": True
33
        },
34
        "time": {
35
          "description": "the time of the event, formatted using 24h military time formatting",
36
          "type": "str",
37
          "required": True
38
        },
39
        "duration": {
40
          "description": "the number of hours the event lasts for",
41
          "type": "float",
42
          "required": True
43
        }
44
      }
45
    }
46
]

1
# Step 1: Get user message
2
message = "Any messages about getting setup with IT?"
3
4
preamble="""## Task & Context
5
You are an assistant who assist new employees of Co1t with their first week. You respond to their questions and assist them with their needs. Today is Monday, June 24, 2024"""
6
7
# Step 2: Tool planning and calling
8
response = co.chat(
9
    message=message,
10
    preamble=preamble,
11
    tools=tools)
12
13
if response.tool_calls:
14
    print("Tool plan:")
15
    print(response.text,"\n")
16
        
17
    print("Tool calls:")
18
    for call in response.tool_calls:
19
        print(f"Tool name: {call.name} | Parameters: {call.parameters}")

1
# Step 3: Tool execution
2
tool_results = []
3
for tc in response.tool_calls:
4
    tool_call = {"name": tc.name, "parameters": tc.parameters}
5
    tool_output = functions_map[tc.name](**tc.parameters)
6
    tool_results.append({"call": tool_call, "outputs": [tool_output]})
7
8
print("Tool results:") 
9
for result in tool_results:
10
    print(result)

1
# Step 4: Response and citation generation
2
response = co.chat(
3
    message="", # In response generation, we set the message as empty
4
    preamble=preamble,
5
    tools=tools,
6
    tool_results=tool_results,
7
    chat_history=response.chat_history
8
)
9
10
# Print final response
11
print("Final response:")
12
print(response.text)
13
print("="*50)
14
15
# Print citations (if any)
16
if response.citations:
17
    print("\nCITATIONS:")
18
    for citation in response.citations:
19
        print(citation)
20
21
    print("\nCITED REFERENCES:")
22
    for document in response.documents:
23
        print(document)

1
model = "command-r-plus-08-2024"
2
3
preamble="""## Task & Context
4
You are an assistant who assists new employees of Co1t with their first week. You respond to their questions and assist them with their needs. Today is Monday, June 24, 2024"""
5
6
# A function that runs multi-step tool use
7
def run_assistant(message, chat_history=[]):
8
    # Step 1: get user message
9
    print(f"Question:\n{message}")
10
    print("="*50)
11
12
    # Step 2: Generate tool calls (if any)    
13
    response = co.chat(
14
        message=message,
15
        model=model,
16
        preamble=preamble,
17
        tools=tools,
18
        chat_history=chat_history
19
    )
20
21
    # Tool execution loop
22
    while response.tool_calls:
23
        tool_calls = response.tool_calls
24
        
25
        if response.text:
26
            print("Intermediate response:")
27
            print(response.text,"\n")
28
        print("Tool calls:")
29
        for call in tool_calls:
30
            print(f"Tool name: {call.name} | Parameters: {call.parameters}")
31
        print("="*50)
32
        
33
        # Step 3: Get tool results
34
        tool_results = []
35
        for tc in tool_calls:
36
            tool_call = {"name": tc.name, "parameters": tc.parameters}
37
            tool_output = functions_map[tc.name](**tc.parameters)
38
            tool_results.append({"call": tool_call, "outputs": [tool_output]})
39
        
40
        # Step 4: Generate response and citations                
41
        response = co.chat(
42
            message="",
43
            model=model,
44
            preamble=preamble,
45
            tools=tools,
46
            tool_results=tool_results,
47
            chat_history=response.chat_history
48
        )
49
50
        chat_history = response.chat_history
51
        
52
    # Print final response
53
    print("Final response:")
54
    print(response.text)
55
    print("="*50)
56
    
57
    # Print citations (if any)
58
    if response.citations:
59
        print("\nCITATIONS:")
60
        for citation in response.citations:
61
            print(citation)
62
63
        print("\nCITED REFERENCES:")
64
        for document in response.documents:
65
            print(document)
66
                
67
    return chat_history

1
chat_history = run_assistant("Can you check if there are any lunch invites, and for those days, block an hour on my calendar from 12-1PM.")