Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
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)
|