diff options
author | Navan Chauhan <navanchauhan@gmail.com> | 2024-02-11 07:52:32 -0700 |
---|---|---|
committer | Navan Chauhan <navanchauhan@gmail.com> | 2024-02-11 07:52:40 -0700 |
commit | 4632f8bc1e0318dd9d5057828a2fd8c851efb858 (patch) | |
tree | 77907fc0c8803de8a7c9a5b017f53bd57e30b013 /api-magic/speller_agent.py | |
parent | 4a2577004e9c9600be06ecd9cbe3f81685f6bfe9 (diff) |
more
Diffstat (limited to 'api-magic/speller_agent.py')
-rw-r--r-- | api-magic/speller_agent.py | 75 |
1 files changed, 75 insertions, 0 deletions
diff --git a/api-magic/speller_agent.py b/api-magic/speller_agent.py new file mode 100644 index 0000000..32d7b67 --- /dev/null +++ b/api-magic/speller_agent.py @@ -0,0 +1,75 @@ +import logging +from typing import Optional, Tuple +import typing +from vocode.streaming.agent.chat_gpt_agent import ChatGPTAgent +from vocode.streaming.models.agent import AgentConfig, AgentType, ChatGPTAgentConfig +from vocode.streaming.agent.base_agent import BaseAgent, RespondAgent +from vocode.streaming.agent.factory import AgentFactory + +import os +import sys +import typing +from dotenv import load_dotenv + +from langchain.memory import ConversationBufferMemory +from langchain.utilities import SerpAPIWrapper + +from langchain.agents import load_tools + +from stdout_filterer import RedactPhoneNumbers + +load_dotenv() + +from langchain.chat_models import ChatOpenAI +# from langchain.chat_models import BedrockChat +from langchain.agents import initialize_agent +from langchain.agents import AgentType as LangAgentType + + +llm = ChatOpenAI(temperature=0, model_name="gpt-3.5-turbo") # type: ignore +#llm = BedrockChat(model_id="anthropic.claude-instant-v1", model_kwargs={"temperature":0}) # type: ignore +memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True) + # Logging of LLMChains +verbose = True +agent = initialize_agent( + tools=[], + llm=llm, + agent=LangAgentType.CHAT_CONVERSATIONAL_REACT_DESCRIPTION, + verbose=verbose, + memory=memory, +) + + +class SpellerAgentConfig(AgentConfig, type="agent_speller"): + pass + + +class SpellerAgent(RespondAgent[SpellerAgentConfig]): + def __init__(self, agent_config: SpellerAgentConfig): + super().__init__(agent_config=agent_config) + + async def respond( + self, + human_input, + conversation_id: str, + is_interrupt: bool = False, + ) -> Tuple[Optional[str], bool]: + print("SpellerAgent: ", human_input) + res = agent.run(human_input) + return res, False + + +class SpellerAgentFactory(AgentFactory): + def create_agent( + self, agent_config: AgentConfig, logger: Optional[logging.Logger] = None + ) -> BaseAgent: + print("Setting up agent", agent_config, agent_config.type) + if agent_config.type == AgentType.CHAT_GPT: + return ChatGPTAgent( + agent_config=typing.cast(ChatGPTAgentConfig, agent_config) + ) + elif agent_config.type == "agent_speller": + return SpellerAgent( + agent_config=typing.cast(SpellerAgentConfig, agent_config) + ) + raise Exception("Invalid agent config")
\ No newline at end of file |