github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-05-30 11:39:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bugfix - allow custom input variables in chat zero shot agent's prompt (#1624)

Browse Source 
I was trying out the `chat-zero-shot-react-description` agent for
[qabot](dbbd31bb27/qabot/agents/data_query_chain.py (L35-L52))
but langchain 0.0.108 doesn't correctly use custom 'input_variables` in
the prompt template.
This commit is contained in:
Brian Thorne 2023-03-14 19:07:35 +13:00 committed by GitHub
parent b3234bf3b0
commit 9ee2713272
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
langchain/agents/chat/base.py
View File

@ -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(