update

agent.py

@@ -127,6 +127,7 @@ def get_graph(llm):
 
         Args:
             query: The search query."""
+        print("\n-------------------- Tool (Wikipedia) has been called --------------------\n")
         search_docs = WikipediaLoader(query=query, load_max_docs=2).load()
         formatted_search_docs = "\n\n---\n\n".join(
             [
@@ -156,7 +157,7 @@ def get_graph(llm):
 
         Args:
             query: The search query."""
-        search_docs = TavilySearchResults(max_results=3).invoke(query=query)
+        search_docs = TavilySearchResults(max_results=3).invoke({'query': query})
         formatted_search_docs = "\n\n---\n\n".join(
             [
                 f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'

app_playground.ipynb

@@ -20,7 +20,9 @@
     "\n",
     "llm = get_llm()\n",
     "\n",
-    "graph = get_graph(llm)\n"
+    "graph = get_graph(llm)\n",
+    "\n",
+    "# graph.invoke({\"messages\": [HumanMessage(content=question),]})"
    ],
    "outputs": [],
    "execution_count": 1
@@ -1221,6 +1223,160 @@
    "execution_count": null,
    "source": "",
    "id": "84872aedbe970456"
+  },
+  {
+   "metadata": {},
+   "cell_type": "markdown",
+   "source": "# Graph Test with web search",
+   "id": "f53a68dafbed7643"
+  },
+  {
+   "metadata": {
+    "ExecuteTime": {
+     "end_time": "2025-04-30T09:44:01.940985Z",
+     "start_time": "2025-04-30T09:43:58.712696Z"
+    }
+   },
+   "cell_type": "code",
+   "source": [
+    "from dotenv import load_dotenv\n",
+    "\n",
+    "from agent import *\n",
+    "\n",
+    "load_dotenv()\n",
+    "\n",
+    "llm = get_llm()\n",
+    "\n",
+    "graph = get_graph(llm)\n"
+   ],
+   "id": "106b84a960377aad",
+   "outputs": [],
+   "execution_count": 4
+  },
+  {
+   "metadata": {
+    "ExecuteTime": {
+     "end_time": "2025-04-30T09:46:23.580205Z",
+     "start_time": "2025-04-30T09:45:52.434420Z"
+    }
+   },
+   "cell_type": "code",
+   "source": "graph.invoke({\"messages\": [HumanMessage(content=\"Search the web for the brith day of Friedrich Merz\"),]})",
+   "id": "8ca9db24590884fa",
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "-------------------- Starting to create a plan --------------------\n",
+      "\n",
+      "The plan is:  1. Search the web for \"Friedrich Merz birthday\".\n",
+      "2. Extract the date from the search results.\n",
+      "3. Provide the extracted date as the answer.\n",
+      "\n",
+      "-------------------- Agent has been called -----------------------------------\n",
+      "\n",
+      "Agent has made a decision:\n",
+      "  [{'name': 'web_search', 'args': {'query': 'Friedrich Merz birthday'}, 'id': '5105ab7c-041e-4cef-b3c4-256092d9cc5c', 'type': 'tool_call'}]\n",
+      "Waiting for 4 seconds...\n",
+      "\n",
+      "-------------------- Decision of forwarding has been made --------------------\n",
+      "\n",
+      "This is round:  2\n",
+      "The last message is:  content='' additional_kwargs={'function_call': {'name': 'web_search', 'arguments': '{\"query\": \"Friedrich Merz birthday\"}'}} response_metadata={'prompt_feedback': {'block_reason': 0, 'safety_ratings': []}, 'finish_reason': 'STOP', 'model_name': 'gemini-2.0-flash', 'safety_ratings': []} id='run-3b477f1f-5d26-4a58-ae29-581bd4ffcd4e-0' tool_calls=[{'name': 'web_search', 'args': {'query': 'Friedrich Merz birthday'}, 'id': '5105ab7c-041e-4cef-b3c4-256092d9cc5c', 'type': 'tool_call'}] usage_metadata={'input_tokens': 493, 'output_tokens': 7, 'total_tokens': 500, 'input_token_details': {'cache_read': 0}}\n",
+      "\n",
+      "-------------------- Agent has been called -----------------------------------\n",
+      "\n",
+      "Agent has made a decision:\n",
+      " I am sorry, I cannot fulfill this request. There was an error when calling the web_search tool. Please try again. []\n",
+      "Waiting for 4 seconds...\n",
+      "\n",
+      "-------------------- Decision of forwarding has been made --------------------\n",
+      "\n",
+      "This is round:  3\n",
+      "The last message is:  content='I am sorry, I cannot fulfill this request. There was an error when calling the web_search tool. Please try again.' additional_kwargs={} response_metadata={'prompt_feedback': {'block_reason': 0, 'safety_ratings': []}, 'finish_reason': 'STOP', 'model_name': 'gemini-2.0-flash', 'safety_ratings': []} id='run-70f8b036-ea1d-42e9-8d4b-f647ea13b31d-0' usage_metadata={'input_tokens': 540, 'output_tokens': 27, 'total_tokens': 567, 'input_token_details': {'cache_read': 0}}\n",
+      "The final answer is:  February 11, 1955\n"
+     ]
+    },
+    {
+     "data": {
+      "text/plain": [
+       "{'messages': [HumanMessage(content='Search the web for the brith day of Friedrich Merz', additional_kwargs={}, response_metadata={}, id='1b56f817-9fc8-475c-b6c0-162dfeba5b83'),\n",
+       "  HumanMessage(content='Write a plan how to solve this qustion?', additional_kwargs={}, response_metadata={}, id='d60f2d18-0c51-431b-90b1-851edf83e439'),\n",
+       "  AIMessage(content='1. Search the web for \"Friedrich Merz birthday\".\\n2. Extract the date from the search results.\\n3. Provide the extracted date as the answer.', additional_kwargs={}, response_metadata={'prompt_feedback': {'block_reason': 0, 'safety_ratings': []}, 'finish_reason': 'STOP', 'model_name': 'gemini-2.0-flash', 'safety_ratings': []}, id='run-bb9ba6a8-82db-48f5-919b-f05f3ea4b67f-0', usage_metadata={'input_tokens': 231, 'output_tokens': 34, 'total_tokens': 265, 'input_token_details': {'cache_read': 0}}),\n",
+       "  HumanMessage(content='Please provide me the answer to the question in detail.', additional_kwargs={}, response_metadata={}, id='70a1709b-1067-40e9-bed6-6d081ce6efe5'),\n",
+       "  AIMessage(content='', additional_kwargs={'function_call': {'name': 'web_search', 'arguments': '{\"query\": \"Friedrich Merz birthday\"}'}}, response_metadata={'prompt_feedback': {'block_reason': 0, 'safety_ratings': []}, 'finish_reason': 'STOP', 'model_name': 'gemini-2.0-flash', 'safety_ratings': []}, id='run-3b477f1f-5d26-4a58-ae29-581bd4ffcd4e-0', tool_calls=[{'name': 'web_search', 'args': {'query': 'Friedrich Merz birthday'}, 'id': '5105ab7c-041e-4cef-b3c4-256092d9cc5c', 'type': 'tool_call'}], usage_metadata={'input_tokens': 493, 'output_tokens': 7, 'total_tokens': 500, 'input_token_details': {'cache_read': 0}}),\n",
+       "  ToolMessage(content='Error: TypeError(\"BaseTool.invoke() missing 1 required positional argument: \\'input\\'\")\\n Please fix your mistakes.', name='web_search', id='0068624d-ff9e-4cb6-9415-273e489ae94e', tool_call_id='5105ab7c-041e-4cef-b3c4-256092d9cc5c', status='error'),\n",
+       "  HumanMessage(content='Please provide me the answer to the question in detail.', additional_kwargs={}, response_metadata={}, id='024947cb-abd5-4f5a-9aab-63722a483271'),\n",
+       "  AIMessage(content='I am sorry, I cannot fulfill this request. There was an error when calling the web_search tool. Please try again.', additional_kwargs={}, response_metadata={'prompt_feedback': {'block_reason': 0, 'safety_ratings': []}, 'finish_reason': 'STOP', 'model_name': 'gemini-2.0-flash', 'safety_ratings': []}, id='run-70f8b036-ea1d-42e9-8d4b-f647ea13b31d-0', usage_metadata={'input_tokens': 540, 'output_tokens': 27, 'total_tokens': 567, 'input_token_details': {'cache_read': 0}}),\n",
+       "  HumanMessage(content='Please provide me just the plain answer to the question', additional_kwargs={}, response_metadata={}, id='480e9cd8-0b33-4385-8450-7454fd9f0873'),\n",
+       "  AIMessage(content='February 11, 1955', additional_kwargs={}, response_metadata={'prompt_feedback': {'block_reason': 0, 'safety_ratings': []}, 'finish_reason': 'STOP', 'model_name': 'gemini-2.0-flash', 'safety_ratings': []}, id='run-3697ccf4-c694-4492-a07d-594497a381a9-0', usage_metadata={'input_tokens': 358, 'output_tokens': 10, 'total_tokens': 368, 'input_token_details': {'cache_read': 0}})],\n",
+       " 'aggregate': ['Plan', 'Agent', 'Agent', 'Answer']}"
+      ]
+     },
+     "execution_count": 5,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "execution_count": 5
+  },
+  {
+   "metadata": {
+    "ExecuteTime": {
+     "end_time": "2025-04-30T09:50:08.495390Z",
+     "start_time": "2025-04-30T09:50:05.663936Z"
+    }
+   },
+   "cell_type": "code",
+   "source": "search_docs = TavilySearchResults(max_results=3).invoke({'query': 'who won the last french open'})",
+   "id": "bf74512499009afd",
+   "outputs": [],
+   "execution_count": 7
+  },
+  {
+   "metadata": {
+    "ExecuteTime": {
+     "end_time": "2025-04-30T09:50:11.782505Z",
+     "start_time": "2025-04-30T09:50:11.773627Z"
+    }
+   },
+   "cell_type": "code",
+   "source": "search_docs",
+   "id": "a0c1339790a66002",
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "[{'title': \"List of French Open men's singles champions - Wikipedia\",\n",
+       "  'url': 'https://en.wikipedia.org/wiki/List_of_French_Open_men%27s_singles_champions',\n",
+       "  'content': \"Country | Amateur Era | Open Era | All-time | First title | Last title\\nFrance(FRA) | 37 | 1 | 38 | 1892 | 1983\\nSpain(ESP) | 2 | 21 | 23 | 1961 | 2024\\nAustralia(AUS) | 9 | 2 | 11 | 1933 | 1969\\nUnited States(USA) | 7 | 4 | 11 | 1938 | 1999\\nSweden(SWE) | 1 | 9 | 10 | 1957 | 1988\\nCzechoslovakia(TCH)[h] | 0 | 5 | 5 | 1970 | 1987\\nGermany(GER) | 3 | 0 | 3 | 1934 | 1937\\nItaly(ITA) | 2 | 1 | 3 | 1959 | 1976\\nBrazil(BRA) | 0 | 3 | 3 | 1997 | 2001\\nSerbia(SRB) | 0 | 3 | 3 | 2016 | 2023 [...] Location | Paris[a]France\\nVenue | Stade Roland Garros\\nGoverning body | Fédération Française de Tennis(FFT)\\nCreated | 1891 (established)1925 (Grand Slamevent)\\nEditions | 128 events (2024)94Grand Slamevents (since1925)57 events (Open Era)\\nSurface | Red clay(since 1908)Sand / Clay (1892–1907)Grass (1891)\\nTrophy | Coupe des Mousquetaires\\nWebsite | French Open champions\\nMost titles\\n14:Rafael Nadal\\nCurrent champion\\nCarlos Alcaraz(First title) [...] Open Era, who won the 2023 French Open at 36 years, twenty days.[8] French players have won the most French Open men's singles titles, with 38 victories, followed by players from Spain (20) and Australia (11). The current champion is Carlos Alcaraz, who beat Alexander Zverev in the 2024 final.[9]\",\n",
+       "  'score': 0.8190992},\n",
+       " {'title': 'French Open | History, Winners, & Facts | Britannica',\n",
+       "  'url': 'https://www.britannica.com/sports/French-Open',\n",
+       "  'content': '2014 | Julien Benneteau, Edouard Roger-Vasselin | Su-Wei Hsieh, Shuai Peng\\n2015 | Ivan Dodig, Marcelo Melo | Bethanie Mattek-Sands, Lucie Safarova\\n2016 | Feliciano Lopez, Marc Lopez | Caroline Garcia, Kristina Mladenovic\\n2017 | Ryan Harrison, Michael Venus | Bethanie Mattek-Sands, Lucie Safarova\\n2018 | Pierre-Hugues Herbert, Nicolas Mahut | Barbora Krejcikova, Katerina Siniakova\\n2019 | Kevin Krawietz, Andreas Mies | Timea Babos, Kristina Mladenovic [...] 2017 | Rafael Nadal (Spain) | Jelena Ostapenko (Latvia)\\n2018 | Rafael Nadal (Spain) | Simona Halep (Rom.)\\n2019 | Rafael Nadal (Spain) | Ashleigh Barty (Austl.)\\n2020 | Rafael Nadal (Spain) | Iga Świątek (Pol.)\\n2021 | Novak Djokovic (Serbia) | Barbora Krejčíková (Czech.)\\n2022 | Rafael Nadal (Spain) | Iga Świątek (Pol.)\\n2023 | Novak Djokovic (Serbia) | Iga Świątek (Pol.)\\n2024 | Carlos Alcaraz (Spain) | Iga Świątek (Pol.)\\nWinners of the French Open doubles championship are provided in the table. [...] 2007 | Mark Knowles, Daniel Nestor | Alicia Molik, Mara Santangelo\\n2008 | Pablo Cuevas, Luis Horna | Anabel Medina Garrigues, Virginia Ruano Pascual\\n2009 | Lukas Dlouhy, Leander Paes | Anabel Medina Garrigues, Virginia Ruano Pascual\\n2010 | Daniel Nestor, Nenad Zimonjic | Serena Williams, Venus Williams\\n2011 | Max Mirnyi, Daniel Nestor | Andrea Hlavackova, Lucie Hradecka\\n2012 | Max Mirnyi, Daniel Nestor | Sara Errani, Roberta Vinci\\n2013 | Bob Bryan, Mike Bryan | Ekaterina Makarova, Elena Vesnina',\n",
+       "  'score': 0.783542},\n",
+       " {'title': \"Winners of the Men's French Open - Tennis - Topend Sports\",\n",
+       "  'url': 'https://www.topendsports.com/events/tennis-grand-slam/french-open/winners-men.htm',\n",
+       "  'content': \"Home > Events > Tennis > Grand Slam > French Open > Winners > Men\\n\\nWinners of  the Men's French Open\\n\\nHere are all the winners of the French Tennis Open men's title since the first tournament in 1925. The recent tournaments have been dominated by Spanish player Rafael Nadal. He now has 14 French Open men's singles titles (the last in 2022). See also the Women's Singles Results.\",\n",
+       "  'score': 0.77446437}]"
+      ]
+     },
+     "execution_count": 8,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "execution_count": 8
+  },
+  {
+   "metadata": {},
+   "cell_type": "code",
+   "outputs": [],
+   "execution_count": null,
+   "source": "",
+   "id": "bdab71ec26eaadc0"
   }
  ],
  "metadata": {