From 9f914f8d479d58d7ccc045784fe614dfea9a1219 Mon Sep 17 00:00:00 2001
From: Navan Chauhan <navanchauhan@gmail.com>
Date: Fri, 13 Oct 2023 23:50:10 -0600
Subject: langchain

---
 tools/contacts.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)
 create mode 100644 tools/contacts.py

(limited to 'tools/contacts.py')

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
-- 
cgit v1.2.3