Pass kwargs from initialize_agent into agent classmethod

[jxnl]

Problem

I noticed that in order to change the prefix of the prompt in the zero-shot-react-description agent
we had to dig around to subset strings deep into the agent's attributes. It requires the user to inspect a long chain of attributes and classes.

initialize_agent -> AgentExecutor -> Agent -> LLMChain -> Prompt from Agent.create_prompt

agent = initialize_agent(
    tools=tools,
    llm=fake_llm,
    agent="zero-shot-react-description"
)
prompt_str = agent.agent.llm_chain.prompt.template
new_prompt_str = change_prefix(prompt_str)
agent.agent.llm_chain.prompt.template = new_prompt_str

Implemented Solution

initialize_agent accepts **kwargs but passes it to AgentExecutor but not ZeroShotAgent, by simply giving the kwargs to the agent class methods we can support changing the prefix and suffix for one agent while allowing future agents to take advantage of initialize_agent.

agent = initialize_agent(
    tools=tools,
    llm=fake_llm,
    agent="zero-shot-react-description",
    agent_kwargs={"prefix": prefix, "suffix": suffix}
)

To be fair, this was before finding docs around custom agents here: https://langchain.readthedocs.io/en/latest/modules/agents/examples/custom_agent.html?highlight=custom%20#custom-llmchain but i find that my use case just needed to change the prefix a little.

Changes

• Pass kwargs to Agent class method
• Added a test to check suffix and prefix

[hwchase17]

these are the same kwargs also passed to the AgentExecutor though right? thats kind of an issue...

could we add another parameter that is agent_kwargs = Optional[dict] = None that we pass through here

[jxnl]

done

[hwchase17]

awesome addition - thanks!