Update rag_langgraph.py

rag_langgraph.py

@@ -6,6 +6,7 @@ from typing import Annotated, Any, Dict, List, Optional, Sequence, Tuple, TypedD
 
 from langchain.agents import AgentExecutor, create_openai_tools_agent
 from langchain_community.tools.tavily_search import TavilySearchResults
+from langchain_community.utilities import ArxivAPIWrapper
 from langchain_core.messages import BaseMessage, HumanMessage
 from langchain_core.output_parsers.openai_functions import JsonOutputFunctionsParser
 from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
@@ -42,6 +43,7 @@ def today_tool(text: str) -> str:
     return (str(date.today()) + "\n\nIf you have completed all tasks, respond with FINAL ANSWER.")
     
 def create_graph(model, topic):
+    arxiv_tool = ArxivAPIWrapper()
     tavily_tool = TavilySearchResults(max_results=10)
     
     members = ["Researcher"]
@@ -94,10 +96,11 @@ def create_graph(model, topic):
         | JsonOutputFunctionsParser()
     )
 
-    researcher_agent = create_agent(llm, [tavily_tool, today_tool], system_prompt="1. Research content on topic: " + topic + ", prioritizing research papers. "
-                                                                                  "2. Based on your research, write a 2000-word article on the topic. "
-                                                                                  "3. At the beginning of the article, add current date and author: Multi-AI-Agent System. "
-                                                                                  "4. At the end of the article, add a references section with research papers.")
+    researcher_agent = create_agent(llm, [arxiv_tool, tavily_tool, today_tool], system_prompt=
+                                    "1. Research content on topic: " + topic + ", prioritizing research papers. "
+                                    "2. Based on your research, write a 2000-word article on the topic. "
+                                    "3. At the beginning of the article, add current date and author: Multi-AI-Agent System. "
+                                    "4. At the end of the article, add a references section with research papers.")
     researcher_node = functools.partial(agent_node, agent=researcher_agent, name="Researcher")
 
     workflow = StateGraph(AgentState)