BabyAGI implementation with tools with multiple inputs (requiring Structured Tool Chat Agent)

[luca-git]

My goal is to extend the tools used by baby AGI, more specifically to use at least the basic WriteFileTool() and ReadFileTool(). They use two inputs though, so I cannot stick with the vanilla ZeroShotAgent. So i resorted to using: AgentType.STRUCTURED_CHAT_ZERO_SHOT_REACT_DESCRIPTION and replace it in the original code in the link below. Agent initialization is the only modification I've made. Can anyone kindly help me understand what I'm missing in the code in order to be able to leverage multiple input tools or provide guidance/resources?

https://python.langchain.com/en/latest/use_cases/agents/baby_agi_with_agent.html?highlight=babyagi%20with%20tools

from langchain.agents import AgentType 

from langchain.agents import initialize_agent 

    @classmethod 

    def from_llm( 

        cls, llm: BaseLLM, vectorstore: VectorStore, verbose: bool = False, **kwargs 

    ) -> "BabyAGI": 

        """Initialize the BabyAGI Controller.""" 

        task_creation_chain = TaskCreationChain.from_llm(llm, verbose=verbose) 

        task_prioritization_chain = TaskPrioritizationChain.from_llm( 

            llm, verbose=verbose 

        ) 

        llm_chain = LLMChain(llm=llm, prompt=prompt) 

        tool_names = [tool.name for tool in tools] 

        agent = initialize_agent(agent=AgentType.STRUCTURED_CHAT_ZERO_SHOT_REACT_DESCRIPTION,llm=llm_chain, tools=tool_names) 

        agent_executor = AgentExecutor.from_agent_and_tools( 

            agent=agent, tools=tools, verbose=True 

        ) 

        return cls( 

            task_creation_chain=task_creation_chain, 

            task_prioritization_chain=task_prioritization_chain, 

            execution_chain=agent_executor, 

            vectorstore=vectorstore, 

            **kwargs, 

        ) 

I get the error

 File ~\anaconda3\envs\aagi\lib\site-packages\langchain\agents\structured_chat\base.py:83 in create_prompt  args_schema = re.sub("}", "}}}}", re.sub("{", "{{{{", str(tool.args))) 
AttributeError: 'str' object has no attribute 'args' 

[toadFrogski]

I got a similar error while trying to create my own agent.