diff --git a/docs/docs/expression_language/cookbook/agent.ipynb b/docs/docs/expression_language/cookbook/agent.ipynb index 452c4762f7..325ce992c1 100644 --- a/docs/docs/expression_language/cookbook/agent.ipynb +++ b/docs/docs/expression_language/cookbook/agent.ipynb @@ -12,18 +12,20 @@ }, { "cell_type": "code", - "execution_count": 1, + "execution_count": 8, "id": "af4381de", "metadata": {}, "outputs": [], "source": [ - "from langchain.agents import AgentExecutor, XMLAgent, tool\n", + "from langchain import hub\n", + "from langchain.agents import AgentExecutor, tool\n", + "from langchain.agents.output_parsers import XMLAgentOutputParser\n", "from langchain.chat_models import ChatAnthropic" ] }, { "cell_type": "code", - "execution_count": 2, + "execution_count": 3, "id": "24cc8134", "metadata": {}, "outputs": [], @@ -33,7 +35,7 @@ }, { "cell_type": "code", - "execution_count": 3, + "execution_count": 4, "id": "67c0b0e4", "metadata": {}, "outputs": [], @@ -46,7 +48,7 @@ }, { "cell_type": "code", - "execution_count": 4, + "execution_count": 5, "id": "7203b101", "metadata": {}, "outputs": [], @@ -56,18 +58,18 @@ }, { "cell_type": "code", - "execution_count": 5, + "execution_count": 6, "id": "b68e756d", "metadata": {}, "outputs": [], "source": [ - "# Get prompt to use\n", - "prompt = XMLAgent.get_default_prompt()" + "# Get the prompt to use - you can modify this!\n", + "prompt = hub.pull(\"hwchase17/xml-agent-convo\")" ] }, { "cell_type": "code", - "execution_count": 6, + "execution_count": 7, "id": "61ab3e9a", "metadata": {}, "outputs": [], @@ -107,27 +109,27 @@ }, { "cell_type": "code", - "execution_count": 7, + "execution_count": 12, "id": "e92f1d6f", "metadata": {}, "outputs": [], "source": [ "agent = (\n", " {\n", - " \"question\": lambda x: x[\"question\"],\n", - " \"intermediate_steps\": lambda x: convert_intermediate_steps(\n", + " \"input\": lambda x: x[\"input\"],\n", + " \"agent_scratchpad\": lambda x: convert_intermediate_steps(\n", " x[\"intermediate_steps\"]\n", " ),\n", " }\n", " | prompt.partial(tools=convert_tools(tool_list))\n", " | model.bind(stop=[\"\", \"\"])\n", - " | XMLAgent.get_default_output_parser()\n", + " | XMLAgentOutputParser()\n", ")" ] }, { "cell_type": "code", - "execution_count": 8, + "execution_count": 13, "id": "6ce6ec7a", "metadata": {}, "outputs": [], @@ -137,7 +139,7 @@ }, { "cell_type": "code", - "execution_count": 9, + "execution_count": 14, "id": "fb5cb2e3", "metadata": {}, "outputs": [ @@ -148,10 +150,8 @@ "\n", "\n", "\u001b[1m> Entering new AgentExecutor chain...\u001b[0m\n", - "\u001b[32;1m\u001b[1;3m search\n", - "weather in new york\u001b[0m\u001b[36;1m\u001b[1;3m32 degrees\u001b[0m\u001b[32;1m\u001b[1;3m\n", - "\n", - "The weather in New York is 32 degrees\u001b[0m\n", + "\u001b[32;1m\u001b[1;3m searchweather in New York\u001b[0m\u001b[36;1m\u001b[1;3m32 degrees\u001b[0m\u001b[32;1m\u001b[1;3m search\n", + "weather in New York\u001b[0m\u001b[36;1m\u001b[1;3m32 degrees\u001b[0m\u001b[32;1m\u001b[1;3m The weather in New York is 32 degrees\u001b[0m\n", "\n", "\u001b[1m> Finished chain.\u001b[0m\n" ] @@ -159,17 +159,17 @@ { "data": { "text/plain": [ - "{'question': 'whats the weather in New york?',\n", + "{'input': 'whats the weather in New york?',\n", " 'output': 'The weather in New York is 32 degrees'}" ] }, - "execution_count": 9, + "execution_count": 14, "metadata": {}, "output_type": "execute_result" } ], "source": [ - "agent_executor.invoke({\"question\": \"whats the weather in New york?\"})" + "agent_executor.invoke({\"input\": \"whats the weather in New york?\"})" ] }, {