From 188fd4ad0aec53e883aed81f687c7391106d90ac Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 13 Jun 2023 15:04:23 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(custom.py):=20remove=20unnec?= =?UTF-8?q?essary=20type=20ignore=20comments=20The=20type=20ignore=20comme?= =?UTF-8?q?nts=20are=20unnecessary=20as=20the=20code=20is=20already=20type?= =?UTF-8?q?=20annotated.=20Removing=20them=20improves=20code=20readability?= =?UTF-8?q?=20and=20maintainability.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../langflow/interface/agents/custom.py | 20 +++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/src/backend/langflow/interface/agents/custom.py b/src/backend/langflow/interface/agents/custom.py index 8ff61c62b..24e8fe1f5 100644 --- a/src/backend/langflow/interface/agents/custom.py +++ b/src/backend/langflow/interface/agents/custom.py @@ -83,8 +83,8 @@ class JsonAgent(CustomAgentExecutor): prompt=prompt, ) agent = ZeroShotAgent( - llm_chain=llm_chain, allowed_tools=tool_names - ) # type: ignore + llm_chain=llm_chain, allowed_tools=tool_names # type: ignore + ) return cls.from_agent_and_tools(agent=agent, tools=tools, verbose=True) def run(self, *args, **kwargs): @@ -132,8 +132,8 @@ class CSVAgent(CustomAgentExecutor): ) tool_names = {tool.name for tool in tools} agent = ZeroShotAgent( - llm_chain=llm_chain, allowed_tools=tool_names, **kwargs - ) # type: ignore + llm_chain=llm_chain, allowed_tools=tool_names, **kwargs # type: ignore + ) return cls.from_agent_and_tools(agent=agent, tools=tools, verbose=True) @@ -171,8 +171,8 @@ class VectorStoreAgent(CustomAgentExecutor): ) tool_names = {tool.name for tool in tools} agent = ZeroShotAgent( - llm_chain=llm_chain, allowed_tools=tool_names, **kwargs - ) # type: ignore + llm_chain=llm_chain, allowed_tools=tool_names, **kwargs # type: ignore + ) return AgentExecutor.from_agent_and_tools( agent=agent, tools=tools, verbose=True ) @@ -241,8 +241,8 @@ class SQLAgent(CustomAgentExecutor): ) tool_names = {tool.name for tool in tools} # type: ignore agent = ZeroShotAgent( - llm_chain=llm_chain, allowed_tools=tool_names, **kwargs - ) # type: ignore + llm_chain=llm_chain, allowed_tools=tool_names, **kwargs # type: ignore + ) return AgentExecutor.from_agent_and_tools( agent=agent, tools=tools, # type: ignore @@ -286,8 +286,8 @@ class VectorStoreRouterAgent(CustomAgentExecutor): ) tool_names = {tool.name for tool in tools} agent = ZeroShotAgent( - llm_chain=llm_chain, allowed_tools=tool_names, **kwargs - ) # type: ignore + llm_chain=llm_chain, allowed_tools=tool_names, **kwargs # type: ignore + ) return AgentExecutor.from_agent_and_tools( agent=agent, tools=tools, verbose=True )