Update multi_agent.py

multi_agent.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(llm_manager, llm_agent, topic):
+def create_graph(llm, topic):
     tavily_tool = TavilySearchResults(max_results=10)
     
     members = ["Researcher"]
@@ -87,7 +87,7 @@ def create_graph(llm_manager, llm_agent, topic):
         ]
     ).partial(options=str(options), members=", ".join(members))
     
-    llm = ChatOpenAI(model=llm_manager)
+    llm = ChatOpenAI(model=llm)
     
     supervisor_chain = (
         prompt
@@ -95,8 +95,6 @@ def create_graph(llm_manager, llm_agent, topic):
         | JsonOutputFunctionsParser()
     )
 
-    llm = ChatOpenAI(model=llm_agent)
-    
     researcher_agent = create_agent(llm, [tavily_tool, today_tool], system_prompt=
                                     "1. Research content on topic: " + topic + ". "
                                     "2. Based on your research, write an in-depth article on the topic. " 
@@ -121,8 +119,8 @@ def create_graph(llm_manager, llm_agent, topic):
     
     return workflow.compile()
 
-def run_multi_agent(llm_manager, llm_agent, topic):
-    graph = create_graph(llm_manager, llm_agent, topic)
+def run_multi_agent(llm, topic):
+    graph = create_graph(llm, topic)
     
     result = graph.invoke({
         "messages": [