sqfoo commited on
Commit
e93d4ea
·
verified ·
1 Parent(s): 3c965e9

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -1
agent.py CHANGED
@@ -94,7 +94,8 @@ class BasicAgent:
94
  self.tools = [web_search, visit_webpage, read_file, image_caption]
95
  self.prompt = ChatPromptTemplate.from_messages([
96
  ("system", self.sys_prompt),
97
- ("human", "{input}")
 
98
  ])
99
  self.agent = create_tool_calling_agent(self.model, self.tools, self.prompt)
100
  self.agent_exe = AgentExecutor(agent=self.agent, tools=self.tools, verbose=True)
 
94
  self.tools = [web_search, visit_webpage, read_file, image_caption]
95
  self.prompt = ChatPromptTemplate.from_messages([
96
  ("system", self.sys_prompt),
97
+ ("human", "{input}"),
98
+ ("placeholder", "{agent_scratchpad}")
99
  ])
100
  self.agent = create_tool_calling_agent(self.model, self.tools, self.prompt)
101
  self.agent_exe = AgentExecutor(agent=self.agent, tools=self.tools, verbose=True)