Fix agents' notebooks outputs (#3517)

Fix agents' notebooks to make the answer reflect what is being asked by
the user.
This commit is contained in:
Tiago De Gaspari 2023-04-25 20:06:47 -03:00 committed by GitHub
parent 5680fb6894
commit 4d53cefbe9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 19 additions and 19 deletions

View File

@ -49,7 +49,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 4, "execution_count": 3,
"id": "a33e2f7e", "id": "a33e2f7e",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@ -97,7 +97,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 5, "execution_count": 4,
"id": "655d72f6", "id": "655d72f6",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@ -107,7 +107,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 6, "execution_count": 5,
"id": "490604e9", "id": "490604e9",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@ -117,7 +117,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 7, "execution_count": 6,
"id": "653b1617", "id": "653b1617",
"metadata": {}, "metadata": {},
"outputs": [ "outputs": [
@ -128,7 +128,7 @@
"\n", "\n",
"\n", "\n",
"\u001b[1m> Entering new AgentExecutor chain...\u001b[0m\n", "\u001b[1m> Entering new AgentExecutor chain...\u001b[0m\n",
"\u001b[32;1m\u001b[1;3m\u001b[0m\u001b[36;1m\u001b[1;3mFoo Fighters is an American rock band formed in Seattle in 1994. Foo Fighters was initially formed as a one-man project by former Nirvana drummer Dave Grohl. Following the success of the 1995 eponymous debut album, Grohl recruited a band consisting of Nate Mendel, William Goldsmith, and Pat Smear.\u001b[0m\u001b[32;1m\u001b[1;3m\u001b[0m\n", "\u001b[32;1m\u001b[1;3m\u001b[0m\u001b[36;1m\u001b[1;3mThe current population of Canada is 38,669,152 as of Monday, April 24, 2023, based on Worldometer elaboration of the latest United Nations data.\u001b[0m\u001b[32;1m\u001b[1;3m\u001b[0m\n",
"\n", "\n",
"\u001b[1m> Finished chain.\u001b[0m\n" "\u001b[1m> Finished chain.\u001b[0m\n"
] ]
@ -136,10 +136,10 @@
{ {
"data": { "data": {
"text/plain": [ "text/plain": [
"'Foo Fighters is an American rock band formed in Seattle in 1994. Foo Fighters was initially formed as a one-man project by former Nirvana drummer Dave Grohl. Following the success of the 1995 eponymous debut album, Grohl recruited a band consisting of Nate Mendel, William Goldsmith, and Pat Smear.'" "'The current population of Canada is 38,669,152 as of Monday, April 24, 2023, based on Worldometer elaboration of the latest United Nations data.'"
] ]
}, },
"execution_count": 7, "execution_count": 6,
"metadata": {}, "metadata": {},
"output_type": "execute_result" "output_type": "execute_result"
} }

View File

@ -31,7 +31,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 21, "execution_count": 2,
"id": "d7c4ebdc", "id": "d7c4ebdc",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@ -43,7 +43,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 22, "execution_count": 3,
"id": "becda2a1", "id": "becda2a1",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@ -66,7 +66,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 23, "execution_count": 4,
"id": "a33e2f7e", "id": "a33e2f7e",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@ -96,8 +96,8 @@
" \"\"\"\n", " \"\"\"\n",
" if len(intermediate_steps) == 0:\n", " if len(intermediate_steps) == 0:\n",
" return [\n", " return [\n",
" AgentAction(tool=\"Search\", tool_input=\"foo\", log=\"\"),\n", " AgentAction(tool=\"Search\", tool_input=kwargs[\"input\"], log=\"\"),\n",
" AgentAction(tool=\"RandomWord\", tool_input=\"foo\", log=\"\"),\n", " AgentAction(tool=\"RandomWord\", tool_input=kwargs[\"input\"], log=\"\"),\n",
" ]\n", " ]\n",
" else:\n", " else:\n",
" return AgentFinish(return_values={\"output\": \"bar\"}, log=\"\")\n", " return AgentFinish(return_values={\"output\": \"bar\"}, log=\"\")\n",
@ -117,8 +117,8 @@
" \"\"\"\n", " \"\"\"\n",
" if len(intermediate_steps) == 0:\n", " if len(intermediate_steps) == 0:\n",
" return [\n", " return [\n",
" AgentAction(tool=\"Search\", tool_input=\"foo\", log=\"\"),\n", " AgentAction(tool=\"Search\", tool_input=kwargs[\"input\"], log=\"\"),\n",
" AgentAction(tool=\"RandomWord\", tool_input=\"foo\", log=\"\"),\n", " AgentAction(tool=\"RandomWord\", tool_input=kwargs[\"input\"], log=\"\"),\n",
" ]\n", " ]\n",
" else:\n", " else:\n",
" return AgentFinish(return_values={\"output\": \"bar\"}, log=\"\")" " return AgentFinish(return_values={\"output\": \"bar\"}, log=\"\")"
@ -126,7 +126,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 24, "execution_count": 5,
"id": "655d72f6", "id": "655d72f6",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@ -136,7 +136,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 25, "execution_count": 6,
"id": "490604e9", "id": "490604e9",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@ -146,7 +146,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 26, "execution_count": 7,
"id": "653b1617", "id": "653b1617",
"metadata": {}, "metadata": {},
"outputs": [ "outputs": [
@ -157,7 +157,7 @@
"\n", "\n",
"\n", "\n",
"\u001b[1m> Entering new AgentExecutor chain...\u001b[0m\n", "\u001b[1m> Entering new AgentExecutor chain...\u001b[0m\n",
"\u001b[32;1m\u001b[1;3m\u001b[0m\u001b[36;1m\u001b[1;3mFoo Fighters is an American rock band formed in Seattle in 1994. Foo Fighters was initially formed as a one-man project by former Nirvana drummer Dave Grohl. Following the success of the 1995 eponymous debut album, Grohl recruited a band consisting of Nate Mendel, William Goldsmith, and Pat Smear.\u001b[0m\u001b[32;1m\u001b[1;3m\u001b[0m\n", "\u001b[32;1m\u001b[1;3m\u001b[0m\u001b[36;1m\u001b[1;3mThe current population of Canada is 38,669,152 as of Monday, April 24, 2023, based on Worldometer elaboration of the latest United Nations data.\u001b[0m\u001b[32;1m\u001b[1;3m\u001b[0m\n",
"Now I'm doing this!\n", "Now I'm doing this!\n",
"\u001b[33;1m\u001b[1;3mfoo\u001b[0m\u001b[32;1m\u001b[1;3m\u001b[0m\n", "\u001b[33;1m\u001b[1;3mfoo\u001b[0m\u001b[32;1m\u001b[1;3m\u001b[0m\n",
"\n", "\n",
@ -170,7 +170,7 @@
"'bar'" "'bar'"
] ]
}, },
"execution_count": 26, "execution_count": 7,
"metadata": {}, "metadata": {},
"output_type": "execute_result" "output_type": "execute_result"
} }