agent notebook fixes (v1) (run-llama#10610)

docs/examples/agent/react_agent.ipynb

@@ -1,7 +1,6 @@
 {
  "cells": [
   {
-   "attachments": {},
    "cell_type": "markdown",
    "id": "5cfa0417",
    "metadata": {},
@@ -22,7 +21,6 @@
    ]
   },
   {
-   "attachments": {},
    "cell_type": "markdown",
    "id": "81d6fba5",
    "metadata": {},
@@ -55,7 +53,22 @@
    "execution_count": null,
    "id": "e8ac1778-0585-43c9-9dad-014d13d7460d",
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "[nltk_data] Downloading package stopwords to /Users/jerryliu/Programmi\n",
+      "[nltk_data]     ng/gpt_index/.venv/lib/python3.10/site-\n",
+      "[nltk_data]     packages/llama_index/legacy/_static/nltk_cache...\n",
+      "[nltk_data]   Unzipping corpora/stopwords.zip.\n",
+      "[nltk_data] Downloading package punkt to /Users/jerryliu/Programming/g\n",
+      "[nltk_data]     pt_index/.venv/lib/python3.10/site-\n",
+      "[nltk_data]     packages/llama_index/legacy/_static/nltk_cache...\n",
+      "[nltk_data]   Unzipping tokenizers/punkt.zip.\n"
+     ]
+    }
+   ],
    "source": [
     "from llama_index.core.agent import ReActAgent\n",
     "from llama_index.llms.openai import OpenAI\n",
@@ -135,12 +148,12 @@
      "output_type": "stream",
      "text": [
       "\u001b[1;3;38;5;200mThought: I need to use a tool to help me answer the question.\n",
-      "assistant: Action: multiply\n",
-      "assistant: Action Input: {\"a\": 2, \"b\": 4}\n",
+      "Action: multiply\n",
+      "Action Input: {\"a\": 2, \"b\": 4}\n",
       "Observation: 8\n",
-      "assistant: Thought: I need to use a tool to help me answer the question.\n",
-      "assistant: Action: add\n",
-      "assistant: Action Input: {\"a\": 20, \"b\": 8}\n",
+      "Thought: I need to use a tool to help me answer the question.\n",
+      "Action: add\n",
+      "Action Input: {\"a\": 20, \"b\": 8}\n",
       "Observation: 28\n",
       "Thought: I can answer without using any more tools.\n",
       "Answer: 28\n",
@@ -162,7 +175,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      " 28"
+      "28"
      ]
     }
    ],
@@ -200,11 +213,11 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "\u001b[1;3;38;5;200mThought: I need to use the tools to help me answer the question. According to the order of operations in mathematics (BIDMAS/BODMAS), multiplication should be done before addition. So, I will first multiply 2 and 4, then add the result to 2.\n",
+      "\u001b[1;3;38;5;200mThought: I need to use the tools to help me answer the question. According to the order of operations in mathematics (BIDMAS/BODMAS), multiplication should be done before addition. So, I will first multiply 2 and 4, and then add the result to 2.\n",
       "Action: multiply\n",
       "Action Input: {'a': 2, 'b': 4}\n",
       "\u001b[0m\u001b[1;3;34mObservation: 8\n",
-      "\u001b[0m\u001b[1;3;38;5;200mThought: Now that I have the result of the multiplication, I need to add this to 2.\n",
+      "\u001b[0m\u001b[1;3;38;5;200mThought: Now that I have the result of the multiplication, I need to add this result to 2.\n",
       "Action: add\n",
       "Action Input: {'a': 2, 'b': 8}\n",
       "\u001b[0m\u001b[1;3;34mObservation: 10\n",
@@ -389,6 +402,27 @@
     "react_system_prompt = PromptTemplate(react_system_header_str)"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "c32037ea-6c30-4059-bd32-4b9ba64912ab",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "{'agent_worker:system_prompt': PromptTemplate(metadata={'prompt_type': <PromptType.CUSTOM: 'custom'>}, template_vars=['tool_desc', 'tool_names'], kwargs={}, output_parser=None, template_var_mappings=None, function_mappings=None, template='\\nYou are designed to help with a variety of tasks, from answering questions     to providing summaries to other types of analyses.\\n\\n## Tools\\nYou have access to a wide variety of tools. You are responsible for using\\nthe tools in any sequence you deem appropriate to complete the task at hand.\\nThis may require breaking the task into subtasks and using different tools\\nto complete each subtask.\\n\\nYou have access to the following tools:\\n{tool_desc}\\n\\n## Output Format\\nTo answer the question, please use the following format.\\n\\n```\\nThought: I need to use a tool to help me answer the question.\\nAction: tool name (one of {tool_names}) if using a tool.\\nAction Input: the input to the tool, in a JSON format representing the kwargs (e.g. {{\"input\": \"hello world\", \"num_beams\": 5}})\\n```\\n\\nPlease ALWAYS start with a Thought.\\n\\nPlease use a valid JSON format for the Action Input. Do NOT do this {{\\'input\\': \\'hello world\\', \\'num_beams\\': 5}}.\\n\\nIf this format is used, the user will respond in the following format:\\n\\n```\\nObservation: tool response\\n```\\n\\nYou should keep repeating the above format until you have enough information\\nto answer the question without using any more tools. At that point, you MUST respond\\nin the one of the following two formats:\\n\\n```\\nThought: I can answer without using any more tools.\\nAnswer: [your answer here]\\n```\\n\\n```\\nThought: I cannot answer the question with the provided tools.\\nAnswer: Sorry, I cannot answer your query.\\n```\\n\\n## Current Conversation\\nBelow is the current conversation consisting of interleaving human and assistant messages.\\n\\n')}"
+      ]
+     },
+     "execution_count": null,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "agent.get_prompts()"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -413,17 +447,19 @@
       "Action: add\n",
       "Action Input: {'a': 5, 'b': 3}\n",
       "\u001b[0m\u001b[1;3;34mObservation: 8\n",
-      "\u001b[0m\u001b[1;3;38;5;200mThought: Now I need to add the result from the previous operation with 2.\n",
+      "\u001b[0m\u001b[1;3;38;5;200mThought: Now I need to add the result from the previous operation to 2.\n",
       "Action: add\n",
       "Action Input: {'a': 8, 'b': 2}\n",
       "\u001b[0m\u001b[1;3;34mObservation: 10\n",
       "\u001b[0m\u001b[1;3;38;5;200mThought: I can answer without using any more tools.\n",
       "Answer: The result of 5+3+2 is 10.\n",
-      "- First, I added 5 and 3 using the add tool, which resulted in 8.\n",
-      "- Then, I added the result (8) to 2 using the add tool, which resulted in 10.\n",
+      "\n",
+      "- I first added 5 and 3 using the add tool, which resulted in 8.\n",
+      "- Then I added the result (8) to 2 using the add tool again, which resulted in 10.\n",
       "\u001b[0mThe result of 5+3+2 is 10.\n",
-      "- First, I added 5 and 3 using the add tool, which resulted in 8.\n",
-      "- Then, I added the result (8) to 2 using the add tool, which resulted in 10.\n"
+      "\n",
+      "- I first added 5 and 3 using the add tool, which resulted in 8.\n",
+      "- Then I added the result (8) to 2 using the add tool again, which resulted in 10.\n"
      ]
     }
    ],

llama-index-core/llama_index/core/agent/react/base.py

@@ -30,6 +30,7 @@
 from llama_index.core.objects.base import ObjectRetriever
 from llama_index.core.settings import Settings
 from llama_index.core.tools import BaseTool
+from llama_index.core.prompts.mixin import PromptMixinType
 
 
 class ReActAgent(AgentRunner):
@@ -126,3 +127,7 @@ def from_tools(
             verbose=verbose,
             context=context,
         )
+
+    def _get_prompt_modules(self) -> PromptMixinType:
+        """Get prompt modules."""
+        return {"agent_worker": self.agent_worker}