From 77ba780d0e781d38608d9af9d33882071ea1a80d Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 11 Jul 2023 17:14:50 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(custom.py):=20add=20handle?= =?UTF-8?q?=5Fparsing=5Ferrors=3DTrue=20to=20VectorStoreAgent,=20SQLAgent,?= =?UTF-8?q?=20and=20VectorStoreRouterAgent=20classes=20to=20handle=20parsi?= =?UTF-8?q?ng=20errors=20during=20execution=20=F0=9F=90=9B=20fix(loading.p?= =?UTF-8?q?y):=20add=20handle=5Fparsing=5Ferrors=3DTrue=20to=20instantiate?= =?UTF-8?q?=5Fagent=20and=20load=5Fagent=5Fexecutor=20functions=20to=20han?= =?UTF-8?q?dle=20parsing=20errors=20during=20agent=20instantiation=20and?= =?UTF-8?q?=20loading?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/interface/agents/custom.py | 6 ++++-- src/backend/langflow/interface/initialize/loading.py | 4 ++-- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/interface/agents/custom.py b/src/backend/langflow/interface/agents/custom.py index aa0cfb5db..b4e2b9bac 100644 --- a/src/backend/langflow/interface/agents/custom.py +++ b/src/backend/langflow/interface/agents/custom.py @@ -157,7 +157,7 @@ class VectorStoreAgent(CustomAgentExecutor): llm_chain=llm_chain, allowed_tools=tool_names, **kwargs # type: ignore ) return AgentExecutor.from_agent_and_tools( - agent=agent, tools=tools, verbose=True + agent=agent, tools=tools, verbose=True, handle_parsing_errors=True ) def run(self, *args, **kwargs): @@ -232,6 +232,7 @@ class SQLAgent(CustomAgentExecutor): verbose=True, max_iterations=15, early_stopping_method="force", + handle_parsing_errors=True, ) def run(self, *args, **kwargs): @@ -276,7 +277,7 @@ class VectorStoreRouterAgent(CustomAgentExecutor): llm_chain=llm_chain, allowed_tools=tool_names, **kwargs # type: ignore ) return AgentExecutor.from_agent_and_tools( - agent=agent, tools=tools, verbose=True + agent=agent, tools=tools, verbose=True, handle_parsing_errors=True ) def run(self, *args, **kwargs): @@ -308,6 +309,7 @@ class InitializeAgent(CustomAgentExecutor): agent=agent, # type: ignore memory=memory, return_intermediate_steps=True, + handle_parsing_errors=True, ) def __init__(self, *args, **kwargs): diff --git a/src/backend/langflow/interface/initialize/loading.py b/src/backend/langflow/interface/initialize/loading.py index ba617fe34..b232d089c 100644 --- a/src/backend/langflow/interface/initialize/loading.py +++ b/src/backend/langflow/interface/initialize/loading.py @@ -193,8 +193,7 @@ def instantiate_agent(node_type, class_object: Type[agent_module.Agent], params: agent = class_method(**params) tools = params.get("tools", []) return AgentExecutor.from_agent_and_tools( - agent=agent, - tools=tools, + agent=agent, tools=tools, handle_parsing_errors=True ) return load_agent_executor(class_object, params) @@ -432,6 +431,7 @@ def load_agent_executor(agent_class: type[agent_module.Agent], params, **kwargs) return AgentExecutor.from_agent_and_tools( agent=agent, tools=allowed_tools, + handle_parsing_errors=True, # memory=memory, **kwargs, )