From 42e54a2cb29943b793bf9a47dd7e0121e1c0c87d Mon Sep 17 00:00:00 2001 From: Navan Chauhan Date: Sat, 14 Oct 2023 14:47:49 -0600 Subject: final --- speller_agent.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) (limited to 'speller_agent.py') diff --git a/speller_agent.py b/speller_agent.py index ce7c197..9cc7ca4 100644 --- a/speller_agent.py +++ b/speller_agent.py @@ -11,10 +11,8 @@ import sys import typing from dotenv import load_dotenv -from tools.contacts import get_all_contacts -from tools.vocode import call_phone_number -from tools.email_tool import email_tasks -from tools.summarize import summarize +from lang_prompt_demo import tools + from langchain.memory import ConversationBufferMemory from langchain.utilities import SerpAPIWrapper @@ -25,7 +23,7 @@ from stdout_filterer import RedactPhoneNumbers load_dotenv() from langchain.chat_models import ChatOpenAI -from langchain.chat_models import BedrockChat +# from langchain.chat_models import BedrockChat from langchain.agents import initialize_agent from langchain.agents import AgentType as LangAgentType @@ -36,7 +34,7 @@ memory = ConversationBufferMemory(memory_key="chat_history", return_messages=Tru # Logging of LLMChains verbose = True agent = initialize_agent( - tools=[get_all_contacts, call_phone_number, email_tasks, summarize] + load_tools(["serpapi", "human"]), + tools=tools, llm=llm, agent=LangAgentType.CHAT_CONVERSATIONAL_REACT_DESCRIPTION, verbose=verbose, -- cgit v1.2.3