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