Update rag_langgraph.py

rag_langgraph.py

@@ -43,7 +43,7 @@ def today_tool(text: str) -> str:
        Any date mathematics should occur outside this function."""
     return (str(date.today()) + "\n\nIf you have completed all tasks, respond with FINAL ANSWER.")
     
-def create_graph(topic, word_count):
+def create_graph(topic):
     tavily_tool = TavilySearchResults(max_results=10)
     
     members = ["Researcher", "Writer"]
@@ -99,7 +99,7 @@ def create_graph(topic, word_count):
     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}, including a reference section with research papers. At the top, add current date and author: Multi-AI-Agent System based on GPT-4o.")
+    writer_agent = create_agent(llm, [today_tool], system_prompt=f"Write a 1000-word article on topic: {topic}, including a reference section with research papers. At the top, add current date and author: Multi-AI-Agent System based on GPT-4o.")
     writer_node = functools.partial(agent_node, agent=writer_agent, name="Writer")
 
     workflow = StateGraph(AgentState)
@@ -118,8 +118,8 @@ def create_graph(topic, word_count):
     
     return workflow.compile()
 
-def run_multi_agent(topic, word_count):
-    graph = create_graph(topic, word_count)
+def run_multi_agent(topic):
+    graph = create_graph(topic)
     result = graph.invoke({
         "messages": [
             HumanMessage(content=topic)