From c1477735aaa78ca0c7b01859175868cf6c4ba5df Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Tue, 30 May 2023 17:30:00 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=9A=9A=20chore(base.py):=20rename=20loadi?= =?UTF-8?q?ng=20module=20to=20types=20The=20loading=20module=20was=20renam?= =?UTF-8?q?ed=20to=20types=20to=20better=20reflect=20its=20contents.=20The?= =?UTF-8?q?=20import=20statement=20was=20updated=20to=20reflect=20this=20c?= =?UTF-8?q?hange.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/interface/agents/base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/interface/agents/base.py b/src/backend/langflow/interface/agents/base.py index 1d23b481e..d404f845d 100644 --- a/src/backend/langflow/interface/agents/base.py +++ b/src/backend/langflow/interface/agents/base.py @@ -1,6 +1,6 @@ from typing import Dict, List, Optional -from langchain.agents import loading +from langchain.agents import types from langflow.custom.customs import get_custom_nodes from langflow.interface.agents.custom import CUSTOM_AGENTS @@ -16,7 +16,7 @@ class AgentCreator(LangChainTypeCreator): @property def type_to_loader_dict(self) -> Dict: if self.type_dict is None: - self.type_dict = loading.AGENT_TO_CLASS + self.type_dict = types.AGENT_TO_CLASS # Add JsonAgent to the list of agents for name, agent in CUSTOM_AGENTS.items(): # TODO: validate AgentType