Update rag_langgraph.py

rag_langgraph.py

@@ -46,7 +46,7 @@ def today_tool(text: str) -> str:
     """Returns today's date. Use this for any questions related to knowing today's date. 
        The input should always be an empty string, and this function will always return today's date. 
        Any date mathematics should occur outside this function."""
-    return str(date.today())
+    return (str(date.today()) + "\n\nIf you have completed all tasks, respond with FINAL ANSWER.")
     
 def create_graph(topic, word_count):
     tavily_tool = TavilySearchResults(max_results=10)
@@ -101,10 +101,10 @@ def create_graph(topic, word_count):
         | JsonOutputFunctionsParser()
     )
 
-    researcher_agent = create_agent(llm, [tavily_tool], system_prompt=f"Prioritizing research papers, research content on topic: '{topic}'.")
+    researcher_agent = create_agent(llm, [tavily_tool], system_prompt=f"Prioritizing research papers, research content on topic: {topic}.")
     researcher_node = functools.partial(agent_node, agent=researcher_agent, name="Researcher")
 
-    writer_agent = create_agent(llm, [today_tool], system_prompt=f"Write a {word_count}-word article on topic: '{topic}'. Add a 'references' section with research papers. Add the 'current date' and 'author': Multi-AI-Agent System based on GPT-4o.")
+    writer_agent = create_agent(llm, [today_tool], system_prompt=f"Write a {word_count}-word article on topic: {topic}. At the top, add current date and author: Multi-AI-Agent System based on GPT-4o. At the bottom, add a references section with research papers.")
     writer_node = functools.partial(agent_node, agent=writer_agent, name="Writer")
 
     workflow = StateGraph(AgentState)