blob42/langchain
1
0
Fork 0
forked from Archives/langchain
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ConversationalChatAgent: Allow customizing TEMPLATE_TOOL_RESPONSE (#2361)

Browse Source 
It's currently not possible to change the `TEMPLATE_TOOL_RESPONSE`
prompt for ConversationalChatAgent, this PR changes that.

---------

Co-authored-by: Dev 2049 <dev.dev2049@gmail.com>
This commit is contained in:
Emil Ahlbäck 2023-05-18 02:23:08 +02:00 committed by GitHub
parent 1ff7c958b0
commit 5c9205d5f4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
langchain/agents/conversational_chat/base.py
View File

@ -37,6 +37,7 @@ class ConversationalChatAgent(Agent):
"""An agent designed to hold a conversation in addition to using tools."""
output_parser: AgentOutputParser = Field(default_factory=ConvoOutputParser)
template_tool_response: str = TEMPLATE_TOOL_RESPONSE
@classmethod
def _get_default_output_parser(cls, **kwargs: Any) -> AgentOutputParser:
@ -99,7 +100,7 @@ class ConversationalChatAgent(Agent):
for action, observation in intermediate_steps:
thoughts.append(AIMessage(content=action.log))
human_message = HumanMessage(
content=TEMPLATE_TOOL_RESPONSE.format(observation=observation)
content=self.template_tool_response.format(observation=observation)
)
thoughts.append(human_message)
return thoughts