jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor JsonAgentComponent to use create_json_agent function

Browse source
This commit is contained in:
anovazzi1 2024-01-15 17:52:10 -03:00
commit f322c77c5c
1 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/backend/langflow/components/agents/JsonAgent.py
View file

@ -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])