Update agent/planning_agent.py

agent/planning_agent.py

@@ -22,12 +22,13 @@ chat_memory = None
 query_memory = None
 agent = None
 
-def initialize_planning_agent(llm_instance, chat_memory_instance, query_memory_instance):
-    global llm, chat_memory, query_memory, agent
+def initialize_planning_agent(llm_instance, chat_memory_instance, query_memory_instance, callback_fn):
+    global llm, chat_memory, query_memory, agent, callback
     
     llm = llm_instance
     chat_memory = chat_memory_instance
     query_memory = query_memory_instance
+    callback = callback_fn
     
     # Initialize agents
     router_agent.initialize_router_agent(llm, chat_memory)