feedback

Author: adnaans

vocode/streaming/agent/default_factory.py

@@ -3,14 +3,12 @@
 from vocode.streaming.agent.base_agent import BaseAgent
 from vocode.streaming.agent.chat_gpt_agent import ChatGPTAgent
 from vocode.streaming.agent.echo_agent import EchoAgent
-from vocode.streaming.agent.langchain_agent import LangchainAgent
 from vocode.streaming.agent.restful_user_implemented_agent import RESTfulUserImplementedAgent
 from vocode.streaming.models.agent import (
     AgentConfig,
     AnthropicAgentConfig,
     ChatGPTAgentConfig,
     EchoAgentConfig,
-    LangchainAgentConfig,
     RESTfulUserImplementedAgentConfig,
 )
 
@@ -25,6 +23,4 @@ def create_agent(self, agent_config: AgentConfig) -> BaseAgent:
             return RESTfulUserImplementedAgent(agent_config=agent_config)
         elif isinstance(agent_config, AnthropicAgentConfig):
             return AnthropicAgent(agent_config=agent_config)
-        elif isinstance(agent_config, LangchainAgentConfig):
-            return LangchainAgent(agent_config=agent_config)
         raise Exception("Invalid agent config", agent_config.type)

vocode/streaming/models/agent.py

@@ -48,6 +48,7 @@ class AgentType(str, Enum):
     RESTFUL_USER_IMPLEMENTED = "agent_restful_user_implemented"
     WEBSOCKET_USER_IMPLEMENTED = "agent_websocket_user_implemented"
     ACTION = "agent_action"
+    LANGCHAIN = "agent_langchain"
 
 
 class FillerAudioConfig(BaseModel):
@@ -131,7 +132,7 @@ class AnthropicAgentConfig(AgentConfig, type=AgentType.ANTHROPIC.value):  # type
     max_tokens: int = LLM_AGENT_DEFAULT_MAX_TOKENS
     temperature: float = LLM_AGENT_DEFAULT_TEMPERATURE
 
-class LangchainAgentConfig(AgentConfig, type=AgentType.CHAT_GPT.value):  # type: ignore
+class LangchainAgentConfig(AgentConfig, type=AgentType.LANGCHAIN.value):  # type: ignore
     prompt_preamble: str
     model_name: str
     provider: str