1
# pip install cohere
2
3
import cohere
4
import json
5
6
co = cohere.ClientV2(api_key="COHERE_API_KEY") # Get your free 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
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
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
        "type": "function",
5
        "function": {
6
            "name": "search_faqs",
7
            "description": "Given a user query, searches a company's frequently asked questions (FAQs) list and returns the most relevant matches to the query.",
8
            "parameters": {
9
                "type": "object",
10
                "properties": {
11
                    "query": {
12
                        "type": "string",
13
                        "description": "The query from the user"
14
                    }
15
                },
16
                "required": ["query"]
17
            }
18
        }
19
    },
20
    {
21
        "type": "function",
22
        "function": {
23
            "name": "search_emails",
24
            "description": "Given a user query, searches a person's emails and returns the most relevant matches to the query.",
25
            "parameters": {
26
                "type": "object",
27
                "properties": {
28
                    "query": {
29
                        "type": "string",
30
                        "description": "The query from the user"
31
                    }
32
                },
33
                "required": ["query"]
34
            }
35
        }
36
    },
37
    {
38
        "type": "function",
39
        "function": {
40
            "name": "create_calendar_event",
41
            "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.",
42
            "parameters": {
43
                "type": "object",
44
                "properties": {
45
                    "date": {
46
                        "type": "string",
47
                        "description": "the date on which the event starts, formatted as mm/dd/yy"
48
                    },
49
                    "time": {
50
                        "type": "string",
51
                        "description": "the time of the event, formatted using 24h military time formatting"
52
                    },
53
                    "duration": {
54
                        "type": "number",
55
                        "description": "the number of hours the event lasts for"
56
                    }
57
                },
58
                "required": ["date", "time", "duration"]
59
            }
60
        }
61
    }
62
]

1
# Create custom system message
2
system_message="""## Task and Context
3
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"""
4
5
6
# Step 1: Get user message
7
message = "Is there any message about getting setup with IT?"
8
9
# Add the system and user messages to the chat history
10
messages = [{"role": "system", "content": system_message},
11
            {"role": "user", "content": message}]
12
13
# Step 2: Tool planning and calling
14
response = co.chat(
15
        model="command-r-plus-08-2024",
16
        messages=messages,
17
        tools=tools
18
    )
19
20
if response.message.tool_calls:    
21
    print("Tool plan:")
22
    print(response.message.tool_plan,"\n")
23
    print("Tool calls:")
24
    for tc in response.message.tool_calls:
25
        print(f"Tool name: {tc.function.name} | Parameters: {tc.function.arguments}")
26
        
27
    # Append tool calling details to the chat history
28
    messages.append({"role": "assistant", "tool_calls": response.message.tool_calls, "tool_plan": response.message.tool_plan})

1
# Step 3: Tool execution
2
for tc in response.message.tool_calls:
3
    tool_result = functions_map[tc.function.name](**json.loads(tc.function.arguments))
4
    tool_content = []
5
    for data in tool_result:
6
        tool_content.append({"type": "document", "document": {"data": json.dumps(data)}})
7
        # Optional: add an "id" field in the "document" object, otherwise IDs are auto-generated
8
    # Append tool results to the chat history
9
    messages.append({"role": "tool", "tool_call_id": tc.id, "content": tool_content})
10
11
    print("Tool results:")
12
    for result in tool_content:
13
        print(result)

1
# Step 4: Response and citation generation
2
response = co.chat(
3
    model="command-r-plus-08-2024",
4
    messages=messages,
5
    tools=tools
6
)
7
8
# Append assistant response to the chat history
9
messages.append({"role": "assistant", "content": response.message.content[0].text})
10
11
# Print final response
12
print("Response:")
13
print(response.message.content[0].text)
14
print("="*50)
15
16
# Print citations (if any)
17
if response.message.citations:
18
    print("\nCITATIONS:")
19
    for citation in response.message.citations:
20
        print(citation, "\n")

1
model = "command-r-plus-08-2024"
2
3
system_message="""## Task and 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
def run_assistant(query, messages=None):
7
    if messages is None:
8
        messages = []
9
        
10
    if "system" not in {m.get("role") for m in messages}:
11
        messages.append({"role": "system", "content": system_message})
12
    
13
    # Step 1: get user message
14
    print(f"Question:\n{query}")
15
    print("="*50)
16
    
17
    messages.append({"role": "user", "content": query})
18
19
    # Step 2: Generate tool calls (if any)
20
    response = co.chat(
21
        model=model,
22
        messages=messages,
23
        tools=tools
24
    )
25
26
    while response.message.tool_calls:
27
        
28
        print("Tool plan:")
29
        print(response.message.tool_plan,"\n")
30
        print("Tool calls:")
31
        for tc in response.message.tool_calls:
32
            print(f"Tool name: {tc.function.name} | Parameters: {tc.function.arguments}")
33
        print("="*50)
34
35
        messages.append({"role": "assistant", "tool_calls": response.message.tool_calls, "tool_plan": response.message.tool_plan})        
36
        
37
        # Step 3: Get tool results
38
        for idx, tc in enumerate(response.message.tool_calls):
39
            tool_result = functions_map[tc.function.name](
40
                **json.loads(tc.function.arguments)
41
            )
42
            tool_content = []
43
            for data in tool_result:
44
                tool_content.append({"type": "document", "document": {"data": json.dumps(data)}})
45
                # Optional: add an "id" field in the "document" object, otherwise IDs are auto-generated
46
            messages.append(
47
                {"role": "tool", "tool_call_id": tc.id, "content": tool_content}
48
            )
49
        
50
        # Step 4: Generate response and citations 
51
        response = co.chat(
52
            model=model,
53
            messages=messages,
54
            tools=tools
55
        )
56
    
57
    messages.append({"role": "assistant", "content": response.message.content[0].text})
58
        
59
    # Print final response
60
    print("Response:")
61
    print(response.message.content[0].text)
62
    print("="*50)
63
    
64
    # Print citations (if any)
65
    if response.message.citations:
66
        print("\nCITATIONS:")
67
        for citation in response.message.citations:
68
            print(citation, "\n")
69
    
70
    return messages

1
messages = run_assistant("Can you check if there are any lunch invites, and for those days, create a one-hour event on my calendar at 12PM.")