Fix parameter in create_json_agent function

This commit is contained in:
anovazzi1 2024-01-15 17:54:40 -03:00
commit a9086a4e41

View file

@ -20,4 +20,4 @@ class JsonAgentComponent(CustomComponent):
llm: BaseLanguageModel,
toolkit: BaseToolkit,
) -> AgentExecutor:
return create_json_agent(llm=llm, toolkit=toolkit[0])
return create_json_agent(llm=llm, toolkit=toolkit)