Skip to content

Commit fa0c939

Browse files
authored
Update custom_agent.ipynb (#2767)
Fixed an issue the agent is not taking the user's question as input.
1 parent 59d0543 commit fa0c939

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docs/modules/agents/agents/custom_agent.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@
7777
" Returns:\n",
7878
" Action specifying what tool to use.\n",
7979
" \"\"\"\n",
80-
" return AgentAction(tool=\"Search\", tool_input=\"foo\", log=\"\")\n",
80+
" return AgentAction(tool=\"Search\", tool_input=kwargs[\"input\"], log=\"\")\n",
8181
"\n",
8282
" async def aplan(\n",
8383
" self, intermediate_steps: List[Tuple[AgentAction, str]], **kwargs: Any\n",
@@ -92,7 +92,7 @@
9292
" Returns:\n",
9393
" Action specifying what tool to use.\n",
9494
" \"\"\"\n",
95-
" return AgentAction(tool=\"Search\", tool_input=\"foo\", log=\"\")"
95+
" return AgentAction(tool=\"Search\", tool_input=kwargs[\"input\"], log=\"\")"
9696
]
9797
},
9898
{

0 commit comments

Comments
 (0)