aboutsummaryrefslogtreecommitdiff
path: root/tools/contacts.py
blob: afdc8a4cc377fda494b4e653f5a3f13e9c6ab11b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
from typing import List
from langchain.agents import tool

from dotenv import load_dotenv
load_dotenv()

import os

CONTACTS = [
    {
        "name": "Greg",
        "phone" : os.getenv("TEST_PHONE_NUMBER"),
        "email": "grsi2038@colorado.edu"
    }
]

@tool("get_all_contacts")
def get_all_contacts(placeholder: str) -> List[dict]:
    """Returns all contacts in the user's phone book which includes email and phone numbers."""
    return CONTACTS