From 9ee2713272afa837dd7595cb37f864073d65f637 Mon Sep 17 00:00:00 2001 From: Brian Thorne Date: Tue, 14 Mar 2023 19:07:35 +1300 Subject: [PATCH] Bugfix - allow custom input variables in chat zero shot agent's prompt (#1624) I was trying out the `chat-zero-shot-react-description` agent for [qabot](https://github.com/hardbyte/qabot/blob/dbbd31bb2702d433ed573bb177f1d1272d36f2ae/qabot/agents/data_query_chain.py#L35-L52) but langchain 0.0.108 doesn't correctly use custom 'input_variables` in the prompt template. --- langchain/agents/chat/base.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/langchain/agents/chat/base.py b/langchain/agents/chat/base.py index 0935362a..971d8332 100644 --- a/langchain/agents/chat/base.py +++ b/langchain/agents/chat/base.py @@ -73,9 +73,9 @@ class ChatAgent(Agent): SystemMessagePromptTemplate.from_template(template), HumanMessagePromptTemplate.from_template("{input}\n\n{agent_scratchpad}"), ] - return ChatPromptTemplate( - input_variables=["input", "agent_scratchpad"], messages=messages - ) + if input_variables is None: + input_variables = ["input", "agent_scratchpad"] + return ChatPromptTemplate(input_variables=input_variables, messages=messages) @classmethod def from_llm_and_tools(