tools/contacts.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

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")
    }
]

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