Update rag_langgraph.py

rag_langgraph.py

@@ -46,17 +46,17 @@ def today_tool(text: str) -> str:
 def create_graph(topic):
     tavily_tool = TavilySearchResults(max_results=10)
     
-    agents = ["Researcher", "Writer"]
+    members = ["Researcher", "Writer"]
     
     system_prompt = (
         "You are a Manager tasked with managing a conversation between the"
-        " following agents: {agents}. Given the following user request,"
+        " following agents: {members}. Given the following user request,"
         " respond with the agent to act next. Each agent will perform a"
         " task and respond with their results and status. When finished,"
         " respond with FINISH."
     )
 
-    options = ["FINISH"] + agents
+    options = ["FINISH"] + members
 
     function_def = {
         "name": "route",
@@ -86,7 +86,7 @@ def create_graph(topic):
                 " Or should we FINISH? Select one of: {options}",
             ),
         ]
-    ).partial(options=str(options), members=", ".join(agents))
+    ).partial(options=str(options), members=", ".join(members))
     
     llm = ChatOpenAI(model=LLM)
     
@@ -107,10 +107,10 @@ def create_graph(topic):
     workflow.add_node("Writer", writer_node)
     workflow.add_node("Manager", supervisor_chain)
 
-    for agent in agents:
-        workflow.add_edge(agent, "Manager")
+    for member in members:
+        workflow.add_edge(member, "Manager")
 
-    conditional_map = {k: k for k in agents}
+    conditional_map = {k: k for k in members}
     conditional_map["FINISH"] = END
     
     workflow.add_conditional_edges("Manager", lambda x: x["next"], conditional_map)