diff options
Diffstat (limited to 'speller_agent.py')
| -rw-r--r-- | speller_agent.py | 10 |
1 files changed, 4 insertions, 6 deletions
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, |