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(model, topic):
+def create_graph(llm_manager, llm_agent, topic):
     tavily_tool = TavilySearchResults(max_results=10)
     
     members = ["Researcher"]
@@ -87,7 +87,7 @@ def create_graph(model, topic):
         ]
     ).partial(options=str(options), members=", ".join(members))
     
-    llm = ChatOpenAI(model=model)
+    llm = ChatOpenAI(model=llm_manager)
     
     supervisor_chain = (
         prompt
@@ -95,6 +95,8 @@ def create_graph(model, 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. " 
@@ -119,8 +121,8 @@ def create_graph(model, topic):
     
     return workflow.compile()
 
-def run_multi_agent(llm, topic):
-    graph = create_graph(llm, topic)
+def run_multi_agent(llm_manager, llm_agent, topic):
+    graph = create_graph(llm_manager, llm_agent, topic)
     
     result = graph.invoke({
         "messages": [
@@ -130,8 +132,8 @@ def run_multi_agent(llm, topic):
     
     article = result['messages'][-1].content
     
-    #print("---")
-    #print(article)
-    #print("---")
+    print("===")
+    print(article)
+    print("===")
     
     return article