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

---
 lang_prompt_demo.py | 37 +++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)
 create mode 100644 lang_prompt_demo.py

(limited to 'lang_prompt_demo.py')

diff --git a/lang_prompt_demo.py b/lang_prompt_demo.py
new file mode 100644
index 0000000..d63aea3
--- /dev/null
+++ b/lang_prompt_demo.py
@@ -0,0 +1,37 @@
+import os
+import sys
+import typing
+from dotenv import load_dotenv
+
+from tools.contacts import get_all_contacts
+from tools.vocode import call_phone_number
+from langchain.memory import ConversationBufferMemory
+
+from stdout_filterer import RedactPhoneNumbers
+
+load_dotenv()
+
+from langchain.chat_models import ChatOpenAI
+from langchain.agents import initialize_agent
+from langchain.agents import AgentType
+
+if __name__ == "__main__":
+    # Redirect stdout to our custom class
+    sys.stdout = typing.cast(typing.TextIO, RedactPhoneNumbers(sys.stdout))
+
+    OBJECTIVE = (
+        input("Objective: ")
+        or "Find a random person in my contacts and tell them a joke"
+    )
+    llm = ChatOpenAI(temperature=0, model_name="gpt-4")  # type: ignore
+    memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
+    # Logging of LLMChains
+    verbose = True
+    agent = initialize_agent(
+        tools=[get_all_contacts, call_phone_number],
+        llm=llm,
+        agent=AgentType.CHAT_CONVERSATIONAL_REACT_DESCRIPTION,
+        verbose=verbose,
+        memory=memory,
+    )
+    agent.run(OBJECTIVE)
-- 
cgit v1.2.3