From f322c77c5c3ff30a9cedc283e7fd2ebf71f0d15e Mon Sep 17 00:00:00 2001 From: anovazzi1 Date: Mon, 15 Jan 2024 17:52:10 -0300 Subject: [PATCH] Refactor JsonAgentComponent to use create_json_agent function --- src/backend/langflow/components/agents/JsonAgent.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/src/backend/langflow/components/agents/JsonAgent.py b/src/backend/langflow/components/agents/JsonAgent.py index 71b276e26..9f11c5088 100644 --- a/src/backend/langflow/components/agents/JsonAgent.py +++ b/src/backend/langflow/components/agents/JsonAgent.py @@ -1,12 +1,10 @@ from langflow import CustomComponent -from langchain.agents import AgentExecutor -from typing import Callable +from langchain.agents import AgentExecutor,create_json_agent from langflow.field_typing import ( BaseLanguageModel, ) from langchain_community.agent_toolkits.base import BaseToolkit - class JsonAgentComponent(CustomComponent): display_name = "JsonAgent" description = "Construct a json agent from an LLM and tools." @@ -21,5 +19,5 @@ class JsonAgentComponent(CustomComponent): self, llm: BaseLanguageModel, toolkit: BaseToolkit, - ) -> Callable: - return AgentExecutor(llm=llm, toolkit=toolkit) + ) -> AgentExecutor: + return create_json_agent(llm=llm, toolkit=toolkit[0]) \ No newline at end of file