diff options
Diffstat (limited to 'tools/contacts.py')
-rw-r--r-- | tools/contacts.py | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/tools/contacts.py b/tools/contacts.py new file mode 100644 index 0000000..4f4a3be --- /dev/null +++ b/tools/contacts.py @@ -0,0 +1,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
\ No newline at end of file |